LLaDA-o: An Effective and Length-Adaptive Omni Diffusion Model

Zebin You, Xiaolu Zhang, Jun Zhou, Chongxuan Li, Ji-Rong Wen
3/1/2026
cs.CVcs.LG

Official code: ML-GSAI/LLaDA-o (53 stars, Apache-2.0, last commit 2026-03-24).

Abstract

We present \textbf{LLaDA-o}, an effective and length-adaptive omni diffusion model for multimodal understanding and generation. LLaDA-o is built on a Mixture of Diffusion (MoD) framework that decouples discrete masked diffusion for text understanding and continuous diffusion for visual generation, while coupling them through a shared, simple, and efficient attention backbone that reduces redundant computation for fixed conditions. Building on MoD, we further introduce a data-centric length adaptation strategy that enables flexible-length decoding in multimodal settings without architectural changes. Extensive experiments show that LLaDA-o achieves state-of-the-art performance among omni-diffusion models on multimodal understanding and generation benchmarks, and reaches 87.04 on DPG-Bench for text-to-image generation, supporting the effectiveness of unified omni diffusion modeling. Code is available at https://github.com/ML-GSAI/LLaDA-o.

View on arXivView PDF

Code Implementations(1)

ML-GSAI/LLaDA-oOfficial100%
Apache-2.0
534Shell, PythonMar 1, 20267 months agoApache-2.0

Similar Papers

Cite this paper

@article{you2026lladao,
  title  = {LLaDA-o: An Effective and Length-Adaptive Omni Diffusion Model},
  author = {Zebin You and Xiaolu Zhang and Jun Zhou and Chongxuan Li and Ji-Rong Wen},
  year   = {2026},
  eprint = {2603.01068},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2603.01068v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2603.01068.svg)](https://codeofpaper.com/paper/2603.01068)

Discussion