An uncertainty-based interpretable deep learning framework for predicting breast cancer outcome

Hua Chai,Siyin Lin,Junqi Lin,Minfan He,Yuedong Yang,Yongzhong OuYang,Huiying Zhao
DOI: https://doi.org/10.1186/s12859-024-05716-7
IF: 3.307
2024-03-02
BMC Bioinformatics
Abstract:Predicting outcome of breast cancer is important for selecting appropriate treatments and prolonging the survival periods of patients. Recently, different deep learning-based methods have been carefully designed for cancer outcome prediction. However, the application of these methods is still challenged by interpretability. In this study, we proposed a novel multitask deep neural network called UISNet to predict the outcome of breast cancer. The UISNet is able to interpret the importance of features for the prediction model via an uncertainty-based integrated gradients algorithm. UISNet improved the prediction by introducing prior biological pathway knowledge and utilizing patient heterogeneity information.
biochemical research methods,biotechnology & applied microbiology,mathematical & computational biology
What problem does this paper attempt to address?