| --- |
| license: cc-by-4.0 |
| library_name: saelens |
| --- |
| |
| # 1. Gemma Scope |
|
|
| Gemma Scope is a comprehensive, open suite of sparse autoencoders for Gemma 2 9B and 2B. Sparse Autoencoders are a "microscope" of sorts that can help us break down a model’s internal activations into the underlying concepts, just as biologists use microscopes to study the individual cells of plants and animals. |
|
|
| See our [landing page](https://huggingface.co/google/gemma-scope) for details on the whole suite. This is a specific set of SAEs: |
|
|
| # 2. What Is `gemma-scope-9b-pt-res`? |
|
|
| - `gemma-scope-`: See 1. |
| - `9b-pt-`: These SAEs were trained on Gemma v2 9B base model. |
| - `res`: These SAEs were trained on the model's residual stream. |
| - We include experimental SAEs trained on token embeddings in the ./embedding folder. |
|
|
| # 3. How can I use these SAEs straight away? |
|
|
| ```python |
| from sae_lens import SAE # pip install sae-lens |
| |
| sae, cfg_dict, sparsity = SAE.from_pretrained( |
| release = "gemma-scope-9b-pt-res-canonical", |
| sae_id = "layer_0/width_16k/canonical", |
| ) |
| ``` |
|
|
| See https://github.com/jbloomAus/SAELens for details on this library. |
|
|
| # 4. Point of Contact |
|
|
| Point of contact: Arthur Conmy |
|
|
| Contact by email: |
|
|
| ```python |
| ''.join(list('moc.elgoog@ymnoc')[::-1]) |
| ``` |
|
|
| HuggingFace account: |
| https://huggingface.co/ArthurConmyGDM |
|
|
| # 5. Citation |
|
|
| Paper: https://arxiv.org/abs/2408.05147 |
|
|
