Generalization Of Reinforcement Learners With Working And Episodic Memory

Abstract

Memory is an important aspect of intelligence and plays a role in many deep reinforcement learning models. However, little progress has been made in understanding when specific memory systems help more than others and how well they generalize. The field also has yet to see a prevalent consistent and rigorous approach for evaluating agent performance on holdout data. In this paper, we aim to develop a comprehensive methodology to test different kinds of memory in an agent and assess how well the agent can apply what it learns in training to a holdout set that differs from the training set along dimensions that we suggest are relevant for evaluating memory-specific generalization. To that end, we first construct a diverse set of memory tasks that allow us to evaluate test-time generalization across multiple dimensions. Second, we develop and perform multiple ablations on an agent architecture that combines multiple memory systems, observe its baseline models, and investigate its performa

Stats

• citations0
• S2 citations—
• github stars0
• HF likes0
• heat score0.00
• arxiv keyfortunato2019generalization

Related papers

• Unraveling The Complexity Of Memory In RL Agents: An Approach For Classification And Evaluation (2024)0.00
• Generative Memory For Lifelong Reinforcement Learning (2019)0.00
• Integrating Episodic Memory Into A Reinforcement Learning Agent Using Reservoir Sampling (2018)0.00
• Stable Hadamard Memory: Revitalizing Memory-augmented Agents For Reinforcement Learning (2024)0.00
• Assessing Generalization In Deep Reinforcement Learning (2018)0.00
• Towards Mental Time Travel: A Hierarchical Memory For Reinforcement Learning Agents (2021)0.00
• Augmented Replay Memory In Reinforcement Learning With Continuous Control (2019)5.24
• Improving Generalization To New Environments And Removing Catastrophic Forgetting In Reinforcement Learning By Using An Eco-system Of Agents (2022)0.00