Biology-aware mutation-based deep learning for outcome prediction of cancer immunotherapy with immune checkpoint inhibitors

The response rate of cancer immune checkpoint inhibitors (ICI) varies among patients, making it challenging to pre-determine whether a particular patient will respond to immunotherapy. While gene mutation is critical to the treatment outcome, a framework capable of explicitly incorporating biology knowledge has yet to be established. Here we aim to propose and validate a mutation-based deep learning model for survival analysis on 1571 patients treated with ICI. Our model achieves an average concordance index of 0.59 ± 0.13 across nine types of cancer, compared to the gold standard Cox-PH model (0.52 ± 0.10). The “black box” nature of deep learning is a major concern in healthcare field. This model’s interpretability, which results from incorporating the gene pathways and protein interaction (i.e., biology-aware) rather than relying on a ‘black box’ approach, helps patient stratification and provides insight into novel gene biomarkers, advancing our understanding of ICI treatment.