Uni-Dock: GPU-Accelerated Docking Enables Ultralarge Virtual Screening

Molecular docking, a structure-based virtual screening method, is a reliable tool to enrich potential bioactive molecules from molecular databases. With the rapid expansion of compound library sizes, the speed of existing molecular docking programs becomes less than adequate to meet the demand for screening ultralarge libraries containing tens of millions or billions of molecules. Here, we propose Uni-Dock, a GPU-accelerated molecular docking program that supports various scoring functions including vina, vinardo, and ad4. Uni-Dock achieves more than 1000-fold speedup with high accuracy compared with the AutoDock Vina running in single CPU core, outperforming reported GPU-accelerated docking programs including AutoDock-GPU and Vina-GPU based on head-to-head experiments. Uni-Dock docks molecules in batches simultaneously using concurrent threads of each molecule. The data flow between GPU and CPU is optimized to eliminate CPU hotspots and maximize GPU utility. Additionally, Uni-Dock also supports hydrogen bond biased docking for all scoring functions and can be migrated to multiple GPUs of different architectures and manufacturers. We analyzed the improved performance of Uni-Dock on the CASF-2016 and DUD-E datasets and recommend three combinations of hyperparameters corresponding to different docking scenarios. To demonstrate Uni-Dock's capability on routinely screening ultralarge libraries, we performed hierarchical virtual screening experiments with Uni-Dock on the Enamine Diverse REAL druglike set containing 38.2 million molecules to a popular target KRAS G12D in 12 h using 100 NVIDIA V100 GPUs. To the best of our knowledge, Uni-Dock should be the fastest GPU-accelerated docking program to date.