Fingerprint-Based Machine Learning for SARS-CoV-2 and MERS-CoV MproInhibition: Highlighting the Potential of Bayesian Neural Networks

Abstract

Severe acute respiratory syndrome coronavirus 2 (SARS-CoV-2) and Middle East respiratory syndrome coronavirus (MERS-CoV) are two important targets in current drug discovery, mainly due to the COVID-19 pandemic and the MERS-CoV outbreaks in recent years. An important target of both SARS-CoV-2 and MERS-CoV is the main protease (Mpro). Recently, the ASAP Discovery Consortium focused on the acceleration of Mpro inhibitors with a part of this initiative being an open blind challenge in collaboration with Valence lab using the Polaris platform, where data sets of previously undisclosed inhibitors of SARS-CoV-2 Mpro and MERS-CoV Mpro were shared with researchers, to allow the development of machine learning and deep learning models for the prediction of the potency. We used this opportunity to evaluate and compare traditional machine learning models consisting of a random forest (RF) and gradient boosting model (XGBoost) with a bayesian neural network (BNN) model. For this purpose, we created single task models for the predictions of each of the targets. The results obtained showed that the BNN model outperformed both traditional machine learning models for both targets, indicating that BNNs are a promising deep learning framework in low-data regimes.