Optimal SVM parameter selection for non-separable and unbalanced datasets.
Optimal SVM parameter selection for non-separable and unbalanced datasets.
复制标题
针对不可分离和不平衡数据集的最佳 SVM 参数选择。
DOI:
10.1007/s00158-014-1105-z
复制
发表时间:
2014
期刊:
影响因子:
--
通讯作者:
Chen,Zhao
中科院分区:
文献类型:
--
作者:
Jiang,Peng;Missoum,Samy;Chen,Zhao
This article presents a study of three validation metrics used for the selection of optimal parameters of a support vector machine (SVM) classifier in the case of non-separable and unbalanced datasets. This situation is often encountered when the data is obtained experimentally or clinically. The three metrics selected in this work are the area under the ROC curve (AUC), accuracy, and balanced accuracy. These validation metrics are tested using computational data only, which enables the creation of fully separable sets of data. This way, non-separable datasets, representative of a real-world problem, can be created by projection onto a lower dimensional sub-space. The knowledge of the separable dataset, unknown in real-world problems, provides a reference to compare the three validation metrics using a quantity referred to as the “weighted likelihood”. As an application example, the study investigates a classification model for hip fracture prediction. The data is obtained from a parameterized finite element model of a femur. The performance of the various validation metrics is studied for several levels of separability, ratios of unbalance, and training set sizes.