Mechanistic Circuit Identification for Controllable Data Generation
Not on the map. Why?
Abstract (arXiv)
While recent advances in data synthesis aim to curate high-quality datasets, most generation pipelines still rely on heuristic prompt-based control. This black-box paradigm provides limited insight into how individual samples interact with a model's underlying learning dynamics. To bridge this gap, we propose a circuit-grounded framework that connects training-dynamics-based data valuation with mechanistic interpretability (MI). Specifically, we conceptualize data quality along three complementary utility axes, learnability, challenge, and alignment. First, we uncover specialized model-internal circuits that causally govern these utility signals. Then, moving beyond heuristic prompting toward mechanistic control, we leverage these circuits as controllable interfaces, actively steering generation to produce utility-targeted data. Building on this capability, we introduce SAMS (Stage-Aware Mechanistic Scheduling), which schedules circuit-steered data according to the model's evolving optimization needs. Experiments on multiple-choice QA tasks demonstrate that our approach yields precisely controlled data with greater diversity than prompt-based baselines, consistently improving downstream performance and calibration. Ultimately, this work establishes a principled white-box paradigm for interpretable data generation, pioneering the use of MI not just as an analytical tool, but as a practical, controllable interface.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 27 of its 36 references that are in the map. First posted on arXiv 25 Aug 2026.
Builds on
27 of 36 references.
References · 27
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
- 2023WizardLM: Empowering large pre-trained language models to follow complex instructionscited by 1,125
- 2023Enhancing Chat Language Models by Scaling High-quality Instructional Conversationscited by 839
- 2017Crowdsourcing Multiple Choice Science Questionscited by 765
- 2020Estimating Training Data Influence by Tracing Gradient Descentcited by 580
- 2021Deep Learning on a Data Diet: Finding Important Examples Early in Trainingcited by 533
- 2023What Makes Good Data for Alignment? A Comprehensive Study of Automatic Data Selection in Instruction Tuningcited by 364
- 2020Identifying Mislabeled Data using the Area Under the Margin Rankingcited by 261
- 2023Large Language Model as Attributed Training Data Generator: A Tale of Diversity and Biascited by 232
- 2023Attribution Patching Outperforms Automated Circuit Discoverycited by 209
- 2024Have Faith in Faithfulness: Going Beyond Circuit Overlap When Finding Model Mechanismscited by 149
15 more
- 2024A Practical Review of Mechanistic Interpretability for Transformer-Based Language Modelscited by 118
- 2024MATES: Model-Aware Data Selection for Efficient Pretraining with Data Influence Modelscited by 100
- 2024A Primer on the Inner Workings of Transformer-based Language Modelscited by 85
- 2023Data Diversity Matters for Robust Instruction Tuningcited by 76
- 2023D2 Pruning: Message Passing for Balancing Diversity and Difficulty in Data Pruningcited by 69
- 2025Synthetic Data Generation Using Large Language Models: Advances in Text and Codecited by 61
- 2024On the Diversity of Synthetic Data and its Impact on Training Large Language Modelscited by 59
- 2023Self-Evolved Diverse Data Sampling for Efficient Instruction Tuningcited by 52
- 2025Prismatic Synthesis: Gradient-based Data Diversification Boosts Generalization in LLM Reasoningcited by 40
- 2023Empowering Cross-lingual Abilities of Instruction-tuned Large Language Models by Translation-following demonstrationscited by 35
- 2024Surveying the Effects of Quality, Diversity, and Complexity in Synthetic Data From Large Language Modelscited by 35
- 2021Estimating Expected Calibration Errorscited by 24
- 2026GradAlign: Gradient-Aligned Data Selection for LLM Reinforcement Learningcited by 12
- 2026Mechanistic Data Attribution: Tracing the Training Origins of Interpretable LLM Unitscited by 10
- 2026propella-1: Multi-Property Document Annotation for LLM Data Curation at Scalecited by 6
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Rethinking Circuit Evaluation: Do Circuits Explain Model Errors?
- 2025Mechanistic Interpretability as Statistical Estimation: A Variance Analysis
- 2025BlackboxNLP-2025 MIB Shared Task: Improving Circuit Faithfulness via Better Edge Selection
- 2026Formal Mechanistic Interpretability: Automated Circuit Discovery with Provable Guarantees
- 2026S^3martCirc: Self-supervised Smart Circuit Discovery
- 2026Certified Mechanistic Interpretability: Lifting Single-Input Findings to Bounded Neighbourhoods
- 2026Certified Circuits: Stability Guarantees for Mechanistic Circuits
- 2026Are We Recovering Mechanisms? Objective-Level Recovery Gaps in Mechanistic Interpretability