English

CircuitProbe: Predicting Reasoning Circuits in Transformers via Stability Zone Detection

Artificial Intelligence 2026-04-02 v1 Machine Learning

Abstract

Transformer language models contain localized reasoning circuits, contiguous layer blocks that improve reasoning when duplicated at inference time. Finding these circuits currently requires brute-force sweeps costing 25 GPU hours per model. We propose CircuitProbe, which predicts circuit locations from activation statistics in under 5 minutes on CPU, providing a speedup of three to four orders of magnitude. We find that reasoning circuits come in two types: stability circuits in early layers, detected through the derivative of representation change, and magnitude circuits in late layers, detected through anomaly scoring. We validate across 9 models spanning 6 architectures, including 2025 models, confirming that CircuitProbe top predictions match or are within 2 layers of the optimal circuit in all validated cases. A scaling experiment across the Qwen 2.5 family reveals that layer duplication consistently benefits models under 3B parameters but degrades performance in 7B+ models, making this a practical scaling technique for small language models. CircuitProbe requires as few as 10 calibration examples and its predictions are stable across English, Hindi, Chinese, and French.

Keywords

Cite

@article{arxiv.2604.00716,
  title  = {CircuitProbe: Predicting Reasoning Circuits in Transformers via Stability Zone Detection},
  author = {Rajkiran Panuganti},
  journal= {arXiv preprint arXiv:2604.00716},
  year   = {2026}
}

Comments

11 pages, 1 figure, 3 tables. Code available at https://github.com/agenticclass/circuitprobe

R2 v1 2026-07-01T11:47:58.958Z