Association between work-related features and coronary artery disease: A heterogeneous hybrid feature selection integrated with balancing approach

• Published in: Pattern recognition letters 2020-05, Vol.133, p.33-40
• Main Authors: Nasarian, Elham, Abdar, Moloud, Fahami, Mohammad Amin, Alizadehsani, Roohallah, Hussain, Sadiq, Basiri, Mohammad Ehsan, Zomorodi-Moghadam, Mariam, Zhou, Xujuan, Pławiak, Paweł, Acharya, U. Rajendra, Tan, Ru-San, Sarrafzadegan, Nizal
• Format: Article
• Language: English
• Subjects: Adaptive sampling; Algorithms; Automation; Bayesian analysis; Beaches; Cardiovascular disease; Classifiers; Coronary artery; Coronary artery disease; Coronary vessels; Data mining; Datasets; Decision trees; Feature extraction; Feature selection; Heart disease; Heart diseases; Learning algorithms; Machine learning
• ISSN: 0167-8655; 1872-7344
• DOI: 10.1016/j.patrec.2020.02.010

•Automated classification of normal and CAD classes.•A new work-related coronary artery disease (CAD) data set.•Proposed a novel heterogeneous hybrid feature selection (2HFS) algorithm to pre-process the CAD data.•Work place, environmental and clinical features are used.•Obtained maximum performance using various data sets. Coronary artery disease (CAD) is a leading cause of death worldwide and is associated with high healthcare expenditure. Researchers are motivated to apply machine learning (ML) for quick and accurate detection of CAD. The performance of the automated systems depends on the quality of features used. Clinical CAD datasets contain different features with varying degrees of association with CAD. To extract such features, we developed a novel hybrid feature selection algorithm called heterogeneous hybrid feature selection (2HFS). In this work, we used Nasarian CAD dataset, in which work place and environmental features are also considered, in addition to other clinical features. Synthetic minority over-sampling technique (SMOTE) and Adaptive synthetic (ADASYN) are used to handle the imbalance in the dataset. Decision tree (DT), Gaussian Naive Bayes (GNB), Random Forest (RF), and XGBoost classifiers are used. 2HFS-selected features are then input into these classifier algorithms. Our results show that, the proposed feature selection method has yielded the classification accuracy of 81.23% with SMOTE and XGBoost classifier. We have also tested our approach with other well-known CAD datasets: Hungarian dataset, Long-beach-va dataset, and Z-Alizadeh Sani dataset. We have obtained 83.94%, 81.58% and 92.58% for Hungarian dataset, Long-beach-va dataset, and Z-Alizadeh Sani dataset, respectively. Hence, our experimental results confirm the effectiveness of our proposed feature selection algorithm as compared to the existing state-of-the-art techniques which yielded outstanding results for the development of automated CAD systems.