Code Less, Align More: Efficient LLM Fine-tuning for Code Generation with Data Pruning

Yun-Da Tsai, Mingjie Liu, Haoxing Ren
7/6/2024
cs.SEcs.LG

No confident code implementation found for this paper. 5 possible matches can be shown below.

Abstract

Recent work targeting large language models (LLMs) for code generation demonstrated that increasing the amount of training data through synthetic code generation often leads to exceptional performance. In this paper we explore data pruning methods aimed at enhancing the efficiency of model training specifically for code LLMs. We present techniques that integrate various clustering and pruning metrics to selectively reduce training data without compromising the accuracy and functionality of the generated code. We observe significant redundancies in synthetic training data generation, where our experiments demonstrate that benchmark performance can be largely preserved by training on only 10% of the data. Moreover, we observe consistent improvements in benchmark results through moderate pruning of the training data. Our experiments show that these pruning strategies not only reduce the computational resources needed but also enhance the overall quality code generation.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 5 weaker matches are hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Cite this paper

@article{tsai2024code,
  title  = {Code Less, Align More: Efficient LLM Fine-tuning for Code Generation with Data Pruning},
  author = {Yun-Da Tsai and Mingjie Liu and Haoxing Ren},
  year   = {2024},
  eprint = {2407.05040},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2407.05040v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2407.05040.svg)](https://codeofpaper.com/paper/2407.05040)

Discussion