SAELens
ArthurConmyGDM's picture
Update README.md
3c0ccba verified
|
raw
history blame
1.36 kB
metadata
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 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?

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:

''.join(list('moc.elgoog@ymnoc')[::-1])

HuggingFace account: https://huggingface.co/ArthurConmyGDM

5. Citation

Paper: https://arxiv.org/abs/2408.05147