atlas

Research map / Reinforcement learning / Policy gradient and actor-critic

Augmenting Decision with Hypothesis in Reinforcement Learning

ICML 2024 · Reinforcement learning › Policy gradient and actor-critic · approach: Temporal-difference and discounting theory

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 ALH, which augments TD3 actor and critic with a learned hypothesis embedding (weak-model-based learning plus adaptive rollout) to fix slow exploitation and bias sensitivity in value-based RL, evaluated on MultiNormEnv and eight MuJoCo tasks.

Problem

Value-based RL (e.g., TD3) exploits slowly early in training because optimistic critic estimates favor exploration, and its policies are sensitive to critic bias from non-representative data.

Approach

Augment actor and critic with a hypothesis embedding produced by an independent weak model trained on observation sets, detached from actor/critic optimization, and update that hypothesis incrementally during rollout so decisions adapt to the latest interactions without changing network parameters.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
augment TD3 with hypothesis representations and adaptive rolloutALH-a · HalfCheetah-v3average return: 7563.45±996.14 -> 7924.09±636.61Helped
augment TD3 with hypothesis representations and adaptive rolloutALH-g · Hopper-v3average return: 3273.77±285.26 -> 3439.50±145.63Helped
augment TD3 with hypothesis representations and adaptive rolloutALH-g · Walker2d-v3average return: 3612.88±860.95 -> 4225.65±382.86Helped
augment TD3 with hypothesis representations and adaptive rolloutALH-a · Ant-v3average return: 4884.83±675.88 -> 5395.55±353.28Helped
augment TD3 with hypothesis representations and adaptive rolloutALH-a · Humanoid-v3average return: 5031.64±182.01 -> 5086.73±212.45Helped
augment TD3 with hypothesis representations and adaptive rolloutALH-g · Reacher-v2average return: -6.06±0.63 -> -5.79±0.76Hurt
augment TD3 with hypothesis representations and adaptive rolloutALH-a · InvertedDoublePendulum-v2average return: 7492.54±3732.47 -> 9358.61±2.57No clear effect
augment TD3 with hypothesis representations and adaptive rolloutALH-a · InvertedPendulum-v2average return: 1000.00±0.00 -> 1000.00±0.00No clear effect
add detachment of hypothesis representation from Q and policy objective gradients (ablation)ALH-g · HalfCheetah-v3 (Average aggregate)normalized score: 91.71% -> 100%Helped
add detachment of hypothesis representation from Q and policy objective gradients (ablation)ALH-a · HalfCheetah-v3 (Average aggregate)normalized score: 95.26% -> 100%Helped
use localized hypotheses instead of one hypothesis from the current batch (ablation)ALH-g · HalfCheetah-v3 (Average aggregate)normalized score: 88.51% -> 100%Helped
use localized hypotheses instead of one hypothesis from the current batch (ablation)ALH-a · HalfCheetah-v3 (Average aggregate)normalized score: 95.53% -> 100%Helped

Surprising findings

Limitations

Experimental setup

Datasets: MultiNormEnv MuJoCo gym tasks (HalfCheetah, Hopper, Walker2d, Ant, Humanoid, Reacher, InvertedDoublePendulum, InvertedPendulum)

Benchmarks: MultiNormEnv MuJoCo HalfCheetah-v3 Hopper-v3 Walker2d-v3 Ant-v3 Humanoid-v3 Reacher-v2 InvertedDoublePendulum-v2 InvertedPendulum-v2

Metrics: average return normalized score across methods standard deviation over 10 trials

Baselines: TD3 DDPG PPO MBPO

Builds on: TD3 MBPO DDPG PPO

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