Horizon Reduction Makes RL Scalable
About
In this work, we study the scalability of offline reinforcement learning (RL) algorithms. In principle, a truly scalable offline RL algorithm should be able to solve any given problem, regardless of its complexity, given sufficient data, compute, and model capacity. We investigate if and how current offline RL algorithms match up to this promise on diverse, challenging, previously unsolved tasks, using datasets up to 1000x larger than typical offline RL datasets. We observe that despite scaling up data, many existing offline RL algorithms exhibit poor scaling behavior, saturating well below the maximum performance. We hypothesize that the horizon is the main cause behind the poor scaling of offline RL. We empirically verify this hypothesis through several analysis experiments, showing that long horizons indeed present a fundamental barrier to scaling up offline RL. We then show that various horizon reduction techniques substantially enhance scalability on challenging tasks. Based on our insights, we also introduce a minimal yet scalable method named SHARSA that effectively reduces the horizon. SHARSA achieves the best asymptotic performance and scaling behavior among our evaluation methods, showing that explicitly reducing the horizon unlocks the scalability of offline RL. Code: https://github.com/seohongpark/horizon-reduction
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Locomotion | OG-Bench humanoidmaze-medium-navigate-oraclerep v0 | Success Rate98 | 10 | |
| Locomotion | OG-Bench humanoidmaze-giant-navigate-oraclerep v0 | Success Rate82 | 10 | |
| Manipulation | OG-Bench puzzle-3x3-play-oraclerep v0 | Success Rate1 | 10 | |
| Offline Goal-Conditioned Reinforcement Learning | cube-octuple-1B | Success Rate3.40e+3 | 10 | |
| Manipulation | OG-Bench cube-double-play-oraclerep v0 | Success Rate95 | 10 | |
| Manipulation | OG-Bench cube-octuple-play-oraclerep v0 | Success Rate1.90e+3 | 10 | |
| Manipulation | OG-Bench puzzle-4x5-play-oraclerep v0 | Success Rate91 | 10 | |
| Offline Goal-Conditioned Reinforcement Learning | cube-quadruple 100M | Success Rate64 | 10 | |
| Offline Goal-Conditioned Reinforcement Learning | cube-triple 100M | Success Rate83 | 10 | |
| Offline Goal-Conditioned Reinforcement Learning | puzzle-4x6-1B | Success Rate6.40e+3 | 10 |