Streaming Federated Learning with Markovian Data

Tan-Khiem HUYNH, Malcolm Egan, Giovanni Neglia, Jean-Marie Gorce

Keywords: Optimization - Non-convex, Federated Learning, Markovian Processes, Stochastic Approximation

TL;DR: We show that collaboration is beneficial in Federated Learning with Markovian streaming data.

Abstract: Federated learning (FL) is now recognized as a key framework for communication-efficient collaborative learning. Most theoretical and empirical studies, however, rely on the assumption that clients have access to pre-collected data sets, with limited investigation into scenarios where clients continuously collect data. In many real-world applications, particularly when data is generated by physical or biological processes, client data streams are often modeled by non-stationary Markov processes. Unlike standard i.i.d. sampling, the performance of FL with Markovian data streams remains poorly understood due to the statistical dependencies between client samples over time. In this paper, we investigate whether FL can still support collaborative learning with Markovian data streams. Specifically, we analyze the performance of Minibatch SGD, Local SGD, and a variant of Local SGD with momentum. We answer affirmatively under standard assumptions and smooth non-convex client objectives: the sample complexity is proportional to the inverse of the number of clients with a communication complexity comparable to the i.i.d. scenario. However, the sample complexity for Markovian data streams remains higher than for i.i.d. sampling. Our analysis is validated via experiments with real pollution monitoring time series data.

Primary Area: Optimization (e.g., convex and non-convex, stochastic, robust)

Submission Number: 12686