arXiv:2409.03460cs.CV2024-09中稿 · WACV 2025被引 11

提出更贴近实际速度的高效视觉骨干网络设计方法。

LowFormer: Hardware Efficient Design for Convolutional Transformer Backbones

  • 基于真实吞吐与延迟优化模型架构,而非仅看计算量
  • 新架构在多硬件上实现更高吞吐和更低延迟
  • 适合部署于GPU、移动GPU及ARM CPU的高效模型

视觉骨干网络研究正朝着卷积与Transformer模块混合的方向发展。合理地从架构和组件层面融合二者,是实现速度与精度平衡的关键。现有研究多以提升精度为目标,并使用乘加操作(MACs)作为效率指标,但该指标常无法准确反映实际运行速度,因受内存访问开销和并行度等因素影响。本文从实际吞吐量和延迟出发,分析常见模块与架构设计选择,认为两者的结合更能体现模型在真实应用中的效率。基于此分析,我们提出提升宏观设计硬件效率的方案,并引入一种简化版多头自注意力机制,契合上述分析。将宏观与微观设计结合,构建出名为LowFormer的新一代高效骨干网络。LowFormer在保持甚至超越当前先进高效骨干网络精度的同时,在吞吐量和延迟上实现显著加速。为验证设计的普适性,我们在GPU、移动GPU和ARM CPU上进行评估,结果表明下游任务如目标检测和语义分割均能从中受益。代码与模型已开源:https://github.com/altair199797/LowFormer。

原文摘要 · Abstract (English)

Research in efficient vision backbones is evolving into models that are a mixture of convolutions and transformer blocks. A smart combination of both, architecture-wise and component-wise is mandatory to excel in the speedaccuracy trade-off. Most publications focus on maximizing accuracy and utilize MACs (multiply accumulate operations) as an efficiency metric. The latter however often do not measure accurately how fast a model actually is due to factors like memory access cost and degree of parallelism. We analyzed common modules and architectural design choices for backbones not in terms of MACs, but rather in actual throughput and latency, as the combination of the latter two is a better representation of the efficiency of models in real applications. We applied the conclusions taken from that analysis to create a recipe for increasing hardware-efficiency in macro design. Additionally we introduce a simple slimmed-down version of MultiHead Self-Attention, that aligns with our analysis. We combine both macro and micro design to create a new family of hardware-efficient backbone networks called LowFormer. LowFormer achieves a remarkable speedup in terms of throughput and latency, while achieving similar or better accuracy than current state-of-the-art efficient backbones. In order to prove the generalizability of our hardware-efficient design, we evaluate our method on GPU, mobile GPU and ARM CPU. We further show that the downstream tasks object detection and semantic segmentation profit from our hardware-efficient architecture. Code and models are available at https://github.com/ altair199797/LowFormer.

高效模型骨干网络硬件优化

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