On the Feature Selection Methods and Reject Option Classifiers for Robust Cancer Prediction

Muhammad Hammad Waseem, Malik Sajjad Ahmed Nadeem, Assad Abbas*, Aliya Shaheen, Wajid Aziz, Adeel Anjum, Umar Manzoor, Muhammad A. Balubaid, Seong O. Shim

*Corresponding author for this work

    Research output: Contribution to journalArticlepeer-review

    Abstract

    Cancer is the second leading cause of mortality across the globe. Approximately 9.6 million people are estimated to have died due to cancer disease in 2019. Accurate and early prediction of cancer can assist healthcare professionals to devise timely therapeutic innervations to control sufferings and the risk of mortality. Generally, a machine learning (ML) based predictive system in healthcare uses data (genetic profile or clinical parameters) and learning algorithms to predict target values for cancer detection. However, optimization of predictive accuracy is an important endeavor for accurate decision making. Reject Option (RO) classifiers have been used to improve the predictive accuracy of classifiers for cancer like complex problems. In a gene profile all of the features are not important and should be shaved off. ML offers different techniques with their own methodology for feature selection (FS) and the classification results are dependent on the datasets each having its own distribution and features. Therefore, both FS methods and ML algorithms with RO need to be considered for robust classification. The main objective of this study is to optimize three parameters (learning algorithm, FS method and rejection rate) for robust cancer prediction rather than considering two traditional parameters (learning algorithm and rejection rate). The analysis of different FS methods (including t-Test, Las Vegas Filter (LVF), Relief, and Information Gain (IG)) and RO classifiers on different rejection thresholds is performed to investigate the robust predictability of cancer. The three cancer datasets (Colon cancer, Leukemia and Breast cancer) were reduced using different FS methods and each of them were used to analyze the predictability of cancer using different RO classifiers. The results reveal that for each dataset predictive accuracies of RO classifiers were different for different FS methods. The findings based on proposed scheme indicate that, the ML algorithms along with their dependence on suitable FS methods need to be taken into consideration for accurate prediction.

    Original languageEnglish
    Pages (from-to)141072-141082
    Number of pages11
    JournalIEEE Access
    Volume7
    Early online date27 Sept 2019
    DOIs
    Publication statusPublished - 9 Oct 2019

    Bibliographical note

    © 2019. This work is licensed under a Creative Commons Attribution 4.0 License. For more information, see http://creativecommons.org/licenses/by/4.0/ .

    Keywords

    • Cancer
    • classification
    • feature selection
    • genetic profile
    • machine learning
    • reject option

    Fingerprint

    Dive into the research topics of 'On the Feature Selection Methods and Reject Option Classifiers for Robust Cancer Prediction'. Together they form a unique fingerprint.

    Cite this