Saved in:
Bibliographic Details
Main Authors: Lyu, Weijie, Huang, Sheng-Jun, Xia, Xuan
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2507.02378
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911036406235136
author Lyu, Weijie
Huang, Sheng-Jun
Xia, Xuan
author_facet Lyu, Weijie
Huang, Sheng-Jun
Xia, Xuan
contents Recent advancements in large language models (LLMs) have significantly improved code generation and program comprehension, accelerating the evolution of software engineering. Current methods primarily enhance model performance by leveraging vast amounts of data, focusing on data quantity while often overlooking data quality, thereby reducing training efficiency. To address this, we introduce an approach that utilizes a parametric model for code data selection, aimed at improving both training efficiency and model performance. Our method optimizes the parametric model to ensure distribution consistency and diversity within the selected subset, guaranteeing high-quality data. Experimental results demonstrate that using only 10K samples, our method achieves gains of 2.4% (HumanEval) and 2.3% (MBPP) over 92K full-sampled baseline, outperforming other sampling approaches in both performance and efficiency. This underscores that our method effectively boosts model performance while significantly reducing computational costs.
format Preprint
id arxiv_https___arxiv_org_abs_2507_02378
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Efficient Code LLM Training via Distribution-Consistent and Diversity-Aware Data Selection
Lyu, Weijie
Huang, Sheng-Jun
Xia, Xuan
Computation and Language
Recent advancements in large language models (LLMs) have significantly improved code generation and program comprehension, accelerating the evolution of software engineering. Current methods primarily enhance model performance by leveraging vast amounts of data, focusing on data quantity while often overlooking data quality, thereby reducing training efficiency. To address this, we introduce an approach that utilizes a parametric model for code data selection, aimed at improving both training efficiency and model performance. Our method optimizes the parametric model to ensure distribution consistency and diversity within the selected subset, guaranteeing high-quality data. Experimental results demonstrate that using only 10K samples, our method achieves gains of 2.4% (HumanEval) and 2.3% (MBPP) over 92K full-sampled baseline, outperforming other sampling approaches in both performance and efficiency. This underscores that our method effectively boosts model performance while significantly reducing computational costs.
title Efficient Code LLM Training via Distribution-Consistent and Diversity-Aware Data Selection
topic Computation and Language
url https://arxiv.org/abs/2507.02378