首达注意力机制(First Passage Attention)
🧭 注意力机制 · 选择性上下文擦除
University of Bergen
用路径极值替代累加衰减,实现选择性上下文擦除并压缩 KV cache 到 25% 仍保持长程检索
📌 工作参考价值(4 点)
- 长序列用户行为建模:FPA 的“路径极值”允许选择性擦除中间事件(如误触、临时会话)而保留早期长期偏好,适合电商用户序列编码。
- 多轮 Agent/对话上下文管理:可借鉴其栈式存活 key 机制,在 RAG 长对话中自动丢弃已撤销指令或中间推理,降低噪声并压缩 KV cache。
- 工程优化:FPA 在约 25% KV cache 压缩下保持 8 倍训练长度 passkey 检索,可直接用于长上下文推理降本,对搜索推荐 LLM 服务有实用价值。
- 可尝试将 FPA 头部作为用户序列编码器,替代 additive forget gate,提高对层级事件(加入/删除购物车、订单状态变更)的敏感度。
预训练中量化注意力 Softmax:校准与直通梯度设计的影响
🧭 低精度注意力 Softmax 训练 ★ 深度解读
University of Illinois Urbana-Champaign
系统研究量化 softmax 前向与反向规则对预训练稳定性的影响,给出可复现的设计选择
📌 工作参考价值(4 点)
- 工业推荐/Agent 里若对 Transformer 做 QAT 或量化 softmax 部署,不要用 .detach() 隔离 calibration 统计量(row max/min);必须让 row extrema 反向传播,否则训练前 25-30M tokens 看似正常,之后 loss 突然恶化,末尾可差 0.65-3.07 nats。零和投影不能修复。
- 做 hard rounding 量化时,优先 Prob-STE(在归一化后概率上放 STE),或校准从 MinMax 换成 fixed window to max(FWM)。在 K=4、124M/2.5B tokens 下 MinMax+Weight-STE 差 softmax 0.89 nats,而 FWM+Prob-STE 只差 0.019;K=16 时 FWM+Weight-STE 差 0.004。
- 如果允许可微实现,优先 LERP(分段线性插值)而非 Nearest hard rounding;K=4/16/32 在 2.5B tokens 都能收敛到 softmax 0.005 nats 以内。
- 评估时必须用训练时的 native forward;训练后替换成 exact softmax 会使 gap 扩大,LERP K=2 从 +0.005 跳到 +1.814 nats,说明权重已经依赖训练算子。
弃权与噪声过滤:Softmax 注意力的两个缺失原语
🧭 注意力机制原语 · 模型训练改进
St. John Fisher University
显式加入弃权(sink logit)与值门控噪声过滤,两者结合在10M-350M匹配模型中均优于softmax基线
📌 工作参考价值(4 点)
- 若在电商搜索/推荐中自研或微调 Transformer 排序模型(如 LLM4Rec、行为序列建模),可直接在 attention 中加入 per-head sink logit 实现 abstention,几乎零参数且兼容 KV cache,适合长序列中忽略无关历史行为。
- 在用户行为序列包含大量噪声(点击噪声、误触、跨域行为)时,对 value 侧加可学习 gate 过滤 residual stream 中的叠加特征干扰;模型规模越大,该噪声过滤收益越明显,10M→350M 趋势提示大规模 CTR/CVR 模型优先考虑 value gating。
- 两种原语有互补盲区,建议同时加入而非只选一种;在 Agent 多步推理或 RAG 检索上下文拼接时,attention 可能被迫把注意力质量分配给无关 token,sink logit 可以让头显式弃权,减少错误信息聚合。
- 若用 LoRA 微调已有 LLM 推荐/Agent 模型,这些改动不增加缓存开销,可作为结构正则化 trick 加在 attention 子层,注意 warmup 和初始门控设置。
KV Cache 淘汰的概率视角:期望估计与解码期校正
🧭 LLM 推理 · KV cache 压缩 ★ 深度解读
University of California, Los Angeles
将 KV cache 淘汰形式化为期望估计问题,提出可纠偏的概率淘汰与解码期校正
📌 工作参考价值(4 点)
- 部署长上下文 LLM 服务(Agent 记忆、商品描述、用户行为序列)时,现有 H2O/SnapKV 等确定式 top-k 淘汰在特定任务上可能灾难性失效;改用论文的 probabilistic eviction + decode-time correction 能提升多任务鲁棒性,适合同一模型同时服务检索、问答、推荐解释等场景。
- 工程迁移成本低:可把已有 score-based 淘汰策略改造成 proposal 分布(如 πH2O=normalize H2O scores),按 head/group 采样并记录 counts,解码时用 Algorithm 1 的自归一化重要性采样校正,改动集中在 attention 计算,无需重训或改模型。
- 温度缩放 τ 是简单可用的 bias-variance 旋钮:高压缩、低采样数时增大 τ 抑制方差,低压缩时保持 τ=1 降低偏差;可按业务 latency/quality 目标做 per-head 或全局 budget 调优。
- 全局压缩率分配可利用论文的采样数与 coupon collector 关系做预算分配,让不同层/head 自适应获得不同压缩率;GQA 模型需用 group 内 mixture proposal 保持支持集一致。
ALiBi 位置编码的浮点下溢失明问题与缓解策略
🧭 ALiBi 位置编码数值缺陷分析与缓解
Leipzig University (InfAI) · ScaDS.AI Dresden/Leipzig · Seltz · University of Kassel · hessian.AI
发现 ALiBi 的线性偏置在长距离下因浮点下溢导致注意力权重为零,提出 log 缩放距离等缓解方法
📌 工作参考价值(4 点)
- 若在电商搜索推荐等场景使用基于 ALiBi 的 decoder-only LLM(如生成 query、长用户序列建模),需注意长距离 token 的注意力权重可能因浮点下溢变为零,导致头部“失明”,影响长文本检索能力。可通过监控 attention underflow 比例来诊断。
- 训练时可改用 log-scaled distances 作为位置偏置,该方法在 passkey retrieval 任务中表现最稳定,能显著缓解下溢;也可结合 clamping 或混合偏置。
- 评估长上下文模型时,不能仅看常规困惑度或标准基准,需加入 needle-in-a-haystack 类 token 检索任务,避免 ALiBi 的隐性缺陷影响线上长序列性能。
- 若必须沿用默认 ALiBi 斜率,应意识到其作为强基线在 needle-in-a-haystack 任务上仍有竞争力,但需权衡训练稳定性与检索能力损失。
有限显存下的长上下文微调:结合分层全局注意力与分段反向传播
🧭 长文本微调显存优化
BMW Group
用分层全局注意力、分段反向传播和分级KV缓存,在16GB显卡上将可训练上下文长度从2K扩展到16K,评估时支持131K tokens
📌 工作参考价值(4 点)
- 在电商推荐场景中,用户长期行为序列或长商品描述经常超过2K tokens,本方法可直接用于消费级GPU(16GB)上微调长上下文模型,大幅降低硬件门槛。
- 分段反向传播与历史KV转存RAM/NVMe的策略,可以借鉴到其它长序列训练中(如序列推荐模型),将不可微历史状态移出显存,只保留当前段可训练,节省显存。
- HGA通过为每个查询块加载有限的历史token,保持每个token的关注量近似恒定,训练速度随上下文长度线性增长,而密集注意力是二次增长,这一特性可在在线学习或增量训练Agent时提升效率。
- 训练得到的适配器可与标准密集注意力生成框架兼容,这意味着可以将高效的HGA训练与成熟的推理引擎结合,在保证模型质量的同时让推理也受益于长上下文。
信息感知的 KV 缓存压缩用于长推理
🧭 LLM 推理优化 · 熵感知 KV 压缩
Shanghai Jiao Tong University · University of Edinburgh
提出熵感知的 KV 缓存压缩框架 InfoKV,结合预测不确定性与注意力得分,提升长推理效率。
📌 工作参考价值(4 点)
- 长对话/文档推理的部署加速:在电商搜索推荐系统的多轮助手、售后长文档分析等场景,可直接用 InfoKV 压缩 KV 缓存,减少显存占用与解码延迟,无需重新训练模型。
- Agent 长期记忆的高效管理:Agent 在维护对话历史或用户行为序列时,利用预测不确定性筛选对远期推理关键的信息 token,既保留关键上下文又控制缓存大小,提升持续交互的响应速度。
- 混合重要性评估范式:将 token 级熵信号与传统的注意力权重结合,是一种低成本、即插即用的重要性评分增强方式,可迁移到其它需要上下文剪枝的 LLM 应用中(如 RAG 中的长文档分块选择)。
- 解码阶段动态压缩:InfoKV 在解码阶段同样有效,适合流式生成时不断增长的 KV 缓存管理,可降低线上长时生成过程的尾部延迟。