挑战主流观点:大模型影响估计中,中间层比首层更准
First is Not Really Better Than Last: Evaluating Layer Choice and Aggregation Strategies in Language Model Data Influence Estimation
- 发现首层影响估计依赖的抵消效应不可靠,改用中间注意力层更优
- 提出排序与投票聚合策略,使影响估计性能显著提升
- 设计新评估指标NDR,无需重训练即可有效衡量影响得分
识别训练样本对大语言模型决策的影响对解释模型行为和审计大规模数据集至关重要。现有影响函数方法通过模型梯度信息流来实现,但因模型参数量达数十亿,通常仅在部分层计算以保证可行性。先前研究认为首层(嵌入层)最适于影响估计,基于影响分数抵消的假设。本文从理论与实证角度证明该抵消效应不可靠,指出中间注意力层才是更优的估计器。同时,针对跨层影响分数聚合问题,展示了排序与投票等替代平均法可显著提升性能。最后,提出无需重训练的新评估方法及噪声检测率(NDR)指标,其预测能力优于传统方法。在多种类型与规模的大模型上广泛实验表明,首层并非必然优于末层,推翻了领域内既有认知。
原文摘要 · Abstract (English)
Identifying how training samples influence/impact Large Language Model (LLM) decision-making is essential for effectively interpreting model decisions and auditing large-scale datasets. Current training sample influence estimation methods (also known as influence functions) undertake this goal by utilizing information flow through the model via its first-order and higher-order gradient terms. However, owing to the large model sizes of today consisting of billions of parameters, these influence computations are often restricted to some subset of model layers to ensure computational feasibility. Prior seminal work by Yeh et al. (2022) in assessing which layers are best suited for computing language data influence concluded that the first (embedding) layers are the most informative for this purpose, using a hypothesis based on influence scores canceling out (i.e., the cancellation effect). In this work, we propose theoretical and empirical evidence demonstrating how the cancellation effect is unreliable, and that middle attention layers are better estimators for influence. Furthermore, we address the broader challenge of aggregating influence scores across layers, and showcase how alternatives to standard averaging (such as ranking and vote-based methods) can lead to significantly improved performance. Finally, we propose better methods for evaluating influence score efficacy in LLMs without undertaking model retraining, and propose a new metric known as the Noise Detection Rate (NDR) that exhibits strong predictive capability compared to the cancellation effect. Through extensive experiments across LLMs of varying types and scales, we concretely determine that the first (layers) are not necessarily better than the last (layers) for LLM influence estimation, contrasting with prior knowledge in the field.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。