Nemotron-TwoTower: Diffusion Language Modeling with Pretrained Autoregressive Context
Abstract
A novel diffusion language model architecture decouples context representation and denoising roles into separate towers, achieving high-generation throughput while maintaining autoregressive quality.
Diffusion language models offer a promising alternative to autoregressive models due to their potential for parallel and iterative generation. However, existing approaches use a single network for both context representation and iterative denoising, forcing one model to serve both roles and limiting its capacity for either role. We propose TwoTower, a block-wise autoregressive diffusion model that decouples these roles into two towers: a frozen AR context tower that causally processes clean tokens, and a trainable diffusion denoiser tower with bidirectional block attention that refines noisy blocks via cross-attention to the context. Built on Nemotron-3-Nano-30B-A3B, an open-weight 30B hybrid Mamba-Transformer MoE model, and trained on approximately 2.1T tokens, Nemotron-TwoTower retains 98.7% of the autoregressive baseline's quality while offering 2.42X higher wall-clock generation throughput. We release the code and model weights at https://huggingface.co/collections/nvidia/nemotron-twotower.
Community
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- Don't Retrain, Align: Adapting Autoregressive LMs to Diffusion LMs via Representation Alignment (2026)
- SAID: Accelerating Diffusion-Based Language Models via Scaffold-Aware Iterative Decoding (2026)
- BlockGen: Flexible Blockwise Sequence Modeling with Hybrid Samplers (2026)
- Data-Efficient Autoregressive-to-Diffusion Language Models via On-Policy Distillation (2026)
- Triplet-Block Diffusion RWKV (2026)
- Orthrus: Memory-Efficient Parallel Token Generation via Dual-View Diffusion (2026)
- SimSD: Simple Speculative Decoding in Diffusion Language Models (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Get this paper in your agent:
hf papers read 2606.26493 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 1
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper