arXiv Machine Learning By Ameen Patel, Max Zhang, Nathan Hu

LLMs Can Annotate Attribution Graphs

Read the original on arXiv Machine Learning →

arXiv:2608. 02632v1 Announce Type: new Abstract: Circuit tracing is an exciting technique for revealing the internal computation of language models, but it requires a time-intensive manual step of grouping individual features or MLP neurons into supernodes.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.

arXiv Computation and Language
Nov 4

Circuit Oracle: Automating Attribution Graph Analysis via Natural-Language Queries

Circuit Oracle is an off‑the‑shelf large language model that, using a multi‑agent harness and task‑specific skills, can answer natural‑language questions about a target model by reading its attribution graph via tool calls. The system was evaluated on three safety‑relevant proxy tasks—detecting spurious features in probe circuits, eliciting hidden knowledge from taboo‑finetuned models, and suppression jailbreaking via feature‑level interventions—achieving 78.2 % accuracy on spurious‑feature detection and mixed results on the other tasks. In addition to task scores, Circuit Oracle produces pruned subgraphs and context‑specific feature explanations, which the authors release for further study.

By Shariar Kabir, Hong Kiat Tan, Swastik Agrawal, Sai V R Chereddy, Sriram Balasubramanian
arXiv Machine Learning
Oct 23

On the Token Collapse Phenomenon in Textual Graph Soft Prompting

The paper investigates graph soft prompting, where a graph neural network generates graph-specific tokens for large language model prompts. It identifies a problem called token collapse, in which these tokens become nearly identical across different inputs, rendering them ineffective at conveying graph information. The authors show that even simple constant prompts can match or outperform collapsed tokens, and propose a two‑stage method to produce more varied tokens while maintaining performance.

By Andrea Giuseppe Di Francesco, Rishabh Jain, Franco Maria Nardini, Nicola Tonellotto, Pietro Li\`o, Fabrizio Silvestri
arXiv AI
Sep 2

S^3martCirc: Self-supervised Smart Circuit Discovery

S^3martCirc is a self‑supervised framework that jointly discovers and interprets neural circuits in large language models, rather than treating circuit discovery and functional interpretation as separate stages. It abstracts node behavior into two general computational roles that generalize across tasks and introduces a quantitative metric for assigning these roles, enabling simultaneous identification of importance and function. Experiments demonstrate that S^3martCirc outperforms existing methods in circuit discovery.

By Wendy Zheng, Yinhan He, Liang Wu, Jundong Li