Deep learning from label proportions for emphysema quantification
We propose an end-to-end deep learning method that learns to estimate emphysema extent from proportions of the diseased tissue. These proportions were visually estimated by experts using a standard grading system, in which grades correspond to intervals (label example: 1–5% of diseased tissue). The proposed architecture encodes the knowledge that the labels represent a volumetric proportion. A custom loss is designed to learn with intervals. Thus, during training, our network learns to segment the diseased tissue such that its proportions fit the ground truth intervals. Our architecture and loss combined improve the performance substantially (8% ICC) compared to a more conventional regression network. We outperform traditional lung densitometry and two recently published methods for emphysema quantification by a large margin (at least 7% AUC and 15% ICC), and achieve near-human-level performance. Moreover, our method generates emphysema segmentations that predict the spatial distribution of emphysema at human level.
|Emphysema quantification, Learning from label proportions, Multiple instance learning, Weak labels|
|Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)|
|No subscription - This is an eBook|
|Organisation||Erasmus MC: University Medical Center Rotterdam|
Bortsova, G. (Gerda), Dubost, F. (Florian), Ørting, S. (Silas), Katramados, I. (Ioannis), Hogeweg, L. (Laurens), Thomsen, L.H, … de Bruijne, M. (2018). Deep learning from label proportions for emphysema quantification. In Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics). doi:10.1007/978-3-030-00934-2_85