38 citations · 124 across the 14 of their papers we have counts for
8 papers · 1 filter
Understanding Hessian Alignment for Domain Generalization
Sobhan Hemati, Guojun Zhang, Amir Estiri +1
Out-of-distribution (OOD) generalization is a critical ability for deep learning models in many real-world scenarios including healthcare and autonomous vehicles. Recently, differe…
Understanding the Role of Layer Normalization in Label-Skewed Federated Learning
Guojun Zhang, Mahdi Beitollahi, Alex Bie +1
Layer normalization (LN) is a widely adopted deep learning technique especially in the era of foundation models. Recently, LN has been shown to be surprisingly effective in federat…
DP-VAE: Differentially Private Pre-trained Variational Autoencoders
Dihong Jiang, Guojun Zhang, Mahdi Karami +3
Modern machine learning systems achieve great success when trained on large datasets. However, these datasets usually contain sensitive information (e.g. medical records, face imag…
Mitigating Data Heterogeneity in Federated Learning with Data Augmentation
Artur Back de Luca, Guojun Zhang, Xi Chen +1
Federated Learning (FL) is a prominent framework that enables training a centralized model while securing user privacy by fusing local, decentralized models. In this setting, one m…
Robust One Round Federated Learning with Predictive Space Bayesian Inference
Mohsin Hasan, Zehao Zhang, Kaiyang Guo +4
Making predictions robust is an important challenge. A separate challenge in federated learning (FL) is to reduce the number of communication rounds, particularly since doing so re…
Federated Bayesian Neural Regression: A Scalable Global Federated Gaussian Process
Haolin Yu, Kaiyang Guo, Mahdi Karami +3
In typical scenarios where the Federated Learning (FL) framework applies, it is common for clients to have insufficient training data to produce an accurate model. Thus, models tha…
Proportional Fairness in Federated Learning
Guojun Zhang, Saber Malekmohammadi, Xi Chen +1
With the increasingly broad deployment of federated learning (FL) systems in the real world, it is critical but challenging to ensure fairness in FL, i.e. reasonably satisfactory p…
No Weighted-Regret Learning in Adversarial Bandits with Delays
Ilai Bistritz, Zhengyuan Zhou, Xi Chen +2
Consider a scenario where a player chooses an action in each round out of rounds and observes the incurred cost after a delay of rounds. The cost functions and the…