imbalanced-datasets

This paper proposes a class-frequency guided noise schedule for diffusion models that assigns larger-scale noises to low-frequency classes to improve generation quality on imbalanced datasets, demonstrating substantial improvements over baselines.