边端网络中按优先级分配模型层,提升多源推理效率。
Priority-Aware Model-Distributed Inference at Edge Networks
- 根据数据源优先级动态分配模型层,优化资源利用。
- 在真实设备上实测,相比基线推理速度提升显著。
- 适合有不同重要性数据源的边缘智能系统部署。
分布式推理分为数据分布式(DDI)和模型分布式(MDI)两类。在DDI中,每个工作节点携带完整模型但仅处理部分数据,导致大数据量时通信开销高。新兴的MDI模式下,每个节点仅持有模型的部分层,数据源设备处理前几层后将中间输出传给邻近设备继续处理,直至完成全部计算。本文研究多数据源共存下的MDI设计,考虑各数据源具有不同重要性即优先级,提出并求解了优先级感知的模型分配优化问题。基于最优解结构,设计了实用的优先级感知模型分布式推理(PA-MDI)算法,综合考虑源优先级实现模型分发与分配。实验在NVIDIA Jetson Xavier与Nano边缘设备及Colosseum测试平台进行,采用ResNet-50、ResNet-56和GPT-2模型,结果表明PA-MDI成功实现优先级感知模型分配,同时较基线显著降低推理时间。
原文摘要 · Abstract (English)
Distributed inference techniques can be broadly classified into data-distributed and model-distributed schemes. In data-distributed inference (DDI), each worker carries the entire Machine Learning (ML) model but processes only a subset of the data. However, feeding the data to workers results in high communication costs, especially when the data is large. An emerging paradigm is model-distributed inference (MDI), where each worker carries only a subset of ML layers. In MDI, a source device that has data processes a few layers of ML model and sends the output to a neighboring device, i.e., offloads the rest of the layers. This process ends when all layers are processed in a distributed manner. In this paper, we investigate the design and development of MDI when multiple data sources co-exist. We consider that each data source has a different importance and, hence, a priority. We formulate and solve a priority-aware model allocation optimization problem. Based on the structure of the optimal solution, we design a practical Priority-Aware Model- Distributed Inference (PA-MDI) algorithm that determines model allocation and distribution over devices by taking into account the priorities of different sources. Experiments were conducted on a real-life testbed of NVIDIA Jetson Xavier and Nano edge devices as well as in the Colosseum testbed with ResNet-50, ResNet- 56, and GPT-2 models. The experimental results show that PA-MDI performs priority-aware model allocation successfully while reducing the inference time as compared to baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。