Diffusion models recover accurate mixture weights despite score function insensitivity

arXiv:2607.15485 2026 Training 1 ideas extracted · analyzed Aug 30, 2026

What the math gives to ML

The paper identifies a concrete failure mode in diffusion training: the clean-data score can be nearly insensitive to mixture weights even when intermediate noisy scores contain strong information about them. Its diffusion score sensitivity index (DSSI) measures a lower bound on how much the DSM objective changes under perturbations of a target distribution parameter, and this quantity depends on the noise schedule. A transferable method is therefore to select or regularize the noise schedule using an empirical minimum DSSI, favoring schedules that make important parameters identifiable at intermediate times rather than merely optimizing generic denoising quality.

Ideas from this paper

Unverified 2026

Sensitivity-aware diffusion noise schedules

Choose the diffusion noise schedule to maximize the minimum DSM sensitivity to important distribution parameters, such as mixture weights. This should reduce mode amplification and improve recovery of rare modes without changing the score-network architecture.

Useful6/10
Difficulty6/10
Novelty7/10
Paper: Diffusion models recover accurate mixture weights despite score function insensitivity arXiv:2607.15485