Optimal feature selection for heart disease prediction using modified Artificial Bee colony (M-ABC) and K-nearest neighbors (KNN)
Heart disease is a complex and widespread illness that affects a significant number of people worldwide. Machine learning provides a way forward for early heart disease diagnosis. A classification model has been developed for the present study to predict heart disease. The attribute selection was done using a modified bee algorithm. Using the proposed model, practitioners can accurately predict heart disease and make informed decisions about patient health. In our study, we have proposed a framework based on Modified Artificial Bee Colony (M-ABC) and k-Nearest Neighbors (KNN) for predicting the optimal feature selection to obtain better accuracy. Using a modified bee algorithm, this paper focuses on identifying the optimal subset of attributes from the dataset. Specifically, during the classification-training phase, only the features that provide significant information are retained. The proposed study not only improves classification accuracy but also reduces training time for classifiers.
Authors
- Tehseen Mazhar (ORCID: https://orcid.org/0000-0002-4649-2376)
- Muhammad Mateen Yaqoob (ORCID: https://orcid.org/0000-0002-3030-7113)
- Umar Farooq Khattak
- Abdul Khader Jilani Saudagar (ORCID: https://orcid.org/0000-0003-4205-3621)
- Mohammad Shahid
- Muhammad Badruddin Khan (ORCID: https://orcid.org/0000-0002-4628-3240)
- Muhammad Amir Khan (ORCID: https://orcid.org/0000-0003-3669-2080)
- Yazeed Yasin Ghadi (ORCID: https://orcid.org/0000-0002-7121-495X)
Institutions
- Al Ain University (AE)
- Institute of Technology of Cambodia (KH)
- National University of Computer and Emerging Sciences (PK)
- Imam Mohammad ibn Saud Islamic University (SA)
- Government of Pakistan (PK)
- Government of Punjab (PK)
- Universiti Teknologi MARA (MY)
- Government College of Science (PK)
Publication Details
- Journal
- Scientific Reports
- Published
- 2024-10-31
- DOI
- https://doi.org/10.1038/s41598-024-78021-1
- Citations
- 19
- Primary Topic
- Artificial Intelligence in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 13.45