Preprint
Machine Learning

WizardCoder

June 1, 2023

0

Citations

0

Influential Citations

Venue

2023

Year

Abstract

Enhances the performance of the open-source Code LLM, StarCoder, through the application of Code Evol-Instruct.

Analysis

Why This Paper Matters

WizardCoder addresses a critical bottleneck in open-source code LLMs: the lack of high-quality, diverse instruction data for fine-tuning. By introducing Code Evol-Instruct, the paper provides a scalable method to automatically generate complex coding tasks, reducing reliance on manual curation. This is particularly significant as the AI community increasingly values open-source alternatives to proprietary models like GPT-4 and Codex. The approach demonstrates that evolutionary techniques can effectively bridge the performance gap between open and closed models.

Furthermore, the paper's focus on StarCoder, a widely used open-source code LLM, ensures immediate practical relevance. The results show that instruction evolution can unlock substantial gains without requiring massive computational resources or proprietary data. This makes the methodology accessible to a broad range of researchers and practitioners, potentially accelerating progress in code generation and software automation.

Technical Contributions

  • Code Evol-Instruct: An evolutionary algorithm that iteratively mutates and combines seed instructions to create more complex and diverse coding problems. This includes operations like adding constraints, increasing difficulty, and combining multiple tasks.
  • Fine-tuning Pipeline: A straightforward pipeline that uses the evolved instructions to fine-tune StarCoder, demonstrating that data quality is a key lever for performance.
  • Open-source Release: The model and training data are released, enabling reproducibility and further research.

Results

WizardCoder achieves a pass@1 score of 57.3% on HumanEval, compared to StarCoder's 33.6%, representing a 70% relative improvement. On the MBPP benchmark, it reaches 61.2% pass@1, outperforming other open-source models like CodeGen and InCoder. These results place WizardCoder competitively with some proprietary models, though still below GPT-4's 82% on HumanEval.

Significance

WizardCoder's impact lies in demonstrating that automated instruction evolution can dramatically improve code LLM performance, offering a cost-effective alternative to manual data creation. This opens avenues for continuous improvement of open-source models and may inspire similar approaches in other domains like math reasoning or scientific writing. The work also highlights the importance of data diversity and complexity in fine-tuning, shifting focus from model architecture to data generation strategies.