How Does Pruning Impact Long-Tailed Multi-label Medical Image Classifiers?

TitleHow Does Pruning Impact Long-Tailed Multi-label Medical Image Classifiers?
Publication TypeJournal Article
Year of Publication2023
AuthorsHolste G, Jiang Z, Jaiswal A, Hanna M, Minkowitz S, Legasto AC, Escalon JG, Steinberger S, Bittman M, Shen TC, Ding Y, Summers RM, Shih G, Peng Y, Wang Z
JournalMed Image Comput Comput Assist Interv
Volume14224
Pagination663-673
Date Published2023 Oct
Abstract

Pruning has emerged as a powerful technique for compressing deep neural networks, reducing memory usage and inference time without significantly affecting overall performance. However, the nuanced ways in which pruning impacts model behavior are not well understood, particularly for long-tailed, multi-label datasets commonly found in clinical settings. This knowledge gap could have dangerous implications when deploying a pruned model for diagnosis, where unexpected model behavior could impact patient well-being. To fill this gap, we perform the first analysis of pruning's effect on neural networks trained to diagnose thorax diseases from chest X-rays (CXRs). On two large CXR datasets, we examine which diseases are most affected by pruning and characterize class "forgettability" based on disease frequency and co-occurrence behavior. Further, we identify individual CXRs where uncompressed and heavily pruned models disagree, known as pruning-identified exemplars (PIEs), and conduct a human reader study to evaluate their unifying qualities. We find that radiologists perceive PIEs as having more label noise, lower image quality, and higher diagnosis difficulty. This work represents a first step toward understanding the impact of pruning on model behavior in deep long-tailed, multi-label medical image classification. All code, model weights, and data access instructions can be found at https://github.com/VITA-Group/PruneCXR.

DOI10.1007/978-3-031-43904-9_64
Alternate JournalMed Image Comput Comput Assist Interv
PubMed ID37829549
PubMed Central IDPMC10568970
Grant ListR00 LM013001 / LM / NLM NIH HHS / United States