Model Directions, Not Words: Mechanistic Topic Models Using Sparse Autoencoders
Zheng, Carolina, Beltran-Velez, Nicolas, Karlekar, Sweta, Shi, Claudia, Nazaret, Achille, Mallik, Asif, Feder, Amir, Blei, David M.
–arXiv.org Artificial Intelligence
Traditional topic models are effective at uncovering latent themes in large text collections. However, due to their reliance on bag-of-words representations, they struggle to capture semantically abstract features. While some neural variants use richer representations, they are similarly constrained by expressing topics as word lists, which limits their ability to articulate complex topics. We introduce Mechanistic Topic Models (MTMs), a class of topic models that operate on interpretable features learned by sparse autoencoders (SAEs). By defining topics over this semantically rich space, MTMs can reveal deeper conceptual themes with expressive feature descriptions. Moreover, uniquely among topic models, MTMs enable controllable text generation using topic-based steering vectors. To properly evaluate MTM topics against word-list-based approaches, we propose \textit{topic judge}, an LLM-based pairwise comparison evaluation framework. Across five datasets, MTMs match or exceed traditional and neural baselines on coherence metrics, are consistently preferred by topic judge, and enable effective steering of LLM outputs.
arXiv.org Artificial Intelligence
Aug-1-2025
- Country:
- Asia
- China > Hubei Province
- Wuhan (0.04)
- Middle East > Jordan (0.04)
- Russia (0.14)
- China > Hubei Province
- Europe
- Greece (0.04)
- Latvia > Lubāna Municipality
- Lubāna (0.04)
- Russia (0.14)
- North America > United States
- Arizona (0.04)
- California > San Bernardino County (0.04)
- Colorado (0.04)
- Gulf of Mexico > Central GOM (0.04)
- Louisiana (0.04)
- Montana > Lewis and Clark County
- Helena (0.04)
- Oceania > Australia (0.04)
- Asia
- Genre:
- Research Report (1.00)
- Industry:
- Government > Regional Government (0.67)
- Health & Medicine > Therapeutic Area
- Immunology (0.67)
- Infections and Infectious Diseases (1.00)
- Law (1.00)
- Leisure & Entertainment > Sports (1.00)
- Technology: