Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Carolina Zheng, Nicolas Beltran-Velez, Sweta Karlekar, Claudia Shi, Achille Nazaret, Asif Mallik, Amir Feder, David M. Blei• 2025

Related benchmarks

TaskDatasetResultRank
Topic ModelingWiki
Topic Judge Elo Score1.66e+3
24
Topic ModelingBills
P1 Score0.52
16
Topic Modeling20NG (20 Newsgroups)
Topic Judge Elo Score1.61e+3
16
Topic ModelingYelp Polarity
Topic Judge Elo Score1.72e+3
16
Topic ModelingAG-News
Topic Judge Elo Score1.61e+3
16
Topic ModelingGoEmotions
Elo Score (Topic Judge)1.73e+3
16
Topic ModelingPoem PoemSum
Topic Judge Elo Score1.68e+3
16
Topic ModelingWriPro (WritingPrompts)
Topic Judge Elo Score1.81e+3
16
Topic ModelingBills U.S. Congress summaries
Topic Judge Elo Score1.64e+3
16
Topic CoherenceGoEmotions
Rating Score2.8
8
Showing 10 of 22 rows

Other info

Follow for update