跳到正文
arXiv · ML· Charles Herr, Gabriele Spadaro, Enzo Tartaglione·· 17 小时前AI 评分25

利用注意力偏差优化 Transformer 计算效率

Exploiting Biases in Attention for Transformers

AI 导读

该研究提出在 Transformer 注意力公式中引入输入相关偏差项,以明确 token 的绝对重要性并实现注意力图剪枝。 这一方法在多种模型尺寸下显著降低了 FLOPs 和推理延迟,同时保留了大部分分类精度。 代码和模型已公开,论文被 WACV 2027 接收。

来源:arXiv · ML · arxiv.org