Feature Selection for Lung and Breast Cancer Disease Prediction Using Machine Learning Techniques

Abstract

Early detection of cancer is essential for a favorable prognosis because it is the biggest cause of death globally. After lung cancer, breast cancer ranks as the second most prevalent cause of death. With the fast expansion of the populace, the risk of mortality from lung and breast cancer is increasing rapidly. Early cancer prediction is challenging because there are few signs of this disease at an early stage. An automated sickness identification system provide accurate, efficient and quick response while assisting medical workers in identifying disorders and decreases death rates. In this research, we proposed PSO-FS (particle swarm optimization-based feature selection) method to select the features for several machine learning techniques to categorize accessible lung and breast cancer data. The best classifier approach for predicting both cancer diseases is considered to be the forest (RF) and deep learning (DL) classifier, which has high accuracy of 99.7% and 97%, respectively. Hence feature selection approach can increase performance by selecting only significant features.