Searching. Please wait…
1582
37
171
29406
4423
2606
347
392
Abstract: This study evaluates several feature ranking techniques together with some classifiers based on machine learning to identify relevant factors regarding the probability of contracting breast cancer and improve the performance of risk prediction models for breast cancer in a healthy population. The dataset with 919 cases and 946 controls comes from the MCC-Spain study and includes only environmental and genetic features. Breast cancer is a major public health problem. Our aim is to analyze which factors in the cancer risk prediction model are the most important for breast cancer prediction. Likewise, quantifying the stability of feature selection methods becomes essential before trying to gain insight into the data. This paper assesses several feature selection algorithms in terms of performance for a set of predictive models. Furthermore, their robustness is quantified to analyze both the similarity between the feature selection rankings and their own stability. The ranking provided by the SVM-RFE approach leads to the best performance in terms of the area under the ROC curve (AUC) metric. Top-47 ranked features obtained with this approach fed to the Logistic Regression classifier achieve an AUC = 0.616. This means an improvement of 5.8% in comparison with the full feature set. Furthermore, the SVM-RFE ranking technique turned out to be highly stable (as well as Random Forest), whereas relief and the wrapper approaches are quite unstable. This study demonstrates that the stability and performance of the model should be studied together as Random Forest and SVM-RFE turned out to be the most stable algorithms, but in terms of model performance SVM-RFE outperforms Random Forest.
Fuente: International Journal of Environmental Research and Public Health, 2021, 18(20), 10670
Publisher: MDPI
Publication date: 01/10/2021
No. of pages: 28
Publication type: Article
DOI: 10.3390/ijerph182010670
ISSN: 1661-7827,1660-4601
Publication Url: https://doi.org/10.3390/ijerph182010670
SCOPUS
Citations
Google Scholar
Metrics
UCrea Repository Read publication
CUETO LÓPEZ, NAHÚM
GARCÍA ORDÁS, MARÍA TERESA
VITELLI STORELLI, FACUNDO
FERNÁNDEZ NAVARRO, PABLO
CAMILO PALAZUELOS CALDERON
ALAIZ RODRÍGUEZ, ROCÍO
Back