Optimal Poisson Subsampling for Softmax Regression  

作　　者：YAO Yaqiong ZOU Jiahui WANG Haiying 

机构地区：[1]Department of Statistics,University of Connecticut,Storrs,CT,06269,USA [2]School of Statistics,Capital University of Economics and Business,Beijing 100070,China

出　　处：《Journal of Systems Science & Complexity》2023年第4期1609-1625,共17页系统科学与复杂性学报（英文版）

基　　金：Wang Haiying’s research was partially supported by the National Science Foundation under Grant No.CCF 2105571.

摘　　要：Softmax regression,which is also called multinomial logistic regression,is widely used in various fields for modeling the relationship between covariates and categorical responses with multiple levels.The increasing volumes of data bring new challenges for parameter estimation in softmax regression,and the optimal subsampling method is an effective way to solve them.However,optimal subsampling with replacement requires to access all the sampling probabilities simultaneously to draw a subsample,and the resultant subsample could contain duplicate observations.In this paper,the authors consider Poisson subsampling for its higher estimation accuracy and applicability in the scenario that the data exceed the memory limit.The authors derive the asymptotic properties of the general Poisson subsampling estimator and obtain optimal subsampling probabilities by minimizing the asymptotic variance-covariance matrix under both A-and L-optimality criteria.The optimal subsampling probabilities contain unknown quantities from the full dataset,so the authors suggest an approximately optimal Poisson subsampling algorithm which contains two sampling steps,with the first step as a pilot phase.The authors demonstrate the performance of our optimal Poisson subsampling algorithm through numerical simulations and real data examples.

关 键 词：Multinomial logistic regression optimality criterion optimal subsampling 

分 类 号：O212.1[理学—概率论与数理统计]