A Framework for Multi-Prototype Based Federated Learning: Towards the Edge Intelligence | IEEE Conference Publication | IEEE Xplore

A Framework for Multi-Prototype Based Federated Learning: Towards the Edge Intelligence


Abstract:

Edge intelligence becomes the enabler to fulfill the privacy-preserving intelligent services and applications for next-generation networking. However, the heterogeneous d...Show More

Abstract:

Edge intelligence becomes the enabler to fulfill the privacy-preserving intelligent services and applications for next-generation networking. However, the heterogeneous data distribution of distributed edge clients often hinders the convergence rate and test accuracy. Federated Learning (FL), as a new paradigm for privacy-preserving distributed edge-artificial intelligence (edge-AI) that enables model training without the raw data of clients leaving their local sides. The differences in the data distribution of clients can easily lead to biased model inference results, especially when inferring through classifiers. In this paper, to enhance robustness against heterogeneity, a novel multiple-prototype based federated learning (MPFed) framework is proposed, in which clients communicate with server as typical federated training, but the model inference is performed by measuring the distance between the target prototype and multiple weighted prototypes. The weighted prototype of each class is calculated by executing the clustering algorithm (e.g., k-means) and weighted strategy at the client side before finishing the last federated iteration. The server aggregates these weighted prototypes collected from all clients, and then distributes to them for model inferences. Experimental analyses on multiple baseline datasets, such as MNIST, Fashion-MNIST, and CIFAR10 demonstrate our method has a higher test accuracy, at least 10%, and is relatively efficient in communication than baselines and state-of-the-art algorithms.
Date of Conference: 11-14 January 2023
Date Added to IEEE Xplore: 22 February 2023
ISBN Information:
Print on Demand(PoD) ISSN: 1976-7684
Conference Location: Bangkok, Thailand

Contact IEEE to Subscribe

References

References is not available for this document.