Residual Context Diffusion Language Models
Key point
By recycling information from discarded tokens through RCD technology, the accuracy and inference speed of diffusion language models have been dramatically improved.
Details
Existing block-wise Diffusion Large Language Models (dLLMs) have a limitation of wasting computational resources by using a 'remasking' mechanism that decodes only the most confident tokens and discards the rest.
The newly proposed Residual Context Diffusion (RCD) improves computational efficiency by converting the representations of discarded tokens into contextual residuals and re-injecting them into the next denoising step. RCD adopts a decoupled two-stage training pipeline to prevent memory bottlenecks.
This approach has demonstrated the following performance improvements:
- Accuracy improvement: Raises accuracy by 5-10 points over existing dLLMs on major benchmarks
- Inference efficiency: On the most challenging AIME task, nearly doubles accuracy while reducing denoising steps by 4-5x at equivalent accuracy
- Fast conversion: A standard dLLM can be efficiently converted to the RCD paradigm with only about 1 billion tokens
This summary was generated automatically by AI. Check the original for the author's claims and context. Copyright belongs to the original author.
Our guide explains how the AI works. Report summary errors, attribution issues, or removal requests via Contact.