arXiv:2603.18585cs.CV2026-03

让视觉Transformer层层传递注意力,提升模型表现

HAViT: Historical Attention Vision Transformer

  • 跨层传播历史注意力矩阵,逐步优化关注模式
  • 在CIFAR-100上准确率提升1.33%,TinyImageNet上提升1.25%
  • 无需改动结构,适合各类Transformer模型快速部署

视觉Transformer在计算机视觉中表现优异,但其注意力机制各层独立运作,限制了信息流动与特征学习。本文提出一种有效的跨层注意力传播方法,保留并融合编码器各层的历史注意力矩阵,为Vision Transformer的层间信息流提供原理性改进。该方法实现注意力模式在变换层次中的渐进式优化,增强特征获取与优化动态。仅需添加注意力矩阵存储和混合操作,几乎不改变原有架构。在CIFAR-100和TinyImageNet上的实验表明,ViT准确率分别从75.74%提升至77.07%(+1.33%),从57.82%提升至59.07%(+1.25%)。跨架构验证显示,多种Transformer变体均获得相似增益,CaiT提升1.01%。系统分析确定历史注意力混合超参α=0.45为最优,平衡当前与历史注意力信息。随机初始化优于零初始化,表明多样化的初始注意力模式可加速收敛并提升最终性能。代码已公开于https://github.com/banik-s/HAViT。

原文摘要 · Abstract (English)

Vision Transformers have excelled in computer vision but their attention mechanisms operate independently across layers, limiting information flow and feature learning. We propose an effective cross-layer attention propagation method that preserves and integrates historical attention matrices across encoder layers, offering a principled refinement of inter-layer information flow in Vision Transformers. This approach enables progressive refinement of attention patterns throughout the transformer hierarchy, enhancing feature acquisition and optimization dynamics. The method requires minimal architectural changes, adding only attention matrix storage and blending operations. Comprehensive experiments on CIFAR-100 and TinyImageNet demonstrate consistent accuracy improvements, with ViT performance increasing from 75.74% to 77.07% on CIFAR-100 (+1.33%) and from 57.82% to 59.07% on TinyImageNet (+1.25%). Cross-architecture validation shows similar gains across transformer variants, with CaiT showing 1.01% enhancement. Systematic analysis identifies the blending hyperparameter of historical attention (alpha = 0.45) as optimal across all configurations, providing the ideal balance between current and historical attention information. Random initialization consistently outperforms zero initialization, indicating that diverse initial attention patterns accelerate convergence and improve final performance. Our code is publicly available at https://github.com/banik-s/HAViT.

视觉Transformer注意力机制模型优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。