arXiv Machine Learning

LLMs Can Annotate Attribution Graphs

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.

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
arXiv Machine Learning
Aug 31

How Do Linear Probes Emerge? A Circuit-Tracing Framework with Concept-Targeted Attribution

The paper introduces Concept-Targeted Attribution (CTA), a method that trains attribution graphs to explain the emergence of internal concept representations in language models, rather than just the final token prediction. CTA produces probe-specific circuits that reveal which internal computations drive a linear probe’s accuracy, and cross-layer transcoders demonstrate that these graphs contain predictive structure across multiple concept categories. Causal ablations show that probe-targeted and logit-targeted graphs capture distinct mechanisms, with probe-relevant features affecting internal concept scores and logit-relevant features altering generated tokens.

By Vedant Palit, Florent Draye, Terry Jingchen Zhang, Bernhard Sch\"olkopf, Zhijing Jin
arXiv Machine Learning
Oct 19

RAW-Explainer: Post-hoc Explanations of Path Reasoning Graph Neural Networks for Knowledge Graph Completion

RAW-Explainer is a new explainability method designed for path reasoning graph neural networks used in knowledge graph completion. It produces connected, concise, and interpretable subgraph explanations for link prediction by leveraging heterogeneous information and a random walk objective. The approach also addresses distribution shift when evaluating small explanatory subgraphs with a robust evaluator, achieving a balance between explanation quality and computational efficiency on standard benchmarks.

By Ryoji Kubo, Samyam Lamichhane, Djellel Difallah
Hugging Face Trending Papers
Jun 9

Generative Explainability for Next-Generation Networks: LLM-Augmented XAI with Mutual Feature Interactions

As artificial intelligence and machine learning (AI/ML) models become integral to network operations, their lack of transparency poses a significant barrier to operator trust. Existing explainable artificial intelligence (XAI) techniques often fail to bridge this gap for non-specialists, producing technical outputs that are difficult to translate into actionable insights.