atlas

Research map / Reinforcement learning / Visual and representation RL

Explaining RL Decisions with Trajectories': A Reproducibility Study

NeurIPS 2024 · Reinforcement learning › Visual and representation RL · approach: In-context and adaptive RL

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.

A reproducibility study of the trajectory-attribution XRL method ("Explaining RL decisions with trajectories"), partially confirming its claims on Grid-World and extending experiments to Seaquest, HalfCheetah, Breakout and Q*Bert.

Problem

Whether the claims of an offline RL explainability method that attributes agent decisions to clusters of training trajectories hold up under independent reproduction across environments.

Approach

Re-implements the trajectory-attribution pipeline (trajectory encoding, XMeans clustering, complementary datasets retraining policies, attribution frequency) for Grid-World, partially from authors' code and from scratch for other environments, then tests four claims with extra metrics, DBSCAN clustering, and alternative encoders.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
train on all trajectories rather than excluding a clustertrajectory-attribution policies · Grid-WorldISV: 0.287 ± 5e-03 -> 0.307 ± 2e-04Helped
train on all trajectories rather than excluding a clustertrajectory-attribution policies · SeaquestISV: 2.010 ± 4e-02 -> 3.569 ± 1e-04Helped
train on all trajectories rather than excluding a clustertrajectory-attribution policies · HalfCheetahISV: 3.3615 -> 3.5280Hurt
train on all trajectories rather than excluding a clustertrajectory-attribution policies · Q*BertISV: 1.075 -> 1.118Hurt
train on all trajectories rather than excluding a clustertrajectory-attribution policies · BreakoutISV: 0.764 -> 0.812Helped
add cluster 5 trajectories to training (ablation)HalfCheetah trajectory-attribution policy · HalfCheetahISV: 3.5280 -> 3.3615Hurt
add cluster 1 trajectories to training (ablation)Q*Bert trajectory-attribution policy · Q*BertISV: 1.118 -> 1.075Hurt
replace XMeans with DBSCAN (ablation)Grid-World trajectory clustering · Grid-WorldISV: 0.3059No clear effect
replace the LSTM encoder with a Trajectory Transformer (ablation)Grid-World trajectory encoding · Grid-WorldISV: 0.3451 -> 0.3413Hurt
replace the LSTM encoder with BERT (ablation)Grid-World trajectory encoding · Grid-WorldISV: 0.3451 -> 0.3427Hurt
use 250 trajectories instead of 60 (ablation)Grid-World clustering · Grid-Worldpresence of the same-length trajectory patternHurt
use DBSCAN instead of XMeans (ablation)Grid-World clustering · Grid-Worldvisual cluster overlapHelped

Surprising findings

Limitations

Experimental setup

Datasets: d4rl-Atari Seaquest Expert-offline RL Breakout and Q*Bert D4RL HalfCheetah Grid-World (generated with Dyna-Q)

Benchmarks: Grid-World 7x7 Grid-World Four Room 11x11 Seaquest HalfCheetah Breakout Q*Bert

Metrics: initial state value local mean absolute action-value difference action contrast measure Wasserstein distance cluster attribution frequency average distance of attributed trajectories

Baselines: DBSCAN BERT

Builds on: Explaining RL decisions with trajectories Dyna-Q XMeans DBSCAN SAC Decision Transformer

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