arXiv:2606.03085cs.LGcs.CL2026-06ACL

同时追踪大模型多个组件的因果路径,精准定位影响性能的关键模块。

Multi-component Causal Tracing in Large Language Models

论文配图:Multi-component Causal Tracing in Large Language Models
图 1 · 摘自论文原文
  • 设计统一框架,可同时干预多个组件(如注意力头、MLP神经元)
  • 通过软干预与度量变换,高效找到对指标影响最大的组件子集
  • 适用于提升模型准确性与公平性分析,适合模型可解释性研究者

因果追踪通过干预大语言模型(LLM)内部表征,系统揭示并量化输入或计算与特定目标指标之间的因果路径,从而理解模型行为。本文在以往单组件或单层研究基础上,提出一个统一的多组件因果追踪框架,可系统识别对目标性能指标(如准确率、公平性)最关键的组件子集(如注意力头和多层感知机神经元)。该框架引入灵活干预机制,适配多种目标指标。为应对多组件组合带来的复杂性,设计了一种高效算法,利用软干预与精心设计的度量变换,将组合搜索问题转化为连续优化问题,在约束条件下高效求解,生成组件选择的二值决策。实验表明,该方法能有效识别对目标指标有显著影响的组件子集,优于现有基线方法。代码已公开于 https://github.com/ZiruiYan/multi-component-causal-tracing。

原文摘要 · Abstract (English)

Causal tracing systematically intervenes on a large language model's (LLM's) internal representations to uncover and quantify the causal pathways linking specific inputs or computations to specific metrics of interest, quantifying the LLM's behavior. Building on previous single-component or single-layer studies, this paper presents a unified framework for causally tracing multiple components simultaneously. This framework systematically identifies the subsets of components (e.g., attention heads and multi-layer perceptron neurons) most critical to a desired target performance metric (e.g., accuracy and fairness). This is achieved by incorporating flexible interventions applied to a wide range of desired metrics. To address the combinatorial complexity of the multi-component problem, an efficient algorithm is designed that leverages soft interventions and a carefully designed metric transformation, converting the combinatorial search problem into a continuous one that can be solved efficiently under proper constraints, thereby generating proper binary decisions for selecting components. Experimental results demonstrate that the proposed method efficiently identifies subsets of the model's components that have a high impact on the target metric, outperforming existing baseline approaches. Our code is available at https://github.com/ZiruiYan/multi-component-causal-tracing.

因果追踪大模型可解释性组件筛选神经网络分析

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