Learning and Understanding a Disentangled Feature Representation for Hidden Parameters in Reinforcement Learning

التفاصيل البيبلوغرافية
العنوان: Learning and Understanding a Disentangled Feature Representation for Hidden Parameters in Reinforcement Learning
المؤلفون: Reale, Christopher, Russell, Rebecca
سنة النشر: 2022
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Machine Learning
الوصف: Hidden parameters are latent variables in reinforcement learning (RL) environments that are constant over the course of a trajectory. Understanding what, if any, hidden parameters affect a particular environment can aid both the development and appropriate usage of RL systems. We present an unsupervised method to map RL trajectories into a feature space where distance represents the relative difference in system behavior due to hidden parameters. Our approach disentangles the effects of hidden parameters by leveraging a recurrent neural network (RNN) world model as used in model-based RL. First, we alter the standard world model training algorithm to isolate the hidden parameter information in the world model memory. Then, we use a metric learning approach to map the RNN memory into a space with a distance metric approximating a bisimulation metric with respect to the hidden parameters. The resulting disentangled feature space can be used to meaningfully relate trajectories to each other and analyze the hidden parameter. We demonstrate our approach on four hidden parameters across three RL environments. Finally we present two methods to help identify and understand the effects of hidden parameters on systems.
Comment: Appears in Proceedings of AAAI FSS-22 Symposium "Lessons Learned for Autonomous Assessment of Machine Abilities (LLAAMA)"
نوع الوثيقة: Working Paper
الوصول الحر: http://arxiv.org/abs/2211.16315Test
رقم الانضمام: edsarx.2211.16315
قاعدة البيانات: arXiv