AI导航

AWS提出Priming方法:旧Transformer改混合架构,成本降99%

AI资讯
2 min read
0 次阅读

Priming 方法在 2026 年 10 月 12 日由亚马逊 AWS 团队通过 Amazon Science 博客公布:它能把已经预训练好的 Transformer 模型改造成混合状态空间架构,不需要从头训练,预训练成本可降低 99% 以上,全流程消耗的 token 预算不到原模型的 0.5%。团队同时开源了 8B 和 32B 两个规模的改造模型、训练代码和 vLLM 服务插件,采用 Apache 2.0 许可。

它解决的问题:换个架构就要重烧一遍训练预算

Transformer 的注意力机制会逐字保存全部历史,上下文越长,KV 缓存的内存开销越大;状态空间模型则把历史压缩进一个固定大小的状态,省内存但表达方式不同。业界早就在试把两者混合的架构,可每试一种组合,都要拿数万亿 token、数十万 GPU 小时从头预训练,对比实验贵到没人做得起。Priming 的出发点是:Transformer 本身可以看作状态空间实现的一个特例,部分注意力层的行为能被固定大小的状态近似,于是这些层的权重可以直接迁移过去复用,知识并不锁死在原架构里。

三步改造:先选层,再迁移,最后整体适配

流程分三步。第一步选层:用滑动窗口注意力做廉价探针,替换某层后精度掉得少的,说明它主要处理局部信息,适合换成状态空间层。第二步迁移:把选中层的查询、键、值投影权重拿来初始化状态空间层,再用一段短蒸馏让新层对齐旧层的行为。第三步整体适配:在下一词预测任务上微调全模型,让保留的注意力层和新状态层学会配合。这一流程在 Qwen、Llama、Mistral 等不同家族,稠密和 MoE 两种类型,8B 到 32B 及以上规模上都跑通了。

实验结论:GKA 最好,32B 混合模型解码最高约 2 倍

团队在同等数据和预算下对比了三种状态空间层:Gated KalmaNet(GKA)最好,Gated DeltaNet 次之,Mamba-2 再次,且状态空间层普遍优于拿到两倍内存预算的滑动窗口注意力。32B 的 GKA 混合模型在推理基准上与原 Transformer 基本持平,KV 缓存约减半,在 128K 上下文时解码吞吐最高约 2.06 倍;在 AIME 2025 数学基准上,它达到同一精度门槛的速度快 1.2 到 1.4 倍,最难的题目上快 1.6 倍。GKA 还有一个部署旋钮:推理时把迭代次数从 30 降到 10,只损失约 1% 精度、解码快约 10%,同一模型可以切出不同的延迟档位。需要注意的是,这些数字来自团队自家论文的受控实验,改造收益在其他任务分布上能否保持,还要等社区在自有负载上复现。

推荐工具

NVIDIA Chat with RTX AI聊天 Chat with RTX 是 NVIDIA 面向 RTX 电脑的本地 AI 聊天工具,可围绕本地文档和视频资料做问答,适合重视隐私、离线检索并具备硬件条件的用户更适合资料不便上传云 HuggingChat AI聊天 HuggingChat 是 Hugging Face 的开源模型聊天应用,支持 Omni 自动选模型,也可手动选社区开放模型对话。它适合体验开源模型、技术探索和问答,结果可能不稳定,重要内容需复核。 纳米AI搜索 AI搜索 纳米AI 是 360 旗下 AI 搜索和智能体入口,支持文字、语音、拍照提问、多模型协作与内容创作,适合中文用户做日常搜索、学习问答、移动查询、热点追踪、生活决策、知识整理和轻量创作。 Qwen Chat(通义千问) AI聊天 Qwen Chat 是Qwen 大模型聊天助手,支持Qwen 模型聊天、写作和编程问答,适合中文用户和开发者完成 AI 对话、资料问答和任务协作,适合上线前核对权限、成本和资料质量。 DuckDuckGo AI Chat AI聊天 DuckDuckGo AI Chat 适合重视隐私的普通用户使用,围绕隐私 AI 聊天、免账户使用和多模型问答提供对话式 AI 支持和任务协作能力,适合上线前核对权限、成本和资料质量。 Goody-2 AI聊天 Goody-2 是以极端安全和伦理对齐为特色的 AI 聊天助手,常用于观察模型拒答边界、内容安全策略、保守输出体验和 AI 风险讨论,更适合作为安全边界样本使用前应理解它偏实验性质,不适合。