← Search

Hossein Zakerinia

5 accepted papers

2026

Federated Learning with Unlabeled Clients: Personalization Can Happen in Low Dimensions

ICML 2026poster

Personalized federated learning has emerged as a popular approach to training on devices holding statistically heterogeneous data, known as clients. However, most existing approaches require a client to have labeled data for training or finetuning in order to obtain their own personalized model. In …

Cited by 0SourceScholar
2025

Fast Rate Bounds for Multi-Task and Meta-Learning with Different Sample Sizes

NeurIPS 2025poster

We present new fast-rate PAC-Bayesian generalization bounds for multi-task and meta-learning in the unbalanced setting, i.e. when the tasks have training sets of different sizes, as is typically the case in real-world scenarios. Previously, only standard-rate bounds were known for this situation, wh…

Cited by 0SourceScholar
2024

Communication-Efficient Federated Learning With Data and Client Heterogeneity

AISTATS 2024poster

Federated Learning (FL) enables large-scale distributed training of machine learning models, while still allowing individual nodes to maintain data locally. However, executing FL at scale comes with inherent practical challenges: 1) heterogeneity of the local node data distributions, 2) heterogeneit…

2024

More Flexible PAC-Bayesian Meta-Learning by Learning Learning Algorithms

ICML 2024poster

We introduce a new framework for studying meta-learning methods using PAC-Bayesian theory. Its main advantage over previous work is that it allows for more flexibility in how the transfer of knowledge between tasks is realized. For previous approaches, this could only happen indirectly, by means of…

2024

PeFLL: Personalized Federated Learning by Learning to Learn

ICLR 2024poster

We present PeFLL, a new personalized federated learning algorithm that improves over the state-of-the-art in three aspects: 1) it produces more accurate models, especially in the low-data regime, and not only for clients present during its training phase, but also for any that may emerge in the futu…