arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2601.08025cs.DC

在哪里分割?DNN边缘推理中的帕累托前沿分析

Where to Split? A Pareto-Front Analysis of DNN Partitioning for Edge Inference

Adiba Masud, Nicholas Foley, Pragathi Durga Rajarajan, Palden Lama

AI总结:

本文提出ParetoPipe框架,通过帕累托前沿分析优化DNN边缘推理中的延迟与吞吐量权衡,提供灵活的开源工具支持分布式推理和基准测试。

摘要:

将深度神经网络(DNN)部署到资源受限的边缘设备上经常受到其显著的计算和内存需求的阻碍。虽然将DNN分布在多个设备上是缓解这一挑战的成熟策略,但以往的研究大多集中在单一目标优化上,如最小化延迟或最大化吞吐量。本文挑战这种观点,将DNN划分重新定义为多目标优化问题。我们主张在现实场景中,存在延迟和吞吐量之间的复杂权衡,这进一步被网络变化所复杂化。为此,我们引入了ParetoPipe,一个开源框架,利用帕累托前沿分析系统地识别平衡这些竞争目标的最佳划分策略。我们的贡献有三方面:我们对异构测试平台上的流水线划分推理进行了基准测试,包括Raspberry Pi和配备GPU的边缘服务器;我们识别了帕累托最优点,以分析在不同网络条件下延迟吞吐量的权衡;我们发布了一个灵活的、开源的框架,以促进分布式推理和基准测试。该工具链具有双通信后端,PyTorch RPC和一个定制的轻量级实现,以最小化开销并支持广泛的实验。

英文摘要:

The deployment of deep neural networks (DNNs) on resource-constrained edge devices is frequently hindered by their significant computational and memory requirements. While partitioning and distributing a DNN across multiple devices is a well-established strategy to mitigate this challenge, prior research has largely focused on single-objective optimization, such as minimizing latency or maximizing throughput. This paper challenges that view by reframing DNN partitioning as a multi-objective optimization problem. We argue that in real-world scenarios, a complex trade-off between latency and throughput exists, which is further complicated by network variability. To address this, we introduce ParetoPipe, an open-source framework that leverages Pareto front analysis to systematically identify optimal partitioning strategies that balance these competing objectives. Our contributions are threefold: we benchmark pipeline partitioned inference on a heterogeneous testbed of Raspberry Pis and a GPU-equipped edge server; we identify Pareto-optimal points to analyze the latency-throughput trade-off under varying network conditions; and we release a flexible, open-source framework to facilitate distributed inference and benchmarking. This toolchain features dual communication backends, PyTorch RPC and a custom lightweight implementation, to minimize overhead and support broad experimentation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑