Nikita Koriagin
Author directory2026
A Model with No Head and Many Thoughts
Nikita Koriagin | Yaroslav Aksenov | George Bredis | Gleb Gerasimov | Nikita Balagansky | Daniil Gavrilov
Findings of the Association for Computational Linguistics: EMNLP 2026
Nikita Koriagin | Yaroslav Aksenov | George Bredis | Gleb Gerasimov | Nikita Balagansky | Daniil Gavrilov
Findings of the Association for Computational Linguistics: EMNLP 2026
Large language models decode by projecting hidden states through a large vocabulary head at every step. This operation is computationally costly and forces all reasoning to be expressed in discrete tokens. We introduce Soft Latent Thinking, a method that replaces the LM head during reasoning with a lightweight projector, enabling autoregressive rollout in embedding space where reasoning steps remain continuous rather than tokenized. Experiments on DeepSeek-Qwen-1.5B and LLaMA-3.2-3B show that Soft Latent Thinking consistently improves pass@k across all k while reducing per-step compute during chain-of-thought. Our method achieves the highest pass@32 among all soft-thinking approaches, demonstrating that effective reasoning can be carried out in continuous space without discrete token generation.
2025
Train One Sparse Autoencoder Across Multiple Sparsity Budgets to Preserve Interpretability and Accuracy
Nikita Balagansky | Yaroslav Aksenov | Daniil Laptev | Vadim Kurochkin | Gleb Gerasimov | Nikita Koriagin | Daniil Gavrilov
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Nikita Balagansky | Yaroslav Aksenov | Daniil Laptev | Vadim Kurochkin | Gleb Gerasimov | Nikita Koriagin | Daniil Gavrilov
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Sparse Autoencoders (SAEs) have proven to be powerful tools for interpreting neural networks by decomposing hidden representations into disentangled, interpretable features via sparsity constraints. However, conventional SAEs are constrained by the fixed sparsity level chosen during training; meeting different sparsity requirements therefore demands separate models and increases the computational footprint during both training and evaluation. We introduce a novel training objective, HierarchicalTopK, which trains a single SAE to optimise reconstructions across multiple sparsity levels simultaneously. Experiments with Gemma-2 2B demonstrate that our approach achieves Pareto-optimal trade-offs between sparsity and explained variance, outperforming traditional SAEs trained at individual sparsity levels. Further analysis shows that HierarchicalTopK preserves high interpretability scores even at higher sparsity. The proposed objective thus closes an important gap between flexibility and interpretability in SAE design.