Feature Information Dynamics in Diffusion
stat.ML, cs.AI, cs.LG
Submitted: 2026-10-06
Updated: 2026-10-06
Comments: Accepted as poster at NeurIPS 2026. 28 pages, including references, appendices, and checklist
Code: https://github.com/AI4Science-WestlakeU/feature-information-dynamics
License: http://creativecommons.org/licenses/by/4.0/
The gist: Diffusion models generate data through a continuum of denoising problems, and are widely observed to reveal coarse structure before fine detail.
Abstract
Diffusion models generate data through a continuum of denoising problems, and are widely observed to reveal coarse structure before fine detail. Yet, this intuition is mostly empirical and qualitative. We introduce feature information dynamics, an information-theoretic framework for localizing when a feature is generated during diffusion. Using the I-MMSE identity, we connect the rate of feature mutual information change to a gap between optimal unconditional and feature-conditional denoising losses, yielding practical estimators for feature information density. We further develop a chained decomposition that separates shared from incremental information in a feature hierarchy. We use this framework first to quantitatively confirm spectral autoregression in pixel diffusion, and then to extend the analysis beyond frequency: under a class to mask to Canny conditioning chain, the per-feature information densities differ across pixel, SDVAE, VAVAE, and RAE, exposing fundamental differences between these representations and suggesting that ordered generation could be beneficial for training diffusion models. Our code is available at https://github.com/AI4Science-WestlakeU/feature-information-dynamics.
Sources
- Latent Forcing: Reordering the Diffusion Trajectory for Pixel-Space Image Generation
- SAM 3: Segment Anything with Concepts
- The Entropic Signature of Class Speciation in Diffusion Models
- Classifier-Free Diffusion Guidance
- Back to Basics: Let Denoising Generative Models Denoise
- Flow Matching for Generative Modeling
- Representation Learning with Contrastive Predictive Coding
- Semantics Lead the Way: Harmonizing Semantic and Texture Modeling with Asynchronous Latent Diffusion
- The information bottleneck method
- Diffusion Transformers with Representation Autoencoders
Related papers
- Behavior of prediction performance metrics with rare events
- Optimal Estimation of Generic Dynamics by Path-Dependent Neural Jump ODEs
- A Posterior-Dynamics Framework for Imaging Inverse Problems with Pretrained Diffusion Priors
- Efficient quadratic entropy with distance sketches
- Embedding-Bias in Conditional Independence Testing
- From Log-Odds to Shapley Values: An Explanatory Geometry for the Weighted Naive Bayes Classifier