arXiv:2506.10178cs.CV2025-06中稿 · ICLR被引 11

提出轻量高效注意力探测方法,提升大模型评估精度与效率

Attention, Please! Revisiting Attentive Probing Through the Lens of Efficiency

  • 设计多查询交叉注意力机制,减少冗余参数
  • 在多个数据集上优于线性探测和原有注意力探测方法
  • 适合需要高效评估大模型性能的研究者使用

随着微调在大规模场景下变得不切实际,探测(probing)正成为主流评估方法。然而,标准的线性探测可能低估那些预训练优化局部表示而非显式全局表示的模型能力。这促使了注意力探测(attentive probing)的发展,该方法利用注意力机制选择性聚合图像块特征。尽管应用日益广泛,现有方法仍存在参数过多、计算效率低的问题。本文从准确率与参数效率的权衡角度重新审视注意力探测,首次系统分析现有方法的设计选择并进行基准测试。基于这些发现,我们提出高效探测(EP),一种轻量级但高效的多查询交叉注意力机制,消除冗余投影,显著减少可训练参数。在多个基准和预训练范式下,EP始终优于线性探测及先前的注意力探测方法,且在结合参数高效微调时依然有效。此外,我们的分析揭示了EP的互补注意力图等新特性,为探测方法的应用拓展提供了新方向。

原文摘要 · Abstract (English)

As fine-tuning becomes impractical at scale, probing is emerging as the preferred evaluation protocol. However, standard linear probing can understate the capability of models whose pre-training optimizes local representations rather than an explicit global representation. This motivates attentive probing, an alternative that uses attention to selectively aggregate patch-level features. Despite growing adoption, attentive probing is still underexplored: existing approaches are often over-parameterized and computationally inefficient. In this work, we revisit attentive probing through the lens of the accuracy vs. parameter-efficiency trade-off. We present the first comprehensive study of existing methods, analyzing their design choices and benchmarking their performance. Building on these insights, we propose efficient probing (EP), a lightweight yet effective multi-query cross-attention mechanism that eliminates redundant projections and reduces the number of trainable parameters. Across multiple benchmarks and pre-training paradigms, EP consistently outperforms linear probing and previous attentive probing methods, and remains effective when combined with parameter-efficient fine-tuning. Beyond evaluation, our analysis uncovers emerging properties of EP, including complementary attention maps, which open new directions for leveraging probing beyond protocol design. Project page: https://vrg.fel.cvut.cz/ep/.

模型评估注意力机制高效学习

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