MEGA Hub

Scaling Inherently Interpretable Language Models

Authors

Do you know Guide Labs Team?You can claim authorship or link another user.Do you know Andreas Madsen?You can claim authorship or link another user.Do you know Aya Abdelsalam Ismail?You can claim authorship or link another user.Do you know Giang Nguyen?You can claim authorship or link another user.Do you know Isaac Plant?You can claim authorship or link another user.Do you know Muawiz Chaudhary?You can claim authorship or link another user.Do you know Nathaniel Monson?You can claim authorship or link another user.Do you know Saqib Azim?You can claim authorship or link another user.Do you know Zhichen Guo?You can claim authorship or link another user.Do you know Julius Adebayo?You can claim authorship or link another user.

Abstract

Interpretability is often treated as a tax on capability: language models are trained as opaque systems, then explained after the fact, with methods whose reliability is difficult to establish. In this work, we challenge this premise. Rather than reverse-engineering a model, we make interpretability a constraint of the training pipeline, optimized alongside the language modeling objective. Across three orders of magnitude of compute, on both autoregressive and diffusion language models, interpretability scales with capability rather than against it. Surprisingly, model representations become more disentangled and aligned with human-understandable concepts with scale. We instantiate the training-time recipe with Steerling-8B, a diffusion language model with a causal attention mask. For any group of generated tokens, Steerling-8B attributes the output to relevant input tokens, human-understandable concepts, and training data. This enables closed-loop intervention: diagnose an output through its concept or feature attribution, retrieve similar training data, and correct the behavior through concept steering without retraining. Steerling-8B remains competitive with open peer models trained on substantially 2-16x more compute, suggesting a different scaling paradigm: interpretability can be designed into training, and it improves with scale.

Community

00