atlas

Research map / Reinforcement learning / Visual and representation RL

Efficient Multi-task Reinforcement Learning with Cross-Task Policy Guidance

NeurIPS 2024 · Reinforcement learning › Visual and representation RL · approach: Skill learning and diffusion policies · 4 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 Cross-Task Policy Guidance (CTPG), where a per-task guide policy selects which task's control policy generates training trajectories every K steps, with policy-filter and guide-block gates, and shows gains when combined with five implicit-sharing MTRL methods on MetaWorld and HalfCheetah.

Problem

Multi-task RL typically shares information implicitly through parameters or optimization, but ignores explicit reuse of other tasks' control policies to generate better training data for unmastered tasks.

Approach

A learned guide policy per task chooses which task's control policy acts for every K timesteps to collect training data; a policy-filter gate masks policies whose guide Q-value is below the task's own V-value, and a guide-block gate disables guidance for easy tasks identified via SAC temperature, plus hindsight off-policy correction for guide training.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add CTPG to SMSM · HalfCheetah-MT5episode return (× 1e3): 9.57 ± 0.21 -> 10.47 ± 0.34Helped
Add CTPG to PCGradPCGrad · HalfCheetah-MT8episode return (× 1e3): 10.17 ± 1.06 -> 11.09 ± 0.50Helped
Add CTPG to SMSM · MetaWorld-MT10success rate (%): 74.52 ± 2.29 -> 78.97 ± 2.41Helped
Add CTPG to PaCoPaCo · MetaWorld-MT50success rate (%): 59.46 ± 5.14 -> 68.10 ± 3.44Helped
Add the policy-filter gate (ablation)MHSAC w/ CTPG · MetaWorld-MT10success rateHelped
Add the guide-block gate using SAC temperature or success rate (ablation)MHSAC w/ CTPG · MetaWorld-MT10success rateHelped
Add hindsight off-policy correction (ablation)MHSAC w/ CTPG · MetaWorld-MT10training efficiency and stabilityHelped
Use guide steps that are short or long instead of K = 10 (ablation)MHSAC w/ CTPG · HalfCheetah-MT8 and MetaWorld-MT10performanceHurt
Estimate the V-value with H = 1 Monte Carlo sample (ablation)MHSAC w/ CTPG · MetaWorld-MT10performanceHurt
Use different Monte Carlo sampling counts except H = 1 (ablation)MHSAC w/ CTPG · MetaWorld-MT10performanceNo clear effect
Replace the SAC temperature metric with a success-rate threshold for the guide-block gate (ablation)MHSAC w/ CTPG · MetaWorld-MT10success rateNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: MetaWorld-MT10 MetaWorld-MT50 HalfCheetah-MT5 HalfCheetah-MT8

Benchmarks: MetaWorld-MT10 MetaWorld-MT50 HalfCheetah-MT5 HalfCheetah-MT8

Metrics: mean success rate episode return

Baselines: MTSAC MHSAC PCGrad SM PaCo QMP BPT Single-Task SAC

Builds on: Soft Actor-Critic Q-filter (QMP) PCGrad soft module PaCo MTRL codebase

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