arXiv · 2502.15662
Automating Curriculum Learning for Reinforcement Learning using a Skill-Based Bayesian Network
Abstract
A major challenge for reinforcement learning is automatically generating curricula to reduce training time or improve performance in some target task. We introduce SEBNs (Skill-Environment Bayesian Networks) which model a probabilistic relationship between a set of skills, a set of goals that relate to the reward structure, and a set of environment features to predict policy performance on (possibly unseen) tasks. We develop an algorithm that uses the inferred estimates of agent success from SEBN to weigh the possible next tasks by expected improvement. We evaluate the benefit of the resulting curriculum on three environments: a discrete gridworld, continuous control, and simulated robotics. The results show that curricula constructed using SEBN frequently outperform other baselines.
Explore related subjects
Keep this discovery
Vincent Hsiao, Mark Roberts, Laura M. Hiatt, George Konidaris, Dana Nau. 2025-02-21. Automating Curriculum Learning for Reinforcement Learning using a Skill-Based Bayesian Network. https://arxiv.org/abs/2502.15662
Cite the original work for its findings. Save a collection to share your selection of sources.