Automatic Pituitary Adenoma Segmentation and Identification of Cavernous Sinus Invasion Via Multi-task Learning

Wenting Rui,Wenyuan Gao,Nidan Qiao,Xin Chen,Min Han,Yue Wu,Tao Xin,Jun ang,Yao Zhao,Zhenwei Yao
DOI: https://doi.org/10.1016/j.crad.2024.106756
IF: 3.389
2024-01-01
Clinical Radiology
Abstract:Objective To develop a multi-task deep learning model for pituitary macroadenoma (PMA) segmentation and identification of cavernous sinus (CS) invasion. Methods A total of 926 patients with PMAs (n=816 from Institution 1 for model training, and n=110 from Institution 2 for model validation) were included. Manual segmentation for tumor and bilateral internal carotid arteries (ICAs) was conducted on coronal T1WI + C images using ITK-SNAP. Two performing neurosurgeons evaluated the CS invasion during operation. A Multi-Task Multiaxis-Attention UNet (MTMAU-Net) framework that combined segmentation and CS invasion classification tasks was proposed. Dice similarity coefficient and 95% Hausdorff distance (HD95) were used to evaluate the segmentation performance. Accuracy and area under the curve (AUC) were used to assess the classification performance. Results Compared with several single-task models, MTMAU-Net model achieved higher Dice coefficient (90.84 ± 0.55%) and lower HD95 values (2.13 ± 0.37 mm) in segmentation tasks, and performed better in the classification task (accuracy: 88.05 ± 2.84%, AUC: 0.89 ± 0.08). Compared with the Knosp grading system, this model showed overall consistent performance (accuracy: 88.05% vs 87.68%) and outperformed human grading evaluation at Knosp grade 3 (83.33% vs 57.62%). MTMAU-Net showed great capability of generalization in both segmentation (Dice coefficient: 83.71 ± 5.93%) and classification tasks (accuracy: 84.55%, AUC: 0.87) in the validation cohort. Conclusion MTMAU-Net outperformed single-task models in PMA segmentation and CS invasion classification, and showed advantages over the Knosp grading system in identifying CS invasion, which might guide clinicians to make appropriate surgical strategies.
What problem does this paper attempt to address?