RT info:eu-repo/semantics/article
T1 Evaluation of Feature Selection Techniques for Breast Cancer Risk Prediction
A1 Cueto López, Nahum
A1 García Ordás, María Teresa
A1 Vitelli Storelli, Facundo Ezequiel
A1 Fernández Navarro, Pablo
A1 Palazuelos Calderón, Camilo
A1 Alaiz Rodríguez, Rocío
A2 Ingenieria de Sistemas y Automatica
K1 Ingeniería de sistemas
K1 Medicina. Salud
K1 Breast cancer
K1 Risk prediction model
K1 Feature selection
K1 Stability
K1 3313 Tecnología E Ingeniería Mecánicas
K1 3210 Medicina Preventiva
AB [EN] 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.
PB MDPI
LK https://hdl.handle.net/10612/21276
UL https://hdl.handle.net/10612/21276
NO López, N. C., García-Ordás, M. T., Vitelli-Storelli, F., Fernández-Navarro, P., Palazuelos, C., & Alaiz-Rodríguez, R. (2021). Evaluation of feature selection techniques for breast cancer risk prediction. International Journal of Environmental Research and Public Health, 18(20). https://doi.org/10.3390/IJERPH182010670
DS BULERIA. Repositorio Institucional de la Universidad de León
RD Jul 12, 2024