CodeBoost: Boosting Code LLMs by Squeezing Knowledge from Code Snippets with RL

2508.05242v1 cs.CL 2025-08-09
Авторы:

Sijie Wang, Quanjiang Guo, Kai Zhao, Yawei Zhang, Xin Li, Xiang Li, Siqi Li, Rui She, Shangshu Yu, Wee Peng Tay

Резюме на русском

**Резюме** Современные коде-LLMs широко используются для создания эффективных автоматизированных кодинг-процессов, но их обучение через пост-тренировку с использованием reinforcement learning (RL) сталкивается с проблемой коллекции качественных кодинг-инструкций. Эти инструкции часто трудоемко собираются вручную и сложно масштабировать. Однако богатый ресурс — коды — лежит в остатках и недостаточно эффективно используется. Мы предлагаем CodeBoost, рамочную программу, которая улучшает коде-LLMs только на основе доступных кодовых сниппетов, без требуемых инструкций. Она включает максимальный клик курсторинга, двунаправленную предсказательную модель, учитывающую как правильные, так и неверные выводы, а также гетерогенную аугментацию и награждение. Эти компоненты обеспечивают разнообразие, улучшают обучение и увеличивают точность. Исследования показали, что CodeBoost постоянно улучшает производительность LLMs, делая его эффективным и масштабируемым подходом.

Abstract

Code large language models (LLMs) have become indispensable tools for building efficient and automated coding pipelines. Existing models are typically post-trained using reinforcement learning (RL) from general-purpose LLMs using "human instruction-final answer" pairs, where the instructions are usually from manual annotations. However, collecting high-quality coding instructions is both labor-intensive and difficult to scale. On the other hand, code snippets are abundantly available from various sources. This imbalance presents a major bottleneck in instruction-based post-training. We propose CodeBoost, a post-training framework that enhances code LLMs purely from code snippets, without relying on human-annotated instructions. CodeBoost introduces the following key components: (1) maximum-clique curation, which selects a representative and diverse training corpus from code; (2) bi-directional prediction, which enables the model to learn from both forward and backward prediction objectives; (3) error-aware prediction, which incorporates learning signals from both correct and incorrect outputs; (4) heterogeneous augmentation, which diversifies the training distribution to enrich code semantics; and (5) heterogeneous rewarding, which guides model learning through multiple reward types including format correctness and execution feedback from both successes and failures. Extensive experiments across several code LLMs and benchmarks verify that CodeBoost consistently improves performance, demonstrating its effectiveness as a scalable and effective training pipeline.

Ссылки и действия