arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2506.19935cs.LGcs.CVstat.ML

任意阶GPT作为掩码扩散模型:解耦公式与架构

Any-Order GPT as Masked Diffusion Model: Decoupling Formulation and Architecture

University of Chinese Academy of Sciences(中国科学院大学);Academy of Mathematics and Systems Science, Chinese Academy of Sciences(中国科学院数学与系统科学研究院);School of Mathematical Sciences, Peking University(北京大学数学科学学院);National University of Singapore(新加坡国立大学);Huawei Noah’s Ark Lab(华为诺亚实验室)

Shuchen Xue, Tianyu Xie, Tianyang Hu, Zijin Feng, Jiacheng Sun, Kenji Kawaguchi, Zhenguo Li, Zhi-Ming Ma

AI总结:

本研究通过在仅解码器框架内评估掩码扩散模型(MDMs),解耦算法与架构因素,对比MDM与自回归范式的权衡,实现约25倍推理加速,为高效基础模型开发提供见解。

摘要:

高效扩展大型语言模型(LLMs)需探索主流自回归(AR)方法的替代方案,掩码扩散模型(MDMs)成为候选。但AR(通常为仅解码器)与MDM(通常为仅编码器)范式的比较因架构差异而混淆,掩盖了真正的算法与效率权衡。本研究通过在仅解码器框架内评估MDMs来解耦这些因素,以:(1)通过阶数差异公平比较MDM(作为任意阶AR)与标准AR范式;(2)研究MDM架构对计算效率的影响。我们表明,仅解码器MDM虽建模空间更大,但结合温度退火等技术可实现约25倍的推理加速与相当的困惑度,为减少推理计算提供了途径。本研究通过将核心建模选择与架构影响解耦,为开发更具计算效率的基础模型提供了见解,代码可在该https URL获取。

英文摘要:

Efficiently scaling Large Language Models (LLMs) necessitates exploring alternatives to dominant autoregressive (AR) methods, with Masked Diffusion Models (MDMs) emerging as candidates. However, comparing AR (typically decoder-only) and MDM (often encoder-only) paradigms is confounded by differing architectures, obscuring true algorithmic and efficiency trade-offs. This research decouples these factors by evaluating MDMs within a decoder-only framework to: (1) Equitably compare MDM (as Any-Order AR) and standard AR paradigms through discrepancies on orders. (2) Investigate MDM architectural impacts on computational efficiency. We show decoder-only MDMs, despite a larger modeling space, can achieve significant inference speedups ($\sim25\times$) and comparable perplexity with techniques like temperature annealing, offering a path to reduced inference compute. This work provides insights for developing more computationally efficient foundation models by disentangling core modeling choices from architectural influences. Code is available at https://github.com/scxue/AO-GPT-MDM.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑