Journal
DISPLAYS
Volume 78, Issue -, Pages -Publisher
ELSEVIER
DOI: 10.1016/j.displa.2023.102402
Keywords
Multi-task learning; Semi-supervised segmentation; Survival analysis
Ask authors/readers for more resources
Accurate survival prediction is crucial in precision oncology for glioma patients. Current deep learning-based survival analysis methods heavily depend on segmented tumor regions, requiring manual annotation. We propose a multi-task learning approach that combines both survival prediction and semi-supervised tumor segmentation, achieving comparable results to state-of-the-art methods.
Accurate survival prediction is essential for precision oncology in patients with glioma. However, current deep learning-based survival analysis methods highly rely on segmented tumor regions, which requires tedious manual annotation. Semi-supervised segmentation offers an efficient way to reduce the annotation burden. However, most studies consider survival prediction and semi-supervised segmentation as two separated problems. Here, we proposed a multi-task learning approach for concurrent survival prediction and semi -supervised tumor segmentation. We train a shared multi-modal Transformer encoder to extract features from multiple modalities and fuse them at different levels. The extracted features are employed to construct contrast learning loss and survival analysis loss to implement semi-supervised segmentation and survival analysis, respectively. Experiments are conducted on two datasets from two local hospitals. Our method achieves comparable or slightly better results than state-of-the-art semi-supervised segmentation methods and achieves acceptable survival analysis results. Our data suggests that the proposed multi-task architecture can enhance both segmentation and survival prediction tasks in a semi-supervised learning manner.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available