atlas

Research map / Reinforcement learning / RL from feedback

ArCHer: Training Language Model Agents via Hierarchical Multi-Turn RL

ICML 2024 · Reinforcement learning › RL from feedback · approach: RL for LLM reasoning · 2 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 ArCHer, a hierarchical actor-critic framework for multi-turn LLM agent RL that trains an utterance-level TD critic and a token-level policy-gradient actor, achieving far better sample efficiency than token-level PPO on language agent tasks.

Problem

Training LLM agents for multi-turn tasks with delayed long-horizon rewards is hard: on-policy PPO is sample-inefficient, token-level off-policy methods suffer long horizons, and utterance-level methods face huge action spaces for maximization.

Approach

A two-level hierarchical MDP: an off-policy TD-trained utterance-level Q/V critic at the high level, and a token-level on-policy policy gradient actor at the low level that uses the high-level advantage as its terminal reward, so the critic avoids token-level Bellman backups and maximization over huge utterance spaces.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use IQL + AWR instead of IQL + REINFORCE (ablation)ArCHer offline · Twenty Questionsreturn: -20 -> -14.1Helped
Add a behavior-cloning loss to IQL + REINFORCE (ablation)ArCHer offline · Twenty Questionsreturn: -20 -> -15.3Helped
Use AWR instead of REINFORCE for the actor (ablation)ArCHer with IQL critic · Twenty Questionsreturn: -15.3 -> -14.1Helped
Use IQL instead of SARSA for the critic (ablation)ArCHer with AWR actor · Twenty Questionsreturn: -14.5 -> -14.1Helped
Use IQL + AWR rather than BCArCHer offline · Twenty Questionsreturn: -16.8 -> -14.1Helped
Use IQL + AWR rather than filtered behavior cloningArCHer offline · Twenty Questionsreturn: -15.4 -> -14.1Helped
Reuse off-policy data rather than rely on PPO's on-policy rolloutsArCHer versus PPO · Twenty Questionssamples to average return just higher than -17: more than 100k samples -> fewer than 1000 samplesHelped
Use a replay buffer containing only the most recent 48 rollouts (ablation)ArCHer · Guess My Citylearning stabilityHurt
Increase replay-buffer size beyond the point needed for stable learning (ablation)ArCHer · Guess My CityperformanceNo clear effect
Replace the RoBERTa critic with a GPT-2 critic (ablation)ArCHer · Online tasksperformanceNo clear effect
Use a Mistral 7B actor instead of a GPT-2 actor (ablation)ArCHer · Twenty Questions Subsetlearning speedHelped
Add a token-level value baseline to the policy gradient (ablation)ArCHer · Guess My CityperformanceHelped

Surprising findings

Limitations

Experimental setup

Datasets: Detective Game Twenty Questions Guess My City WebShop

Benchmarks: Detective Game Twenty Questions Twenty Questions Subset Guess My City WebShop

Metrics: task return sample efficiency (samples to reach return)

Baselines: PPO Filtered BC CHAI ILQL ReAct BC

Builds on: PPO REINFORCE implicit Q-learning advantage-weighted regression CHAI

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