English

Code Comparison Tuning for Code Large Language Models

Computation and Language 2024-06-06 v2

Abstract

We present Code Comparison Tuning (CCT), a simple and effective tuning method for code large language models (Code LLMs) to better handle subtle code errors. Specifically, we integrate the concept of comparison into instruction tuning, both at the token and sequence levels, enabling the model to discern even the slightest deviations in code. To compare the original code with an erroneous version containing manually added code errors, we use token-level preference loss for detailed token-level comparisons. Additionally, we combine code segments to create a new instruction tuning sample for sequence-level comparisons, enhancing the model's bug-fixing capability. Experimental results on the HumanEvalFix benchmark show that CCT surpasses instruction tuning in pass@1 scores by up to 4 points across diverse code LLMs, and extensive analysis demonstrates the effectiveness of our method.

Keywords

Cite

@article{arxiv.2403.19121,
  title  = {Code Comparison Tuning for Code Large Language Models},
  author = {Yufan Jiang and Qiaozhi He and Xiaomin Zhuang and Zhihua Wu},
  journal= {arXiv preprint arXiv:2403.19121},
  year   = {2024}
}

Comments

Preprint