MobilePTX: Sparse Coding for Pneumothorax Detection Given Limited Training Examples
Proceedings of the AAAI Conference on Artificial Intelligence
2023
Abstract
Point-of-Care Ultrasound (POCUS) refers to clinician-performed and interpreted ultrasonography at the patient's bedside. Interpreting these images requires a high level of expertise, which may not be available during emergencies. In this paper, we support POCUS by developing classifiers that can aid medical professionals by diagnosing whether or not a patient has pneumothorax. We decomposed the task into multiple steps, using YOLOv4 to extract relevant regions of the video and a 3D sparse coding model to represent video features. Given the difficulty in acquiring positive training videos, we trained a small-data classifier with a maximum of 15 positive and 32 negative examples. To counteract this limitation, we leveraged subject matter expert (SME) knowledge to limit the hypothesis space, thus reducing the cost of data collection. We present results using two lung ultrasound datasets and demonstrate that our model is capable of achieving performance on par with SMEs in pneumothorax identification. We then developed an iOS application that runs our full system in less than 4 seconds on an iPad Pro, and less than 8 seconds on an iPhone 13 Pro, labeling key regions in the lung sonogram to provide interpretable diagnoses.
BibTeX
@inproceedings{hannan-iaai-2023,
title = {MobilePTX: Sparse Coding for Pneumothorax Detection Given Limited Training Examples},
author = {Hannan, Darryl and Nesbit, Steven C. and Wen, Ximing and Smith, Glen and Zhang, Qiao and Goffi, Alberto and Chan, Vincent and Morris, Michael J. and Hunninghake, John C. and Villalobos, Nicholas E. and Kim, Edward and Weber, Rosina O. and MacLellan, Christopher J.},
booktitle = {Proceedings of the AAAI Conference on Artificial Intelligence},
volume = {37},
number = {13},
pages = {15675-15681},
year = {2023},
doi = {10.1609/aaai.v37i13.26859},
}