Efficient Lung Ultrasound Classification.
Antonio BrunoGiacomo IgnestiOvidio SalvettiDavide MoroniMassimo MartinelliPublished in: Bioengineering (Basel, Switzerland) (2023)
A machine learning method for classifying lung ultrasound is proposed here to provide a point of care tool for supporting a safe, fast, and accurate diagnosis that can also be useful during a pandemic such as SARS-CoV-2. Given the advantages (e.g., safety, speed, portability, cost-effectiveness) provided by the ultrasound technology over other examinations (e.g., X-ray, computer tomography, magnetic resonance imaging), our method was validated on the largest public lung ultrasound dataset. Focusing on both accuracy and efficiency, our solution is based on an efficient adaptive ensembling of two EfficientNet-b0 models reaching 100% of accuracy, which, to our knowledge, outperforms the previous state-of-the-art models by at least 5%. The complexity is restrained by adopting specific design choices: ensembling with an adaptive combination layer, ensembling performed on the deep features, and minimal ensemble using two weak models only. In this way, the number of parameters has the same order of magnitude of a single EfficientNet-b0 and the computational cost (FLOPs) is reduced at least by 20%, doubled by parallelization. Moreover, a visual analysis of the saliency maps on sample images of all the classes of the dataset reveals where an inaccurate weak model focuses its attention versus an accurate one.
Keyphrases
- magnetic resonance imaging
- sars cov
- machine learning
- deep learning
- high resolution
- healthcare
- computed tomography
- mental health
- coronavirus disease
- artificial intelligence
- convolutional neural network
- contrast enhanced ultrasound
- working memory
- magnetic resonance
- contrast enhanced
- diffusion weighted imaging
- emergency department
- electronic health record
- electron microscopy