arXiv · 2411.16725
$\textit{Revelio}$: Interpreting and leveraging semantic information in diffusion models
Abstract
We study $\textit{how}$ rich visual semantic information is represented within various layers and denoising timesteps of different diffusion architectures. We uncover monosemantic interpretable features by leveraging k-sparse autoencoders (k-SAE). We substantiate our mechanistic interpretations via transfer learning using light-weight classifiers on off-the-shelf diffusion models' features. On $4$ datasets, we demonstrate the effectiveness of diffusion features for representation learning. We provide an in-depth analysis of how different diffusion architectures, pre-training datasets, and language model conditioning impacts visual representation granularity, inductive biases, and transfer learning capabilities. Our work is a critical step towards deepening interpretability of black-box diffusion models. Code and visualizations available at: https://github.com/revelio-diffusion/revelio
Explore related subjects
Keep this discovery
Dahye Kim, Xavier Thomas, Deepti Ghadiyaram. 2024-11-23. $\textit{Revelio}$: Interpreting and leveraging semantic information in diffusion models. https://arxiv.org/abs/2411.16725
Cite the original work for its findings. Save a collection to share your selection of sources.