Ambient Discrete Diffusion: Using the Wrong Data at the Right Time for Data Efficient Learning
Not on the map. Why?
Abstract (arXiv)
We introduce RefineMix, a framework for training discrete diffusion models under severe data scarcity, a common constraint in scientific applications. RefineMix uses out-of-distribution data at selected diffusion times to improve generalization without biasing the sampling distribution. Although this strategy has been explored in continuous diffusion, discrete diffusion presents a distinct challenge: unlike Gaussian noise, masking preserves domain information in surviving tokens, limiting the use of related data at high noise levels. At low noise levels, however, the domains effectively disjoint supports become an advantage, allowing the model to learn from both in-domain and out-of-distribution data without biasing the sampler. We formalize these intuitions and provide a theoretical analysis for the proposed method. Experimentally, across five domain-shift settings, RefineMix matches or outperforms in-domain finetuning and data mixing. For protein sequence generation, finetuning with just 197 in-domain examples nearly doubles the fraction of generated proteins that are simultaneously novel, foldable, and in-family compared to standard finetuning.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 36 of its 64 references that are in the map. First posted on arXiv 8 Oct 2026.
Builds on
36 of 64 references.
References · 36
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2022Scalable Diffusion Models with Transformerscited by 5,931
- 2022Classifier-Free Diffusion Guidancecited by 5,717
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
- 2024Simple and Effective Masked Diffusion Language Modelscited by 800
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 712
- 2024Simplified and Generalized Masked Diffusion for Discrete Datacited by 606
- 2025Dream 7B: Diffusion Large Language Modelscited by 515
- 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Datacited by 371
- 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 273
- 2025Mercury: Ultra-Fast Language Models Based on Diffusioncited by 225
24 more
- 2025d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learningcited by 178
- 2025Seed Diffusion: A Large-Scale Diffusion Language Model with High-Speed Inferencecited by 173
- 2024Simple Guidance Mechanisms for Discrete Diffusion Modelscited by 148
- 2023Generative Diffusion Models on Graphs: Methods and Applicationscited by 79
- 2023Efficient Online Data Mixing For Language Model Pre-Trainingcited by 77
- 2024Fine-Tuning Discrete Diffusion Models via Reward Optimization with Applications to DNA and Protein Designcited by 72
- 2025Diffusion Beats Autoregressive in Data-Constrained Settingscited by 61
- 2025GenMol: A Drug Discovery Generalist with Discrete Diffusioncited by 47
- 2025Diffusion Language Models are Super Data Learnerscited by 45
- 2023GSURE-Based Diffusion Model Training with Corrupted Datacited by 40
- 2025Esoteric Language Models: A Family of Any-Order Diffusion LLMscited by 33
- 2025Ambient Diffusion Omni: Training Good Models with Bad Datacited by 18
- 2024Transfer Learning for Diffusion Modelscited by 10
- 2025What Makes Diffusion Language Models Super Data Learners?cited by 8
- 2025Measurement Score-Based Diffusion Modelcited by 7
- 2025DiffEM: Learning from Corrupted Data with Diffusion Models via Expectation Maximizationcited by 6
- 2025Generative Modeling from Black-box Corruptions via Self-Consistent Stochastic Interpolantscited by 5
- 2026Ambient Diffusion Policy: Imitation Learning from Suboptimal Data in Roboticscited by 4
- 2025Score Distillation Beyond Acceleration: Generative Modeling from Corrupted Datacited by 3
- 2025Guided Transfer Learning for Discrete Diffusion Modelscited by 3
- 2025SFBD-OMNI: Bridge models for lossy measurement restoration with limited clean samplescited by 2
- 2026Ambient Dataloops: Generative Models for Dataset Refinementcited by 2
- 2026SCoRe: Clean Image Generation from Diffusion Models Trained on Noisy Imagescited by 2
- 2026NaRA: Noise-Aware LoRA for Parameter-Efficient Fine-Tuning of Diffusion LLMscited by 2
Built on by
Not cited yet.