TitleOptimal SVM Parameter Selection for Non-separable and Unbalanced Datasets
Publication TypeJournal Article
Year of Publication2014
AuthorsJiang, P, Missoum, S, Chen, Z
JournalStructural and Multidisciplinary Optimization
Volume50
Issue4
Pagination523-535
Date Published10/2014
KeywordsCross validation, Non-separable and unbalanced datasets, Support vector machines, Validation metrics
Abstract

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 nonseparable 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.

URLhttp://codes.arizona.edu/sites/default/files/pdf/Jiang2014.pdf
DOI10.1007/s00158-014-1105-z
Citation Keyjiang2014optimal

Theme by Danetsoft and Danang Probo Sayekti inspired by Maksimer