Hybrid-Parallel: Achieving High Performance and Energy Efficient Distributed Inference on Robots

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Sun, Zekai, Guan, Xiuxian, Wang, Junming, Song, Haoze, Qing, Yuhao, Shen, Tianxiang, Huang, Dong, Liu, Fangming, Cui, Heming
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911892612579328
author Sun, Zekai
Guan, Xiuxian
Wang, Junming
Song, Haoze
Qing, Yuhao
Shen, Tianxiang
Huang, Dong
Liu, Fangming
Cui, Heming
author_facet Sun, Zekai
Guan, Xiuxian
Wang, Junming
Song, Haoze
Qing, Yuhao
Shen, Tianxiang
Huang, Dong
Liu, Fangming
Cui, Heming
contents The rapid advancements in machine learning techniques have led to significant achievements in various real-world robotic tasks. These tasks heavily rely on fast and energy-efficient inference of deep neural network (DNN) models when deployed on robots. To enhance inference performance, distributed inference has emerged as a promising approach, parallelizing inference across multiple powerful GPU devices in modern data centers using techniques such as data parallelism, tensor parallelism, and pipeline parallelism. However, when deployed on real-world robots, existing parallel methods fail to provide low inference latency and meet the energy requirements due to the limited bandwidth of robotic IoT. We present Hybrid-Parallel, a high-performance distributed inference system optimized for robotic IoT. Hybrid-Parallel employs a fine-grained approach to parallelize inference at the granularity of local operators within DNN layers (i.e., operators that can be computed independently with the partial input, such as the convolution kernel in the convolution layer). By doing so, Hybrid-Parallel enables different operators of different layers to be computed and transmitted concurrently, and overlap the computation and transmission phases within the same inference task. The evaluation demonstrate that Hybrid-Parallel reduces inference time by 14.9% ~41.1% and energy consumption per inference by up to 35.3% compared to the state-of-the-art baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2405_19257
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Hybrid-Parallel: Achieving High Performance and Energy Efficient Distributed Inference on Robots
Sun, Zekai
Guan, Xiuxian
Wang, Junming
Song, Haoze
Qing, Yuhao
Shen, Tianxiang
Huang, Dong
Liu, Fangming
Cui, Heming
Robotics
Distributed, Parallel, and Cluster Computing
The rapid advancements in machine learning techniques have led to significant achievements in various real-world robotic tasks. These tasks heavily rely on fast and energy-efficient inference of deep neural network (DNN) models when deployed on robots. To enhance inference performance, distributed inference has emerged as a promising approach, parallelizing inference across multiple powerful GPU devices in modern data centers using techniques such as data parallelism, tensor parallelism, and pipeline parallelism. However, when deployed on real-world robots, existing parallel methods fail to provide low inference latency and meet the energy requirements due to the limited bandwidth of robotic IoT. We present Hybrid-Parallel, a high-performance distributed inference system optimized for robotic IoT. Hybrid-Parallel employs a fine-grained approach to parallelize inference at the granularity of local operators within DNN layers (i.e., operators that can be computed independently with the partial input, such as the convolution kernel in the convolution layer). By doing so, Hybrid-Parallel enables different operators of different layers to be computed and transmitted concurrently, and overlap the computation and transmission phases within the same inference task. The evaluation demonstrate that Hybrid-Parallel reduces inference time by 14.9% ~41.1% and energy consumption per inference by up to 35.3% compared to the state-of-the-art baselines.
title Hybrid-Parallel: Achieving High Performance and Energy Efficient Distributed Inference on Robots
topic Robotics
Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2405.19257