论文

分层连续扩散语言模型

Hierarchical Continuous Diffusion Language Models

精选理由

新提出的HC-DLM模型解决了扩散语言模型的统计依赖问题,在多个任务上表现优异。

HC-DLM模型将离散令牌生成与连续潜在轨迹耦合,在单一去噪过程中处理。在数独和倒计时等结构化推理任务中,该模型在相同模型规模下优于离散和连续扩散基线。在LM1B语言建模基准上,HC-DLM降低了生成困惑度。项目页面已上线。

原文 · arXiv cs.LG

Hierarchical Continuous Diffusion Language Models

Discrete diffusion language models offer a compelling alternative to autoregressive generation for tasks demanding bidirectional reasoning and global constraint satisfaction. Yet they share a structural bottleneck: when decoding in parallel, each token is sampled independently from its marginal, severing the statistical dependencies among the tokens decoded together. Continuous diffusion language models avoid this by denoising a shared continuous state, but their denoiser sees only that state, so nothing ties it to a valid token configuration until it is finally decoded. To address this, we propose Hierarchical Continuous Diffusion Language Models (HC-DLM), which couple discrete token generation with a continuous latent trajectory in a single, principled denoising process, whose training objective is derived from a variational bound on the token likelihood. In contrast to recent methods that attach continuous context to a self-contained discrete chain, HC-DLM makes the latent the only persistent generative state: tokens are read out from it at every step and feed back as a scaffold for the next latent update. On structured reasoning (Sudoku), mathematical planning (Countdown) and language modeling (LM1B), HC-DLM improves over discrete and continuous diffusion baselines at matched model size, in puzzle accuracy on Sudoku and Countdown and in generative perplexity on LM1B. Project page: https://hc-dlm.github.io/.