Optimizing Convolutional Neural Networks for Lung Disease Classification Using Bayesian Optimization
摘要
Pulmonary diseases constitute a great threat to global health and therefore, their early detection remains very important for improving survival rates. This paper demonstrates the effectiveness of convolutional neural networks in classifying certain lung diseases from chest X-ray images. We introduced Bayesian optimization in hyperparameter tuning to maximize the performance of the model. The experimental findings reveal that, generally, the performance would experience significant improvement after optimization, with MobileNet-BO achieving an accuracy of 93.53%. These results further confirm the important role played by hyperparameter optimization in improving the performance of CNNs, particularly in critical medical applications adhering to accuracy.