Abstract
Semi-supervised learning has recently been attracting attention as an alternative to fully supervised models that require large pools of labeled data. Moreover, optimizing a model for multiple tasks can provide better generalizability than single-task learning. Leveraging self-supervision and adversarial training, we propose a novel, general purpose semi-supervised, multiple-task model - namely, self-supervised, semi-supervised, multi-task learning (S4MTL) - for accomplishing two important medical image analysis tasks: segmentation and diagnostic classification. Experimental results on chest and spine X-ray datasets confirm that our S4MTL model significantly outperforms semi-supervised single-task, semi/fully-supervised multi-task, and fully-supervised single-task models, even with a 50% reduction in class and segmentation labels.
Original language | English |
---|---|
Title of host publication | Proceedings - 19th IEEE International Conference on Machine Learning and Applications, ICMLA 2020 |
Editors | M. Arif Wani, Feng Luo, Xiaolin Li, Dejing Dou, Francesco Bonchi |
Pages | 769-774 |
Number of pages | 6 |
ISBN (Electronic) | 9781728184708 |
DOIs | |
State | Published - Dec 2020 |
Event | 19th IEEE International Conference on Machine Learning and Applications, ICMLA 2020 - Virtual, Miami, United States Duration: Dec 14 2020 → Dec 17 2020 |
Publication series
Name | Proceedings - 19th IEEE International Conference on Machine Learning and Applications, ICMLA 2020 |
---|
Conference
Conference | 19th IEEE International Conference on Machine Learning and Applications, ICMLA 2020 |
---|---|
Country/Territory | United States |
City | Virtual, Miami |
Period | 12/14/20 → 12/17/20 |
Bibliographical note
Publisher Copyright:© 2020 IEEE.
Keywords
- X-ray
- chest
- classification
- multi-task learning
- segmentation
- self-supervision
- semi-supervised learning
- spine
ASJC Scopus subject areas
- Artificial Intelligence
- Computer Science Applications
- Computer Vision and Pattern Recognition
- Hardware and Architecture