AI Paper Atlas

Research map / Reinforcement learning / Visual and representation RL

Learned state and action representations

Learn compact representations of states, actions, or contexts to speed up and generalize RL.

148 accepted papers at ICML, NeurIPS, ICLR, CVPR and AAAI (2016–2026) · an approach in Visual and representation RL, keyword cluster “representation · latent · generalization”

State History Tried Checklist Preprints Who Export

Where it stands

Not enough benchmark data Fewer than four papers report the same benchmark, so there is no record to follow.

39 papers in 2023–24 and 40 in 2025–26 (at ICML, NeurIPS, ICLR, CVPR and AAAI); its share of the topic shrank, from 6.8% to 5.7%.

How this is decided

A benchmark is tracked when at least four papers of this approach report their own method on the same dataset and metric in at least two different years. Its record is the best number reported up to 2024 compared with the best reported in 2025–26: still improving means at least 1% better, flat means at most 0.2% better with at least two papers from 2025–26 trying. Numbers on a different scale from the rest (a fraction among percentages) and far outliers are left out. Results come from different papers with different setups, so they are signals, not head-to-head comparisons. The share of papers is counted at ICML, NeurIPS, ICLR, CVPR and AAAI.

How it got here

2016: 1162017: 0172018: 5182019: 10192020: 11202021: 18212022: 24222023: 23232024: 16242025: 14252026: 2626

Most cited paper of each year

Earliest papers here

What has been tried

113 papers here report 1,096 measured changes (a component added, removed or swapped, compared with a baseline): helped 61%, no clear effect 11%, hurt 23%, mixed 5%. Papers mostly publish changes that helped, so "helped" is over-represented.

Ablations in recent papers

What was changedMeasured onResultEffectPaper
remove epoch embeddings and recurrent environment representationCIFAR-10, ResNet-56 · ∆-Acc+0.45% -> −0.03%HurtJointly Training and Pruning CNNs via Learnable Agent Guidance and Alignment (CVPR 2024 · 20 citations)
increase training dataset sizePushT · success rate0.08 -> 0.92HelpedDINO-WM: World Models on Pre-trained Visual Features enable Zero-shot Planning (ICML 2025 · 5 citations)
Use implicit rather than explicit dynamics modelsPick, Stack, Series, and DMC tasks · sample efficiencyHelpedBuilding Minimal and Reusable Causal State Abstractions for Reinforcement Learning (AAAI 2024 · 2 citations)
Add the Last Transition Label to CTL+C-PREPGN, MP, and PD; six EL/CM context settings · ttt_aucHelpedContextual Pre-planning on Reward Machine Abstractions for Enhanced Transfer in Deep Reinforcement Learning (AAAI 2024 · 2 citations)
use only the current-step reward instead of accumulated Q-value rewards (γ=0)FashionMNIST and CIFAR10 · accuracy75.63±0.11 vs 78.23±0.13; 41.91±0.15 vs 46.46±0.15HurtImproving Unsupervised Hierarchical Representation with Reinforcement Learning (CVPR 2024 · 2 citations)
remove chronological embedding ψDM_Control distraction, cheetah-run and walker-walk · scoreHurtState Chrono Representation for Enhancing Generalization in Reinforcement Learning (NeurIPS 2024 · 2 citations)
Replace basis-function reward reconstruction with an MLP conditioned on the representationMs. Pacman · success rateHurtZero-Shot Reinforcement Learning via Function Encoders (ICML 2024 · 1 citation)
Add LESR state representationMujoco (mean, 1M steps) · RPI21% -> 29%HelpedLLM-Empowered State Representation for Reinforcement Learning (ICML 2024 · 1 citation)
remove decoder gradientsImageNet-1k validation, pools 1, 5, and 10 · million steps to 80% success0.42 0.06 -> 1.13 0.12; 1.23 0.20 -> 1.79 0.61; 1.44 0.58 -> 2.57 0.91HurtSliding Puzzles Gym: A Scalable Benchmark for State Representation in Visual Reinforcement Learning (ICML 2025 · 0 citations)
Down-scale Gaussian component variances compared with direct GMM samplingKitchen · Task success7%HelpedQuantization-Free Autoregressive Action Transformer (NeurIPS 2025 · 0 citations)
Add static/dynamic state factorizationLibero, compositional generalization · success rate0.64 -> 0.70HelpedLearning Interactive World Model for Object-Centric Reinforcement Learning (NeurIPS 2025 · 0 citations)
Replace relative trajectory actions with absolute trajectory actionsMimicGen, four-task ablation · mean success rate67.0 -> 55.9; reduces performance by 11.1%HurtA Practical Guide for Incorporating Symmetry in Diffusion Policy (NeurIPS 2025 · 0 citations)

What reviewers will expect

Datasets and baselines named in the 21 papers of this approach from 2025–2026 that have a full-text summary. Leaving out the common ones is a frequent review question.

Datasets

  • Franka Kitchen · 2 of 21 papers
  • Meta-World · 2 of 21 papers
  • Atari · 2 of 21 papers

Baselines

  • DQN · 4 of 21 papers
  • DreamerV3 · 4 of 21 papers
  • TD-MPC2 · 3 of 21 papers
  • PPO · 3 of 21 papers
  • SPR · 3 of 21 papers

Recent preprints

Loading recent arXiv preprints…

    Who publishes on it

    Institutions on the most papers of this approach since 2023 (of 79 papers; from OpenAlex and the papers' affiliations).

    Use it

    Download BibTeX of 40 key papersOpen the topic in the interactive map

    Get alerts. Describe your idea in scime atlas and save it (free scime account): you get a weekly email when new preprints land close to it.

    Same keyword cluster

    Elsewhere in the topic

    Benchmark numbers, ablations and baseline numbers are extracted by AI from each paper's full text and can be wrong; the paper is the authority. Data: results index of 2026-11-02, citation counts from OpenAlex (2026-10-09).

    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