Tab-Cox: An Interpretable Deep Survival Analysis Model for Patients with Nasopharyngeal Carcinoma based on TabNet

IEEE J Biomed Health Inform. 2024 May 13:PP. doi: 10.1109/JBHI.2024.3397955. Online ahead of print.

Abstract

The nutritional status of cancer patients is closely associated with the clinical progression of the disease. A survival analysis model combined with a neural network can predict future disease trends in patients, facilitating early prevention and assisting physicians in making diagnoses. However, the complexity of neural networks and their incompatibility with medical tabular data can reduce the interpretability of the model. To address this issue, thr paper propose a novel survival analysis model called Tab-Cox, which combines TabNet and Cox models. This model is specifically designed to predict the survival outcomes of patients with nasopharyngeal carcinoma. The model utilizes TabNet's sequential attention mechanism to extract more interpretable features, providing an interpretable method for identifying disease risk factors. Consequently, the model ensures accurate survival prediction while also making the results more comprehensible for both patients and doctors. The paper tested the efficacy of the model by conducting experiments on various diverse datasets in comparison with other commonly used survival models. The results showed that the proposed model delivered the highest or second-highest accuracy across all datasets. Furthermore, the paper conducted a comparative interpretability analysis against the classical Cox model. In addition and compare the interpretability of the Tab-Cox model with the classical Cox model and discuss the advantages and disadvantages of its interpretability. This demonstrates that Tab-Cox can assist doctors in identifying risk factors that are challenging to capture using artificial methods.