Exploring Low-Dimensional Subspaces in Diffusion Models for Controllable Image Editing
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Recently, diffusion models have emerged as a powerful class of generative models. Despite their success, there is still limited understanding of their semantic spaces. This makes it challenging to achieve precise and disentangled image generation without additional training, especially in an unsupervised way. In this work, we improve the understanding of their semantic spaces from intriguing observations: among a certain range of noise levels, (1) the learned posterior mean predictor (PMP) in the diffusion model is locally linear, and (2) the singular vectors of its Jacobian lie in low-dimensional semantic subspaces. We provide a solid theoretical basis to justify the linearity and low-rankness in the PMP. These insights allow us to propose an unsupervised, single-step, training-free LOw-rank COntrollable image editing (LOCO Edit) method for precise local editing in diffusion models. LOCO Edit identified editing directions with nice properties: homogeneity, transferability, composability, and linearity. These properties of LOCO Edit benefit greatly from the low-dimensional semantic subspace. Our method can further be extended to unsupervised or text-supervised editing in various text-to-image diffusion models (T-LOCO Edit). Finally, extensive empirical experiments demonstrate the effectiveness and efficiency of LOCO Edit.
Key facts. Cited by 15 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 64 of its 87 references that are in the map. First posted on arXiv 4 Sep 2024.
Builds on
64 of 87 references.
Closest ancestors · 20
Same direction, newest first
- 2024Enabling Local Editing in Diffusion Models by Joint and Individual Component Analysis29 Aug
- 2024Diffusion Models and Representation Learning: A Survey30 Jun
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesis5 Mar
- 2023NoiseCLR: A Contrastive Learning Approach for Unsupervised Discovery of Interpretable Directions in Diffusion Models8 Dec
- 2023The Hidden Linear Structure in Score-Based Models and its Application17 Nov
- 2023Latent Consistency Models: Synthesizing High-Resolution Images with Few-Step Inference6 Oct
- 2023On the Posterior Distribution in Denoising: Application to Uncertainty Quantification24 Sep
- 2023Understanding the Latent Space of Diffusion Models through the Lens of Riemannian Geometry24 Jul
- 2023Discovering Interpretable Directions in the Semantic Latent Space of Diffusion Models20 Mar
- 2022Diffusion Models already have a Semantic Latent Space20 Oct
- 2022All are Worth Words: A ViT Backbone for Diffusion Models25 Sep
- 2022Understanding Diffusion Models: A Unified Perspective25 Aug
8 more
- 2022Classifier-Free Diffusion Guidance26 Jul
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps2 Jun
- 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
- 2022Perception Prioritized Training of Diffusion Models1 Apr
- 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Implicit Models6 Oct
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 44
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
- 2023Segment Anythingcited by 8,158
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,539
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
- 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 5,986
- 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,466
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,131
- 2014Deep Learning Face Attributes in the Wildcited by 3,679
- 2022DreamBooth: Fine Tuning Text-to-Image Diffusion Models for Subject-Driven Generationcited by 3,286
32 more
- 2022InstructPix2Pix: Learning to Follow Image Editing Instructionscited by 2,591
- 2023Stable Video Diffusion: Scaling Latent Video Diffusion Models to Large Datasetscited by 2,526
- 2022Prompt-to-Prompt Image Editing with Cross Attention Controlcited by 2,409
- 2022Diffusion Posterior Sampling for General Noisy Inverse Problemscited by 1,375
- 2015LSUN: Construction of a Large-scale Image Dataset using Deep Learning with Humans in the Loopcited by 1,350
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,339
- 2020Training Generative Adversarial Networks with Limited Datacited by 1,184
- 2022Null-text Inversion for Editing Real Images using Guided Diffusion Modelscited by 1,168
- 2019StarGAN v2: Diverse Image Synthesis for Multiple Domainscited by 1,095
- 2021Blended Diffusion for Text-driven Editing of Natural Imagescited by 924
- 2024Eyes Wide Shut? Exploring the Visual Shortcomings of Multimodal LLMscited by 719
- 2023Text2Video-Zero: Text-to-Image Diffusion Models are Zero-Shot Video Generatorscited by 666
- 2022DiffEdit: Diffusion-based semantic image editing with mask guidancecited by 643
- 2021DiffusionCLIP: Text-Guided Diffusion Models for Robust Image Manipulationcited by 640
- 2020WaveGrad: Estimating Gradients for Waveform Generationcited by 620
- 2022Improving Diffusion Models for Inverse Problems using Manifold Constraintscited by 501
- 2023Instruct-NeRF2NeRF: Editing 3D Scenes with Instructionscited by 482
- 2021The Intrinsic Dimension of Images and Its Impact on Learningcited by 307
- 2023Solving Inverse Problems with Latent Diffusion Models via Hard Data Consistencycited by 180
- 2023Video Probabilistic Diffusion Models in Projected Latent Spacecited by 171
- 2022Uncovering the Disentanglement Capability in Text-to-Image Diffusion Modelscited by 116
- 2023InstructEdit: Improving Automatic Masks for Diffusion-based Image Editing With User Instructionscited by 64
- 2023ZONE: Zero-Shot Instruction-Guided Local Editingcited by 61
- 2024Zero-Shot Unsupervised and Text-Based Audio Editing Using DDPM Inversioncited by 58
- 2021Low-Rank Subspaces in GANscited by 52
- 2022Understanding 3D Object Articulation in Internet Videoscited by 36
- 2023Understanding Deep Representation Learning via Layerwise Feature Compression and Discriminationcited by 31
- 2024Compressible Dynamics in Deep Overparameterized Low-Rank Learning & Adaptationcited by 26
- 2023Boundary Guided Learning-Free Semantic Control with Diffusion Modelscited by 23
- 2024PaCE: Parsimonious Concept Engineering for Large Language Modelscited by 19
- 2024A Global Geometric Analysis of Maximal Coding Rate Reductioncited by 13
- 2023Robust Physics-based Deep MRI Reconstruction Via Diffusion Purificationcited by 4
Built on by
Citing papers per year: 2024: 8; 2025: 6; 2026: 1.
15 papers.
Sort:
By area:
- 2024O(d/T) Convergence Theory for Diffusion Probabilistic Models under Minimal Assumptionscited by 48
- 2024Denoising diffusion probabilistic models are optimally adaptive to unknown low dimensionalitycited by 44
- 2024Understanding Generalizability of Diffusion Models Requires Rethinking the Hidden Gaussian Structurecited by 38
- 2024Manifolds, Random Matrices and Spectral Gaps: The geometric phases of generative diffusioncited by 31
- 2024SITCOM: Step-wise Triple-Consistent Diffusion Sampling for Inverse Problemscited by 23
- 2025Emergence and Evolution of Interpretable Concepts in Diffusion Modelscited by 20
- 2025A Closer Look at Model Collapse: From a Generalization-to-Memorization Perspectivecited by 18
- 2025Point Prompting: Counterfactual Tracking with Video Diffusion Modelscited by 5
- 2025Concept Lancet: Image Editing with Compositional Representation Transplantcited by 4
- 2026Personalized Image Generation via Human-in-the-loop Bayesian Optimizationcited by 2
- 2024Shallow Diffuse: Robust and Invisible Watermarking through Low-Dimensional Subspaces in Diffusion Modelscited by 1
- 2025Instructing Text-to-Image Diffusion Models via Classifier-Guided Semantic Optimizationcited by 1
- 2026Personalized Image Generation via Human-in-the-loop Bayesian Optimization2 Feb 2026
- 2025Generative human motion mimicking through feature extraction in denoising diffusion settings16 Oct 2025
- 2025Point Prompting: Counterfactual Tracking with Video Diffusion Models13 Oct 2025
- 2025A Closer Look at Model Collapse: From a Generalization-to-Memorization Perspective20 Sep 2025
- 2025Instructing Text-to-Image Diffusion Models via Classifier-Guided Semantic Optimization20 May 2025
- 2025Emergence and Evolution of Interpretable Concepts in Diffusion Models21 Apr 2025
- 2025Concept Lancet: Image Editing with Compositional Representation Transplant3 Apr 2025
- 2024Subspace Langevin Monte Carlo18 Dec 2024
- 2024Understanding Generalizability of Diffusion Models Requires Rethinking the Hidden Gaussian Structure31 Oct 2024
- 2024Shallow Diffuse: Robust and Invisible Watermarking through Low-Dimensional Subspaces in Diffusion Models28 Oct 2024
- 2024Unsupervised Panoptic Interpretation of Latent Spaces in GANs Using Space-Filling Vector Quantization27 Oct 2024
- 2024Denoising diffusion probabilistic models are optimally adaptive to unknown low dimensionality24 Oct 2024
3 more
- 2024O(d/T) Convergence Theory for Diffusion Probabilistic Models under Minimal Assumptionscited by 48
- 2024Denoising diffusion probabilistic models are optimally adaptive to unknown low dimensionalitycited by 44
- 2024Understanding Generalizability of Diffusion Models Requires Rethinking the Hidden Gaussian Structurecited by 38
- 2024Manifolds, Random Matrices and Spectral Gaps: The geometric phases of generative diffusioncited by 31
- 2024SITCOM: Step-wise Triple-Consistent Diffusion Sampling for Inverse Problemscited by 23
- 2025Emergence and Evolution of Interpretable Concepts in Diffusion Modelscited by 20
- 2025A Closer Look at Model Collapse: From a Generalization-to-Memorization Perspectivecited by 18
- 2025Point Prompting: Counterfactual Tracking with Video Diffusion Modelscited by 5
- 2025Concept Lancet: Image Editing with Compositional Representation Transplantcited by 4
- 2024Shallow Diffuse: Robust and Invisible Watermarking through Low-Dimensional Subspaces in Diffusion Modelscited by 1
- 2025Instructing Text-to-Image Diffusion Models via Classifier-Guided Semantic Optimizationcited by 1
- 2025Generative human motion mimicking through feature extraction in denoising diffusion settingscited by 0
- 2025Generative human motion mimicking through feature extraction in denoising diffusion settings16 Oct 2025
- 2025Point Prompting: Counterfactual Tracking with Video Diffusion Models13 Oct 2025
- 2025A Closer Look at Model Collapse: From a Generalization-to-Memorization Perspective20 Sep 2025
- 2025Instructing Text-to-Image Diffusion Models via Classifier-Guided Semantic Optimization20 May 2025
- 2025Emergence and Evolution of Interpretable Concepts in Diffusion Models21 Apr 2025
- 2025Concept Lancet: Image Editing with Compositional Representation Transplant3 Apr 2025
- 2024Understanding Generalizability of Diffusion Models Requires Rethinking the Hidden Gaussian Structure31 Oct 2024
- 2024Shallow Diffuse: Robust and Invisible Watermarking through Low-Dimensional Subspaces in Diffusion Models28 Oct 2024
- 2024Denoising diffusion probabilistic models are optimally adaptive to unknown low dimensionality24 Oct 2024
- 2024Manifolds, Random Matrices and Spectral Gaps: The geometric phases of generative diffusion8 Oct 2024
- 2024SITCOM: Step-wise Triple-Consistent Diffusion Sampling for Inverse Problems6 Oct 2024
- 2024O(d/T) Convergence Theory for Diffusion Probabilistic Models under Minimal Assumptions27 Sep 2024
Similar, not cited
Close in wording, no citation either way
- 2024Unsupervised Region-Based Image Editing of Denoising Diffusion Models
- 2025Self-Attention Decomposition For Training Free Diffusion Editing
- 2025Exploring the latent space of diffusion models directly through singular value decomposition
- 2026Guidance for Low-Level Perceptual Editing in Unconditional Diffusion Models
- 2023High-Fidelity Diffusion-based Image Editing
- 2023Unsupervised Discovery of Semantic Latent Directions in Diffusion Models
- 2022Image Embedding for Denoising Generative Models
- 2024Latent Space Disentanglement in Diffusion Transformers Enables Precise Zero-shot Semantic Editing