Xiangyuan Wang
Author directory2026
Beyond Analysis: Training Language Models with Internal Mechanistic Feedback
Hanlin Xue | Gang Zhang | Xiangyuan Wang | Yunhao Bai | Yulong Wan | Xiang Li | Jiaqi Zhou | Ruochong Xiong | Junfei Liu
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Hanlin Xue | Gang Zhang | Xiangyuan Wang | Yunhao Bai | Yulong Wan | Xiang Li | Jiaqi Zhou | Ruochong Xiong | Junfei Liu
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Recent advances in mechanistic interpretability have revealed how language models process information, yet these insights rarely improve model performance. We propose Interpretable Feature-Space Regularization (IFSR), a training-time framework that transforms mechanistic insights into optimization signals. IFSR identifies error-prone feature interactions through circuit attribution and penalizes them during training, encouraging the model to discover alternative computational pathways. Unlike inference-time interventions that operate on individual features, IFSR targets feature-to-feature edges representing internal computational patterns, and permanently encodes improvements into model parameters. Experiments across ten classification tasks show consistent improvements. Cross-task evaluation demonstrates that IFSR training can transfer positively to unrelated tasks, suggesting benefits to general model capabilities. Our analysis reveals that most identified error patterns resist human interpretation, yet penalizing them still improves performance, suggesting that automatic error identification at the level of feature interactions is feasible and effective. This work demonstrates that mechanistic interpretability can directly enhance task performance through training-time optimization.