Segmentation-guided multi-modal brain tumors survival prediction model using pseudo-labeling approach

Predicting a patient's survival period with a brain tumor can enable doctors to more efficiently assess the efficacy of various treatment modalities and adjust clinical treatment plans under the actual circumstances. This approach aims to enhance patients' prognosis and survival period. In recent years, there have been significant advancements in predicting the survival period of brain tumors using deep learning techniques. However, most of these methods require the use of two or more independent networks for subsequent knowledge sharing, often overlooking population information. Moreover, analyzing censored data presents an additional challenge, as incomplete information can make it difficult to apply standard analytical techniques and often leads to suboptimal performance in predictive models. To address these challenges, we propose an end-to-end multi-model pseudo-label approach for brain tumor survival prediction, designed to accurately estimate patient survival periods. First, our framework integrates patient population information for optimal use of population information. Next, to enhance data correction, we introduce a method for generating class labels, effectively enlarging the original sample size and improving data utilization efficiency. Additionally, we collected and supplemented survival labels from the BraTS 2021 dataset, which now serves as a new dataset for both brain tumor segmentation and survival prediction. Experimental results demonstrate that our model can enhance the precision of predicting the survival rate of brain tumor patients and exhibit superior generalization. The source code is available at: https://github.com/linciel/Brain-Tumor-Survival-Prediction.