← Search

Marianne Arriola

5 accepted papers

2026

Set Diffusion: Interpolating Token Orderings between Autoregression and Diffusion for Fast and Flexible Decoding

ICML 2026poster

Masked discrete diffusion models have improved steadily, but still lag behind autoregressive (AR) models in quality, require fixed-length generation, and cannot exploit key-value (KV) caching. Block Diffusion partially bridges diffusion and AR by unmasking left-to-right token blocks, but sacrifices …

Cited by 0SourceScholar
2026

d2: Improved Techniques for Training Reasoning Diffusion Language Models

ICML 2026poster

While diffusion language models (DLMs) have achieved competitive performance in text generation, improving their reasoning ability with reinforcement learning remains an active research area. Here, we introduce d2, a reasoning framework tailored for masked DLMs. Central to our framework is a new pol…

Cited by 0SourceScholar
2025

Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models

ICLR 2025oral

Diffusion language models offer unique benefits over autoregressive models due to their potential for parallelized generation and controllability, yet they lag in likelihood modeling and are limited to fixed-length generation. In this work, we introduce a class of block diffusion language models tha…

2025

Encoder-Decoder Diffusion Language Models for Efficient Training and Inference

NeurIPS 2025poster

Discrete diffusion models enable parallel token sampling for faster inference than autoregressive approaches. However, prior diffusion models use a decoder-only architecture, which requires sampling algorithms that invoke the full network at every denoising step and incur high computational cost. Ou…

Cited by 0SourceScholar
2024

Simple and Effective Masked Diffusion Language Models

NeurIPS 2024poster

While diffusion models excel at generating high-quality images, prior work reports a significant performance gap between diffusion and autoregressive (AR) methods in language modeling. In this work, we show that simple masked discrete diffusion is more performant than previously thought. We apply an…