Accelerating Transfer Learning with Near-Data Computation on Cloud Object Stores

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Petrescu, Diana, Guirguis, Arsany, Quoc, Do Le, Picorel, Javier, Guerraoui, Rachid, Dinu, Florin
Natura: Preprint
Pubblicazione: 2022
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915001194774528
author Petrescu, Diana
Guirguis, Arsany
Quoc, Do Le
Picorel, Javier
Guerraoui, Rachid
Dinu, Florin
author_facet Petrescu, Diana
Guirguis, Arsany
Quoc, Do Le
Picorel, Javier
Guerraoui, Rachid
Dinu, Florin
contents Storage disaggregation underlies today's cloud and is naturally complemented by pushing down some computation to storage, thus mitigating the potential network bottleneck between the storage and compute tiers. We show how ML training benefits from storage pushdowns by focusing on transfer learning (TL), the widespread technique that democratizes ML by reusing existing knowledge on related tasks. We propose HAPI, a new TL processing system centered around two complementary techniques that address challenges introduced by disaggregation. First, applications must carefully balance execution across tiers for performance. HAPI judiciously splits the TL computation during the feature extraction phase yielding pushdowns that not only improve network time but also improve total TL training time by overlapping the execution of consecutive training iterations across tiers. Second, operators want resource efficiency from the storage-side computational resources. HAPI employs storage-side batch size adaptation allowing increased storage-side pushdown concurrency without affecting training accuracy. HAPI yields up to 2.5x training speed-up while choosing in 86.8% of cases the best performing split point or one that is at most 5% off from the best.
format Preprint
id arxiv_https___arxiv_org_abs_2210_08650
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Accelerating Transfer Learning with Near-Data Computation on Cloud Object Stores
Petrescu, Diana
Guirguis, Arsany
Quoc, Do Le
Picorel, Javier
Guerraoui, Rachid
Dinu, Florin
Machine Learning
Databases
Storage disaggregation underlies today's cloud and is naturally complemented by pushing down some computation to storage, thus mitigating the potential network bottleneck between the storage and compute tiers. We show how ML training benefits from storage pushdowns by focusing on transfer learning (TL), the widespread technique that democratizes ML by reusing existing knowledge on related tasks. We propose HAPI, a new TL processing system centered around two complementary techniques that address challenges introduced by disaggregation. First, applications must carefully balance execution across tiers for performance. HAPI judiciously splits the TL computation during the feature extraction phase yielding pushdowns that not only improve network time but also improve total TL training time by overlapping the execution of consecutive training iterations across tiers. Second, operators want resource efficiency from the storage-side computational resources. HAPI employs storage-side batch size adaptation allowing increased storage-side pushdown concurrency without affecting training accuracy. HAPI yields up to 2.5x training speed-up while choosing in 86.8% of cases the best performing split point or one that is at most 5% off from the best.
title Accelerating Transfer Learning with Near-Data Computation on Cloud Object Stores
topic Machine Learning
Databases
url https://arxiv.org/abs/2210.08650