Anticipating the Consequences of Curriculum Decisions with Large Language Models
Organizations: University College London
Abstract
Automatic curriculum learning can improve the effectiveness of reinforcement learning by selecting the training experiences presented to the agent over time. Predicting the consequences of such decisions can, however, be difficult. We analyze automatic curriculum learning as a sequential decision-making problem, highlighting a gap between the quantities that determine the value of curriculum decisions and the information captured by local learning signals commonly used to guide them. We then investigate whether Large Language Models (LLMs) can exploit richer information about the learning problem to better anticipate the consequences of curriculum decisions. We introduce a method that combines online learning-progress estimates with LLM-informed estimates of (i) the potential downstream benefits of learning on each task and (ii) whether direct training on a task is currently likely to produce progress. We evaluate the approach on a custom benchmark of 256 textual goals in Craftax under different curriculum objectives. We observe the strongest gains when optimizing for individual target tasks. When optimizing across the full task set, the benefits vary across learners with different mechanisms for cross-task transfer, ranging from modest improvements in learning speed to larger gains that persist through the end of training.
Figures & tables
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
| Goal specification | Values of | Cumulative count |
|---|---|---|
| Have at least wood. | 7 | |
| Have at least stone. | 14 | |
| Have at least coal. | 21 | |
| Have at least iron. | 27 | |
| Have at least diamonds. | 32 | |
| Have at least sapphires. | 36 |
| Hyperparameter | Values evaluated | Selected |
|---|---|---|
| Learning rate | ||
| Entropy coefficient | ||
| Discount factor | ||
| GAE parameter | ||
| PPO minibatches | ||
| Rollout length per environment |
| Uniform task weights (3 seeds) | |||||
| B | B | B | |||
| Default | 2 | 1 | |||
| 1 | 1 | ||||
| 3 | 1 | ||||
| 2 | 3 | ||||
| 2 | 10 | ||||