Complex KDA: Understanding and Enhancing the Expressivity of Kimi Delta Attention
Complex KDA:理解并增强 Kimi Delta Attention 的表达能力
🧭 线性 RNN 表达能力增强
University of Freiburg · Microsoft Research · University of Tübingen · MPI-IS Tübingen · EPFL
通过扩展门控与 delta 系数范围,提出 CKDA,在不增加秩或成本下实现 2D 旋转并达到 DeltaProduct2 的表达能力
📌 工作参考价值(4 点)
- 对需要轻量序列建模的电商/广告场景(如用户行为序列、点击流预测),CKDA 提供了一种保持状态非膨胀的对角加秩一线性 RNN,计算开销低,可作为长序列用户建模的候选架构。
- 参数范围扩展(门控 [-1,1],beta [0,2])简单且不增加推理成本,可以尝试在已有 KDA 或 delta-rule 模块中直接调整范围,观察对长尾序列或周期行为(如促销周期)的建模提升。
- 文中证明 CKDA 可跟踪 SO(3) 子群,表明其具备更强的状态跟踪能力,适合需要精确维护内部状态的 Agent 或对话系统,可能减少层数或状态维度需求。
- 注意:论文核心是基础架构创新,业务直接迁移价值有限,主要在需要高效长上下文处理且对表达能力有要求的场景(如用户长期兴趣演化)可考虑替换 Transformer 层或作为轻量 backbone。