arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2601.01002cs.CV

轻量级通道注意力机制用于高效的CNNs

Lightweight Channel Attention for Efficient CNNs

Rochester Institute of Technology(罗切斯特技术研究所)

Prem Babu Kanaparthi, Tulasi Venkata Sri Varshini Padamata

AI总结:

本文提出轻量级通道注意力机制LCA,通过减少参数使用并保持有效注意力行为,在ResNet 18和MobileNetV2上实现了高准确率和低延迟的性能。

摘要:

注意力机制已成为现代卷积神经网络(CNNs)的重要组成部分,通过最小的计算开销带来显著的性能提升。然而,不同通道注意力设计的效率与精度的权衡仍处于探索阶段。本文通过在CIFAR 10数据集上对ResNet 18和MobileNetV2架构进行实证研究,比较了Squeeze and Excitation(SE)、Efficient Channel Attention(ECA)以及所提出的Lite Channel Attention(LCA)模块。LCA采用自适应的一维卷积和分组操作来减少参数使用,同时保持有效的注意力行为。实验结果表明,LCA在ResNet 18上达到94.68%的准确率,在MobileNetV2上达到93.10%的准确率,同时在参数效率方面与ECA相当,并保持了有利的推理延迟。全面的基准测试包括FLOPs、参数计数和GPU延迟测量,为在资源受限环境中部署增强注意力的CNNs提供了实用见解。

英文摘要:

Attention mechanisms have become integral to modern convolutional neural networks (CNNs), delivering notable performance improvements with minimal computational overhead. However, the efficiency accuracy trade off of different channel attention designs remains underexplored. This work presents an empirical study comparing Squeeze and Excitation (SE), Efficient Channel Attention (ECA), and a proposed Lite Channel Attention (LCA) module across ResNet 18 and MobileNetV2 architectures on CIFAR 10. LCA employs adaptive one dimensional convolutions with grouped operations to reduce parameter usage while preserving effective attention behavior. Experimental results show that LCA achieves competitive accuracy, reaching 94.68 percent on ResNet 18 and 93.10 percent on MobileNetV2, while matching ECA in parameter efficiency and maintaining favorable inference latency. Comprehensive benchmarks including FLOPs, parameter counts, and GPU latency measurements are provided, offering practical insights for deploying attention enhanced CNNs in resource constrained environments.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑