SheepNav
新上线今天0 投票

精准恢复神经网络权值机制:靶向参数分解方法大幅降低计算成本

近日,一项发表于 ICML 2026 机械可解释性研讨会的研究提出了 靶向参数分解(targeted PD, tPD) 方法,旨在高效地从大型神经网络中恢复与特定输入相关的可解释计算组件。该工作由 Antoine Vigouroux 和 Lee Sharkey 完成,论文编号 arXiv:2607.13047。

背景:参数分解的规模瓶颈

传统的 参数分解(Parameter Decomposition, PD) 技术能够将神经网络分解为忠实反映原始网络操作的、可解释的计算组件。然而,当应用于大型模型时,PD 需要巨大的计算资源,这使得其成本高昂且风险较大。例如,对一个大模型进行完整分解可能需要数万 GPU 小时,严重限制了该方法的实用性。

tPD 的核心创新

tPD 通过引入一个 高秩“兜底”组件(catch-all component) 来突破这一瓶颈。该组件专门负责处理所有非目标数据,从而使得分解过程可以只聚焦于处理特定输入(从单个提示到大型子任务)的组件。这样一来,tPD 能够跳过对全局无关部分的冗余计算,大幅降低资源需求。

验证与实验结果

研究团队在玩具模型以及基于 The Pile 数据集训练的 Transformer 语言模型上验证了 tPD 的有效性。实验结果表明,tPD 能够恢复可复现、机制上忠实的神经回路。

  • 效率提升:在一个 4 层 Transformer 模型中,tPD 仅使用其完整分解所需 7% 的 FLOPs 就提取出了一个仅处理 CSS 代码的子模型。
  • 精准干预:在一个 12 层 Transformer 中,研究人员使用 tPD 成功消融并重写了模型中的记忆化序列,而对其他输入的副作用可忽略不计。

行业意义

随着 AI 模型规模持续增长,可解释性已成为安全部署的关键挑战。tPD 提供了一种经济且精准的工具,使得开发者可以“手术刀”式地分析模型在特定任务上的行为,而无需对整个模型进行昂贵的大规模分解。这对于调试模型偏见、移除不良记忆以及理解模型内部机制具有重要意义。该工作有望推动可解释性技术从学术研究走向实际工业应用。

延伸阅读

  1. 解构知识追踪:PAKT模型如何区分学生的“能力”与“熟练度”
  2. TSSM: Triaxial State Space Model for Global Station Weather Forecasting with Temporal-Variable-Historical Modeling
  3. 流式系统中事件触发LLM调用的不确定性感知顺序决策规则
查看原文