Federated Learning on Stochastic Neural Networks

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tang, Jingqiao, Bausback, Ryan, Bao, Feng, Archibald, Richard
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915335172521984
author Tang, Jingqiao
Bausback, Ryan
Bao, Feng
Archibald, Richard
author_facet Tang, Jingqiao
Bausback, Ryan
Bao, Feng
Archibald, Richard
contents Federated learning is a machine learning paradigm that leverages edge computing on client devices to optimize models while maintaining user privacy by ensuring that local data remains on the device. However, since all data is collected by clients, federated learning is susceptible to latent noise in local datasets. Factors such as limited measurement capabilities or human errors may introduce inaccuracies in client data. To address this challenge, we propose the use of a stochastic neural network as the local model within the federated learning framework. Stochastic neural networks not only facilitate the estimation of the true underlying states of the data but also enable the quantification of latent noise. We refer to our federated learning approach, which incorporates stochastic neural networks as local models, as Federated stochastic neural networks. We will present numerical experiments demonstrating the performance and effectiveness of our method, particularly in handling non-independent and identically distributed data.
format Preprint
id arxiv_https___arxiv_org_abs_2506_08169
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Federated Learning on Stochastic Neural Networks
Tang, Jingqiao
Bausback, Ryan
Bao, Feng
Archibald, Richard
Machine Learning
Distributed, Parallel, and Cluster Computing
Federated learning is a machine learning paradigm that leverages edge computing on client devices to optimize models while maintaining user privacy by ensuring that local data remains on the device. However, since all data is collected by clients, federated learning is susceptible to latent noise in local datasets. Factors such as limited measurement capabilities or human errors may introduce inaccuracies in client data. To address this challenge, we propose the use of a stochastic neural network as the local model within the federated learning framework. Stochastic neural networks not only facilitate the estimation of the true underlying states of the data but also enable the quantification of latent noise. We refer to our federated learning approach, which incorporates stochastic neural networks as local models, as Federated stochastic neural networks. We will present numerical experiments demonstrating the performance and effectiveness of our method, particularly in handling non-independent and identically distributed data.
title Federated Learning on Stochastic Neural Networks
topic Machine Learning
Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2506.08169