Button Text
Back

P31 - Improving Chest X-ray Image Classification via Parallelized Generative Neural Architecture Search

This is some text inside of a div block.
This is some text inside of a div block.
-
This is some text inside of a div block.
CEST
Climate, Weather and Earth Sciences
Chemistry and Materials
Computer Science, Machine Learning, and Applied Mathematics
Applied Social Sciences and Humanities
Engineering
Life Sciences
Physics
This is some text inside of a div block.

Description

Explore GenNAS for chest X-ray classification in lung diseases, leveraging novel parallel training methods for enhanced accuracy and efficiency. Medical image classification for pulmonary pathologies from chest X-rays is traditionally time-consuming. GenNAS, using GPT-4's generative capabilities, automates optimal architecture learning from data. This study investigates parallelization and generative algorithms to optimize neural network architectures for chest X-ray classification, analyzing their impact on the NAS algorithm using the ChexPert dataset.The study uses the CheXpert dataset with 224,316 chest X-rays, focusing on classifying five lung disease pathologies. GenNASXRays evaluates 6561 architecture possibilities in an 8-layer search space, with AUC-ROC and Precision-Recall plots as metrics. Training on 187,641 images, the sequential algorithm took 190.2 hours for an accuracy of 0.869. In parallel execution on two GPUs, an accuracy of 0.87 was achieved in 127.09 hours, highlighting the efficiency of parallelization. The experiments were executed with well-known neural network architectures for image classification such as DenseNet-121 obtaining an accuracy of 0.8678, ResNet-152 0.875 and EfficientNet-B0 0.7494 being very close to the architectures generated by GenNAS.. GenNAS demonstrates precision in defining deep learning models. Parallelization significantly accelerates Neural Architecture Search, potentially improving patient outcomes through timely and accurate diagnoses.

Presenter(s)

Authors