462 citations · 2.5k across the 127 of their papers we have counts for
12 papers · 1 filter
Sparse PCA with Oracle Property
Quanquan Gu, Zhaoran Wang, Han Liu
In this paper, we study the estimation of the -dimensional sparse principal subspace of covariance matrix in the high-dimensional setting. We aim to recover the oracle princ…
The Implicit Bias of Batch Normalization in Linear Models and Two-layer Linear Convolutional Neural Networks
Yuan Cao, Difan Zou, Yuanzhi Li +1
We study the implicit bias of batch normalization trained by gradient descent. We show that when learning a linear model with batch normalization for binary classification, gradien…
Cooperative Multi-Agent Reinforcement Learning: Asynchronous Communication and Linear Function Approximation
Yifei Min, Jiafan He, Tianhao Wang +1
We study multi-agent reinforcement learning in the setting of episodic Markov decision processes, where multiple agents cooperate via communication through a central server. We pro…
Robust Learning with Progressive Data Expansion Against Spurious Correlation
Yihe Deng, Yu Yang, Baharan Mirzasoleiman +1
While deep learning models have shown remarkable performance in various tasks, they are susceptible to learning non-generalizable spurious features rather than the core features th…
Uniform-PAC Guarantees for Model-Based RL with Bounded Eluder Dimension
Yue Wu, Jiafan He, Quanquan Gu
Recently, there has been remarkable progress in reinforcement learning (RL) with general function approximation. However, all these works only provide regret or sample complexity g…
Personalized Federated Learning under Mixture of Distributions
Yue Wu, Shuaicheng Zhang, Wenchao Yu +5
The recent trend towards Personalized Federated Learning (PFL) has garnered significant attention as it allows for the training of models that are tailored to each client while mai…