Personalized Federated Learning by Structured and Unstructured Pruning under Data Heterogeneity. - Saeed Vahidian; Mahdi Morafah; Bill Lin 001 (ICDCS Workshops, 2021)

Efficient Distribution Similarity Identification in Clustered Federated Learning via Principal Angles between Client Data Subspaces. - Saeed Vahidian; Mahdi Morafah; Weijia Wang 002; Vyacheslav Kungurtsev; Chen Chen 001; Mubarak Shah; Bill Lin 001 (AAAI, 2023)

FLIS: Clustered Federated Learning Via Inference Similarity for Non-IID Data Distribution. - Mahdi Morafah; Saeed Vahidian; Weijia Wang 002; Bill Lin 001 (IEEE Open J. Comput. Soc., 2023)

Decentralized Asynchronous Nonconvex Stochastic Optimization on Directed Graphs. - Vyacheslav Kungurtsev; Mahdi Morafah; Tara Javidi; Gesualdo Scutari (IEEE Trans. Control. Netw. Syst., 2023)

Large Scale Delocalized Federated Learning Over a Huge Diversity of Devices in Emerging Next-Generation Edge Intelligence Environments. - Mahdi Morafah; Hojin Chang; Bill Lin 001 (ICCAD, 2024)

Towards Diverse Device Heterogeneous Federated Learning via Task Arithmetic Knowledge Integration. - Mahdi Morafah; Vyacheslav Kungurtsev; Hojin Chang; Chen Chen 001; Bill Lin 001 (NeurIPS, 2024)

Stable Diffusion-based Data Augmentation for Federated Learning with Non-IID Data. - Mahdi Morafah; Matthias Reisser; Bill Lin 001; Christos Louizos (CoRR, 2024)

Rethinking Data Heterogeneity in Federated Learning: Introducing a New Notion and Standard Benchmarks. - Saeed Vahidian; Mahdi Morafah; Chen Chen 001; Mubarak Shah; Bill Lin 001 (IEEE Trans. Artif. Intell., 2024)

A Practical Recipe for Federated Learning Under Statistical Heterogeneity Experimental Design. - Mahdi Morafah; Weijia Wang 002; Bill Lin 001 (IEEE Trans. Artif. Intell., 2024)

Avoid Forgetting by Preserving Global Knowledge Gradients in Federated Learning with Non-IID Data. - Abhijit Chunduru; Majid Morafah; Mahdi Morafah; Vishnu Pandi Chellapandi; Ang Li (CoRR, 2025)

Federated Learning Client Pruning for Noisy Labels. - Mahdi Morafah; Hojin Chang; Chen Chen 001; Bill Lin 001 (ACM Trans. Model. Perform. Evaluation Comput. Syst., 2025)