atlas

Research map / Reinforcement learning / Visual and representation RL

Rethinking Exploration in Reinforcement Learning with Effective Metric-Based Exploration Bonus

NeurIPS 2024 · Reinforcement learning › Visual and representation RL · approach: Novelty and impact-driven exploration

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 Effective Metric-based Exploration-bonus (EME), a bisimulation-free state-discrepancy metric with KL policy term and an ensemble-variance scaling factor, as an intrinsic exploration bonus for hard-exploration RL tasks.

Problem

Metric-based state-discrepancy exploration bonuses rely on episodic count scaling factors that fail when states are unique, and bisimulation-based metrics (e.g., LIBERTY) suffer from approximation gaps that break theoretical guarantees and degrade in hard exploration tasks.

Approach

Replace the Wasserstein-based bisimulation metric with an EME metric that strictly computes reward-difference expectations (via ensemble-based variance correction), adds a KL divergence between policy distributions, and scales the bonus by the variance of an ensemble of reward model predictions instead of episodic counts.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
remove the episodic count term from RIDE (ablation)RIDE · Robosuitesuccess rate: 0.52 -> 0.05Hurt
remove the episodic count term from RIDE (ablation)RIDE · MiniGridsuccess rate: 0.93 -> 0.00Hurt
remove the episodic count term from RIDE (ablation)RIDE · Habitatsuccess rate: 0.19 -> 0.00Hurt
remove the episodic count term from NovelD (ablation)NovelD · Robosuitesuccess rate: 0.55 -> 0.11Hurt
remove the episodic count term from NovelD (ablation)NovelD · MiniGridsuccess rate: 0.93 -> 0.00Hurt
remove the episodic count term from NovelD (ablation)NovelD · Habitatsuccess rate: 0.17 -> 0.00Hurt
use EME instead of RIDEEME and RIDE · MRN7S8success rate: 0.67 ± 0.001 -> 0.67 ± 0.011No clear effect
use EME instead of RIDEEME and RIDE · MRN7S8-NTsuccess rate: 0.65 ± 0.001 -> 0.65 ± 0.003No clear effect
use EME instead of RIDEEME and RIDE · KCS4R3success rate: 0.93 ± 0.002 -> 0.91 ± 0.011Hurt
use EME instead of RIDEEME and RIDE · KCS4R3-NTsuccess rate: 0.81 ± 0.001 -> 0.91 ± 0.003Helped
use EME instead of NovelDEME and NovelD · OM2D1hsuccess rate: 0.96 ± 0.011 -> 0.95 ± 0.028Hurt
use EME instead of NovelDEME and NovelD · OM2D1h-NTsuccess rate: 0.91 ± 0.008 -> 0.95 ± 0.011Helped

Surprising findings

Limitations

Experimental setup

Datasets: Atari hard exploration games MiniGrid Robosuite Habitat (Matterport3D) MuJoCo

Benchmarks: Robosuite Atari MiniGrid Habitat MuJoCo delayed reward

Metrics: success rate average return exploration coverage mean success rate

Baselines: ICM RND RIDE NovelD E3B LIBERTY

Builds on: LIBERTY RIDE NovelD RND bisimulation metric

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