Private Federated Learning In Real World Application -- A Case Study

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Ji, An, Bandyopadhyay, Bortik, Song, Congzheng, Krishnaswami, Natarajan, Vashisht, Prabal, Smiroldo, Rigel, Litton, Isabel, Mahinder, Sayantan, Chitnis, Mona, Hill, Andrew W
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910819905699840
author Ji, An
Bandyopadhyay, Bortik
Song, Congzheng
Krishnaswami, Natarajan
Vashisht, Prabal
Smiroldo, Rigel
Litton, Isabel
Mahinder, Sayantan
Chitnis, Mona
Hill, Andrew W
author_facet Ji, An
Bandyopadhyay, Bortik
Song, Congzheng
Krishnaswami, Natarajan
Vashisht, Prabal
Smiroldo, Rigel
Litton, Isabel
Mahinder, Sayantan
Chitnis, Mona
Hill, Andrew W
contents This paper presents an implementation of machine learning model training using private federated learning (PFL) on edge devices. We introduce a novel framework that uses PFL to address the challenge of training a model using users' private data. The framework ensures that user data remain on individual devices, with only essential model updates transmitted to a central server for aggregation with privacy guarantees. We detail the architecture of our app selection model, which incorporates a neural network with attention mechanisms and ambiguity handling through uncertainty management. Experiments conducted through off-line simulations and on device training demonstrate the feasibility of our approach in real-world scenarios. Our results show the potential of PFL to improve the accuracy of an app selection model by adapting to changes in user behavior over time, while adhering to privacy standards. The insights gained from this study are important for industries looking to implement PFL, offering a robust strategy for training a predictive model directly on edge devices while ensuring user data privacy.
format Preprint
id arxiv_https___arxiv_org_abs_2502_04565
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Private Federated Learning In Real World Application -- A Case Study
Ji, An
Bandyopadhyay, Bortik
Song, Congzheng
Krishnaswami, Natarajan
Vashisht, Prabal
Smiroldo, Rigel
Litton, Isabel
Mahinder, Sayantan
Chitnis, Mona
Hill, Andrew W
Machine Learning
Cryptography and Security
This paper presents an implementation of machine learning model training using private federated learning (PFL) on edge devices. We introduce a novel framework that uses PFL to address the challenge of training a model using users' private data. The framework ensures that user data remain on individual devices, with only essential model updates transmitted to a central server for aggregation with privacy guarantees. We detail the architecture of our app selection model, which incorporates a neural network with attention mechanisms and ambiguity handling through uncertainty management. Experiments conducted through off-line simulations and on device training demonstrate the feasibility of our approach in real-world scenarios. Our results show the potential of PFL to improve the accuracy of an app selection model by adapting to changes in user behavior over time, while adhering to privacy standards. The insights gained from this study are important for industries looking to implement PFL, offering a robust strategy for training a predictive model directly on edge devices while ensuring user data privacy.
title Private Federated Learning In Real World Application -- A Case Study
topic Machine Learning
Cryptography and Security
url https://arxiv.org/abs/2502.04565