arXiv · 2504.00220
Can Diffusion Models Disentangle? A Theoretical Perspective
Abstract
This paper presents a novel theoretical framework for understanding how diffusion models can learn disentangled representations. Within this framework, we establish identifiability conditions for general disentangled latent variable models, analyze training dynamics, and derive sample complexity bounds for disentangled latent subspace models. To validate our theory, we conduct disentanglement experiments across diverse tasks and modalities, including subspace recovery in latent subspace Gaussian mixture models, image colorization, image denoising, and voice conversion for speech classification. Additionally, our experiments show that training strategies inspired by our theory, such as style guidance regularization, consistently enhance disentanglement performance.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Liming Wang, Muhammad Jehanzeb Mirza, Yishu Gong, Yuan Gong, Jiaqi Zhang, Brian H. Tracey, Katerina Placek, Marco Vilela, James R. Glass. 2025-03-31. Can Diffusion Models Disentangle? A Theoretical Perspective. https://arxiv.org/abs/2504.00220
Cite the original work for its findings. Save a collection to share your selection of sources.