Multiple-Frequencies Population-Based Training
By: Waël Doulazmi , Auguste Lehuger , Marin Toromanoff and more
Potential Business Impact:
Helps AI learn better by trying many ways.
Reinforcement Learning's high sensitivity to hyperparameters is a source of instability and inefficiency, creating significant challenges for practitioners. Hyperparameter Optimization (HPO) algorithms have been developed to address this issue, among them Population-Based Training (PBT) stands out for its ability to generate hyperparameters schedules instead of fixed configurations. PBT trains a population of agents, each with its own hyperparameters, frequently ranking them and replacing the worst performers with mutations of the best agents. These intermediate selection steps can cause PBT to focus on short-term improvements, leading it to get stuck in local optima and eventually fall behind vanilla Random Search over longer timescales. This paper studies how this greediness issue is connected to the choice of evolution frequency, the rate at which the selection is done. We propose Multiple-Frequencies Population-Based Training (MF-PBT), a novel HPO algorithm that addresses greediness by employing sub-populations, each evolving at distinct frequencies. MF-PBT introduces a migration process to transfer information between sub-populations, with an asymmetric design to balance short and long-term optimization. Extensive experiments on the Brax suite demonstrate that MF-PBT improves sample efficiency and long-term performance, even without actually tuning hyperparameters.
Similar Papers
Iterated Population Based Training with Task-Agnostic Restarts
Machine Learning (CS)
**Finds best computer learning settings automatically.**
PB$^2$: Preference Space Exploration via Population-Based Methods in Preference-Based Reinforcement Learning
Artificial Intelligence
Teaches computers to learn from choices better.
Evolutionary Policy Optimization
Machine Learning (CS)
Teaches robots to learn faster and better.