On The Effect of Auxiliary Tasks on Representation Dynamics
Not on the map. Why?
Abstract (arXiv)
While auxiliary tasks play a key role in shaping the representations learnt by reinforcement learning agents, much is still unknown about the mechanisms through which this is achieved. This work develops our understanding of the relationship between auxiliary tasks, environment structure, and representations by analysing the dynamics of temporal difference algorithms. Through this approach, we establish a connection between the spectral decomposition of the transition operator and the representations induced by a variety of auxiliary tasks. We then leverage insights from these theoretical results to inform the selection of auxiliary tasks for deep reinforcement learning agents in sparse-reward environments.
Key facts. Cited by 56 papers in the map (data through 7 Oct 2026). Builds on 37 of its 72 references that are in the map. First posted on arXiv 25 Feb 2021.
Builds on
37 of 72 references.
References · 37
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2020CURL: Contrastive Unsupervised Representations for Reinforcement Learningcited by 745
- 2017A Distributional Perspective on Reinforcement Learningcited by 743
- 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 562
- 2016Deep Exploration via Bootstrapped DQNcited by 518
- 2017Distributional Reinforcement Learning with Quantile Regressioncited by 464
- 2019Implicit Regularization in Deep Matrix Factorizationcited by 402
- 2016Reinforcement Learning with Unsupervised Auxiliary Taskscited by 400
- 2016Successor Features for Transfer in Reinforcement Learningcited by 331
- 2017Revisiting the Arcade Learning Environment: Evaluation Protocols and Open Problems for General Agentscited by 267
- 2020Decoupling Representation Learning from Reinforcement Learningcited by 263
25 more
- 2018Randomized Prior Functions for Deep Reinforcement Learningcited by 218
- 2019DeepMDP: Learning Continuous Latent Space Models for Representation Learningcited by 204
- 2016Learning to Navigate in Complex Environmentscited by 192
- 2020Self-Distillation Amplifies Regularization in Hilbert Spacecited by 171
- 2018Learning by Playing - Solving Sparse Reward Tasks from Scratchcited by 164
- 2019Is a Good Representation Sufficient for Sample Efficient Reinforcement Learning?cited by 140
- 2020Implicit Under-Parameterization Inhibits Data-Efficient Deep Reinforcement Learningcited by 134
- 2019Learning with Good Feature Representations in Bandits and in RL with a Generative Modelcited by 124
- 2017A Laplacian Framework for Option Discovery in Reinforcement Learningcited by 108
- 2016Averaged-DQN: Variance Reduction and Stabilization for Deep Reinforcement Learningcited by 106
- 2020Bootstrap Latent-Predictive Representations for Multitask Reinforcement Learningcited by 101
- 2018Adapting Auxiliary Losses Using Gradient Similaritycited by 98
- 2017Eigenoption Discovery through the Deep Successor Representationcited by 85
- 2018An Analysis of Categorical Distributional Reinforcement Learningcited by 68
- 2020The Value-Improvement Path: Towards Better Representations for Reinforcement Learningcited by 60
- 2019Hyperbolic Discounting and Learning over Multiple Horizonscited by 58
- 2019A Geometric Perspective on Optimal Representations for Reinforcement Learningcited by 54
- 2019SciPy 1.0--Fundamental Algorithms for Scientific Computing in Pythoncited by 49
- 2019A Comparative Analysis of Expected and Distributional Reinforcement Learningcited by 47
- 2020Representations for Stable Off-Policy Reinforcement Learningcited by 31
- 2019Comments on the Du-Kakade-Wang-Yang Lower Boundscited by 16
- 2017Shallow Updates for Deep Reinforcement Learningcited by 14
- 2018Feature-Based Aggregation and Deep Reinforcement Learning: A Survey and Some New Implementationscited by 12
- 2019Terminal Prediction as an Auxiliary Task for Deep Reinforcement Learningcited by 11
- 2017Unsupervised Basis Function Adaptation for Reinforcement Learningcited by 1
Built on by
Citing papers per year: 2021: 5; 2022: 16; 2023: 14; 2024: 13; 2025: 5; 2026: 3.
56 papers.
Sort:
By area:
- 2022Multi-Game Decision Transformerscited by 212
- 2024Stop Regressing: Training Value Functions via Classification for Scalable Deep RLcited by 136
- 2022Understanding and Preventing Capacity Loss in Reinforcement Learningcited by 124
- 2021Stabilizing Deep Q-Learning with ConvNets and Vision Transformers under Data Augmentationcited by 101
- 2024Bridging State and History Representations: Understanding Self-Predictive RLcited by 52
- 2023Proto-Value Networks: Scaling Representation Learning with Auxiliary Taskscited by 41
- 2022Understanding Self-Predictive Learning for Reinforcement Learningcited by 40
- 2022On the Generalization of Representations in Reinforcement Learningcited by 28
- 2023Deep Laplacian-based Options for Temporally-Extended Explorationcited by 28
- 2022An Empirical Study of Implicit Regularization in Deep Offline RLcited by 24
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradientcited by 24
- 2024Plasticity Loss in Deep Reinforcement Learning: A Surveycited by 24
38 more
- 2025floq: Training Critics via Flow-Matching for Scaling Compute in Value-Based RLcited by 20
- 2022Transfer RL across Observation Feature Spaces via Model-Based Regularizationcited by 18
- 2025A Survey of State Representation Learning for Deep Reinforcement Learningcited by 17
- 2023TD Convergence: An Optimization Perspectivecited by 15
- 2024When does Self-Prediction help? Understanding Auxiliary Tasks in Reinforcement Learningcited by 13
- 2022Learning Dynamics and Generalization in Reinforcement Learningcited by 11
- 2023Bootstrapped Representations in Reinforcement Learningcited by 11
- 2024Investigating the Histogram Loss in Regressioncited by 11
- 2022Exploring through Random Curiosity with General Value Functionscited by 10
- 2022Frustratingly Easy Regularization on Representation Can Boost Deep Reinforcement Learningcited by 8
- 2023Policy Gradient Methods in the Presence of Symmetries and State Abstractionscited by 8
- 2022A Novel Stochastic Gradient Descent Algorithm for Learning Principal Subspacescited by 7
- 2023Predictive auxiliary objectives in deep RL mimic learning in the braincited by 7
- 2024Adaptive Regularization of Representation Rank as an Implicit Constraint of Bellman Equationcited by 7
- 2024A Unifying Framework for Action-Conditional Self-Predictive Reinforcement Learningcited by 7
- 2021Model-Value Inconsistency as a Signal for Epistemic Uncertaintycited by 6
- 2024Random Latent Exploration for Deep Reinforcement Learningcited by 6
- 2021An Open-Source Multi-Goal Reinforcement Learning Environment for Robotic Manipulation with Pybulletcited by 5
- 2022What makes useful auxiliary tasks in reinforcement learning: investigating the effect of the target policycited by 5
- 2023The RL Perceptron: Generalisation Dynamics of Policy Learning in High Dimensionscited by 5
- 2024Understanding What Affects the Generalization Gap in Visual Reinforcement Learning: Theory and Empirical Evidencecited by 5
- 2023PcLast: Discovering Plannable Continuous Latent Statescited by 4
- 2022Engineering flexible machine learning systems by traversing functionally-invariant pathscited by 3
- 2023Towards a Better Understanding of Representation Dynamics under TD-learningcited by 3
- 2023Investigating Navigation Strategies in the Morris Water Maze through Deep Reinforcement Learningcited by 3
- 2022Auxiliary task discovery through generate-and-testcited by 2
- 2023Representations and Exploration for Deep Reinforcement Learning using Singular Value Decompositioncited by 2
- 2023Eigensubspace of Temporal-Difference Dynamics and How It Improves Value Approximation in Reinforcement Learningcited by 2
- 2024A Role of Environmental Complexity on Representation Learning in Deep Reinforcement Learning Agentscited by 2
- 2021Balancing Value Underestimation and Overestimation with Realistic Actor-Criticcited by 1
- 2022VI-PINNs: Variance-involved Physics-informed Neural Networks for Fast and Accurate Prediction of Partial Differential Equationscited by 1
- 2023The Ladder in Chaos: A Simple and Effective Improvement to General DRL Algorithms by Policy Path Trimming and Boostingcited by 1
- 2024Learning the Target Network in Function Spacecited by 1
- 2025Generalized Task-Driven Medical Image Quality Enhancement with Gradient Promotioncited by 1
- 2025Deep Double Q-learningcited by 1
- 2026Task-Induced Representational Invariances Depend on Learning Objective in Deep RLcited by 1
- 2021Uncertainty-aware Low-Rank Q-Matrix Estimation for Deep Reinforcement Learningcited by 0
- 2022The Sandbox Environment for Generalizable Agent Research (SEGAR)cited by 0
50 most cited of 56 All 56 (JSON)
- 2026What a World Model Represents Is Three Questions7 Jul 2026
- 2026Task-Induced Representational Invariances Depend on Learning Objective in Deep RL1 Jun 2026
- 2026PREFAB: PREFerence-based Affective Modeling for Low-Budget Self-Annotation20 Jan 2026
- 2025Predictive Coding Enhances Meta-RL To Achieve Interpretable Bayes-Optimal Belief Representation Under Partial Observability24 Oct 2025
- 2025floq: Training Critics via Flow-Matching for Scaling Compute in Value-Based RL8 Sep 2025
- 2025Deep Double Q-learning30 Jun 2025
- 2025A Survey of State Representation Learning for Deep Reinforcement Learning20 Jun 2025
- 2025Generalized Task-Driven Medical Image Quality Enhancement with Gradient Promotion2 Jan 2025
- 2024Plasticity Loss in Deep Reinforcement Learning: A Survey7 Nov 2024
- 2024Random Latent Exploration for Deep Reinforcement Learning18 Jul 2024
- 2024A Role of Environmental Complexity on Representation Learning in Deep Reinforcement Learning Agents3 Jul 2024
- 2024When does Self-Prediction help? Understanding Auxiliary Tasks in Reinforcement Learning25 Jun 2024
38 more
- 2024A Unifying Framework for Action-Conditional Self-Predictive Reinforcement Learning4 Jun 2024
- 2024Learning the Target Network in Function Space3 Jun 2024
- 2024An Advanced Reinforcement Learning Framework for Online Scheduling of Deferrable Workloads in Cloud Computing3 Jun 2024
- 2024Adaptive Regularization of Representation Rank as an Implicit Constraint of Bellman Equation19 Apr 2024
- 2024MTLight: Efficient Multi-Task Reinforcement Learning for Traffic Signal Control1 Apr 2024
- 2024Stop Regressing: Training Value Functions via Classification for Scalable Deep RL6 Mar 2024
- 2024Investigating the Histogram Loss in Regression20 Feb 2024
- 2024Understanding What Affects the Generalization Gap in Visual Reinforcement Learning: Theory and Empirical Evidence5 Feb 2024
- 2024Bridging State and History Representations: Understanding Self-Predictive RL17 Jan 2024
- 2023PcLast: Discovering Plannable Continuous Latent States6 Nov 2023
- 2023Predictive auxiliary objectives in deep RL mimic learning in the brain9 Oct 2023
- 2023TD Convergence: An Optimization Perspective30 Jun 2023
- 2023Eigensubspace of Temporal-Difference Dynamics and How It Improves Value Approximation in Reinforcement Learning29 Jun 2023
- 2023The RL Perceptron: Generalisation Dynamics of Policy Learning in High Dimensions17 Jun 2023
- 2023Bootstrapped Representations in Reinforcement Learning16 Jun 2023
- 2023Investigating Navigation Strategies in the Morris Water Maze through Deep Reinforcement Learning1 Jun 2023
- 2023Towards a Better Understanding of Representation Dynamics under TD-learning29 May 2023
- 2023Policy Gradient Methods in the Presence of Symmetries and State Abstractions9 May 2023
- 2023Representations and Exploration for Deep Reinforcement Learning using Singular Value Decomposition1 May 2023
- 2023Proto-Value Networks: Scaling Representation Learning with Auxiliary Tasks25 Apr 2023
- 2023The Ladder in Chaos: A Simple and Effective Improvement to General DRL Algorithms by Policy Path Trimming and Boosting2 Mar 2023
- 2023CrystalBox: Future-Based Explanations for Input-Driven Deep RL Systems27 Feb 2023
- 2023Deep Laplacian-based Options for Temporally-Extended Exploration26 Jan 2023
- 2022A Novel Stochastic Gradient Descent Algorithm for Learning Principal Subspaces8 Dec 2022
- 2022Understanding Self-Predictive Learning for Reinforcement Learning6 Dec 2022
- 2022VI-PINNs: Variance-involved Physics-informed Neural Networks for Fast and Accurate Prediction of Partial Differential Equations30 Nov 2022
- 2022Exploring through Random Curiosity with General Value Functions18 Nov 2022
- 2022Auxiliary task discovery through generate-and-test25 Oct 2022
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradient15 Sep 2022
- 2022An Empirical Study of Implicit Regularization in Deep Offline RL5 Jul 2022
- 2022Learning Dynamics and Generalization in Reinforcement Learning5 Jun 2022
- 2022Multi-Game Decision Transformers30 May 2022
- 2022Frustratingly Easy Regularization on Representation Can Boost Deep Reinforcement Learning29 May 2022
- 2022Engineering flexible machine learning systems by traversing functionally-invariant paths30 Apr 2022
- 2022Understanding and Preventing Capacity Loss in Reinforcement Learning20 Apr 2022
- 2022What makes useful auxiliary tasks in reinforcement learning: investigating the effect of the target policy1 Apr 2022
- 2022The Sandbox Environment for Generalizable Agent Research (SEGAR)19 Mar 2022
- 2022On the Generalization of Representations in Reinforcement Learning1 Mar 2022
50 newest of 56 All 56 (JSON)
- 2022Multi-Game Decision Transformerscited by 212
- 2022An Empirical Study of Implicit Regularization in Deep Offline RLcited by 24
- 2022An Empirical Study of Implicit Regularization in Deep Offline RL5 Jul 2022
- 2022Multi-Game Decision Transformers30 May 2022
- 2026What a World Model Represents Is Three Questionscited by 0
- 2026What a World Model Represents Is Three Questions7 Jul 2026
- 2024Stop Regressing: Training Value Functions via Classification for Scalable Deep RLcited by 136
- 2022Understanding and Preventing Capacity Loss in Reinforcement Learningcited by 124
- 2021Stabilizing Deep Q-Learning with ConvNets and Vision Transformers under Data Augmentationcited by 101
- 2024Bridging State and History Representations: Understanding Self-Predictive RLcited by 52
- 2023Proto-Value Networks: Scaling Representation Learning with Auxiliary Taskscited by 41
- 2022Understanding Self-Predictive Learning for Reinforcement Learningcited by 40
- 2022On the Generalization of Representations in Reinforcement Learningcited by 28
- 2023Deep Laplacian-based Options for Temporally-Extended Explorationcited by 28
- 2022Continuous MDP Homomorphisms and Homomorphic Policy Gradientcited by 24
- 2024Plasticity Loss in Deep Reinforcement Learning: A Surveycited by 24
- 2022Transfer RL across Observation Feature Spaces via Model-Based Regularizationcited by 18
- 2025A Survey of State Representation Learning for Deep Reinforcement Learningcited by 17
12 most cited of 52 All 52 (JSON)
- 2026Task-Induced Representational Invariances Depend on Learning Objective in Deep RL1 Jun 2026
- 2026PREFAB: PREFerence-based Affective Modeling for Low-Budget Self-Annotation20 Jan 2026
- 2025Predictive Coding Enhances Meta-RL To Achieve Interpretable Bayes-Optimal Belief Representation Under Partial Observability24 Oct 2025
- 2025Deep Double Q-learning30 Jun 2025
- 2025A Survey of State Representation Learning for Deep Reinforcement Learning20 Jun 2025
- 2025Generalized Task-Driven Medical Image Quality Enhancement with Gradient Promotion2 Jan 2025
- 2024Plasticity Loss in Deep Reinforcement Learning: A Survey7 Nov 2024
- 2024Random Latent Exploration for Deep Reinforcement Learning18 Jul 2024
- 2024A Role of Environmental Complexity on Representation Learning in Deep Reinforcement Learning Agents3 Jul 2024
- 2024When does Self-Prediction help? Understanding Auxiliary Tasks in Reinforcement Learning25 Jun 2024
- 2024A Unifying Framework for Action-Conditional Self-Predictive Reinforcement Learning4 Jun 2024
- 2024Learning the Target Network in Function Space3 Jun 2024
12 newest of 52 All 52 (JSON)