Denoising Hierarchical Representations: Joint Continuous Diffusion for Language Modeling
cs.CL, cs.LG
Submitted: 2026-10-06
Updated: 2026-10-06
Comments: 27 pages, 10 figures
Code: https://github.com/matol-16/HCDLM
Project page: http://skylion007.github.io/OpenWebTextCorpus
License: http://creativecommons.org/licenses/by/4.0/
The gist: Diffusion Language Models (DLMs) hold the promise of order-agnostic, parallel text generation.
Abstract
Diffusion Language Models (DLMs) hold the promise of order-agnostic, parallel text generation. Recently, continuous diffusion and flow matching models have seen substantial gains, driven by carefully crafted token representations and diffusion/flow spaces. In this work, we introduce Hierarchical Continuous Diffusion Language Models (H-CDLMs), a simple framework that further improves continuous DLMs with minimal compute and parameter overhead. Drawing on the discrete DLM and continuous image diffusion literature on joint diffusion, we diffuse multiple modalities in parallel. These modalities represent tokens at different semantic granularities: in our instantiation, the tokens themselves and coarser clusters obtained by clustering pretrained token embeddings. We propose a general setup that allows per-modality samplers and schedules to enhance the interplay between modalities. Applied to CoBit, this yields H-CoBit, which delivers large empirical gains across benchmarks. At dataset entropy, H-CoBit improves MAUVE and reaches a generative perplexity (GenPPL) of 49.4 on LM1B and 50.4 on OWT, improving on the baseline by 24.2 and 20.7 points and surpassing even discrete DLMs of comparable size. On GSM8K, it reaches 27.4% accuracy, outperforming prior continuous diffusion and flow-based models. We further apply H-CDLM to the flow matching model FLM, obtaining consistent gains with H-FLM and demonstrating that the framework generalizes across continuous generative paradigms. Our code will be made publicly available at https://github.com/matol-16/HCDLM.git.
Sources
- CoBit: Language Modeling with Bitstream Diffusion
- LangFlow: Continuous Diffusion Rivals Discrete in Language Modeling
- Language Modeling with Hyperspherical Flows
- ELF: Embedded Language Flows
- Generative Frontiers: Why Evaluation Matters for Diffusion Language Models
- Noise Scheduling as Information-Guided Allocation in Diffusion Training
- Why Gaussian Diffusion Models Fail on Discrete Data and How to Prevent It?
- Llama 2: Open Foundation and Fine-Tuned Chat Models
- Rethinking Token Prediction: Tree-Structured Diffusion Language Model
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering