nla-gemma3-12b-L32-av
The AV (activation verbalizer, vector โ text) half of a Natural Language Autoencoder (NLA) pair,
fine-tuned from google/gemma-3-12b-it. The
other half is kitft/nla-gemma3-12b-L32-ar; both are
released together and are intended to be used as a pair.
NLA pairs are interpretability tools: the AV (activation verbalizer) maps a hidden-state vector to a natural-language description; the AR (activation reconstructor) maps that description back to a vector. Together they let you read out what a residual-stream activation "means" and measure how much of it the description captured. These checkpoints are not useful as general-purpose language models โ the fine-tuning repurposes them entirely for activation decoding.
- ๐ Paper: Natural Language Autoencoders Produce Unsupervised Explanations of LLM Activations
- Inference code + worked examples:
kitft/nla-inference - Training code:
kitft/natural_language_autoencoders - Extraction layer: residual stream output of block 32
- In-distribution fve_nrm: 0.768 (training set, 50/50 WildChat + Ultra-FineWeb)
Usage
See the nla-inference README for the
full recipe (SGLang launch, NLAClient/NLACritic, embedding-injection
details).
Citation
@article{frasertaliente2026nla,
author = {Fraser-Taliente, Kit and Kantamneni, Subhash and Ong, Euan and Mossing, Dan and Lu, Christina and Bogdan, Paul C. and Ameisen, Emmanuel and Chen, James and Kishylau, Dzmitry and Pearce, Adam and Tarng, Julius and Wu, Alex and Wu, Jeff and Zhang, Yang and Ziegler, Daniel M. and Hubinger, Evan and Batson, Joshua and Lindsey, Jack and Zimmerman, Samuel and Marks, Samuel},
title = {Natural Language Autoencoders Produce Unsupervised Explanations of LLM Activations},
journal = {Transformer Circuits Thread},
year = {2026},
url = {https://transformer-circuits.pub/2026/nla/index.html}
}
License & use restrictions
This model is a derivative of Gemma 3 and is provided under and subject to the Gemma Terms of Use. By using this model you agree to those terms and the Gemma Prohibited Use Policy. See NOTICE in this repository.
Training data attribution
The fine-tuning data was derived from two public datasets:
- WildChat-1M (allenai/WildChat-1M). Contains information from WildChat-1M which is made available under the ODC Attribution License.
- Ultra-FineWeb (openbmb/Ultra-FineWeb, Apache-2.0), a filtered derivative of HuggingFaceFW/fineweb (ODC-BY).
- Downloads last month
- 791