CAP: A General Algorithm for Online Selective Conformal Prediction with FCR Control Yajie Bao, Yuyang Huo, Haojie Ren and Changliang Zou. arxiv, 2024+.
Selective Conformal Inference with False Coverage-statement Rate Control Yajie Bao, Yuyang Huo, Haojie Ren and Changliang Zou. Biometrika, 2024.
Semi-Profiled Distributed Estimation for High-Dimensional Partially Linear Model Yajie Bao and Haojie Ren. Computational Statistics & Data Analysis, 2023.
Varying Coefficient Linear Discriminant Analysis for Dynamic Data Yajie Bao and Yuyang Liu. Electronic Journal of Statistics, 2022.
Provable Benefits of Local Steps in Heterogeneous Federated Learning for Neural Networks: A Feature Learning Perspective Yajie Bao, Michael Crawshaw and Mingrui Liu. International Conference on Machine Learning (ICML), 2024.
Global Convergence Analysis of Local SGD for Two-layer Neural Network without Overparameterization Yajie Bao, Amarda Shehu, Mingrui Liu. Advances in Neural Information Processing Systems (NeurIPS), 2023.
Federated Learning with Client Subsampling, Data Heterogeneity, and Unbounded Smoothness: A New Algorithm and Lower Bounds Michael Crawshaw, Yajie Bao and Mingrui Liu. Advances in Neural Information Processing Systems (NeurIPS), 2023.
EPISODE: Episodic Gradient Clipping with Periodic Resampled Corrections for Federated Learning with Heterogeneous Data Michael Crawshaw, Yajie Bao and Mingrui Liu. International Conference on Learning Representations (ICLR), 2023.
Fast Composite Optimization and Statistical Recovery in Federated Learning Yajie Bao, Michael Crawshaw, Shan Luo and Mingrui Liu. International Conference on Machine Learning (ICML), 2022.
Byzantine-Tolerant Distributed Multiclass Sparse Linear Discriminant Analysis Yajie Bao, Weidong Liu, Xiaojun Mao and Weijia Xiong. Conference on Uncertainty in Artificial Intelligence (UAI), 2022.
One-Round Communication Efficient Distributed M-Estimation Yajie Bao and Weijia Xiong. International Conference on Artificial Intelligence and Statistics (AISTATS), 2021.
Step into High-dimensional Statistics: Sparse Mean Estimation
Linear Regression and M-estimator for Diverging p