Stable variable ranking and selection in regularized logistic regression for severely imbalanced big binary data

被引:2
作者
Nadeem, Khurram [1 ]
Jabri, Mehdi-Abderrahman [1 ]
机构
[1] Univ Guelph, Guelph, ON, Canada
关键词
MODELS;
D O I
10.1371/journal.pone.0280258
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
We develop a novel covariate ranking and selection algorithm for regularized ordinary logistic regression (OLR) models in the presence of severe class-imbalance in high dimensional datasets with correlated signal and noise covariates. Class-imbalance is resolved using response-based subsampling which we also employ to achieve stability in variable selection by creating an ensemble of regularized OLR models fitted to subsampled (and balanced) datasets. The regularization methods considered in our study include Lasso, adaptive Lasso (adaLasso) and ridge regression. Our methodology is versatile in the sense that it works effectively for regularization techniques involving both hard- (e.g. Lasso) and soft-shrinkage (e.g. ridge) of the regression coefficients. We assess selection performance by conducting a detailed simulation experiment involving varying moderate-to-severe class-imbalance ratios and highly correlated continuous and discrete signal and noise covariates. Simulation results show that our algorithm is robust against severe class-imbalance under the presence of highly correlated covariates, and consistently achieves stable and accurate variable selection with very low false discovery rate. We illustrate our methodology using a case study involving a severely imbalanced high-dimensional wildland fire occurrence dataset comprising 13 million instances. The case study and simulation results demonstrate that our framework provides a robust approach to variable selection in severely imbalanced big binary data.
引用
收藏
页数:26
相关论文
共 51 条
[1]  
Agresti Alan., 2013, CATEGORICAL DATA ANA, V3rd
[2]   Effective Feature Selection Method for Class-Imbalance Datasets Applied to Chemical Toxicity Prediction [J].
Antelo-Collado, Aurelio ;
Carrasco-Velar, Ramon ;
Garcia-Pedrajas, Nicolas ;
Cerruela-Garcia, Gonzalo .
JOURNAL OF CHEMICAL INFORMATION AND MODELING, 2021, 61 (01) :76-94
[3]   Response-Based Sampling for Binary Choice Models With Sample Selection [J].
Arezzo, Maria Felice ;
Guagnano, Giuseppina .
ECONOMETRICS, 2018, 6 (01)
[4]  
Bach F.R., 2008, Proceedings of the 25th International Conference on Machine learning, P33
[5]  
Bavaghar M. Pir, 2015, Journal of Forest Science (Prague), V61, P193, DOI [10.17221/78/2014-jfs, 10.17221/78/2014-JFS]
[6]  
Breiman L., 2017, Trees, P216, DOI [DOI 10.1201/9781315139470-8, DOI 10.1201/9781315139470, 10.1201/9781315139470-8]
[7]  
Breslow N E, 1987, IARC Sci Publ, P1
[8]  
Bühlmann P, 2011, SPRINGER SER STAT, P1, DOI 10.1007/978-3-642-20192-9
[9]  
Candes E, 2007, ANN STAT, V35, P2313, DOI 10.1214/009053606000001523
[10]  
Carini M. T., 1997, International Astronomical Union Circular