Energy-Aware Heterogeneous Federated Learning via Approximate DNN Accelerators

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Pfeiffer, Kilian, Balaskas, Konstantinos, Siozios, Kostas, Henkel, Jörg
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917853174693888
author Pfeiffer, Kilian
Balaskas, Konstantinos
Siozios, Kostas
Henkel, Jörg
author_facet Pfeiffer, Kilian
Balaskas, Konstantinos
Siozios, Kostas
Henkel, Jörg
contents In Federated Learning (FL), devices that participate in the training usually have heterogeneous resources, i.e., energy availability. In current deployments of FL, devices that do not fulfill certain hardware requirements are often dropped from the collaborative training. However, dropping devices in FL can degrade training accuracy and introduce bias or unfairness. Several works have tackled this problem on an algorithm level, e.g., by letting constrained devices train a subset of the server neural network (NN) model. However, it has been observed that these techniques are not effective w.r.t. accuracy. Importantly, they make simplistic assumptions about devices' resources via indirect metrics such as multiply accumulate (MAC) operations or peak memory requirements. We observe that memory access costs (that are currently not considered in simplistic metrics) have a significant impact on the energy consumption. In this work, for the first time, we consider on-device accelerator design for FL with heterogeneous devices. We utilize compressed arithmetic formats and approximate computing, targeting to satisfy limited energy budgets. Using a hardware-aware energy model, we observe that, contrary to the state of the art's moderate energy reduction, our technique allows for lowering the energy requirements (by 4x) while maintaining higher accuracy.
format Preprint
id arxiv_https___arxiv_org_abs_2402_18569
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Energy-Aware Heterogeneous Federated Learning via Approximate DNN Accelerators
Pfeiffer, Kilian
Balaskas, Konstantinos
Siozios, Kostas
Henkel, Jörg
Hardware Architecture
In Federated Learning (FL), devices that participate in the training usually have heterogeneous resources, i.e., energy availability. In current deployments of FL, devices that do not fulfill certain hardware requirements are often dropped from the collaborative training. However, dropping devices in FL can degrade training accuracy and introduce bias or unfairness. Several works have tackled this problem on an algorithm level, e.g., by letting constrained devices train a subset of the server neural network (NN) model. However, it has been observed that these techniques are not effective w.r.t. accuracy. Importantly, they make simplistic assumptions about devices' resources via indirect metrics such as multiply accumulate (MAC) operations or peak memory requirements. We observe that memory access costs (that are currently not considered in simplistic metrics) have a significant impact on the energy consumption. In this work, for the first time, we consider on-device accelerator design for FL with heterogeneous devices. We utilize compressed arithmetic formats and approximate computing, targeting to satisfy limited energy budgets. Using a hardware-aware energy model, we observe that, contrary to the state of the art's moderate energy reduction, our technique allows for lowering the energy requirements (by 4x) while maintaining higher accuracy.
title Energy-Aware Heterogeneous Federated Learning via Approximate DNN Accelerators
topic Hardware Architecture
url https://arxiv.org/abs/2402.18569