Designing Transformers for Continuous Diffusion Language Models· 为连续扩散语言模型设计Transformer
Continuous diffusion language models (CDLMs) offer an alternative approach to parallel language generation; however, most existing studies directly use transformer architectures from either causal large language models (LLMs) or Diffusion Transformers (DiTs) for image generation, which were not optimized for CDLMs. In this paper, we conduct a systematic study on Transformer architectures for CDLMs and introduce Language Diffusion Transformer (LangDiT), combining feature and query-key normalization, wider noise conditioning, likelihood-shortcut gradient routing and calibration, and confidence-gated training self-conditioning. LangDiT significantly reduces generation perplexity in matched LangFlow comparisons, establishing a new state of the art in continuous language generation across diffe
为连续扩散语言模型设计优化的Transformer架构
- 核心方法
- 引入Language Diffusion Transformer(LangDiT),结合特征和查询-键值归一化、更宽的噪声条件、似然捷径梯度路由和校准、自信门控训练自条件
- 适合谁读
- 研究者 / 工程师
- 要解决的问题
- 现有的Transformer架构并不适合连续扩散语言模型(CDLMs)
- 关键实验
- 在匹配的LangFlow比较中显著降低了生成困惑度
- 主要贡献
- 降低了生成过程中的困惑度,成为连续语言生成的新技术前沿
- 意义与局限
- 通过优化Transformer架构,提高了连续扩散语言模型的生成效果和效率;但可能需要更多的计算资源