├── README.md
├── assets
└── teaser.svg
├── datasets
├── __init__.py
└── activations.py
├── dictionary_learning
├── .gitignore
├── LICENSE
├── README.md
├── __init__.py
├── buffer.py
├── config.py
├── dictionary.py
├── evaluation.py
├── grad_pursuit.py
├── interp.py
├── pretrained_dictionary_downloader.sh
├── requirements.txt
├── tests
│ └── test_end_to_end.py
├── trainers
│ ├── __init__.py
│ ├── batch_top_k.py
│ ├── gated_anneal.py
│ ├── gdm.py
│ ├── jumprelu.py
│ ├── matroyshka_batch_top_k.py
│ ├── p_anneal.py
│ ├── standard.py
│ ├── top_k.py
│ └── trainer.py
├── training.py
└── utils.py
├── encode_images.py
├── find_hai_indices.py
├── imagenet_subset.py
├── images
└── white.png
├── inat_depth.py
├── metric.py
├── models
├── clip.py
├── dino.py
├── llava.py
└── siglip.py
├── requirements.txt
├── sae_train.py
├── save_activations.py
├── scripts
├── matryoshka_hierarchy.sh
├── mllm_steering.sh
└── monosemanticity_score.sh
├── similarity_baseline.py
├── steering_qualitative.py
├── steering_score.py
├── uniqueness.py
├── utils.py
└── visualize_neurons.py
/README.md:
--------------------------------------------------------------------------------
1 | ##
2 | Sparse Autoencoders Learn Monosemantic Features in Vision-Language Models
3 |
4 |
15 |
16 | Abstract
17 |
18 |
19 | Sparse Autoencoders (SAEs) have recently been shown to enhance interpretability and steerability in Large Language Models (LLMs). In this work, we extend the application of SAEs to Vision-Language Models (VLMs), such as CLIP, and introduce a comprehensive framework for evaluating monosemanticity in vision representations. Our experimental results reveal that SAEs trained on VLMs significantly enhance the monosemanticity of individual neurons while also exhibiting hierarchical representations that align well with expert-defined structures (e.g., iNaturalist taxonomy). Most notably, we demonstrate that applying SAEs to intervene on a CLIP vision encoder, directly steer output from multimodal LLMs (e.g., LLaVA) without any modifications to the underlying model. These findings emphasize the practicality and efficacy of SAEs as an unsupervised approach for enhancing both the interpretability and control of VLMs.
20 |
21 |
22 |
23 |
24 |