arxivcs.LGcs.AI2026-07-10
From Atoms to Entropy: Optimal Noise Allocation for Diffusion Training in the Convex Regime
Luca Ambrogioni, Giulio Franzese, Alberto Foresti, Gabriel Raya, Bac Nguyen, Georgios Batzolis, et al.
How should a diffusion model decide which noise levels to train on, and how much? Despite the importance of this choice, current noise schedules are based largely on heuristics or empirical tuning. Here, we develop a general statistical framework for studying asymptotically optim…