arXiv:2602.11726cs.LG2026-02

用神经元阈值和增益实现可解释的轻量级模型微调

Dopamine: Brain Modes, Not Brains

  • 通过学习神经元的阈值和增益,实现激活空间的轻量微调
  • 仅用数百参数/层即提升旋转手写数字识别准确率
  • 比LoRA更少参数、更易解释,适合追求可解释性的场景

参数高效微调(PEFT)方法如LoRA通过添加小规模权重更新来适配大预训练模型。然而,权重变化难以机制化解释,且无法直接揭示新任务中哪些内部计算被复用或跳过。受神经调节启发,我们提出一种新视角:将适应视为模式切换——选择并缩放现有计算,而非重写基础权重。我们提出MethodName,一种简单激活空间的PEFT技术,冻结基础权重,学习每个神经元的阈值和增益。训练时,平滑门控决定神经元激活是否参与;推理时可硬化门控,实现显式的条件计算与神经元级归因。以MNIST(0°)与旋转45°为例,我们在50/50混合数据上预训练一个小MLP,冻结权重后用MethodName对旋转模式进行专业化。跨种子实验显示,MethodName在仅使用每层几百个可训练参数的情况下,优于冻结基线的旋转准确率,并表现出部分激活稀疏性(少数单元强激活)。相比LoRA,MethodName以少量精度损失换取更少参数和更清晰的‘哪些神经元激活’机制。我们讨论其局限性,包括当冻结基模型缺乏目标模式所需特征时表达能力下降。

原文摘要 · Abstract (English)

Parameter-efficient fine-tuning (PEFT) methods such as \lora{} adapt large pretrained models by adding small weight-space updates. While effective, weight deltas are hard to interpret mechanistically, and they do not directly expose \emph{which} internal computations are reused versus bypassed for a new task. We explore an alternative view inspired by neuromodulation: adaptation as a change in \emph{mode} -- selecting and rescaling existing computations -- rather than rewriting the underlying weights. We propose \methodname{}, a simple activation-space PEFT technique that freezes base weights and learns per-neuron \emph{thresholds} and \emph{gains}. During training, a smooth gate decides whether a neuron's activation participates; at inference the gate can be hardened to yield explicit conditional computation and neuron-level attributions. As a proof of concept, we study ``mode specialization'' on MNIST (0$^\circ$) versus rotated MNIST (45$^\circ$). We pretrain a small MLP on a 50/50 mixture (foundation), freeze its weights, and then specialize to the rotated mode using \methodname{}. Across seeds, \methodname{} improves rotated accuracy over the frozen baseline while using only a few hundred trainable parameters per layer, and exhibits partial activation sparsity (a minority of units strongly active). Compared to \lora{}, \methodname{} trades some accuracy for substantially fewer trainable parameters and a more interpretable ``which-neurons-fire'' mechanism. We discuss limitations, including reduced expressivity when the frozen base lacks features needed for the target mode.

模型微调可解释性轻量化激活空间

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