arXiv:2502.01962cs.CV2025-02ICLR被引 7

提升视觉Transformer适配器的内存效率,加速密集预测任务推理

Memory Efficient Transformer Adapter for Dense Predictions

论文配图:Memory Efficient Transformer Adapter for Dense Predictions
图 1 · 摘自论文原文
  • 通过共享层归一化和交叉注意力减少内存访问开销
  • 在多个数据集上实现更高精度与更快推理速度的平衡
  • 适合需要高效密集预测的场景,如目标检测与语义分割

现有视觉Transformer适配器虽具良好精度,但推理速度受制于低效的内存访问操作,如标准归一化和频繁重塑。本文提出META,一种简单高效的ViT适配器,通过减少此类操作提升内存效率并降低内存耗时。其核心是内存优化的适配器模块:在自注意力与前馈网络间共享层归一化,减少归一化依赖;采用交叉形自注意力结构,降低频繁重塑开销;额外引入轻量卷积分支,增强局部归纳偏置,尤其适用于目标检测、实例分割和语义分割等密集预测任务。模块以级联方式构建,生成多样头特征,丰富表示能力。实验表明,META在多个代表性数据集上显著提升预测质量,达成新的精度-效率最优平衡。理论分析显示,META具备更优泛化能力和更强适应性。

原文摘要 · Abstract (English)

While current Vision Transformer (ViT) adapter methods have shown promising accuracy, their inference speed is implicitly hindered by inefficient memory access operations, e.g., standard normalization and frequent reshaping. In this work, we propose META, a simple and fast ViT adapter that can improve the model's memory efficiency and decrease memory time consumption by reducing the inefficient memory access operations. Our method features a memory-efficient adapter block that enables the common sharing of layer normalization between the self-attention and feed-forward network layers, thereby reducing the model's reliance on normalization operations. Within the proposed block, the cross-shaped self-attention is employed to reduce the model's frequent reshaping operations. Moreover, we augment the adapter block with a lightweight convolutional branch that can enhance local inductive biases, particularly beneficial for the dense prediction tasks, e.g., object detection, instance segmentation, and semantic segmentation. The adapter block is finally formulated in a cascaded manner to compute diverse head features, thereby enriching the variety of feature representations. Empirically, extensive evaluations on multiple representative datasets validate that META substantially enhances the predicted quality, while achieving a new state-of-the-art accuracy-efficiency trade-off. Theoretically, we demonstrate that META exhibits superior generalization capability and stronger adaptability.

视觉Transformer适配器密集预测内存效率

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