arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.05293cs.LGcs.AI
收藏

结直肠癌风险预测模型的特征选择比较研究

A comparative study on feature selection for a risk prediction model for colorectal cancer

Catalan Institute of Oncology-IDIBELL(加泰罗尼亚肿瘤研究所-IDIBELL);University of Barcelona(巴塞罗那大学);Public Health Division, Department of Health of Madrid(马德里卫生部公共卫生司)

N. Cueto-López, M. T. García-Ordás, V. Dávila-Batista, V. Moreno, N. Aragonés, R. Alaiz-Rodríguez

AI总结:

本研究比较多种特征选择算法对结直肠癌风险预测模型性能与稳定性的影响,发现SVM包装器与皮尔逊方法最优,并提出了可视化稳定性评估方法。

摘要:

背景与目标:风险预测模型旨在识别罹患目标疾病风险较高的人群。特征选择对于提高预测模型性能、避免过拟合以及识别主要的癌症风险(及保护)因素尤为重要。当目标是分析具有更强预测能力的特征时,评估特征选择/排序算法的稳定性便成为一个重要问题。方法:本研究聚焦于结直肠癌,评估了几种特征排序算法在一组风险预测模型(神经网络、支持向量机(SVM)、逻辑回归、k近邻和提升树)中的性能。此外,通过使用标量稳定性度量的常规方法以及本研究提出的可视化方法来评估其稳健性,以研究特征排序技术之间的相似性及其各自的稳定性。本研究还对比分析了研究中发现的最相关特征与专家根据最新知识提供的特征。结果:在受试者工作特征曲线下面积(AUC)方面,两个最佳性能结果分别是使用SVM包装器方法选择的前41个特征的SVM分类器(AUC=0.693)和使用皮尔逊方法选择的前40个特征的逻辑回归(AUC=0.689)。实验表明,与使用全部特征集的结果相比,进行特征选择对SVM和逻辑回归分类器的分类性能分别带来了3.9%和1.9%的AUC提升。本研究提出的可视化方法表明,基于神经网络的包装器排序最不稳定,而随机森林最稳定。

英文摘要:

Background and objective Risk prediction models aim at identifying people at higher risk of developing a target disease. Feature selection is particularly important to improve the prediction model performance avoiding overfitting and to identify the leading cancer risk (and protective) factors. Assessing the stability of feature selection/ranking algorithms becomes an important issue when the aim is to analyze the features with more prediction power. Methods This work is focused on colorectal cancer, assessing several feature ranking algorithms in terms of performance for a set of risk prediction models (Neural Networks, Support Vector Machines (SVM), Logistic Regression, k-Nearest Neighbors and Boosted Trees). Additionally, their robustness is evaluated following a conventional approach with scalar stability metrics and a visual approach proposed in this work to study both similarity among feature ranking techniques as well as their individual stability. A comparative analysis is carried out between the most relevant features found out in this study and features provided by the experts according to the state-of-the-art knowledge. Results The two best performance results in terms of Area Under the ROC Curve (AUC) are achieved with a SVM classifier using the top-41 features selected by the SVM wrapper approach (AUC=0.693) and Logistic Regression with the top-40 features selected by the Pearson (AUC=0.689). Experiments showed that performing feature selection contributes to classification performance with a 3.9% and 1.9% improvement in AUC for the SVM and Logistic Regression classifier, respectively, with respect to the results using the full feature set. The visual approach proposed in this work allows to see that the Neural Network-based wrapper ranking is the most unstable while the Random Forest is the most stable.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑