atlas

Research map / Reinforcement learning / RL from feedback

$\mathcal{B}$-Coder: Value-Based Deep Reinforcement Learning for Program Synthesis

ICLR 2024 · Reinforcement learning › RL from feedback · approach: RL for LLM reasoning · 0 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.

Proposes B-Coder, a value-based deep RL framework for natural-language-to-code generation that initializes Q from a pre-trained LM's logits and uses a conservative Bellman operator, plus a free learned reward model for ranking candidate programs.

Problem

Policy-based RL dominates program synthesis despite program synthesis having cheap verifiable rewards and abundant off-policy data (human and historical programs), which value-based methods could exploit; but value-based RL is hard to train in huge token state-action spaces.

Approach

Initialize a dueling Q-function from the logits of a pre-trained code LM (Q = alpha*log-prob advantage + V), pre-train the value head with TD error, and fine-tune with a conservative Bellman operator that bootstraps through the reference LM's greedy action; the recovered Q yields a reward model used to rank generated programs.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use 𝓑-Coder rather than the strongest reported baselineAPPS test set · APPS test set, AllPass@1: 1.45 -> 2.30Helped
Use 𝓑-Coder rather than the strongest reported baselineAPPS test set · APPS test set, AllPass@5: 3.78 -> 3.80Helped
Use 𝓑-Coder rather than the strongest reported baselineAPPS test set · APPS test set, AllPass@1000: 19.92 -> 18.12Hurt
Use only the basic reward instead of reward engineering used by comparison methods𝓑-Coder on APPS · APPS test set, All; basic reward onlyPass@1: 1.62 -> 2.30Helped
Use only the basic reward instead of reward engineering used by comparison methods𝓑-Coder on APPS · APPS test set, All; basic reward onlyPass@5: 3.50 -> 3.80Helped
Vary ranking budget m (ablation)𝓑-Coder candidate ranking · APPSPass rateHelped
Rank CodeRL-generated programs using the learned cumulative rewardCodeRL-generated candidates · APPSPass rateHelped
Rank candidates using the learned cumulative reward𝓑-Coder candidate ranking · MBPP testPass rateHelped
Use raw generation without candidate ranking rather than the full ranking strategy𝓑-Coder on APPS · APPS test setPass rate: Pass@1: 1.60 vs 1.60; Pass@5: 3.28 vs 2.88; Pass@50: 7.16 vs 7.35; Pass@100: 8.76 vs 9.18Mixed

Surprising findings

Limitations

Experimental setup

Datasets: APPS MBPP

Benchmarks: APPS MBPP

Metrics: pass@1 pass@5 pass@1000

Baselines: CodeRL PPOCoder RLTF Codex AlphaCode GPT-J GPT-Neo

Builds on: CodeRL Dueling DQN CodeT T5

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