← Search

Yijun Wan

3 accepted papers

2024

Implicit Compressibility of Overparametrized Neural Networks Trained with Heavy-Tailed SGD

ICML 2024poster

Neural network compression has been an increasingly important subject, not only due to its practical relevance, but also due to its theoretical implications, as there is an explicit connection between compressibility and generalization error. Recent studies have shown that the choice of the hyperpar…

2024

Lessons from Generalization Error Analysis of Federated Learning: You May Communicate Less Often!

ICML 2024poster

We investigate the generalization error of statistical learning models in a Federated Learning (FL) setting. Specifically, we study the evolution of the generalization error with the number of communication rounds $R$ between $K$ clients and a parameter server (PS), i.e. the effect on the generaliza…

2022

Chaotic Regularization and Heavy-Tailed Limits for Deterministic Gradient Descent

NeurIPS 2022accept

Recent studies have shown that gradient descent (GD) can achieve improved generalization when its dynamics exhibits a chaotic behavior. However, to obtain the desired effect, the step-size should be chosen sufficiently large, a task which is problem dependent and can be difficult in practice. In thi…