Research

Preprints

• CAP: A General Algorithm for Online Selective Conformal Prediction with FCR Control
  Yajie Bao, Yuyang Huo, Haojie Ren and Changliang Zou.
  arxiv, 2024+.

Journal publication

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

Conference publication

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

Writtings

• Step into High-dimensional Statistics: Sparse Mean Estimation
  

• Linear Regression and M-estimator for Diverging p