High-accuracy lung disease classification via logistic regression and advanced feature extraction techniques

Abstract

Lung disease diagnosis through medical imaging integrated with machine learning has seen significant advancements. This study investigates the optimization of lung disease classification by exploring various preprocessing, feature extraction, and machine-learning classifier combinations. Our methodology begins with an input dataset of lung X-ray images, which undergoes preprocessing steps such as image sharpening and histogram equalization to enhance image quality. Subsequently, feature extraction techniques, including Scale-Invariant Feature Transform (SIFT), Histogram of Oriented Gradients (HOG), and Local Binary Patterns (LBP), are applied to the pre-processed images. We evaluate the effectiveness of several machine learning classifiers—Naive Bayes (NB), K Nearest Neighbors (KNN), Decision Tree (DT), Logistic Regression (LR), and Support Vector Machine (SVM)—on both original and pre-processed images to determine the optimal classifier. Following the selection of the best-performing classifier, i.e., Logistic Regression, we further optimize the classification process by applying combinations of the feature extraction techniques (SIFT + HOG, SIFT + LBP, HOG + LBP, SIFT + HOG + LBP). The SIFT + HOG + LBP feature extraction, in combination with Logistic Regression, performed the best on the original images, obtaining an accuracy of 97.12 %, precision of 97.97 %, recall of 97.55 %, and F1-score of 97.76 %. Our study presents the comparative performance of each preprocessing and feature extraction technique, both individually and in combination, in bringing about an improvement in the accuracy of lung disease detection. The study concludes with the identification of the most effective preprocessing and feature extraction combination, coupled with the best machine learning classifier, providing a robust framework for enhanced lung disease diagnosis.