arXiv:2412.16616cs.DCcs.AI2024-12

根据数据复杂度动态分配模型计算,降低移动设备推理成本。

Distributed Inference on Mobile Edge and Cloud: A Data-Cartography based Clustering Approach

  • 用数据地图方法评估样本复杂度,决定在何处运行模型
  • 在GLUE上推理成本降43%以上,准确率损失小于0.5%
  • 适合资源受限的移动端和边缘计算场景

大型神经网络(DNN)部署在移动、边缘和物联网等资源受限设备上面临挑战。为此,可采用分布式推理框架:将小型DNN(初始层)部署在移动设备,较大版本放在边缘设备,完整模型置于云端。简单样本在移动设备处理,中等复杂度样本由边缘设备处理,复杂样本交由云端。由于样本复杂度事先未知,核心问题是判断每类样本应由哪个层级处理。本文提出一种新方法 extit{our{}}, 基于最初用于提升DNN泛化能力的数据地图(Data Cartography)技术,通过该方法评估样本复杂度。该方法在提升精度的同时,兼顾从移动设备向边缘/云端卸载的开销。在涵盖多种自然语言处理任务的GLUE数据集上的实验表明,相比全部在云端推理,本方法推理成本降低超过43%,准确率下降不足0.5%。源代码见https://anonymous.4open.science/r/DIMEC-1B04。

原文摘要 · Abstract (English)

The large size of DNNs poses a significant challenge for deployment on devices with limited resources, such as mobile, edge, and IoT platforms. To address this issue, a distributed inference framework can be utilized. In this framework, a small-scale DNN (initial layers) is deployed on mobile devices, a larger version on edge devices, and the full DNN on the cloud. Samples with low complexity (easy) can be processed on mobile, those with moderate complexity (medium) on edge devices, and high complexity (hard) samples on the cloud. Given that the complexity of each sample is unknown in advance, the crucial question in distributed inference is determining the sample complexity for appropriate DNN processing. We introduce a novel method named \our{}, which leverages the Data Cartography approach initially proposed for enhancing DNN generalization. By employing data cartography, we assess sample complexity. \our{} aims to boost accuracy while considering the offloading costs from mobile to edge/cloud. Our experimental results on GLUE datasets, covering a variety of NLP tasks, indicate that our approach significantly lowers inference costs by more than 43\% while maintaining a minimal accuracy drop of less than 0.5\% compared to performing all inferences on the cloud. The source code is available at https://anonymous.4open.science/r/DIMEC-1B04.

分布式推理边缘计算数据地图NLP

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。