Gated Differentiable Working Memory for Long-Context Language Modeling
Lingrui Mei · Shenghua Liu · Yiwei Wang · Yuyao Ge · Baolong Bi · Jiayu Yao · Jun Wan · Ziling Yin · Jiafeng Guo · Xueqi Cheng
摘要
Long contexts break transformers: attention scores dilute across thousands of tokens, critical information gets lost in the middle, and the model cannot adapt to novel patterns at inference time. Recent work on test-time adaptation addresses this by maintaining a form of working memory—transient...