Dependable Distributed Training of Compressed Machine Learning Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Malandrino, Francesco, Di Giacomo, Giuseppe, Levorato, Marco, Chiasserini, Carla Fabiana
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909116689022976
author Malandrino, Francesco
Di Giacomo, Giuseppe
Levorato, Marco
Chiasserini, Carla Fabiana
author_facet Malandrino, Francesco
Di Giacomo, Giuseppe
Levorato, Marco
Chiasserini, Carla Fabiana
contents The existing work on the distributed training of machine learning (ML) models has consistently overlooked the distribution of the achieved learning quality, focusing instead on its average value. This leads to a poor dependability}of the resulting ML models, whose performance may be much worse than expected. We fill this gap by proposing DepL, a framework for dependable learning orchestration, able to make high-quality, efficient decisions on (i) the data to leverage for learning, (ii) the models to use and when to switch among them, and (iii) the clusters of nodes, and the resources thereof, to exploit. For concreteness, we consider as possible available models a full DNN and its compressed versions. Unlike previous studies, DepL guarantees that a target learning quality is reached with a target probability, while keeping the training cost at a minimum. We prove that DepL has constant competitive ratio and polynomial complexity, and show that it outperforms the state-of-the-art by over 27% and closely matches the optimum.
format Preprint
id arxiv_https___arxiv_org_abs_2402_14346
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Dependable Distributed Training of Compressed Machine Learning Models
Malandrino, Francesco
Di Giacomo, Giuseppe
Levorato, Marco
Chiasserini, Carla Fabiana
Machine Learning
Artificial Intelligence
The existing work on the distributed training of machine learning (ML) models has consistently overlooked the distribution of the achieved learning quality, focusing instead on its average value. This leads to a poor dependability}of the resulting ML models, whose performance may be much worse than expected. We fill this gap by proposing DepL, a framework for dependable learning orchestration, able to make high-quality, efficient decisions on (i) the data to leverage for learning, (ii) the models to use and when to switch among them, and (iii) the clusters of nodes, and the resources thereof, to exploit. For concreteness, we consider as possible available models a full DNN and its compressed versions. Unlike previous studies, DepL guarantees that a target learning quality is reached with a target probability, while keeping the training cost at a minimum. We prove that DepL has constant competitive ratio and polynomial complexity, and show that it outperforms the state-of-the-art by over 27% and closely matches the optimum.
title Dependable Distributed Training of Compressed Machine Learning Models
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2402.14346