Interpreting Large Text-to-Image Diffusion Models with Dictionary Learning
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866908445515448320 |
|---|---|
| author | Shabalin, Stepan Panda, Ayush Kharlapenko, Dmitrii Ali, Abdur Raheem Hao, Yixiong Conmy, Arthur |
| author_facet | Shabalin, Stepan Panda, Ayush Kharlapenko, Dmitrii Ali, Abdur Raheem Hao, Yixiong Conmy, Arthur |
| contents | Sparse autoencoders are a promising new approach for decomposing language model activations for interpretation and control. They have been applied successfully to vision transformer image encoders and to small-scale diffusion models. Inference-Time Decomposition of Activations (ITDA) is a recently proposed variant of dictionary learning that takes the dictionary to be a set of data points from the activation distribution and reconstructs them with gradient pursuit. We apply Sparse Autoencoders (SAEs) and ITDA to a large text-to-image diffusion model, Flux 1, and consider the interpretability of embeddings of both by introducing a visual automated interpretation pipeline. We find that SAEs accurately reconstruct residual stream embeddings and beat MLP neurons on interpretability. We are able to use SAE features to steer image generation through activation addition. We find that ITDA has comparable interpretability to SAEs. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2505_24360 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | Interpreting Large Text-to-Image Diffusion Models with Dictionary Learning Shabalin, Stepan Panda, Ayush Kharlapenko, Dmitrii Ali, Abdur Raheem Hao, Yixiong Conmy, Arthur Machine Learning Sparse autoencoders are a promising new approach for decomposing language model activations for interpretation and control. They have been applied successfully to vision transformer image encoders and to small-scale diffusion models. Inference-Time Decomposition of Activations (ITDA) is a recently proposed variant of dictionary learning that takes the dictionary to be a set of data points from the activation distribution and reconstructs them with gradient pursuit. We apply Sparse Autoencoders (SAEs) and ITDA to a large text-to-image diffusion model, Flux 1, and consider the interpretability of embeddings of both by introducing a visual automated interpretation pipeline. We find that SAEs accurately reconstruct residual stream embeddings and beat MLP neurons on interpretability. We are able to use SAE features to steer image generation through activation addition. We find that ITDA has comparable interpretability to SAEs. |
| title | Interpreting Large Text-to-Image Diffusion Models with Dictionary Learning |
| topic | Machine Learning |
| url | https://arxiv.org/abs/2505.24360 |