Practice Makes Perfect: Planning to Learn Skill Parameter Policies

• Published in: arXiv.org 2024-05
• Main Authors: Kumar, Nishanth, Silver, Tom, McClinton, Willie, Zhao, Linfeng, Proulx, Stephen, Lozano-Pérez, Tomás, Leslie Pack Kaelbling, Barry, Jennifer
• Format: Article
• Language: English
• Subjects: Parameterization; Parameters; Policies; Robot control; Robots; Skills; Task complexity
• ISSN: 2331-8422

One promising approach towards effective robot decision making in complex, long-horizon tasks is to sequence together parameterized skills. We consider a setting where a robot is initially equipped with (1) a library of parameterized skills, (2) an AI planner for sequencing together the skills given a goal, and (3) a very general prior distribution for selecting skill parameters. Once deployed, the robot should rapidly and autonomously learn to improve its performance by specializing its skill parameter selection policy to the particular objects, goals, and constraints in its environment. In this work, we focus on the active learning problem of choosing which skills to practice to maximize expected future task success. We propose that the robot should estimate the competence of each skill, extrapolate the competence (asking: "how much would the competence improve through practice?"), and situate the skill in the task distribution through competence-aware planning. This approach is implemented within a fully autonomous system where the robot repeatedly plans, practices, and learns without any environment resets. Through experiments in simulation, we find that our approach learns effective parameter policies more sample-efficiently than several baselines. Experiments in the real-world demonstrate our approach's ability to handle noise from perception and control and improve the robot's ability to solve two long-horizon mobile-manipulation tasks after a few hours of autonomous practice. Project website: http://ees.csail.mit.edu