Drug–target prediction through self supervised learning with dual task ensemble approach

Abstract

Drug–Target interaction (DTI) prediction, a transformative approach in pharmaceutical research, seeks novel therapeutic applications for computational method based virtual screening, existing drugs to address untreated diseases and discovery of existing drugs side effects. The proposed model predict DTI through Heterogeneous biological network by combining drug, genes and disease related knowledge. For the purpose of embedding extraction Self-supervised learning (SSL) has been used which, trains models through pretext tasks, eliminating the need for manual annotations. The pretext tasks are related to either structural based information or similarity based information. To mitigate GNN vulnerability to non-robustness, ensemble learning can be incorporated into GNNs, harnessing multiple models to enhance robustness. This paper introduces a Graph neural network based architecture consisting of task based module and ensemble module for link prediction of DTI. The ensemble module of dual task combinations, both in cold start and warm start scenarios achieve very good performance as it provide 0.960 in cold start and 0.970 in warm start mean AUCROC score with less deviation.