atlas

Research map / Large language models / Instruction and fine-tuning

ComBack: A Versatile Dataset for Enhancing Compiler Backend Development Efficiency

NeurIPS 2024 · Large language models › Instruction and fine-tuning · approach: Fine-tuning for reasoning data · 5 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Introduces ComBack, a dataset of 178 GCC/LLVM compiler backends with three tasks (statement completion, next-statement suggestion, code generation), and shows fine-tuned small code models, especially CodeT5+ (220M), beat Fork-Flow, ChatGPT-3.5-Turbo and Code-LLaMA-34B-Instruct on new targets.

Problem

Compiler backend development for new processor targets is laborious and manual, and language models lack public training data for this domain.

Approach

A public, multi-target dataset of compiler backend functions from GCC and LLVM, with target-specific values abstracted into placeholder tokens and three derived tasks, used to fine-tune code language models for backend code completion and generation.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Fine-tune CodeT5+ on ComBackCodeT5+ · Statement-Level CompletionEM (%): 0.02 -> 66.93Helped
Fine-tune CodeT5+ on ComBackCodeT5+ · Next-Statement SuggestionEM (%): 0.12 -> 59.57Helped
Fine-tune CodeT5+ on ComBackCodeT5+ · Code GenerationBLEU-4: 0.00 -> 75.29Helped
Fine-tune CodeT5+ on ComBackCodeT5+ · GCC RISC-V, ARC, and NVPTX Code GenerationBLEU-4: CodeT5+: 32.56, 19.94, 25.47; ChatGPT: 1.37, 1.67, 1.57Helped
Train with MPU and GPU data rather than CPU-only data (ablation)CodeT5+ · GCC ARC (MPU) Code GenerationBLEU-4: 19.29 -> 19.94Helped
Train with MPU and GPU data rather than CPU-only data (ablation)CodeT5+ · GCC NVPTX (GPU) Code GenerationBLEU-4: 22.46 -> 25.47Helped
Train with MPU and GPU data rather than CPU-only data (ablation)CodeT5+ · LLVM ARC (MPU) Code GenerationBLEU-4: 49.62 -> 55.38Helped
Train with MPU and GPU data rather than CPU-only data (ablation)CodeT5+ · LLVM NVPTX (GPU) Code GenerationBLEU-4: 42.94 -> 44.33Helped
Add RISC-V training data (ablation)CodeT5+ · RI5CY Statement-Level CompletionEM (%): +7.90Helped
Add RISC-V training data (ablation)CodeT5+ · RI5CY Next-Statement SuggestionEM (%): +9.96Helped
Add RISC-V training data (ablation)CodeT5+ · RI5CY Code GenerationBLEU-4: +25.05Helped

Surprising findings

Limitations

Experimental setup

Datasets: ComBack (GCC and LLVM backends from GitHub and official releases)

Benchmarks: ComBack Statement-Level Completion ComBack Next-Statement Suggestion ComBack Code Generation

Metrics: exact match Levenshtein edit distance similarity BLEU-4

Baselines: Fork-Flow ChatGPT-3.5-Turbo Code-LLaMA-34B-Instruct

Builds on: CodeXGlue Fork-Flow

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy