期刊
ARTIFICIAL INTELLIGENCE IN MEDICINE
卷 63, 期 1, 页码 19-31出版社
ELSEVIER SCIENCE BV
DOI: 10.1016/j.artmed.2014.12.004
关键词
Feature selection methods; Mann-Whitney U-test; uFilter method; Machine learning algorithms; Redundancy analysis; Breast cancer CADx
资金
- Fundacao para a Ciencia e a Tecnologia [SFRH/BD/48896/2008]
- QREN, Mais Centro program [CENTRO-07-ST24-FEDER-002031]
- Fundação para a Ciência e a Tecnologia [SFRH/BD/48896/2008] Funding Source: FCT
Objective: This work addresses the theoretical description and experimental evaluation of a new feature selection method (named uFilter). The uFilter improves the Mann-Whitney U-test for reducing dimensionality and ranking features in binary classification problems. Also, it presented a practical uFilter application on breast cancer computer-aided diagnosis (CADx). Materials and methods: A total of 720 datasets (ranked subsets of features) were formed by the application of the chi-square (CHI2) discretization, information-gain (IG), one-rule (1Rule), Relief, uFilter and its theoretical basis method (named U-test). Each produced dataset was used for training feed-forward backpropagation neural network, support vector machine, linear discriminant analysis and naive Bayes machine learning algorithms to produce classification scores for further statistical comparisons. Results: A head-to-head comparison based on the mean of area under receiver operating characteristics curve scores against the U-test method showed that the uFilter method significantly outperformed the U-test method for almost all classification schemes (p < 0.05); it was superior in 50%; tied in a 37.5% and lost in a 12.5% of the 24 comparative scenarios. Also, the performance of the uFilter method, when compared with CHI2 discretization, IC, 1Rule and Relief methods, was superior or at least statistically similar on the explored datasets while requiring less number of features. Conclusions: The experimental results indicated that uFilter method statistically outperformed the U-test method and it demonstrated similar, but not superior, performance than traditional feature selection methods (CHI2 discretization, IG, 1Rule and Relief). The uFilter method revealed competitive and appealing cost-effectiveness results on selecting relevant features, as a support tool for breast cancer CADx methods especially in unbalanced datasets contexts. Finally, the redundancy analysis as a complementary step to the uFilter method provided us an effective way for finding optimal subsets of features without decreasing the classification performances. (C) 2014 Elsevier B.V. All rights reserved.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据