Statistical Guarantees for Lifelong Reinforcement Learning using PAC-Bayes Theory
On the map: Deep reinforcement learning methods → RL theory and sample complexity
Abstract (arXiv)
Lifelong reinforcement learning (RL) has been developed as a paradigm for extending single-task RL to more realistic, dynamic settings. In lifelong RL, the "life" of an RL agent is modeled as a stream of tasks drawn from a task distribution. We propose EPIC (Empirical PAC-Bayes that Improves Continuously), a novel algorithm designed for lifelong RL using PAC-Bayes theory. EPIC learns a shared policy distribution, referred to as the world policy, which enables rapid adaptation to new tasks while retaining valuable knowledge from previous experiences. Our theoretical analysis establishes a relationship between the algorithm's generalization performance and the number of prior tasks preserved in memory. We also derive the sample complexity of EPIC in terms of RL regret. Extensive experiments on a variety of environments demonstrate that EPIC significantly outperforms existing methods in lifelong RL, offering both theoretical guarantees and practical efficacy through the use of the world policy.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Placed in RL theory and sample complexity, one of 15 directions in Deep reinforcement learning methods. Builds on 37 of its 54 references that are in the map. First posted on arXiv 1 Nov 2024.
Builds on
37 of 54 references.
References · 37
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
- 2015Continuous control with deep reinforcement learningcited by 3,407
- 2015Trust Region Policy Optimizationcited by 2,580
- 2018World Modelscited by 1,513
- 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 763
- 2017Mastering Chess and Shogi by Self-Play with a General Reinforcement Learning Algorithmcited by 734
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 715
- 2017Exploring Generalization in Deep Learningcited by 600
- 2016Weight Normalization: A Simple Reparameterization to Accelerate Training of Deep Neural Networkscited by 543
- 2016Target-driven Visual Navigation in Indoor Scenes using Deep Reinforcement Learningcited by 366
- 2017A PAC-Bayesian Approach to Spectrally-Normalized Margin Bounds for Neural Networkscited by 314
- 2020Towards Continual Reinforcement Learning: A Review and Perspectivescited by 274
25 more
- 2023A Definition of Continual Reinforcement Learningcited by 101
- 2017Meta-Learning by Adjusting Priors Based on Extended PAC-Bayes Theorycited by 76
- 2021A general sample complexity analysis of vanilla policy gradientcited by 76
- 2020learn2learn: A Library for Meta-Learning Researchcited by 69
- 2020On the role of data in PAC-Bayes boundscited by 64
- 2022A Provably Efficient Model-Free Posterior Sampling Method for Episodic Reinforcement Learningcited by 31
- 2015Data-dependent Initializations of Convolutional Neural Networkscited by 29
- 2017Robust and Efficient Transfer Learning with Hidden-Parameter Markov Decision Processescited by 29
- 2020Lifelong Policy Gradient Learning of Factored Policies for Faster Training Without Forgettingcited by 29
- 2021Regret Minimization Experience Replay in Off-Policy Reinforcement Learningcited by 25
- 2020Lipschitz Lifelong Reinforcement Learningcited by 23
- 2018PAC-Bayes Control: Learning Policies that Provably Generalize to Novel Environmentscited by 22
- 2024Efficient Frameworks for Generalized Low-Rank Matrix Bandit Problemscited by 20
- 2023Prediction and Control in Continual Reinforcement Learningcited by 19
- 2019Lifelong Learning with a Changing Action Setcited by 17
- 2022The Effectiveness of World Models for Continual Reinforcement Learningcited by 17
- 2020Probably Approximately Correct Vision-Based Planning using Motion Primitivescited by 16
- 2023Statistical Guarantees for Variational Autoencoders using PAC-Bayesian Theorycited by 16
- 2023Robust Lipschitz Bandits to Adversarial Corruptionscited by 15
- 2022Model-based Lifelong Reinforcement Learning with Bayesian Explorationcited by 12
- 2021Syndicated Bandits: A Framework for Auto Tuning Hyper-parameters in Contextual Bandit Algorithmscited by 10
- 2018Planning and Learning with Stochastic Action Setscited by 9
- 2022Reactive Exploration to Cope with Non-Stationarity in Lifelong Reinforcement Learningcited by 9
- 2022PAC-Bayesian Lifelong Learning For Multi-Armed Banditscited by 4
- 2021Reinforcement Learning With Sparse-Executing Actions via Sparsity Regularizationcited by 2
Built on by
Citing papers per year: 2024: 2; 2025: 5.
7 papers.
- 2024The Computational Limits of State-Space Models and Mamba via the Lens of Circuit Complexitycited by 21
- 2025Breaking the Frozen Subspace: Importance Sampling for Low-Rank Optimization in LLM Pretrainingcited by 7
- 2025Scaling Law Phenomena Across Regression Paradigms: Multiple and Kernel Approachescited by 4
- 2024Unlocking the Potential of Text-to-Image Diffusion with PAC-Bayesian Theorycited by 3
- 2025CoVE: Compressed Vocabulary Expansion Makes Better LLM-based Recommender Systemscited by 3
- 2025Time and Memory Trade-off of KV-Cache Compression in Tensor Transformer Decodingcited by 2
- 2025PAC-Bayesian Reinforcement Learning Trains Generalizable Policiescited by 1
- 2025PAC-Bayesian Reinforcement Learning Trains Generalizable Policies12 Oct 2025
- 2025CoVE: Compressed Vocabulary Expansion Makes Better LLM-based Recommender Systems24 Jun 2025
- 2025Time and Memory Trade-off of KV-Cache Compression in Tensor Transformer Decoding14 Mar 2025
- 2025Scaling Law Phenomena Across Regression Paradigms: Multiple and Kernel Approaches3 Mar 2025
- 2025Breaking the Frozen Subspace: Importance Sampling for Low-Rank Optimization in LLM Pretraining9 Feb 2025
- 2024The Computational Limits of State-Space Models and Mamba via the Lens of Circuit Complexity9 Dec 2024
- 2024Unlocking the Potential of Text-to-Image Diffusion with PAC-Bayesian Theory25 Nov 2024
Similar, not cited
Close in wording, no citation either way
- 2024Fast TRAC: A Parameter-Free Optimizer for Lifelong Reinforcement Learning
- 2026Reinforcement Learning for Reachability: Guaranteeing Asymptotic Optimality
- 2025Instance-Dependent Continuous-Time Reinforcement Learning via Maximum Likelihood Estimation
- 2025Rethinking the Foundations for Continual Reinforcement Learning
- 2026Beyond Single-Model Optimization: Preserving Plasticity in Continual Reinforcement Learning
- 2026Efficient Bayes-Adaptive Reinforcement Learning with Temporal Logic Specifications
- 2026DARLING: Detection Augmented Reinforcement Learning with Non-Stationary Guarantees
- 2026Maximum Likelihood Reinforcement Learning