Back

Interpretable Drug Response Prediction through Molecule Structure-aware and Knowledge-Guided Visible Neural Network

Xie, J.; Zhang, Z.; Li, Y.; Rao, J.; Yang, Y.

2024-02-09 bioinformatics
10.1101/2024.02.07.579280 bioRxiv
Show abstract

Precise prediction of anti-cancer drug responses has become a crucial obstruction in anti-cancer drug design and clinical applications. In recent years, various deep learning methods have been applied to drug response prediction and become more accurate. However, they are still criticized as being non-transparent. To offer reliable drug response prediction in real-world applications, there is still a pressing demand to develop a model with high predictive performance as well as interpretability. In this study, we propose DrugVNN, an end-to-end interpretable drug response prediction framework, which extracts gene features of cell lines through a knowledge-guided visible neural network (VNN), and learns drug representation through a node-edge communicative message passing network (CMPNN). Additionally, between these two networks, a novel drug-aware gene attention gate is designed to direct the drug representation to VNN to simulate the effects of drugs. By evaluating on the GDSC dataset, DrugVNN achieved state-of-the-art performance. Moreover, DrugVNN can identify active genes and relevant signaling pathways for specific drug-cell line pairs with supporting evidence in the literature, implying the interpretability of our model.

Matching journals

The top 6 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.