arXiv:2411.16170cs.CV2024-11CVPR被引 8

提出轻量线性视觉变压器,兼顾移动端效率与精度。

CARE Transformer: Mobile-Friendly Linear Visual Transformer via Decoupled Dual Interaction

  • 解耦局部与长程特征学习,提升计算效率
  • 在ImageNet上仅用0.7 GMACs达78.4%准确率
  • 适合移动端部署,兼顾精度与速度

近期大量工作致力于设计高效线性复杂度的视觉Transformer。然而,现有线性注意力模型因效率提升有限或精度显著下降,难以部署于资源受限的移动设备。本文提出一种去耦双交互线性注意力机制(CARE),揭示特征解耦与交互可充分发挥线性注意力潜力。首先,采用非对称特征解耦策略,分别处理局部归纳偏置与长程依赖的学习过程,有效保留局部与全局信息并提升效率;其次,引入动态记忆单元以保持网络路径中的关键信息;此外,设计双交互模块,促进局部归纳偏置与长程信息间、以及不同层特征间的有效交互。通过解耦学习与特征互补性利用,方法同时实现高效率与高精度。在ImageNet-1K、COCO和ADE20K数据集上的大量实验表明其有效性,例如在ImageNet-1K上仅需0.7/1.9 GMACs即可达到78.4%/82.1%的top-1准确率。代码将公开于https://github.com/zhouyuan888888/CARE-Transformer。

原文摘要 · Abstract (English)

Recently, large efforts have been made to design efficient linear-complexity visual Transformers. However, current linear attention models are generally unsuitable to be deployed in resource-constrained mobile devices, due to suffering from either few efficiency gains or significant accuracy drops. In this paper, we propose a new de\textbf{C}oupled du\textbf{A}l-interactive linea\textbf{R} att\textbf{E}ntion (CARE) mechanism, revealing that features' decoupling and interaction can fully unleash the power of linear attention. We first propose an asymmetrical feature decoupling strategy that asymmetrically decouples the learning process for local inductive bias and long-range dependencies, thereby preserving sufficient local and global information while effectively enhancing the efficiency of models. Then, a dynamic memory unit is employed to maintain critical information along the network pipeline. Moreover, we design a dual interaction module to effectively facilitate interaction between local inductive bias and long-range information as well as among features at different layers. By adopting a decoupled learning way and fully exploiting complementarity across features, our method can achieve both high efficiency and accuracy. Extensive experiments on ImageNet-1K, COCO, and ADE20K datasets demonstrate the effectiveness of our approach, e.g., achieving $78.4/82.1\%$ top-1 accuracy on ImagegNet-1K at the cost of only $0.7/1.9$ GMACs. Codes will be released on \href{https://github.com/zhouyuan888888/CARE-Transformer}{https://github.com/zhouyuan888888/CARE-Transformer}.

视觉Transformer轻量化线性注意力移动端

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