跳到正文
Apple Machine Learning Research·· 3 天前AI 评分45

门控机制的重要性:SSM 中的记忆与上下文学习之争

On the Importance of Gating: Memorization vs. In-Context Learning in State Space Models

AI 导读

Apple 与哈佛大学的研究人员通过理论与实验分析,门控机制导致 SSM 在训练中先学到“权重内记忆”方案,从而延迟甚至阻止其收敛到正确的上下文学习解,即便架构和记忆容量没有根本限制。SSM 以固定内存和线性计算复杂度著称,但在需要上下文学习和精确检索的任务上仍落后于 Transformer。研究同时发现门控有利于提升对长序列的泛化能力,为理解和改进线性时间模型提供了基础。

来源:Apple Machine Learning Research · machinelearning.apple.com