稀疏注意力训练加速系统HyScale:不建分数表,1M长上下文训练快2倍

arXiv 上一篇新论文提出长上下文训练系统 HyScale,针对可学习稀疏注意力中“选择”环节的瓶颈重构实现方式,在素材给出的条件下把 1M token 序列训练提速约 1.5–2 倍。值得关注的是,它把稀疏注意力从架构创新推向了“训练系统能不能跟上”的工程问题。

一句话看懂:arXiv 上一篇新论文提出长上下文训练系统 HyScale,针对可学习稀疏注意力中“选择”环节的瓶颈重构实现方式,在素材给出的条件下把 1M token 序列训练提速约 1.5–2 倍。值得关注的是,它把稀疏注意力从架构创新推向了“训练系统能不能跟上”的工程问题。

事件核心:发生了什么

根据 arXiv 2026 年 10 月 26 日的新论文摘要,HyScale 是一个面向可学习稀疏注意力的长上下文训练系统。摘要称,这类架构被视为支撑百万级 token 上下文的重要方向,并且自 DeepSeek-V3.2 起已在生产中被较多采用,但“选择”操作逐渐成为训练瓶颈:每个 query 都要给全部历史 token 打分,而稀疏注意力实际只读取 k 个;现有系统还会物化完整分数表,并在重计算时重复执行选择。

作者的切入点是:选择的结果不是激活值,而是一组索引。基于此,HyScale 引入 HyTopK,在不构建分数表的前提下即时求出每个 query 的精确 top-k;同时用 selection replay 在重计算阶段复用前向选择结果。摘要给出的数据是:在 512 张 H20 GPU 上,GLM-5.1 的 1M token 序列训练相比 Megatron-LM 提速 2.07 倍;在 32 张 H800 上,DeepSeek-V4-Flash 的 CSA 层和 Qwen3.8-Flash-Next 的 QSA 层分别提速 2.01 倍和 1.57 倍。

为什么重要

长上下文竞争过去多集中在注意力架构和位置编码,而随着稀疏注意力进入大模型训练主链路,打分与重计算这类“配角”操作的算力开销开始显性化。HyScale 的思路说明,训练效率有时不取决于更大的模型改动,而取决于是否重新审视中间数据的表示形式——把分数表换成索引,就能绕开显存和重复计算。若该方法可迁移,可能降低百万 token 上下文的训练成本,并影响训练框架与算力调度的优化方向。

对用户/开发者/创作者的影响

目前公开信息仅限于论文摘要,尚未见完整实验与开源实现,因此对普通用户没有直接可感知的产品变化。对训练侧开发者而言,HyTopK 和 selection replay 属于可在稀疏注意力算子与重计算机制中借鉴的工程技巧,尤其适合在 Megatron-LM 类框架上做长序列训练调优;对使用 GLM、DeepSeek、Qwen 系列模型或 API 的团队来说,短期更可能先体现在训练成本和上下文长度的演化上,而不是立刻出现可调用的新功能。

GamsGo AI

AI 工具推荐

想把多个 AI 模型放在一个入口?

GamsGo AI 集成 ChatGPT、DeepSeek、Gemini、Claude、Midjourney、Veo 等常用模型,适合写作、绘图、视频和日常 AI 工作流。

了解 GamsGo AI

推广链接:通过此链接购买,我可能获得佣金,不影响你的价格。

值得关注的后续

一是论文是否公开代码并与主流训练框架集成,决定其能否被社区复用;二是加速比能否在更多模型、更多 GPU 型号和不同 k 值下保持;三是 DeepSeek、智谱、阿里等团队是否在后续模型中采用类似选择机制,从而把长上下文训练的成本曲线继续压低。

来源:arXiv cs.AI

celebrityanime
celebrityanime
文章: 29986

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注