arXiv:2605.05495cs.LG2026-05

Transformer模型在持续学习中易陷入捷径解,影响泛化能力。

Shortcut Solutions Learned by Transformers Impair Continual Compositional Reasoning

论文配图:Shortcut Solutions Learned by Transformers Impair Continual Compositional Reasoning
图 1 · 摘自论文原文
  • 用持续学习版LEGO框架测试Transformer模型的组合推理能力
  • BERT因捷径解导致泛化差,而ALBERT表现更优
  • 循环结构的ALBERT更适合持续学习,适合研究模型架构与学习机制

识别跨领域的共性特征是人类类比推理的核心,也被认为对持续学习至关重要。为实现这一目标,需发展通用且灵活的计算策略。尽管近年来对Transformer模型进行组合推理的能力已有广泛研究,但其如何利用表征来学习新相关经验仍缺乏系统探讨。为此,本文将原有的学习等式与组操作(LEGO)框架扩展至持续学习(CL)场景(即“持续LEGO”)。通过该实验范式,我们研究了前馈和循环Transformer模型在持续学习中的表现。结果发现,作为典型前馈Transformer的BERT,会学习到限制泛化的捷径解;而相比之下,循环版本的ALBERT展现出类似For循环的解法,带来更好的持续学习性能。当应用于需要跨经验组合的任务时,两类模型均失败。研究还表明,通过整合不同经验的数据训练可提升ALBERT的表现,但对BERT无效——其捷径解在初始训练后已固化。结果表明,循环结构的ALBERT可能具有更适配持续学习的归纳偏置,提示未来应深入探究Transformer架构与现代任务中涌现计算解之间的关系。

原文摘要 · Abstract (English)

Identifying and exploiting common features across domains is at the heart of the human ability to make analogies, and is believed to be crucial for the ability to continually learn. To do this successfully, general and flexible computational strategies must be developed. While the extent to which Transformer neural network models can perform compositional reasoning has been the subject of intensive recent investigation, little work has been done to systematically understand how well these models can leverage their representations to learn new, related experiences. To address this gap, we expand the previously developed Learning Equality and Group Operations (LEGO) framework to a continual learning (CL) setting ("continual LEGO"). Using this continual LEGO experimental paradigm, we study the capability of feedforward and recurrent Transformer models to perform CL. We find that BERT, a canonical feedforward Transformer model, learns shortcut solutions that limits its ability to generalize and prevents strong forward transfer to new experiences. In contrast, we find evidence supporting the hypothesis that ALBERT, a recurrent version of BERT, learns a For loop-esque solution, which leads to better CL performance. When applying BERT and ALBERT models to a CL setting that requires composition across experiences, we find that both model families fail. Our investigation suggests that ALBERT models can have their performance drop rescued by use of training strategies that combine data across experiences, but this is not true for BERT models, where a detrimental shortcut solution becomes entrenched with initial training. Our results demonstrate that the recurrent ALBERT model may have an inductive bias better suited for CL and motivate future investigation of the interplay between Transformer architecture and computational solutions that emerge in modern models and tasks.

Transformer持续学习组合推理模型偏差

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