Selected Publications

All Publications

Show all publications
  • J++ Lens: Jacobian Filtering Enables More Faithful Workspace Lenses
    Kola Ayonrinde , Jack Lindsey - 2026
    LessWrong
    Show Abstract

    The J-Lens was proposed to read verbalisable representations from language model activations. We introduce the J++ Lens: an improvement to the J-Lens that enables more faithful readouts across 9–284B models. The J++ Lens filters noisy gradients before they enter the averaged Jacobian. Using the J++ Lens as a drop-in replacement for the J-Lens, we read workspace representations with 55% success on latent variable extraction tasks (J-Lens: 36%, R-Lens: 38%) with especially strong performance at early layers. We believe that more faithful lenses can be useful in monitoring for eval awareness and unverbalised scheming. We release code, an interactive demo and open-source lenses.


  • RouterInterp: Understanding Superposed Specialisation in Mixture of Experts Routing
    Ilya Lasy , Nora Yinuo Cai , Kola Ayonrinde - 2026
    ICML 2026
    Show Abstract

    Sparse Mixture-of-Experts (MoE) models scale more efficiently than dense models by routing tokens to modular expert networks that are only active when relevant to the task. A leading hypothesis for the performance of MoE models is that each expert specialises in a single, coherent domain. However, interpretability efforts that assume this hypothesis have generally been unsuccessful. We propose and present evidence for an alternative account that we call the Superposed Specialisation Hypothesis (SSH): experts specialise in a disjoint union of fine-grained features rather than one broad domain. Leveraging the SSH, we introduce RouterInterp, a method for interpreting expert routing that identifies Sparse Autoencoder features most predictive of routing decisions and produces unified natural-language explanations. On gpt-oss-20b, explanations from RouterInterp predict expert routing with 77% higher accuracy than prior methods. This work provides a scalable method for generating concise and more accurate explanations of expert routing and increases our understanding of a previously uninterpretable component of foundation models.


  • Interpretability Should Prioritise Use-Inspired Basic Research for AI Safety
    Kola Ayonrinde - 2026
    ICML 2026, Philosophy Meets Machine Learning Workshop
    Show Abstract

    There is currently a debate in the field of AI interpretability about whether pragmatic, incremental research which applies simple techniques is a more promising strategy for AI safety applications than ambitious, riskier research which builds new tools, theory and analysis to explain neural networks. This debate closely mirrors the debate in Innovation Economics about how to balance basic science, aiming at understanding, versus applied science, aiming at use, for increasing the rate of technological innovation. We follow Donald Stokes’ approach to this conflict in proposing the synthesis of Use-Inspired Basic Research, an approach in which researchers focus on practical problems, yet aim to deeply understand the relevant phenomena. Ideally, Use-Inspired Basic Research may provide practical solutions that are generalising rather than brittle, and for which we know the domain in which the solutions should and should not work. We show how, by moving their focus towards Use-Inspired Basic Research, researchers currently focused on basic or applied research respectively can increase their rate of scientific innovation. We believe that understanding the drivers of scientific innovation and the role of use-inspired basic research can increase the impact of interpretability research towards AI safety and other socially beneficial goals.


  • Getting Monosemantic About Monosemanticity
    *Raphaël Millière , *Kola Ayonrinde - 2026
    🌟 ICML 2026, Oral at Philosophy Meets Machine Learning Workshop
    Show Abstract

    Mechanistic interpretability often treats monosemanticity as a central target: successful decompositions of neural networks should replace polysemantic neurons with units that carry single meanings. We argue that this target is underspecified. Current research uses several criteria to determine whether a unit is “monosemantic”: that it fires on a coherent set of inputs, that it reliably detects a particular concept, that its behaviour admits a short natural-language description, that it plays a single causal role in the model’s computation, or that it cannot be split into smaller components. These criteria are related but not equivalent, and the same unit can pass under one and fail under another. We argue that criteria for monosemanticity play three different roles, and that distinguishing the roles dissolves much of the apparent disagreement in the literature. Some criteria supply inconclusive evidence that a unit is monosemantic, while others pick out desirable properties for human-intelligibility rather than monosemanticity directly. We argue that only two criteria are actually constitutive of monosemanticity: the unit must map onto a single causally relevant variable in the model’s computation, and decomposing it further must not improve the explanation at the chosen level of grain. Building on these two criteria, we outline a framework that clarifies what feature-level evaluations and interventions can and cannot establish, and provides better reporting norms for claims about monosemanticity.


  • Language Models Can Coarsely Modulate Entropy Under Instruction
    Luca Baroni , Kola Ayonrinde , Shi Feng , Puria Radmard - 2026
    NeurIPS 2026
    Show Abstract

    Recent work has pointed out current models’ limitations in sampling according to prespecified target distributions. In this work, we show that current models nevertheless possess a coarse form of control over their output distributions: by instructing a range of open-source models to maximise or minimise output certainty in a two-alternative forced-choice task with no correct answer, we find that several models can shift entropy in the instructed direction. Through contrastive activation analysis, we further identify a linear direction in activation space that models recruit under uncertainty-modulation instructions and that can be used to causally modulate entropy via activation steering in the absence of any such instructions, suggesting that deliberate entropy control has an identifiable internal representation.


  • Position: AI Should Verify, Not Judge, Scientific Work
    Prabhant Singh , Thanh Gia Hieu Khuong , Vlasta Sikimić , Benedictus Kent Rachmat , Kola Ayonrinde , Ihsan Ullah , Christina Lioma , Kevin Qinghong Lin , Luis Oala , Neil F. Abernethy , Lele Cao , Hilde Weerts , Joaquin Vanschoren - 2026
    🌟 ICML 2026, Oral at AI for Science Workshop
    Show Abstract

    AI reviewers are gaining more attention with the advent of LLMs. With the increasing volume of conference papers, the use of AI-based reviewers has been suggested and implemented to enable faster review cycles, improve review quality, and help the scientific community. In this position paper, we argue against the replacement of peer review by AI reviewers and advocate the use of AI review tools primarily to verify claims and improve the quality of scientific work. We argue that AI review tools should be utilised primarily by authors during manuscript preparation to improve submission quality and streamline downstream evaluation. We ground our argument in the values of peer review and the scientific process. Finally, we present research directions for the responsible integration of AI in the peer review process.


  • From Mechanistic to Compositional Interpretability
    Ward Gauderis , Thomas Dooms , Steven T. Homer , Kola Ayonrinde , Geraint A. Wiggins - 2026
    arXiv
    Show Abstract

    Mechanistic interpretability aims to explain neural model behaviour by reverse-engineering learned computational structure into human-understandable components. Without a formal framework, however, mechanistic explanations cannot be objectively verified, compared, or composed. We introduce compositional interpretability, a category-theoretic framework grounded in the principles of compositionality and minimum description length. Compositional interpretations are pairs of syntactic and semantic mappings that must commute to enforce consistency between a model’s decomposition and its observed behaviour. We deconstruct explanation quality into measures of faithfulness and complexity to cast interpretability as a constrained optimisation problem, and introduce compressive refinement to systematically restructure models into simpler parts without altering their function. Finally, we derive a parsimony criterion under which syntactic compression theoretically guarantees more concise, human-aligned explanations. Our framework situates prominent mechanistic methods as subclasses of refinement, and clarifies why their compressibility heuristics tend to align with human interpretability. Our work provides a measurable, optimisable blueprint for automating the discovery and evaluation of mechanistic explanations.


  • Multi-shot AutoInterp: Agents Can Explain Complex Features By Refining Explanations
    Kamal Maher , Simon Elias Schrader , Kola Ayonrinde - 2026
    ICLR 2026, HCAIR Workshop
    Show Abstract

    Interpretability studies often show that some features of language models are easily interpretable by humans while others are difficult to explain. If such features are truly beyond human grasp, the field of interpretability may have fundamental limits. We notice, however, that previous human and automated interpretability studies typically seek single-shot explanations, and we hypothesise that the lack of interaction may be a cause for features seeming uninterpretable. To add interaction to feature interpretability, we introduce Multi-Shot AutoInterp (MSA), an agentic framework where LLM agents iteratively refine feature explanations by forming hypotheses, performing targeted experiments on new sequences, and incorporating feedback from feature activations. We find that MSA agents achieve improved AutoInterp performance on difficult-to-interpret sparse autoencoder features with 7.7% greater prediction accuracy than current AutoInterp methods. In addition, MSA agents produced insightful explanations for features that initially appeared opaque to the authors, for example illuminating the meaning of a feature that is activated on tokens representing conceptual parallelism. Our results suggest that features that were previously thought of as uninterpretable may instead be simply not amenable to single-shot explanation, which increases our estimation of how many features are ultimately human-understandable. We argue that through multi-shot approaches humans may be able to understand more of the elusive safety-relevant features in a language model or learn novel, AI-native scientific concepts that were not previously salient to humans.


  • AISI Frontier AI Trends Report (2025)
    Abby D’Cruz, Alastair Pearson , …, Kola Ayonrinde, …, Xander Davies - 2025
    AI Security Institute
    Show Abstract

    The UK AI Security Institute (AISI) has conducted evaluations of frontier AI systems since November 2023 across domains critical to national security and public safety. This report presents our first public analysis of the trends we’ve observed. It seeks to provide accessible, data-driven insights into the frontier of AI capabilities and promote a shared understanding among governments, industry, and the public.


  • Auditing Games for Sandbagging
    Jordan Taylor , Sid Black , Dillon Bowen , Thomas Read , Satvik Golechha , Alex Zelenka-Martin , Oliver Makins , Connor Kissane , Kola Ayonrinde , Jacob Merizian , Samuel Marks , Chris Cundy , Joseph Bloom - 2025
    arXiv
    Show Abstract

    Future AI systems could conceal their capabilities (“sandbagging”) during evaluations, potentially misleading developers and auditors. We stress-tested sandbagging detection techniques using an auditing game. First, a red team fine-tuned five models, some of which conditionally underperformed, as a proxy for sandbagging. Second, a blue team used black-box, model-internals, or training-based approaches to identify sandbagging models. We found that the blue team could not reliably discriminate sandbaggers from benign models. Black-box approaches were defeated by effective imitation of a weaker model. Linear probes, a model-internals approach, showed more promise but their naive application was vulnerable to behaviours instilled by the red team. We also explored capability elicitation as a strategy for detecting sandbagging. Although prompt-based elicitation was not reliable, training-based elicitation consistently elicited full performance from the sandbagging models, using only a single correct demonstration of the evaluation task. However, the performance of benign models was sometimes also raised, so relying on elicitation as a detection strategy was prone to false positives. In the short term, we recommend developers remove potential sandbagging using on-distribution training for elicitation. In the longer term, further research is needed to ensure the efficacy of training-based elicitation and develop robust methods for sandbagging detection.


  • Evaluating Explanations: An Explanatory Virtues Framework for Mechanistic Interpretability -- The Strange Science Part I.ii
    Kola Ayonrinde , Louis Jaburi - 2025
    NeurIPS 2025, Mechanistic Interpretability Workshop
    Show Abstract

    Mechanistic Interpretability (MI) aims to understand neural networks through causal explanations. Though MI has many explanation-generating methods, progress has been limited by the lack of a universal approach to evaluating explanations. Here we analyse the fundamental question “What makes a good explanation?” We introduce a pluralist Explanatory Virtues Framework drawing on four perspectives from the Philosophy of Science - the Bayesian, Kuhnian, Deutschian, and Nomological - to systematically evaluate and improve explanations in MI. We find that Compact Proofs consider many explanatory virtues and are hence a promising approach. Fruitful research directions implied by our framework include (1) clearly defining explanatory simplicity, (2) focusing on unifying explanations and (3) deriving universal principles for neural networks. Improved MI methods enhance our ability to monitor, predict, and steer AI systems.


  • A Mathematical Philosophy of Explanations in Mechanistic Interpretability -- The Strange Science Part I.i
    Kola Ayonrinde , Louis Jaburi - 2025
    AIES 2025
    Show Abstract

    Mechanistic Interpretability aims to understand neural networks through causal explanations. We argue for the Explanatory View Hypothesis: that Mechanistic Interpretability research is a principled approach to understanding models because neural networks contain implicit explanations which can be extracted and understood. We hence show that Explanatory Faithfulness, an assessment of how well an explanation fits a model, is well-defined. We propose a definition of Mechanistic Interpretability (MI) as the practice of producing Model-level, Ontic, Causal-Mechanistic, and Falsifiable explanations of neural networks, allowing us to distinguish MI from other interpretability paradigms and detail MI’s inherent limits. We formulate the Principle of Explanatory Optimism, a conjecture which we argue is a necessary precondition for the success of Mechanistic Interpretability.


  • SAEBench: A Comprehensive Benchmark for Sparse Autoencoders in Language Model Interpretability
    Adam Karvonen , Can Rager , Johnny Lin , Curt Tigges , Joseph Bloom , David Chanin , Yeu-Tong Lau , Eoin Farrell , Callum McDougall , Kola Ayonrinde , Demian Till , Matthew Wearden , Arthur Conmy , Samuel Marks , Neel Nanda - 2025
    ICML 2025
    Show Abstract

    Sparse autoencoders (SAEs) are a popular technique for interpreting language model activations, and there is extensive recent work on improving SAE effectiveness. However, most prior work evaluates progress using unsupervised proxy metrics with unclear practical relevance. We introduce SAEBench, a comprehensive evaluation suite that measures SAE performance across eight diverse metrics, spanning interpretability, feature disentanglement and practical applications like unlearning. To enable systematic comparison, we open-source a suite of over 200 SAEs across eight recently proposed SAE architectures and training algorithms. Our evaluation reveals that gains on proxy metrics do not reliably translate to better practical performance. For instance, while Matryoshka SAEs slightly underperform on existing proxy metrics, they substantially outperform other architectures on feature disentanglement metrics; moreover, this advantage grows with SAE scale. By providing a standardized framework for measuring progress in SAE development, SAEBench enables researchers to study scaling trends and make nuanced comparisons between different SAE architectures and training methodologies. Our interactive interface enables researchers to flexibly visualize relationships between metrics across hundreds of open-source SAEs at: this http URL


  • Position: Interpretability is a Bidirectional Communication Problem
    Kola Ayonrinde - 2025
    ICLR 2025, Bidirectional Alignment Workshop
    Show Abstract

    Interpretability is the process of explaining neural networks in a human-understandable way. A good explanation has three core components: it is (1) faithful to the explained model, (2) understandable to the interpreter, and (3) effectively communicated. We argue that current mechanistic interpretability methods focus primarily on faithfulness and could improve by additionally considering the human interpreter and communication process. We propose and analyse two approaches to \emph{Concept Enrichment} for the human interpreter – \emph{Pre-Explanation Learning} and \emph{Mechanistic Socratic Explanation} – approaches to using the AI’s representations to teach the interpreter novel and useful concepts. We reframe the Interpretability Problem as a Bidirectional Communication Problem between the model and the interpreter, highlighting interpretability’s pedagogical aspects. We suggest that Concept Enrichment may be a key way to aid Conceptual Alignment between AIs and humans for improved mutual understanding.


  • Adaptive Sparse Allocation with Mutual Choice & Feature Choice Sparse Autoencoders
    Kola Ayonrinde - 2024
    arXiv
    Show Abstract

    Sparse autoencoders (SAEs) are a promising approach to extracting features from neural networks, enabling model interpretability as well as causal interventions on model internals. SAEs generate sparse feature representations using a sparsifying activation function that implicitly defines a set of token-feature matches. We frame the token-feature matching as a resource allocation problem constrained by a total sparsity upper bound. For example, TopK SAEs solve this allocation problem with the additional constraint that each token matches with at most k features. In TopK SAEs, the k active features per token constraint is the same across tokens, despite some tokens being more difficult to reconstruct than others. To address this limitation, we propose two novel SAE variants, Feature Choice SAEs and Mutual Choice SAEs, which each allow for a variable number of active features per token. Feature Choice SAEs solve the sparsity allocation problem under the additional constraint that each feature matches with at most m tokens. Mutual Choice SAEs solve the unrestricted allocation problem where the total sparsity budget can be allocated freely between tokens and features. Additionally, we introduce a new auxiliary loss function, 𝚊𝚞𝚡_𝚣𝚒𝚙𝚏_𝚕𝚘𝚜𝚜, which generalises the 𝚊𝚞𝚡_𝚔_𝚕𝚘𝚜𝚜 to mitigate dead and underutilised features. Our methods result in SAEs with fewer dead features and improved reconstruction loss at equivalent sparsity levels as a result of the inherent adaptive computation. More accurate and scalable feature extraction methods provide a path towards better understanding and more precise control of foundation models.


  • Interpretability as Compression: Reconsidering SAE Explanations of Neural Activations with MDL-SAEs
    *Kola Ayonrinde , *Michael Pearce , Lee Sharkey - 2024
    🌟 NeurIPS 2024, Oral at InterpretableAI Workshop
    Show Abstract

    Sparse Autoencoders (SAEs) have emerged as a useful tool for interpreting the internal representations of neural networks. However, naively optimising SAEs for reconstruction loss and sparsity results in a preference for SAEs that are extremely wide and sparse. We present an information-theoretic framework for interpreting SAEs as lossy compression algorithms for communicating explanations of neural activations. We appeal to the Minimal Description Length (MDL) principle to motivate explanations of activations which are both accurate and concise. We further argue that interpretable SAEs require an additional property, “independent additivity”: features should be able to be understood separately. We demonstrate an example of applying our MDL-inspired framework by training SAEs on MNIST handwritten digits and find that SAE features representing significant line segments are optimal, as opposed to SAEs with features for memorised digits from the dataset or small digit fragments. We argue that using MDL rather than sparsity may avoid potential pitfalls with naively maximising sparsity such as undesirable feature splitting and that this framework naturally suggests new hierarchical SAE architectures which provide more concise explanations.