Pneumonia Detection in Chest X-Ray Images Using Enhanced Restricted Boltzmann Machine

Abstract

The process of pneumonia detection has been the focus of researchers as it has proved itself to be one of the most dangerous and life-threatening disorders. In recent years, many machine learning and deep learning algorithms have been applied in an attempt to automate this process but none of them has been successful significantly to achieve the highest possible accuracy. In a similar attempt, we propose an enhanced approach of a deep learning model called restricted Boltzmann machine (RBM) which is named enhanced RBM (ERBM). One of the major drawbacks associated with the standard format of RBM is its random weight initialization which leads to improper feature learning of the model during the training phase, resulting in poor performance of the machine. This problem has been tried to eliminate in this work by finding the differences between the means of a specific feature vector and the means of all features given as inputs to the machine. By performing this process, the reconstruction of the actual features is increased which ultimately reduces the error generated during the training phase of the model. The developed model has been applied to three different datasets of pneumonia diseases and the results have been compared with other state of the art techniques using different performance evaluation parameters. The proposed model gave highest accuracy of 98.56% followed by standard RBM, SVM, KNN, and decision tree which gave accuracies of 97.53%, 92.62%, 91.64%, and 88.77%, respectively, for dataset named dataset 2. Similarly, for the dataset 1, the highest accuracy of 96.66 has been observed for the eRBM followed by srRBM, KNN, decision tree, and SVM which gave accuracies of 90.22%, 89.34%, 87.65%, and 86.55%, respectively. In the same way, the accuracies observed for the dataset 3 by eRBM, standard RBM, KNN, decision tree, and SVM are 92.45%, 90.98%, 87.54%, 85.49%, and 84.54%, respectively. Similar observations can also be seen for other performance parameters showing the efficiency of the proposed model. As revealed in the results obtained, a significant improvement has been observed in the working of the RBM by introducing a new method of weight initialization during the training phase. The results show that the improved model outperforms other models in terms of different performance evaluation parameters, namely, accuracy, sensitivity, specificity, F1-score, and ROC curve.