A Unified View on Solving Objective Mismatch in Model-Based Reinforcement Learning

Published: 08 Feb 2024, Last Modified: 08 Feb 2024Accepted by TMLREveryoneRevisionsBibTeX
Abstract: Model-based Reinforcement Learning (MBRL) aims to make agents more sample-efficient, adaptive, and explainable by learning an explicit model of the environment. While the capabilities of MBRL agents have significantly improved in recent years, how to best learn the model is still an unresolved question. The majority of MBRL algorithms aim at training the model to make accurate predictions about the environment and subsequently using the model to determine the most rewarding actions. However, recent research has shown that model predictive accuracy is often not correlated with action quality, tracing the root cause to the objective mismatch between accurate dynamics model learning and policy optimization of rewards. A number of interrelated solution categories to the objective mismatch problem have emerged as MBRL continues to mature as a research area. In this work, we provide an in-depth survey of these solution categories and propose a taxonomy to foster future research.
Certifications: Survey Certification
Submission Length: Long submission (more than 12 pages of main content)
Assigned Action Editor: ~Amir-massoud_Farahmand1
License: Creative Commons Attribution 4.0 International (CC BY 4.0)
Submission Number: 1665