Variance-Aware Adaptive Weighting for Diffusion Model Training

Nanlong Sun, Lei Shi

Diffusion models have recently achieved remarkable success in generative modeling, yet their training dynamics across different noise levels remain highly imbalanced, which can lea…