ScoreMix: Improving Face Recognition via Score Composition in Diffusion Generators
By: Parsa Rahimi, Sebastien Marcel
Potential Business Impact:
Makes AI better at recognizing things with less data.
In this paper, we propose ScoreMix, a novel yet simple data augmentation strategy leveraging the score compositional properties of diffusion models to enhance discriminator performance, particularly under scenarios with limited labeled data. By convexly mixing the scores from different class-conditioned trajectories during diffusion sampling, we generate challenging synthetic samples that significantly improve discriminative capabilities in all studied benchmarks. We systematically investigate class-selection strategies for mixing and discover that greater performance gains arise when combining classes distant in the discriminator's embedding space, rather than close in the generator's condition space. Moreover, we empirically show that, under standard metrics, the correlation between the generator's learned condition space and the discriminator's embedding space is minimal. Our approach achieves notable performance improvements without extensive parameter searches, demonstrating practical advantages for training discriminative models while effectively mitigating problems regarding collections of large datasets. Paper website: https://parsa-ra.github.io/scoremix
Similar Papers
Score Augmentation for Diffusion Models
Machine Learning (CS)
Makes AI create better pictures with less data.
Improving Compositional Generation with Diffusion Models Using Lift Scores
Machine Learning (CS)
Makes AI pictures match your exact words.
Image Interpolation with Score-based Riemannian Metrics of Diffusion Models
CV and Pattern Recognition
Makes AI art smoother and more realistic.