Faster sorting algorithms discovered using deep reinforcement learning
Daniel J. Mankowitz,Andrea Michi,Anton Zhernov,Marco Gelmi,Marco Selvi,Cosmin Paduraru,Edouard Leurent,Shariq Iqbal,Jean-Baptiste Lespiau,Alex Ahern,Thomas Köppe,Kevin Millikin,Stephen Gaffney,Sophie Elster,Jackson Broshear,Chris Gamble,Kieran Milan,Robert Tung,Minjae Hwang,Taylan Cemgil,Mohammadamin Barekatain,Yujia Li,Amol Mandhane,Thomas Hubert,Julian Schrittwieser,Demis Hassabis,Pushmeet Kohli,Martin Riedmiller,Oriol Vinyals,David Silver
DOI: https://doi.org/10.1038/s41586-023-06004-9
IF: 64.8
2023-06-08
Nature
Abstract:Fundamental algorithms such as sorting or hashing are used trillions of times on any given day 1 . As demand for computation grows, it has become critical for these algorithms to be as performant as possible. Whereas remarkable progress has been achieved in the past 2 , making further improvements on the efficiency of these routines has proved challenging for both human scientists and computational approaches. Here we show how artificial intelligence can go beyond the current state of the art by discovering hitherto unknown routines. To realize this, we formulated the task of finding a better sorting routine as a single-player game. We then trained a new deep reinforcement learning agent, AlphaDev, to play this game. AlphaDev discovered small sorting algorithms from scratch that outperformed previously known human benchmarks. These algorithms have been integrated into the LLVM standard C++ sort library 3 . This change to this part of the sort library represents the replacement of a component with an algorithm that has been automatically discovered using reinforcement learning. We also present results in extra domains, showcasing the generality of the approach.
multidisciplinary sciences