← Search

Shen Nie

10 accepted papers

2026

LLaDA-V: Large Language Diffusion Models with Visual Instruction Tuning

CVPR 2026

In this work, we introduce LLaDA-V, a purely diffusion-based Multimodal Large Language Model (MLLM) that integrates visual instruction tuning with masked diffusion models, representing a departure from the autoregressive paradigms dominant in current multimodal approaches. Built upon LLaDA, a repres

Cited by 0SourcecodeScholar
2026

UltraLLaDA: Scaling the Context Length to 128K for Diffusion Large Language Models

ICLR 2026poster

Diffusion LLMs have attracted growing interest, with plenty of recent work emphasizing their great potential in various downstream tasks; yet the long‑context behavior of diffusion LLMs remains largely uncharted. We present a case study of post‑training techniques for extending the context window of…

Cited by 0SourcecodeScholar
2025

Masked Diffusion Models as Energy Minimization

NeurIPS 2025poster

We present a systematic theoretical framework that interprets masked diffusion models (MDMs) as solutions to energy minimization problems in discrete optimal transport. Specifically, we prove that three distinct energy formulations—kinetic, conditional kinetic, and geodesic energy—are mathematically…

Cited by 0SourceScholar
2025

Scaling up Masked Diffusion Models on Text

ICLR 2025poster

Masked diffusion models (MDMs) have shown promise in language modeling, yet their scalability and effectiveness in core language tasks, such as text generation and language understanding, remain underexplored. This paper establishes the first scaling law for MDMs, demonstrating a scaling rate compar…

2025

Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data

ICLR 2025poster

Discrete diffusion models with absorbing processes have shown promise in language modeling. The key quantities to be estimated are the ratios between the marginal probabilities of two transitive states at all timesteps, called the concrete score. In this paper, we reveal that the concrete score in a…

2024

The Blessing of Randomness: SDE Beats ODE in General Diffusion-based Image Editing

ICLR 2024poster

We present a unified probabilistic formulation for diffusion-based image editing, where a latent variable is edited in a task-specific manner and generally deviates from the corresponding marginal distribution induced by the original stochastic or ordinary differential equation (SDE or ODE). Instead…

2024

Unifying Bayesian Flow Networks and Diffusion Models through Stochastic Differential Equations

ICML 2024poster

Bayesian flow networks (BFNs) iteratively refine the parameters, instead of the samples in diffusion models (DMs), of distributions at various noise levels through Bayesian inference. Owing to its differentiable nature, BFNs are promising in modeling both continuous and discrete data, while simultan…

2023

All Are Worth Words: A ViT Backbone for Diffusion Models

CVPR 2023poster

Vision transformers (ViT) have shown promise in various vision tasks while the U-Net based on a convolutional neural network (CNN) remains dominant in diffusion models. We design a simple and general ViT-based architecture (named U-ViT) for image generation with diffusion models. U-ViT is characteri…

2023

One Transformer Fits All Distributions in Multi-Modal Diffusion at Scale

ICML 2023poster

This paper proposes a unified diffusion framework (dubbed UniDiffuser) to fit all distributions relevant to a set of multi-modal data in one model. Our key insight is -- learning diffusion models for marginal, conditional, and joint distributions can be unified as predicting the noise in the perturb…