paper-with-me

홈 › Papers

Scalable Circuit Learning for Interpreting Large Language Models

2026-06-15 · Naiyu Yin, Dennis Wei, Tian Gao, Amit Dhurandhar, Karthikeyan Natesan Ramamurthy, Yue Yu arxiv

A prominent research direction in mechanistic interpretability is learning sparse circuits over LLM components to reveal how they jointly produce model behavior. However, raw neurons are polysemantic, making learned circuits hard to interpret. Sparse autoencoder (SAE) features alleviate this, but their high dimensionality makes existing intervention-based circuit learning methods computationally prohibitive. We propose CircuitLasso, a scalable circuit-learning approach based on sparse linear regression. CircuitLasso recovers circuits whose structural accuracy matches that of state-of-the-art intervention-based methods on the benchmark data, at a fraction of the computational cost. For interpretability, CircuitLasso efficiently uncovers relationships among SAE features, showing how human-interpretable semantic features propagate through the model and influence its predictions. Finally, we validate the utility of our learned circuits by leveraging their insights to achieve comparable performance at substantially lower cost on a domain-generalization task.

📄 PDF Abstract BibTeX arXiv:2606.16939

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Finding Transformer Circuits with Edge Pruning

2024-06-24 · Adithya Bhaskar, Alexander Wettig, Dan Friedman, Danqi Chen

The path to interpreting a language model often proceeds via analysis of circuits -- sparse computational subgraphs of the model that capture specific aspects of its behavior. Recent work has automated the task of discov…

In-Context LearningLanguage Modelling

Causal Head Gating: A Framework for Interpreting Roles of Attention Heads in Transformers

2025-05-19 · Andrew Nam, Henry Conklin, Yukang Yang, Thomas Griffiths 외

We present causal head gating (CHG), a scalable method for interpreting the functional roles of attention heads in transformer models. CHG learns soft gates over heads and assigns them a causal taxonomy - facilitating, i…

In-Context LearningInstruction FollowingMathematical Reasoning

MuaLLM: A Multimodal Large Language Model Agent for Circuit Design Assistance with Hybrid Contextual Retrieval-Augmented Generation

2025-08-11 · Pravallika Abbineni, Saoud Aldowaish, Colin Liechty, Soroosh Noorzad 외 arxiv

Conducting a comprehensive literature review is crucial for advancing circuit design methodologies. However, the rapid influx of state-of-the-art research, inconsistent data representation, and the complexity of optimizi…

Information Retrieval

Mechanistic Interpretability for Large Language Model Alignment: Progress, Challenges, and Future Directions

2026-01-21 · Usman Naseem arxiv

Large language models (LLMs) have achieved remarkable capabilities across diverse tasks, yet their internal decision-making processes remain largely opaque. Mechanistic interpretability (i.e., the systematic study of how…

Reinforcement Learning

Discovering Transformer Circuits via a Hybrid Attribution and Pruning Framework

2025-09-28 · Hao Gu, Vibhas Nair, Amrithaa Ashok Kumar, Jayvart Sharma 외 arxiv

Interpreting language models often involves circuit analysis, which aims to identify sparse subnetworks, or circuits, that accomplish specific tasks. Existing circuit discovery algorithms face a fundamental trade-off: at…