用细粒度学习率提升长序列记忆能力,效果优于现有方法。
FG$^2$-GDN: Enhancing Long-Context Gated Delta Networks with Doubly Fine-Grained Control
- 将学习率从标量改为通道级向量,实现更精细的更新控制
- 在合成与真实数据集上,关联回忆准确率显著高于GDN和KDA
- 适合需要长上下文建模的模型优化,如大语言模型
线性注意力机制因其推理时线性时间复杂度成为softmax注意力的有力替代。近期研究如门控增量网络(GDN)和Kimi增量注意力(KDA)表明,基于在线梯度下降的增量规则,在关联回忆任务上优于简单加法更新。尽管KDA将粗粒度头级衰减门细化为通道级衰减,但增量更新中的学习率β_t仍为标量,限制了维度特异性适应能力。本文提出FG²-GDN,将标量β_t替换为通道级向量,类比于从SGD到AdaGrad、Adam等自适应优化器的演进。进一步提出FG²-GDN+,解耦键与值的缩放,实现对擦除强度与写入强度的独立控制。在合成及真实世界基准测试中,FG²-GDN及其变体在保持相当计算效率的同时,显著提升了关联回忆性能与长上下文理解能力。
原文摘要 · Abstract (English)
Linear attention mechanisms have emerged as promising alternatives to softmax attention, offering linear-time complexity during inference. Recent advances such as Gated DeltaNet (GDN) and Kimi Delta Attention (KDA) have demonstrated that the delta rule, an online gradient descent update, enables superior associative recall compared to simple additive updates. While KDA refined the coarse head-wise decay gate into channel-wise decay, the learning rate $β_t$ in the delta update remains a scalar, limiting the model's capacity for dimension-specific adaptation. We introduce FG$^2$-GDN, which replaces the scalar $β_t$ with a channel-wise vector analogous to the transition from SGD to per-coordinate adaptive optimizers such as AdaGrad and Adam. We further propose FG$^2$-GDN+, which decouples the scaling for keys and values, enabling independent control of erasure strength and write strength. Experiments on synthetic and real-world benchmarks show that FG$^2$-GDN and its variant improve associative recall and long-context understanding over GDN and KDA, with comparable computational efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。