Abstract
In this paper, we delineate a comparative classification of Pneumonia using Machine Learning and Deep Learning models. The data used was the dataset of Chest X-Ray Images for Classification made available by Kermany (2018) with a total of 5863 images, with 2 classes: normal and pneumonia. For the purpose of correcting the class imbalance between normal and pneumonia images, we use General Adversarial Networks to generate pneumonia ridden images. The comparative classification is built on a final dataset of 19,784 images. Logistic regression, SVC, KNN, Random Forest and other machine learning models such as XgBoost and CatBoost are compared with deep learning models such as Mobile Net and VGG-16. The machine learning model is based on blob segmentation and detecting the difference between blobs of pneumonia ridden and normal individuals. Finally, a new deep learning model with convolutional and artificial neural networks has been proposed for the classification purpose which increases the accuracy significantly and has a classification report which is best suitable for medical analysis.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have