Devil is in the Details: Density Guidance for Detail-Aware Generation with Flow Models
Rafal Karczewski, Markus Heinonen, Vikas K. Garg
Abstract
Diffusion models have emerged as a powerful class of generative models, capable of producing high-quality images by mapping noise to a data distribution. However, recent findings suggest that image likelihood does not align with perceptual quality: high-likelihood samples tend to be smooth, while lower-likelihood ones are more detailed. Controlling sample density is thus crucial for balancing realism and detail. In this paper, we analyze an existing technique, Prior Guidance, which scales the latent code to influence image detail. We introduce score alignment, a condition that explains why this method works and show that it can be tractably checked for any continuous normalizing flow model. We then propose Density Guidance, a principled modification of the generative ODE that enables exact log-density control during sampling. Finally, we extend Density Guidance to stochastic sampling, ensuring precise log-density control while allowing controlled variation in structure or fine details. Our experiments demonstrate that these techniques provide fine-grained control over image detail without compromising sample quality.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 992c7ccf-d2fb-474f-9c23-aa7e89377e60Cited by top-tier papers5
- The Spacetime of Diffusion Models: An Information Geometry PerspectiveRafal Karczewski, Markus Heinonen, Alison Pouplin, Søren Hauberg et al.ICLR 2026 · 7 citations
- Kernel Density Steering: Inference-Time Scaling via Mode Seeking for Image RestorationYuyang Hu, Kangfu Mei, Mojtaba Sahraee-Ardakan, Ulugbek Kamilov et al.NeurIPS 2025 · 6 citations
- MAMBO-G: Magnitude-Aware Mitigation for Boosted GuidanceShangwen Zhu, Qianyu Peng, Zhilei Shu, Yuting Hu et al.ICML 2026 · 1 citation
- Flow-Based Density Ratio Estimation for Intractable Distributions with Applications in GenomicsEgor Antipov, Alessandro Palma, Lorenzo Consoli, Stephan Günnemann et al.ICML 2026
- Composition of Pretrained Diffusion Models: A Logic-Based CalculusPeter Blohm, Vikas K GargICLR 2026
Builds on19
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- Denoising Diffusion Implicit ModelsJiaming Song, Chenlin Meng, Stefano ErmonICLR 2021 · 11,743 citations
- Elucidating the Design Space of Diffusion-Based Generative ModelsTero Karras, Miika Aittala, Timo Aila, Samuli LaineNeurIPS 2022 · 3,959 citations
- Consistency ModelsYang Song, Prafulla Dhariwal, Mark Chen, Ilya SutskeverICML 2023 · 1,720 citations
Related papers
- Reflected Diffusion ModelsAaron Lou, Stefano ErmonICML 2023 · 84 citations
- Temporal Score Rescaling for Temperature Sampling in Diffusion and Flow ModelsYanbo Xu, Yu Wu, Sungjae Park, Zhizhuo Zhou et al.ICML 2026 · 7 citations
- Rényi Diffusion ModelsYirong Shen, Lu GAN, Cong LingICML 2026 · 4 citations
- HiGS: History-Guided Sampling for Plug-and-Play Enhancement of Diffusion ModelsSeyedmorteza Sadat, Farnood Salehi, Romann M. WeberICLR 2026 · 2 citations
- Steering Guidance for Personalized Text-to-Image Diffusion ModelsSunghyun Park, Seokeon Choi, Hyoungwoo Park, Sungrack YunICCV 2025 · 2 citations
