arXiv:2605.02323cs.LGcs.AI2026-05

解决注意力模型在叠加信号中无法区分多个成分的问题

When Attention Collapses: Residual Evidence Modeling for Compositional Inference

论文配图:When Attention Collapses: Residual Evidence Modeling for Compositional Inference
图 1 · 摘自论文原文
  • 引入残差证据建模,通过消除已解释信息避免注意力重复聚焦
  • 在合成数据和真实音频混合中,槽位坍缩减少一个数量级
  • 适合需要分离多重信号的科学数据分析场景

组合推理——将观测分解为未知数量的潜在成分——是感知与科学数据分析的核心。基于注意力的模型在成分近似可分时表现良好,如物体中心视觉任务。但在加性叠加情形下(多个成分共同影响每次观测),我们发现一种结构性失败模式:槽位坍缩——多个注意力槽收敛至同一主导成分,而较弱成分未被表示。其根源在于注意力对已解释证据缺乏记忆:所有槽反复作用于相同输入,未考虑已有解释,导致梯度被最强成分主导,引发槽间共享不动点。因此注意力无法在加性叠加下实现非冗余分配。我们提出残差证据建模,通过证据耗尽(结合乘法耗尽与注意力偏置)实现。消融实验表明,平行注意力、顺序处理或损失正则化均无效;仅证据耗尽(在顺序注意力中加入残差状态)持续有效。在合成基准与真实音频混合(FUSS)中,证据耗尽使槽位坍缩减少达一个数量级,且泛化能力超越合成环境。在欧空局/美国宇航局LISA任务的引力波源推断中,相同架构、数据与损失下,标准注意力失效,而证据耗尽防止坍缩并实现多源后验估计。结果表明,在加性叠加下,追踪残差证据是防止坍缩、实现组合推理的关键。

原文摘要 · Abstract (English)

Compositional inference - the decomposition of observations into an unknown number of latent components - is central to perception and scientific data analysis. Attention-based models perform well when components are approximately separable, as in object-centric vision. Under additive superposition, however - where multiple components contribute to every observation - we identify a structural failure mode we term slot collapse: multiple slots converge to the same dominant component while weaker ones remain unrepresented. We trace this to a general limitation: attention is memoryless with respect to explained evidence. All slots repeatedly operate on the same input without accounting for what has already been explained, so gradients are dominated by the strongest component, inducing shared fixed points across slots. As a result, attention fails to enforce non-redundant allocation under additive superposition. We address this by introducing residual evidence modeling, instantiated via evidence depletion - a minimal modification combining multiplicative depletion with an attention bias. Controlled ablations show that parallel attention, sequential processing alone, and loss-based regularization fail to resolve collapse; evidence depletion, which adds residual state to sequential attention, consistently succeeds. Across synthetic benchmarks and real-world audio mixtures (FUSS), evidence depletion reduces slot collapse by up to an order of magnitude, generalizing beyond synthetic settings. On gravitational-wave source inference for the ESA/NASA LISA mission, under identical architectures, data, and losses, standard attention fails while evidence depletion prevents collapse and enables multi-source posterior estimation. These results show that under additive superposition, residual evidence tracking is the operative ingredient for preventing collapse and enabling compositional inference.

注意力机制组合推理信号分离残差建模

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