Kevin Wang (Princeton researcher and NeurIPS Best Paper co-author) highlights the benchmark performance gains of their 1,000-layer reinforcement learning architecture.
Insight
Kevin Wang: Cross-entropy trajectory classification enables scalable deep reinforcement learning
“I think it's because we're fundamentally shifting the burden of learning from something like, Q-learning or, like, regressing to, like, TD errors, which we know is quite spurious and noisy and biased, to fundamentally, like, a classification problem. We're try…”
Assertion Supported
Kevin Wang: Scaling RL network depth unlocks effective batch size scaling
“We notice that we see that scaling width actually also improves performance, and we also find that actually by scaling depth, we actually unlock the ability to scale along batch size as well.”
Assertion Supported
Kevin Wang: 1000-layer RL networks can train on single 80GB H100
“The nice thing is that all of our experiments, even the thousand layer networks, can be run on one single, 80 gigabyte, each 100 GPU.”
Assertion Not checkable as stated
Wang: Traditional value-based reinforcement learning fails to scale
“And so what we did is that we know that traditional RL, like let's say like value value-based RL doesn't really scale, right? This is pretty clear from the literature.”
Assertion Supported
Wang: 64 layers saturate performance in most reinforcement learning tasks
“Within our paper, like, for most environments we are able to, like, saturate, like, get to, like, almost perfect performance within just, you know, we don't even need to get to, like, a thousand layers. Like, maybe just 64 layers, for example, is sufficient.”
Assertion Supported
Wang: GPU environments collect hundreds of millions of RL timesteps hourly
“With these, like, GPU accelerated environments, we can collect hundreds of millions of time steps of data within just a few hours”