APPT: Boosting Automated Patch Correctness Prediction via Fine-tuning Pre-trained Models

TL;DR

This paper introduces APPT, a novel approach that fine-tunes pre-trained models for automated patch correctness prediction, significantly improving accuracy over existing methods by fully leveraging pre-trained representations.

Contribution

The paper proposes a fine-tuning framework for pre-trained models in patch correctness prediction, enhancing representation quality and overall prediction performance.

Findings

APPT achieves 79.7% accuracy and 83.2% recall, outperforming previous methods.

Fine-tuning and LSTM components significantly improve prediction performance.

Advanced pre-trained models further enhance the generalizability and effectiveness of APPT.

Abstract

Automated program repair (APR) aims to fix software bugs automatically without human debugging efforts and plays a crucial role in software development and maintenance. Despite promising, APR is still challenged by a long-standing overfitting problem (i.e., the generated patch is plausible but overfitting). Various techniques have thus been proposed to address the overfitting problem. Recently, researchers have employed BERT to extract code features, which are then used to train a classifier for patch correctness prediction. However, BERT is restricted to feature extraction for classifier training without benefiting from the training process, potentially generating sub-optimal vector representations for patched code snippets. In this paper, we propose APPT, a pre-trained model-based automated patch correctness assessment technique by both pre-training and fine-tuning. APPT adopts a pre-trained model as the encoder stack, followed by an LSTM stack and a deep learning classifier. More importantly, the pre-trained model is fine-tuned in conjunction with other components as a whole pipeline to fully adapt it specifically for reasoning about patch correctness. We conduct an extensive experiment on 1,183 Defects4J patches and the experimental results show that APPT achieves prediction accuracy of 79.7% and recall of 83.2%, outperforming CACHE by 4.3% and 6.7%. Our additional investigation on 49,694 real-world patches shows that APPT achieves the optimum performance compared with existing representation learning techniques. We further investigate the impact of each component and find that they all positively contribute to APPT, e.g., the fine-tuning process and the LSTM stack increase F1-score by 10.22% and 4.11%, respectively. We also prove that adopting advanced pre-trained models can further provide substantial advancement, highlighting the generalizability of APPT.