ITER: Iterative Neural Repair for Multi-Location Patches
He Ye, Martin Monperrus
TL;DR
ITER introduces an iterative neural program repair paradigm that refines partial patches toward correctness by interleaving compilation-error and functional-error repair within a single model. It combines iterative training (self-supervised data augmentation) and iterative inference (re-executed fault localization guided by partial patches) to repair both single- and multi-location bugs. On Defects4J v2.0, ITER fixes 74 bugs and plausibly patches 119, including 9 unique multi-location fixes, outperforming state-of-the-art baselines. The approach also demonstrates notable multi-location repair capabilities and provides insight into time-cost trade-offs, with a pipeline that can be applied to Java programs and potentially other languages. Public code and data are available to support reproducibility and further research.
Abstract
Automated program repair (APR) has achieved promising results, especially using neural networks. Yet, the overwhelming majority of patches produced by APR tools are confined to one single location. When looking at the patches produced with neural repair, most of them fail to compile, while a few uncompilable ones go in the right direction. In both cases, the fundamental problem is to ignore the potential of partial patches. In this paper, we propose an iterative program repair paradigm called ITER founded on the concept of improving partial patches until they become plausible and correct. First, ITER iteratively improves partial single-location patches by fixing compilation errors and further refining the previously generated code. Second, ITER iteratively improves partial patches to construct multi-location patches, with fault localization re-execution. ITER is implemented for Java based on battle-proven deep neural networks and code representation. ITER is evaluated on 476 bugs from 10 open-source projects in Defects4J 2.0. ITER succeeds in repairing 15.5% of them, including 9 uniquely repaired multi-location bugs.
