How to Stay Curious while Avoiding Noisy TVs using Aleatoric Uncertainty Estimation
Not on the map. Why?
Abstract (arXiv)
Exploration in environments with sparse rewards is difficult for artificial agents. Curiosity driven learning -- using feed-forward prediction errors as intrinsic rewards -- has achieved some success in these scenarios, but fails when faced with action-dependent noise sources. We present aleatoric mapping agents (AMAs), a neuroscience inspired solution modeled on the cholinergic system of the mammalian brain. AMAs aim to explicitly ascertain which dynamics of the environment are unpredictable, regardless of whether those dynamics are induced by the actions of the agent. This is achieved by generating separate forward predictions for the mean and variance of future states and reducing intrinsic rewards for those transitions with high aleatoric variance. We show AMAs are able to effectively circumvent action-dependent stochastic traps that immobilise conventional curiosity driven agents. The code for all experiments presented in this paper is open sourced: http://github.com/self-supervisor/Escaping-Stochastic-Traps-With-Aleatoric-Mapping-Agents.
Key facts. Cited by 27 papers in the map (data through 7 Oct 2026). Builds on 21 of its 57 references that are in the map. First posted on arXiv 8 Feb 2021.
Builds on
21 of 57 references.
References · 21
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,335
- 2016Asynchronous Methods for Deep Reinforcement Learningcited by 2,218
- 2017Curiosity-driven Exploration by Self-supervised Predictioncited by 1,173
- 2018Exploration by Random Network Distillationcited by 966
- 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 903
- 2016Unifying Count-Based Exploration and Intrinsic Motivationcited by 632
- 2016Deep Exploration via Bootstrapped DQNcited by 518
- 2018Large-Scale Study of Curiosity-Driven Learningcited by 277
9 more
- 2017Revisiting the Arcade Learning Environment: Evaluation Protocols and Open Problems for General Agentscited by 267
- 2019Self-Supervised Exploration via Disagreementcited by 263
- 2020Never Give Up: Learning Directed Exploration Strategiescited by 210
- 2015Variational Information Maximisation for Intrinsically Motivated Reinforcement Learningcited by 142
- 2020RIDE: Rewarding Impact-Driven Exploration for Procedurally-Generated Environmentscited by 135
- 2018Model-Based Active Explorationcited by 105
- 2019Estimating Risk and Uncertainty in Deep Reinforcement Learningcited by 68
- 2018Gotta Learn Fast: A New Benchmark for Generalization in RLcited by 59
- 2019Epistemic Risk-Sensitive Reinforcement Learningcited by 17
Built on by
Citing papers per year: 2022: 4; 2023: 3; 2024: 6; 2025: 8; 2026: 6.
27 papers.
Sort:
By area:
- 2023Minigrid & Miniworld: Modular & Customizable Reinforcement Learning Environments for Goal-Oriented Taskscited by 320
- 2024Open-Endedness is Essential for Artificial Superhuman Intelligencecited by 79
- 2023A Survey on Uncertainty Quantification Methods for Deep Learningcited by 78
- 2022A Review of Uncertainty for Deep Reinforcement Learningcited by 36
- 2024Highly Efficient Self-Adaptive Reward Shaping for Reinforcement Learningcited by 21
- 2022General Intelligence Requires Rethinking Explorationcited by 20
- 2022Pretraining in Deep Reinforcement Learning: A Surveycited by 18
- 2024NAVIX: Scaling MiniGrid Environments with JAXcited by 16
- 2023Approximate information state based convergence analysis of recurrent Q-learningcited by 5
- 2025Uncertainty Prioritized Experience Replaycited by 5
- 2022Curiosity in Hindsight: Intrinsic Exploration in Stochastic Environmentscited by 4
- 2025Wonder Wins Ways: Curiosity-Driven Exploration through Multi-Agent Contextual Calibrationcited by 4
15 more
- 2024A unified uncertainty-aware exploration: Combining epistemic and aleatory uncertaintycited by 2
- 2025TW-CRL: Time-Weighted Contrastive Reward Learning for Efficient Inverse Reinforcement Learningcited by 2
- 2025NBDI: A Simple and Effective Termination Condition for Skill Extraction from Task-Agnostic Demonstrationscited by 1
- 2025Disentangling Uncertainties by Learning Compressed Data Representationcited by 1
- 2025ScreenExplorer: Training a Vision-Language Model for Diverse Exploration in Open GUI Worldcited by 1
- 2025Exploration by Random Reward Perturbationcited by 1
- 2025Beyond Noisy-TVs: Noise-Robust Exploration Via Learning Progress Monitoringcited by 1
- 2026DreamSAC: Learning Hamiltonian World Models via Symmetry Explorationcited by 1
- 2024Generalized Gaussian Temporal Difference Error for Uncertainty-aware Reinforcement Learningcited by 0
- 2024Centralized Reward Agent for Knowledge Sharing and Transfer in Multi-Task Reinforcement Learningcited by 0
- 2026Decoupling Exploration and Policy Optimization: Uncertainty Guided Tree Search for Hard Explorationcited by 0
- 2026CIG: Exploration via Conditional Information Gaincited by 0
- 2026Principled Direction-Free Intrinsic Motivation through Model-Free Epistemic Free-Energy Estimatorscited by 0
- 2026Risk and Anomaly Identification for Distribution Network Optimal Operation Based on Reinforcement Learning and Uncertainty Quantificationcited by 0
- 2026Escaping Local Views: Discovering Latent Concepts for Interpretable Multi-Agent Reinforcement Learningcited by 0
- 2026Escaping Local Views: Discovering Latent Concepts for Interpretable Multi-Agent Reinforcement Learning28 Sep 2026
- 2026Risk and Anomaly Identification for Distribution Network Optimal Operation Based on Reinforcement Learning and Uncertainty Quantification3 Sep 2026
- 2026Principled Direction-Free Intrinsic Motivation through Model-Free Epistemic Free-Energy Estimators18 Jul 2026
- 2026CIG: Exploration via Conditional Information Gain20 May 2026
- 2026Decoupling Exploration and Policy Optimization: Uncertainty Guided Tree Search for Hard Exploration23 Mar 2026
- 2026DreamSAC: Learning Hamiltonian World Models via Symmetry Exploration8 Mar 2026
- 2025Beyond Noisy-TVs: Noise-Robust Exploration Via Learning Progress Monitoring29 Sep 2025
- 2025Wonder Wins Ways: Curiosity-Driven Exploration through Multi-Agent Contextual Calibration25 Sep 2025
- 2025Uncertainty Prioritized Experience Replay10 Jun 2025
- 2025Exploration by Random Reward Perturbation10 Jun 2025
- 2025ScreenExplorer: Training a Vision-Language Model for Diverse Exploration in Open GUI World25 May 2025
- 2025TW-CRL: Time-Weighted Contrastive Reward Learning for Efficient Inverse Reinforcement Learning8 Apr 2025
15 more
- 2025Disentangling Uncertainties by Learning Compressed Data Representation20 Mar 2025
- 2025NBDI: A Simple and Effective Termination Condition for Skill Extraction from Task-Agnostic Demonstrations22 Jan 2025
- 2024Centralized Reward Agent for Knowledge Sharing and Transfer in Multi-Task Reinforcement Learning20 Aug 2024
- 2024Highly Efficient Self-Adaptive Reward Shaping for Reinforcement Learning6 Aug 2024
- 2024Generalized Gaussian Temporal Difference Error for Uncertainty-aware Reinforcement Learning5 Aug 2024
- 2024NAVIX: Scaling MiniGrid Environments with JAX28 Jul 2024
- 2024Open-Endedness is Essential for Artificial Superhuman Intelligence6 Jun 2024
- 2024A unified uncertainty-aware exploration: Combining epistemic and aleatory uncertainty5 Jan 2024
- 2023Minigrid & Miniworld: Modular & Customizable Reinforcement Learning Environments for Goal-Oriented Tasks24 Jun 2023
- 2023Approximate information state based convergence analysis of recurrent Q-learning9 Jun 2023
- 2023A Survey on Uncertainty Quantification Methods for Deep Learning26 Feb 2023
- 2022Curiosity in Hindsight: Intrinsic Exploration in Stochastic Environments18 Nov 2022
- 2022General Intelligence Requires Rethinking Exploration15 Nov 2022
- 2022Pretraining in Deep Reinforcement Learning: A Survey8 Nov 2022
- 2022A Review of Uncertainty for Deep Reinforcement Learning18 Aug 2022
- 2023Minigrid & Miniworld: Modular & Customizable Reinforcement Learning Environments for Goal-Oriented Taskscited by 320
- 2024Open-Endedness is Essential for Artificial Superhuman Intelligencecited by 79
- 2023A Survey on Uncertainty Quantification Methods for Deep Learningcited by 78
- 2022A Review of Uncertainty for Deep Reinforcement Learningcited by 36
- 2024Highly Efficient Self-Adaptive Reward Shaping for Reinforcement Learningcited by 21
- 2022General Intelligence Requires Rethinking Explorationcited by 20
- 2022Pretraining in Deep Reinforcement Learning: A Surveycited by 18
- 2024NAVIX: Scaling MiniGrid Environments with JAXcited by 16
- 2023Approximate information state based convergence analysis of recurrent Q-learningcited by 5
- 2025Uncertainty Prioritized Experience Replaycited by 5
- 2022Curiosity in Hindsight: Intrinsic Exploration in Stochastic Environmentscited by 4
- 2025Wonder Wins Ways: Curiosity-Driven Exploration through Multi-Agent Contextual Calibrationcited by 4
12 most cited of 24 All 24 (JSON)
- 2026Escaping Local Views: Discovering Latent Concepts for Interpretable Multi-Agent Reinforcement Learning28 Sep 2026
- 2026Risk and Anomaly Identification for Distribution Network Optimal Operation Based on Reinforcement Learning and Uncertainty Quantification3 Sep 2026
- 2026Principled Direction-Free Intrinsic Motivation through Model-Free Epistemic Free-Energy Estimators18 Jul 2026
- 2026CIG: Exploration via Conditional Information Gain20 May 2026
- 2026Decoupling Exploration and Policy Optimization: Uncertainty Guided Tree Search for Hard Exploration23 Mar 2026
- 2025Beyond Noisy-TVs: Noise-Robust Exploration Via Learning Progress Monitoring29 Sep 2025
- 2025Wonder Wins Ways: Curiosity-Driven Exploration through Multi-Agent Contextual Calibration25 Sep 2025
- 2025Uncertainty Prioritized Experience Replay10 Jun 2025
- 2025Exploration by Random Reward Perturbation10 Jun 2025
- 2025Disentangling Uncertainties by Learning Compressed Data Representation20 Mar 2025
- 2025NBDI: A Simple and Effective Termination Condition for Skill Extraction from Task-Agnostic Demonstrations22 Jan 2025
- 2024Centralized Reward Agent for Knowledge Sharing and Transfer in Multi-Task Reinforcement Learning20 Aug 2024
12 newest of 24 All 24 (JSON)