MMCL-Net: Spinal Disease Diagnosis in Global Mode using Progressive Multi-task Joint Learning
Published in Neurocomputing, 2020
Recommended citation: Yanfei Hong, Benzheng Wei, Zhongyi Han, Xiang Li, Yuanjie Zheng, Shuo Li, "MMCL-Net: Spinal Disease Diagnosis in Global Mode using Progressive Multi-task Joint Learning". Neurocomputing, pp.307-316, 2020, doi: 10.1016/j.neucom.2020.01.112.
Abstract
Simultaneous detection, segmentation, and classification of multiple spinal structures on MRI is crucial for the early and pathogenesis-based diagnosis of multiple spine diseases in the clinical setting. It is more assistance for radiologists reflections on the disease based on the pathogenesis when the lesion area and its adjacent structures are detected. Obviously, the multiple structures of the spine are directly interdependent and influential, and the multi-tasks under a deep convolutional neural network framework can also influence each other. Multi-task joint optimization in the spinal global mode is a direct outlet to seek the dynamic balance of the above potential correlation. In this paper, we propose a novel end-to-end Multi-task Multi-structure Correlation Learning Network (MMCL-Net) for the detection, segmentation, and classification (normal, slight, marked, and severe) of three types of spine structure: disc, vertebra, and neural foramen simultaneously. And the model is locally optimized to achieve a more stable dynamic equilibrium state. Extensive experiments on T1/T2-weighted MR scans from 200 subjects demonstrate that MMCL-Net achieves high performance with mAP of 0.9187, the classification accuracy of 90.67%, and dice coefficient of 90.60%. The experimental results show that the performance of our method is comparable to that of the state-of-the-art methods.