Book chapter
Bridging the Task Barriers: Online Knowledge Distillation Across Tasks for Semi-supervised Mediastinal Segmentation in CT
Machine Learning in Medical Imaging, pp.310-319
Lecture Notes in Computer Science, v. 14348, Springer Nature Switzerland
10/15/2023
DOI: 10.1007/978-3-031-45673-2_31
Abstract
Segmentation of the mediastinal vasculature in computed tomography (CT) enables automated extraction of important biomarkers for cardiopulmonary disease characterization and outcome prediction. However, the limited contrast between blood and surrounding soft tissue makes manual segmentation of mediastinal structures challenging in non-contrast CT (NCCT) images, resulting in limited annotations for training deep learning models. To overcome this challenge, we propose a semi-supervised mediastinal vasculature segmentation method that utilizes knowledge distillation from unlabeled training data of contrast-enhanced dual-energy CT to achieve segmentation of the main pulmonary artery, main pulmonary veins, and aorta in NCCT. Our framework incorporates multitask learning with attention feature fusion bridges for online knowledge transfer from a related image-to-image translation task to the target segmentation task. Experimental evaluations demonstrate superior segmentation accuracy of our approach compared to fully supervised methods as well as two sequential approaches that do not leverage distillation between tasks. The proposed approach achieves a Dice similarity coefficient of 0.871 for the main pulmonary artery, 0.920 for the aorta, and 0.824 for the main pulmonary veins. By leveraging a large dataset without annotations through multitask learning and knowledge distillation, our approach improves performance in the target task of mediastinal segmentation with limited annotated training data.
Details
- Title: Subtitle
- Bridging the Task Barriers: Online Knowledge Distillation Across Tasks for Semi-supervised Mediastinal Segmentation in CT
- Creators
- Muhammad F. A. Chaudhary - University of IowaSeyed Soheil Hosseini - University of IowaR. Graham Barr - Columbia University Irving Medical CenterJoseph M. Reinhardt - University of IowaEric A. Hoffman - University of IowaSarah E. Gerard - University of Iowa
- Resource Type
- Book chapter
- Publication Details
- Machine Learning in Medical Imaging, pp.310-319
- Publisher
- Springer Nature Switzerland; Cham
- Series
- Lecture Notes in Computer Science; v. 14348
- DOI
- 10.1007/978-3-031-45673-2_31
- eISSN
- 1611-3349
- ISSN
- 0302-9743
- Language
- English
- Date published
- 10/15/2023
- Academic Unit
- Internal Medicine; Roy J. Carver Department of Biomedical Engineering; Radiology
- Record Identifier
- 9984475075602771
Metrics
9 Record Views