Back to Search Start Over

Stable variable selection of class-imbalanced data with precision-recall criterion.

Authors :
Fu, Guang-Hui
Xu, Feng
Zhang, Bing-Yang
Yi, Lun-Zhao
Source :
Chemometrics & Intelligent Laboratory Systems. Dec2017, Vol. 171, p241-250. 10p.
Publication Year :
2017

Abstract

Screening important variables for class-imbalanced data is still a challenging task. In this study, we propose an algorithm for stably selecting key variables on class-imbalanced data based on the precision-recall curve (PRC), where the PRC is utilized as the assessment criterion in the model building stage, and sparse regularized logistic regression combined with subsampling (SRLRS) is designed to perform stable variable selection. Considering the characteristic of class-imbalanced data, we also proposed classification-based partition for cross validation, as well as leaving half of majority observations out and leaving one minority observation out (LHO-LOO) for subsampling. Simulation results and real data showed that our algorithm is highly suitable for handling class-imbalanced data, and that the PRC can be an alternative evaluation criterion for model selection when handling class-imbalanced data. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
01697439
Volume :
171
Database :
Academic Search Index
Journal :
Chemometrics & Intelligent Laboratory Systems
Publication Type :
Academic Journal
Accession number :
126415705
Full Text :
https://doi.org/10.1016/j.chemolab.2017.10.015