arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2503.06473cs.CVcs.AI

通过剪枝冗余检索增强层注意力效率

Enhancing Layer Attention Efficiency through Pruning Redundant Retrievals

De Artificial Intelligence Lab(德人工智能实验室)

Hanze Li, Yaosong Du, Zhibo Yao, Mengyao Zeng, Xiuqi Ge, Xiande Huang

AI总结:

针对层注意力机制中相邻层权重冗余导致特征重复和训练效率低的问题,提出基于KL散度量化冗余并利用增强Beta分位数映射(EBQM)跳过冗余层的高效层注意力(ELA)架构,在图像分类和目标检测任务中训练时间减少30%且性能提升。

摘要:

越来越多的证据表明,层注意力机制增强了深度神经网络中层间的交互,显著推进了网络架构的发展。然而,现有的层注意力方法存在冗余问题,因为相邻层学习的注意力权重往往变得高度相似。这种冗余导致多个层提取几乎相同的特征,降低了模型的表示能力并增加了训练时间。为了解决这个问题,我们提出了一种新颖的方法,利用相邻层之间的Kullback-Leibler(KL)散度来量化冗余。此外,我们引入了一种增强Beta分位数映射(EBQM)方法,能够准确识别并跳过冗余层,从而保持模型稳定性。我们提出的高效层注意力(ELA)架构提高了训练效率和整体性能,在图像分类和目标检测等任务中实现了30%的训练时间减少,同时提升了性能。

英文摘要:

Growing evidence suggests that layer attention mechanisms, which enhance interaction among layers in deep neural networks, have significantly advanced network architectures. However, existing layer attention methods suffer from redundancy, as attention weights learned by adjacent layers often become highly similar. This redundancy causes multiple layers to extract nearly identical features, reducing the model's representational capacity and increasing training time. To address this issue, we propose a novel approach to quantify redundancy by leveraging the Kullback-Leibler (KL) divergence between adjacent layers. Additionally, we introduce an Enhanced Beta Quantile Mapping (EBQM) method that accurately identifies and skips redundant layers, thereby maintaining model stability. Our proposed Efficient Layer Attention (ELA) architecture, improves both training efficiency and overall performance, achieving a 30% reduction in training time while enhancing performance in tasks such as image classification and object detection.

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

↑