← Search

Subham Sekhar Sahoo

12 accepted papers

2026

Scaling Beyond Masked Diffusion Language Models

ICML 2026poster

Diffusion language models are a promising alternative to autoregressive models due to their potential for faster generation. Among discrete diffusion approaches, Masked diffusion currently dominates, largely driven by strong perplexity on language modeling benchmarks. In this work, we present the fi…

Cited by 0SourceScholar
2026

The Diffusion Duality, Chapter II: $\Psi$-Samplers and Efficient Curriculum

ICLR 2026poster

Uniform-state discrete diffusion models excel at few-step generation and guidance due to their inherent ability to self-correct, making them more preferable than autoregressive or masked diffusion models in these settings. Yet, their sampling efficiency has been limited by reliance on standard poste…

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

Remasking Discrete Diffusion Models with Inference-Time Scaling

NeurIPS 2025poster

Part of the success of diffusion models stems from their ability to perform iterative refinement, i.e., repeatedly correcting outputs during generation. However, modern masked discrete diffusion lacks this capability: when a token is generated, it cannot be updated again, even when it introduces an…

Cited by 0SourceScholar
2025

Simple Guidance Mechanisms for Discrete Diffusion Models

ICLR 2025poster

Diffusion models for continuous data gained widespread adoption owing to their high quality generation and control mechanisms. However, controllable diffusion on discrete data faces challenges given that continuous guidance methods do not directly apply to discrete diffusion. Here, we provide a stra…

2025

The Diffusion Duality

ICML 2025poster

Uniform-state discrete diffusion models hold the promise of fast text generation due to their inherent ability to self-correct. However, they are typically outperformed by autoregressive models and masked diffusion models. In this work, we narrow this performance gap by leveraging a key insight: Un…

2024

Diffusion Models With Learned Adaptive Noise

NeurIPS 2024spotlight

Diffusion models have gained traction as powerful algorithms for synthesizing high-quality images. Central to these algorithms is the diffusion process, a set of equations which maps data to noise in a way that can significantly affect performance. In this paper, we explore whether the diffusion p…

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…

2023

Backpropagation through Combinatorial Algorithms: Identity with Projection Works

ICLR 2023poster

Embedding discrete solvers as differentiable layers has given modern deep learning architectures combinatorial expressivity and discrete reasoning capabilities. The derivative of these solvers is zero or undefined, therefore a meaningful replacement is crucial for effective gradient-based learning.…

2023

Semi-Autoregressive Energy Flows: Exploring Likelihood-Free Training of Normalizing Flows

ICML 2023poster

Training normalizing flow generative models can be challenging due to the need to calculate computationally expensive determinants of Jacobians. This paper studies the likelihood-free training of flows and proposes the energy objective, an alternative sample-based loss based on proper scoring rules.…

2021

Scaling Symbolic Methods using Gradients for Neural Model Explanation

ICLR 2021poster

Symbolic techniques based on Satisfiability Modulo Theory (SMT) solvers have been proposed for analyzing and verifying neural network properties, but their usage has been fairly limited owing to their poor scalability with larger networks. In this work, we propose a technique for combining gradient-…