Frontiers
Browse
Image_10_A Deep-Learning Model With the Attention Mechanism Could Rigorously Predict Survivals in Neuroblastoma.tif (856.06 kB)

Image_10_A Deep-Learning Model With the Attention Mechanism Could Rigorously Predict Survivals in Neuroblastoma.tif

Download (856.06 kB)
figure
posted on 2021-07-14, 08:05 authored by Chenzhao Feng, Tianyu Xiang, Zixuan Yi, Xinyao Meng, Xufeng Chu, Guiyang Huang, Xiang Zhao, Feng Chen, Bo Xiong, Jiexiong Feng
Background

Neuroblastoma is one of the most devastating forms of childhood cancer. Despite large amounts of attempts in precise survival prediction in neuroblastoma, the prediction efficacy remains to be improved.

Methods

Here, we applied a deep-learning (DL) model with the attention mechanism to predict survivals in neuroblastoma. We utilized 2 groups of features separated from 172 genes, to train 2 deep neural networks and combined them by the attention mechanism.

Results

This classifier could accurately predict survivals, with areas under the curve of receiver operating characteristic (ROC) curves and time-dependent ROC reaching 0.968 and 0.974 in the training set respectively. The accuracy of the model was further confirmed in a validation cohort. Importantly, the two feature groups were mapped to two groups of patients, which were prognostic in Kaplan-Meier curves. Biological analyses showed that they exhibited diverse molecular backgrounds which could be linked to the prognosis of the patients.

Conclusions

In this study, we applied artificial intelligence methods to improve the accuracy of neuroblastoma survival prediction based on gene expression and provide explanations for better understanding of the molecular mechanisms underlying neuroblastoma.

History