首页 游戏 软件 资讯 排行榜 专题
首页
AI
阿里Qwen3 RTPurbo:15%全量Attention长文本5倍压缩方案

阿里Qwen3 RTPurbo:15%全量Attention长文本5倍压缩方案

热心网友
86
转载
2025-12-23



机器之心发布

为什么大模型厂商给了 128K 的上下文窗口,却在计费上让长文本显著更贵?

为什么 Claude 能 “吞下整本书”,但最新示例往往只展示几千字的文档?

为什么所有大模型厂商都在卷 “更长上下文”,而真正做落地的产品经理却天天琢磨 “怎么把用户输入变短”?

这些看似矛盾的现象,其实答案藏在一个长期被技术光环遮掩的真相里:

长序列,正在成为大模型应用里最昂贵的奢侈品

在当前主流的 Full Attention 机制下,计算开销会随着输入长度平方增长,序列一长,处理就变得 “又贵又慢”(见图 1)。针对这一核心难题,阿里 RTP-LLM 团队提出了一种全新的后训练压缩方案:RTPurbo。在不损失模型效果的前提下,实现了 Attention 计算5 倍压缩(见图 2)。



左图 1:长序列 Attention 计算成本瓶颈;右图 2:RTPurbo 极大降低 Attention 计算开销

总的来说, RTPurbo 采用了一种非侵入式的压缩方法:通过分辨 LLM 内部的长程 Attention Head,仅保留关键 Head 的全局信息,对于剩下冗余的 Head 直接丢弃远程 Tokens。这种 Headwise 级别的混合算法以其简洁的方案设计和优越的算子兼容性,极大地降低了大模型在长序列下的推理代价,为新一代 LLM 结构设计提供了一个新的视角和演进方向。

目前,项目模型与推理代码已经发布至 Huggingface、ModelScope 平台,感兴趣的读者可以阅读 RTP-LLM 相应的技术 blog 了解更多细节。

https://huggingface.co/RTP-LLM/Qwen3-Coder-30B-A3B-Instruct-RTPurbohttps://modelscope.cn/models/RTP-LLM/Qwen3-Coder-30B-A3B-Instruct-RTPurbo



图 3:RTPurbo 采用混合压缩方案,仅有少数 Attention Head 使用全量 Attention)

化繁为简,被低估的 SWA



但在真实落地中,这两条路线都有较明显的共性代价:一方面,它们通常强依赖大量后训练,工程实现与适配成本也更高;另一方面,Linear Attention 在压缩信息后,长序列下的召回能力显著弱于 Full Attention [1],因此往往需要与 Full Attention 混合使用,带来性能与加速收益的双重上限。此外,Linear / Sparse Attention 的算子与调度设计相对复杂,也进一步影响其在工程生态中的通用性、可维护性与一致性。也正因如此,一些前期工作 [2] 反而把目光投向看似 “简单粗暴” 的 Sliding Window Attention(SWA),例如 gpt-oss 和 MiMo ,这在一定程度上说明 SWA 并非 “权宜之计”,而是一种可规模化复用的工程选择。

在进一步分析现有 LLM 的注意力模式后,团队观察到一个更细粒度的关键现象:绝大多数 Attention Head 天然更偏好局部信息,只有极少数 “长程头” 能在长文本中稳定地定位并召回关键关联信息。基于这一现象,团队提出了一个关键假设:

类似人类的阅读与写作过程,LLM 在处理长文本时,往往会先从长上下文中 “召回” 相关信息,再在相对局部的上下文范围内完成推理并输出答案

换句话说,模型内部可能只有少量 Attention Head 真正在承担 “信息召回” 的职责:它们通过注意力机制把远距离信息搬运到当前的 Residual Stream(信息通道)中,让后续层可以在更局部、更高信噪比的状态下完成推理。这也直接意味着:对于那些并不承担长程依赖的 Attention Head,其实并不需要使用 Full Attention—— 长序列场景存在可观的压缩空间。

为了验证这一假设,团队设计了一个直观的对比试验:

方案一:只有 15% 的长程头使用 Full Attention,剩余 85% 使用 SWA;方案二:15% 的长程头使用 SWA,剩余 85% 使用 Full Attention。

如表 1 所示,尽管方案二保留了 85% 的 KV cache,但是其长文能力仍然显著劣于方案一。



表 1:方案一只用 15% 的 Full Attention,长文效果显著优于方案二

进一步地,在不做任何微调的情况下,方案一在长文本指标上也非常有竞争力(表 2),几乎无损:



表 2:方案一不经过训练,在 Ruler 上无损

不过,在某些特定的长文任务上,未经微调的压缩模型仍会出现明显的性能退化(见表 3)。其根源在于:压缩前后注意力模式的直接切换会对模型输出造成一定扰动,需要额外训练来 “消化” 这种变化。



表 3:方案一在特殊 benchmark 上仍然存在显著负向

因此,为实现更接近 “无损” 的压缩,团队进一步提出了一个面向 RL 后模型的压缩训练范式:在不依赖高质量标注数据的前提下,仅通过轻量级微调,就能显著提升压缩后模型在长文任务上的表现。

自蒸馏,从根本上解决数据问题

当前主流 LLM 通常采用 “预训练 + 后训练 + RL” 的训练范式,如果直接使用长文 SFT / 预训练语料进行续训,会带来两方面挑战:

RL 后模型在经过 SFT 会出现过拟合甚至灾难性遗忘,损伤短文本任务上的原有能力(见表 4);高质量的长文本语料难以获取。



表 4:Qwen3-30B-A3B-Instruct RL 后模型继续 SFT 会过拟合,造成灾难性遗忘

为解决这两点,RTPurbo 使用 “模型自蒸馏” 作为关键训练策略:让压缩后的模型对齐原模型输出,从而同时化解数据与能力保留问题:

仅对模型自身的输出进行对齐,避免依赖特定领域的问答数据,从而确保短文本下游指标基本无损;只需使用长文本预训练语料即可完成训练,使模型快速适应 headwise 稀疏的工作模式。

实测中,仅使用约 1 万条 32k 长度的预训练语料(训练时间小时级),RTPurbo 就能让长文任务表现与原模型持平。

结果对比

在长文本测试场景下,RTPurbo 仅保留约 15% 的 Attention Heads 使用 Full KV cache,压缩后的 Qwen-Coder-Plus、Qwen3-30B-A3B-Instruct 在多项长文指标上可与未压缩模型齐平,充分验证了压缩后模型的精度保障。



更重要的是,这种压缩并非以牺牲通用能力为代价。在多项短文本(通用)Benchmark 上,采用自蒸馏范式训练后的模型并未出现性能衰减,原有对话、推理和代码理解等能力都得到了良好保留。



这表明,RTPurbo 不仅是一种单一模型的 “特定优化技巧”,而是一套具有良好可迁移性和通用性的长序列加速方案,可为更大规模、更多架构的 LLM 提供高性价比的推理加速路径。

从大模型可解释性到 LLM 压缩

早期可解释性工作 [3] 已指出:模型内部存在很强的 “召回” 机制,一部分特定 Attention Head 能稳定定位前文相关信息。团队成员的前期工作 [2] 也观察到这些 Head 在长文场景仍保持类似行为。

与此同时,在 [4] 中,作者指出 Softmax 本身在长序列存在熵增的问题。更具体的,随着序列变长,每个 Token 的注意力不可避免的变得更加弥散(信噪比降低),如下图所示:



图 4:Attention 在长序列下存在信噪比下降的问题

因此,为了避免远程信息干扰模型本身的推理能力,LLM 内部实现了一种非常巧妙的机制:

多数 Head 只处理局部信息,以获得更高信噪比;少数 Head 负责从远处 “召回” 关键信息并搬运到当前位置,使后续层能在局部范围内完成推理。

这与 RTPurbo 的 headwise 设计高度一致:把 “全局召回” 能力集中保留给少量关键 Head,其余 Head 则用工程收益更稳定的 SWA 来承载。

RTP-LLM:RTPurbo 在长文上的极致性能优化



图 5:RTPurbo HeadWise Attention 性能加速结果,图上结果仅使用 15% 的 Full Attention

RTPurbo 按固定比例划分 SWA Head 与 Full Head 虽然直观有效,但工程上必须解决一个问题:不同 Head 计算模式与计算量不一致,会导致负载不均衡,影响 GPU 并行效率与端到端吞吐。

为此,RTP-LLM 围绕该不均衡在算子层与框架层做了针对性优化,核心包括:

Full Attention Head 的 PTX 级优化:对仍需全量计算的 Full Head 深入 PTX 指令层,利用 gmma::mma_async_shmA 等异步拷贝与矩阵乘指令提升效率;融合 IO warps 与 P/V 计算阶段,优化 Ping-Pong 流水与调度,减少空转等待。稀疏度感知的负载均衡调度:针对 Tail Latency,采用稀疏度感知动态调度(如反向拓扑排序),优先分配重 tile 给 SM,使各 SM 更同步完成任务,降低尾延迟、提升吞吐。SWA 的高效实现:避免传统 SWA 常见的 “三段式 KV 拼接” 或 “Custom Mask” 做法(访存与调度开销大),通过重塑数据布局与计算路径减少冗余访存与额外算子开销。用 CP(Context Parallel)替代 TP(Tensor Parallel):在 headwise 稀疏场景下,TP 易导致算力利用率低且不够灵活;采用 CP 让单卡完成全部 head 的 attention 计算,提高 GPU 利用率,并通过计算 - 通信重叠降低通信开销。

综合以上优化,RTP-LLM 能将 Attention 稀疏带来的理论收益稳定、可复现地转化为端到端加速;在 256k 长序列下实现单算子最高 9× 加速(见图 5,图中仅 15% Head 使用 Full Attention)。

团队介绍

RTP-LLM 是阿里巴巴智能引擎团队自研的高性能大模型推理引擎,支持了淘宝、天猫、高德、饿了么等核心业务的大模型推理需求。智能引擎源自阿里巴巴搜索、推荐和广告技术,是阿里 AI 工程领域的先行者和深耕者。团队专注于 AI 工程系统的建设,主导建立了大数据 AI 工程体系 AI・OS,持续为阿里集团各业务提供高质量的 AI 工程服务。

RTP-LLM 项目已开源,欢迎交流共建: https://github.com/alibaba/rtp-llm

参考文献:

[1]: Repeat After Me:Transformers are Better than State Space Models at Copying.

[2]: RazorAttention: Efficient KV Cache Compression Through Retrieval Heads

[3]: In-context Learning and Induction Heads

[4]: 苏建林,“注意力机制真的可以集中注意力吗?”,https://www.spaces.ac.cn/archives/9889

来源:https://www.163.com/dy/article/KHFD1GM40511AQHO.html
免责声明: 游乐网为非赢利性网站,所展示的游戏/软件/文章内容均来自于互联网或第三方用户上传分享,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系youleyoucom@outlook.com。

相关攻略

许哲诚计算性设计展演评析:数字逻辑与物质建构的生成境域
科技数码
许哲诚计算性设计展演评析:数字逻辑与物质建构的生成境域

数字逻辑与物质建构的深度对话 ——评许哲诚“境域·生成”计算性设计展演 □ 丁雅力(江苏省美术馆策展人) 当代设计与造物的核心范式,正经历着由计算性设计带来的深刻变革。2026年3月20日,南京艺术学院教师许哲诚于南京莫玄空间呈现的“境域·生成”个人专场展演,正是这一前沿趋势的集中体现。本次展览超越

热心网友
05.18
具身智能研发框架Dexbotic重塑机器人开发流程
AI
具身智能研发框架Dexbotic重塑机器人开发流程

近日,开源具身智能原生框架Dexbotic宣布正式支持以RLinf作为其分布式强化学习后端。对具身智能开发者而言,这不仅是一次普通的工程适配,更意味着VLA模型研发中长期存在的「SFT与RL割裂」问题,正在被真正打通。 这是一种典型的「乐高式协作」:双方不强行Fork、不粗暴揉合代码,而是保持清晰边

热心网友
05.13
RMS-MoE模型通过检索记忆优化专家路由调度效率
AI
RMS-MoE模型通过检索记忆优化专家路由调度效率

随着大模型参数规模不断增长,混合专家(Mixture-of-Experts, MoE)架构因其稀疏激活特性,成为平衡模型性能与计算开销的主流方案。然而,在实际的Web级应用部署中,一个关键挑战日益凸显:传统MoE的路由机制通常是“无记忆”的。 试想,在搜索引擎、智能问答或多轮对话等高并发场景下,用户

热心网友
05.12
编程入门指南从零基础到理解核心概念
编程语言
编程入门指南从零基础到理解核心概念

编程十年的一点分享 在软件开发的路上走过十几年,从一个爱好者到以此为业,有些体会或许值得聊聊,就当是抛砖引玉吧。 最早接触编程,是从BASIC和C语言开始的。工作后,随着需要,陆续学习了dBase、Access这类桌面数据库的开发。真正以开发为职业,可以说始于FoxPro 5 0,之后技术栈随着项目

热心网友
05.07
编程初学者入门指南与核心思维解析
编程语言
编程初学者入门指南与核心思维解析

引言 编程,是一门实践科学。这意味着,学习它的最佳方式就是动手去敲代码。但这是否意味着,我们可以因此轻视理论的学习呢? 入门编程 如果你去各大技术社区提问“该如何入门编程”,五花八门的答案会瞬间涌来。 不过,仔细梳理一下,无外乎以下几种流派: 学院派 他们推荐从C语言入手,并辅以数据结构、操作系统等

热心网友
05.07

最新APP

宝宝过生日
宝宝过生日
应用辅助 04-07
台球世界
台球世界
体育竞技 04-07
解绳子
解绳子
休闲益智 04-07
骑兵冲突
骑兵冲突
棋牌策略 04-07
三国真龙传
三国真龙传
角色扮演 04-07

热门推荐

全链网解析:欧洲多国领导人面临支持率挑战
web3.0
全链网解析:欧洲多国领导人面临支持率挑战

英国工党领袖斯塔默面临公众信任挑战,支持率低迷。类似困境在欧洲多国领导人中普遍存在,德国总理默茨与法国总统马克龙的支持率同样远低于不支持率,反映出欧洲政界广泛的信任危机。

热心网友
05.18
芝麻开门Gate.io官网入口网址及安全交易平台使用指南
web3.0
芝麻开门Gate.io官网入口网址及安全交易平台使用指南

芝麻开门:安全便捷的数字资产交易平台 在数字货币的世界里,选择一个可靠、便捷的交易入口是第一步。芝麻开门作为一款服务于全球用户的知名交易平台,以其多重安全防护、对主流币种的广泛支持以及现货、杠杆等丰富功能,成为了许多交易者的选择。今天,我们就来详细梳理一下如何通过官方渠道,安全地获取并使用芝麻开门平

热心网友
05.18
通胀担忧加剧全球债市抛售 长期美债收益率创近三年新高
web3.0
通胀担忧加剧全球债市抛售 长期美债收益率创近三年新高

全球债市因通胀担忧遭剧烈抛售,长期美债收益率升至近三年高位。30年期美债收益率一度突破5%,10年期与2年期收益率同步攀升。日本30年期国债收益率单日飙升20基点创新高。油价上涨加剧通胀忧虑,策略师建议关注美债收益率在5 25%-5 5%区间的后续动向。

热心网友
05.18
欧易交易所官方入口网址一键直达 安全登录OKX交易平台
web3.0
欧易交易所官方入口网址一键直达 安全登录OKX交易平台

欧易(OKX):您的官方数字资产交易入口 在加密货币的世界里,选择一个可靠、功能全面的交易平台是第一步。欧易(OKX)作为全球领先的数字资产服务商,早已成为数百万用户的首选。它不仅提供比特币、以太坊等主流币种的现货交易,更将业务延伸至衍生品、DeFi以及NFT市场,构建了一个完整的加密生态。其背后,

热心网友
05.18
Gate.io官方安卓交易APP v7.19.1最新版下载与使用指南
web3.0
Gate.io官方安卓交易APP v7.19.1最新版下载与使用指南

gate io交易APP官方版 v7 19 1 安卓版下载与安装全指南 对于数字资产交易者来说,一个可靠、顺手的交易工具至关重要。Gate io交易APP,正是这样一款专业的平台,它为全球用户提供比特币、以太坊乃至上千种加密货币的实时行情与交易服务。其最新的安卓v7 19 1版本,在用户体验和系统稳

热心网友
05.18