icalens-gemma-2-2b-pile10k

ICA Lens for google/gemma-2-2b

This repository contains a fitted ICA Lens for analyzing internal activations of google/gemma-2-2b. It provides layer-wise ICA transformations for mapping residual-stream activations to independent-component scores and energy shares.

Artifact summary

The analyzed model identity and revision are stored authoritatively in icalens.json; model-card metadata is not used when loading the lens.

Usage

Analyze text end to end:

from icalens import ICALens

lens = ICALens.from_pretrained("REPOSITORY_ID")
result = lens.analyze("She deposited the check at the bank.", layer=0)

print(result.tokens)
print(result.scores)  # signed standard ICA scores
print(result.energy)  # per-token squared-score fractions

Or transform activations captured separately:

scores = lens.transform(activations, layer=0)

Externally captured activations must use the model revision, activation site, layer indexing, and preprocessing recorded in icalens.json.

Score definition

Signed scores are the centered, whitened activations followed by the learned orthogonal ICA rotation. No post-ICA source scaling is applied to v0.2 fits. For a token, component energy share is score² / sum(all component scores²).

Fitting

Fitting provenance

{
  "candidate_tokens": 5255414,
  "context_length": 1024,
  "dataset": {
    "repo_id": "NeelNanda/pile-10k",
    "revision": "127bfedcd5047750df5ccf3a12979a47bfa0bafa",
    "split": "train"
  },
  "document_framing": {
    "evidence_note": "The official tokenizer configuration enables add_bos_token and identifies <bos> as the BOS token.",
    "evidence_url": "https://huggingface.co/google/gemma-2-2b/blob/main/tokenizer_config.json",
    "included_in_fitting_samples": false,
    "policy_source": "project:model_framing.json",
    "registry_model_entry": "google/gemma-2-2b",
    "registry_schema_version": 1,
    "registry_sha256": "c6782bd5d550ed35b5e5b1e63c922907444bf4f40c10fb3f7f8f86858a4093d1",
    "requested": "auto",
    "strategy": "prepend-bos",
    "token": "<bos>",
    "token_id": 2
  },
  "fitting_tokens": 1000000,
  "sampling_seed": 0,
  "text_field": "text",
  "token_scope": "all"
}

Limitations

• Component IDs are specific to a layer and fitted artifact.
• Standard ICA scores are signed and are not probabilities.

Paper

ICA Lens: Interpreting Language Models Without Training Another Dictionary

Citation

@article{liu2026icalens,
  title={ICA Lens: Interpreting Language Models Without Training Another Dictionary},
  author={Liu, Sida and Han, Feijiang},
  journal={arXiv preprint arXiv:2606.11722},
  year={2026}
}