首页 > 计算机科学

刘瑷玮@腾讯微信\u00A0AI研究员:WeDLM:基于因果注意力重构扩散语言模型,实现高效并行推理

机器学习算法与自然语言处理 2026-01-17 00:00
文章摘要
背景:扩散语言模型在生成任务中展现出潜力,但其双向注意力机制导致无法有效利用键值(KV)缓存,限制了推理效率。研究目的:腾讯微信AI研究员刘瑷玮提出的WeDLM框架,旨在通过重构因果注意力,实现扩散语言模型的高效并行推理。结论:WeDLM采用拓扑重排序、双流掩码训练和流式并行解码等创新技术,在Qwen系列模型上继续预训练后,在GSM8K等基准测试中不仅保持了模型能力,还实现了显著的推理加速(最高达10倍以上),成为首个在实际部署中超越自回归推理引擎的扩散语言模型,为扩散式解码与高效缓存结合开辟了新方向。
刘瑷玮@腾讯微信\u00A0AI研究员:WeDLM:基于因果注意力重构扩散语言模型,实现高效并行推理
本站注明稿件来源为其他媒体的文/图等稿件均为转载稿,本站转载出于非商业性的教育和科研之目的,并不意味着赞同其观点或证实其内容的真实性。如转载稿涉及版权等问题,请作者速来电或来函联系。
最新文章
大语言模型的自提升:技术综述与未来展望
大语言模型的自提升:技术综述与未来展望
随着大语言模型(LLMs)的持续演进,单纯依赖人类监督进行模型改进的成本日益高昂,且在可扩展性(Scalabi
2026-04-02
首次!OpenClaw又更新,直接合并QQ官方插件
首次!OpenClaw又更新,直接合并QQ官方插件
任务调度能力进一步完善!
2026-04-02
Book学术官方微信
Book学术文献互助
Book学术文献互助群
群 号:604180095
Book学术
文献互助 智能选刊 最新文献 互助须知 联系我们:info@booksci.cn
Book学术提供免费学术资源搜索服务,方便国内外学者检索中英文文献。致力于提供最便捷和优质的服务体验。
Copyright © 2023 Book学术 All rights reserved.
ghs 京公网安备 11010802042870号 京ICP备2023020795号-1