Model Directions, Not Words: Mechanistic Topic Models Using Sparse Autoencoders
About
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 steering vectors. To properly evaluate MTM topics against word list approaches, we propose \textit{topic judge}, an LLM-based pairwise comparison evaluation framework. Across eight datasets, MTMs match or exceed traditional and neural baselines on coherence metrics, are consistently preferred by topic judge, and enable effective LLM steering.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Topic Modeling | Wiki | Topic Judge Elo Score1.66e+3 | 24 | |
| Topic Modeling | Bills | P1 Score0.52 | 16 | |
| Topic Modeling | 20NG (20 Newsgroups) | Topic Judge Elo Score1.61e+3 | 16 | |
| Topic Modeling | Yelp Polarity | Topic Judge Elo Score1.72e+3 | 16 | |
| Topic Modeling | AG-News | Topic Judge Elo Score1.61e+3 | 16 | |
| Topic Modeling | GoEmotions | Elo Score (Topic Judge)1.73e+3 | 16 | |
| Topic Modeling | Poem PoemSum | Topic Judge Elo Score1.68e+3 | 16 | |
| Topic Modeling | WriPro (WritingPrompts) | Topic Judge Elo Score1.81e+3 | 16 | |
| Topic Modeling | Bills U.S. Congress summaries | Topic Judge Elo Score1.64e+3 | 16 | |
| Topic Coherence | GoEmotions | Rating Score2.8 | 8 |