Peer-Reviewed Federated Learning
by Mattia Passeri, Andrea Agiollo, and Andrea Omicini
While representing the de-facto framework for enabling distributed training of Machine Learning models, Federated Learning (FL) still suffers convergence issues when non-Independent and Identically Distributed (non-IID) data are considered. In this context, the local model optimisation on different data distributions generate dissimilar updates, which are difficult to aggregate and translate into sub-optimal convergence. To tackle this issues, we propose Peer-Reviewed Federated Learning (PRFL), an extension of the traditional FL training process inspired by the peer-review procedure common in the academic field, where model updates are reviewed by several other clients in the federation before being aggregated at the server-side. PRFL aims at enabling the identification of relevant updates, while disregarding the ineffective ones. We implement PRFL on top of the Flower FL library, and make Peer-Reviewed Flower a publicly-available library for the modular implementation of any review-based FL algorithm. A preliminary case study on both regression and classification tasks highlights the potential of PRFL, showcasing how the distributed solution can achieve performance similar to that obtained by the corresponding centralised algorithm, even when non-IID data are considered.
@inproceedings{prfl-woa2023,
author = {Passeri, Mattia and Agiollo, Andrea and Omicini, Andrea},
booktitle = {WOA 2023 -- 24th Workshop ``From Objects to Agents''},
dblp = {conf/woa/PasseriAO23},
editor = {Falcone, Rino and Castelfranchi, Cristiano and Sapienza, Alessandro and Cantucci, Filippo},
iris = {11585/950373},
issn = {1613-0073},
keywords = {Federated Learning; non-IID data; Peer-review},
month = nov,
numpages = 17,
pages = {49--65},
publisher = {Sun SITE Central Europe, RWTH Aachen University},
scholar = {7461839519241304430},
scopus = {2-s2.0-85179627285},
series = {CEUR Workshop Proceedings},
subseries = {AIxIA Series},
title = {Peer-Reviewed Federated Learning},
url = {https://ceur-ws.org/Vol-3579/paper4.pdf},
urlopenaccess = {https://ceur-ws.org/Vol-3579/paper4.pdf},
urlpdf = {https://ceur-ws.org/Vol-3579/paper4.pdf},
volume = 3579,
year = 2023
}