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

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