Inference Load-Aware Orchestration for Hierarchical Federated Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Lackinger, Anna, Frangoudis, Pantelis A., Čilić, Ivan, Furutanpey, Alireza, Murturi, Ilir, Žarko, Ivana Podnar, Dustdar, Schahram
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910539215536128
author Lackinger, Anna
Frangoudis, Pantelis A.
Čilić, Ivan
Furutanpey, Alireza
Murturi, Ilir
Žarko, Ivana Podnar
Dustdar, Schahram
author_facet Lackinger, Anna
Frangoudis, Pantelis A.
Čilić, Ivan
Furutanpey, Alireza
Murturi, Ilir
Žarko, Ivana Podnar
Dustdar, Schahram
contents Hierarchical federated learning (HFL) designs introduce intermediate aggregator nodes between clients and the global federated learning server in order to reduce communication costs and distribute server load. One side effect is that machine learning model replication at scale comes "for free" as part of the HFL process: model replicas are hosted at the client end, intermediate nodes, and the global server level and are readily available for serving inference requests. This creates opportunities for efficient model serving but simultaneously couples the training and serving processes and calls for their joint orchestration. This is particularly important for continual learning, where serving a model while (re)training it periodically, upon specific triggers, or continuously, takes place over shared infrastructure spanning the computing continuum. Consequently, training and inference workloads can interfere with detrimental effects on performance. To address this issue, we propose an inference load-aware HFL orchestration scheme, which makes informed decisions on HFL configuration, considering knowledge about inference workloads and the respective processing capacity. Applying our scheme to a continual learning use case in the transportation domain, we demonstrate that by optimizing aggregator node placement and device-aggregator association, significant inference latency savings can be achieved while communication costs are drastically reduced compared to flat centralized federated learning.
format Preprint
id arxiv_https___arxiv_org_abs_2407_16836
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Inference Load-Aware Orchestration for Hierarchical Federated Learning
Lackinger, Anna
Frangoudis, Pantelis A.
Čilić, Ivan
Furutanpey, Alireza
Murturi, Ilir
Žarko, Ivana Podnar
Dustdar, Schahram
Distributed, Parallel, and Cluster Computing
Hierarchical federated learning (HFL) designs introduce intermediate aggregator nodes between clients and the global federated learning server in order to reduce communication costs and distribute server load. One side effect is that machine learning model replication at scale comes "for free" as part of the HFL process: model replicas are hosted at the client end, intermediate nodes, and the global server level and are readily available for serving inference requests. This creates opportunities for efficient model serving but simultaneously couples the training and serving processes and calls for their joint orchestration. This is particularly important for continual learning, where serving a model while (re)training it periodically, upon specific triggers, or continuously, takes place over shared infrastructure spanning the computing continuum. Consequently, training and inference workloads can interfere with detrimental effects on performance. To address this issue, we propose an inference load-aware HFL orchestration scheme, which makes informed decisions on HFL configuration, considering knowledge about inference workloads and the respective processing capacity. Applying our scheme to a continual learning use case in the transportation domain, we demonstrate that by optimizing aggregator node placement and device-aggregator association, significant inference latency savings can be achieved while communication costs are drastically reduced compared to flat centralized federated learning.
title Inference Load-Aware Orchestration for Hierarchical Federated Learning
topic Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2407.16836