arXiv:2602.05635cs.LG2026-02

用乘法结构提升模型可编辑性,让神经网络更懂数学规律

Structural Disentanglement in Bilinear MLPs via Architectural Inductive Bias

  • 引入双线性MLP架构,通过乘法交互实现表征解耦
  • 在模运算等任务中恢复真实代数算子,准确率显著提升
  • 适合需要精准可编辑与长期推理的高可靠场景

现代神经网络在选择性遗忘和长程外推任务中仍表现脆弱,即使任务具有潜在代数结构。本文认为,问题根源不仅在于优化或遗忘算法,更在于模型训练过程中内部表征的构建方式。我们探索将显式乘法交互作为架构归纳偏置是否有助于结构解耦,采用双线性MLP进行研究。分析表明,在梯度流条件下,双线性参数化具备‘非混合’特性,使功能组件分离为正交子空间表示,为精确模型修改提供数学基础。通过一系列受控实验验证假设,涵盖模运算、循环推理、李群动力学及定向遗忘基准。相较于点状非线性网络,乘法架构能恢复与底层代数结构一致的真实算子。结果表明,模型可编辑性与泛化能力受限于表征结构,而架构归纳偏置在实现可靠遗忘中起核心作用。

原文摘要 · Abstract (English)

Selective unlearning and long-horizon extrapolation remain fragile in modern neural networks, even when tasks have underlying algebraic structure. In this work, we argue that these failures arise not solely from optimization or unlearning algorithms, but from how models structure their internal representations during training. We explore if having explicit multiplicative interactions as an architectural inductive bias helps in structural disentanglement, through Bilinear MLPs. We show analytically that bilinear parameterizations possess a `non-mixing' property under gradient flow conditions, where functional components separate into orthogonal subspace representations. This provides a mathematical foundation for surgical model modification. We validate this hypothesis through a series of controlled experiments spanning modular arithmetic, cyclic reasoning, Lie group dynamics, and targeted unlearning benchmarks. Unlike pointwise nonlinear networks, multiplicative architectures are able to recover true operators aligned with the underlying algebraic structure. Our results suggest that model editability and generalization are constrained by representational structure, and that architectural inductive bias plays a central role in enabling reliable unlearning.

模型可编辑性代数结构双线性网络解耦表征

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