arXiv:2510.23503cs.DCcs.LG2025-10被引 2

用贝叶斯优化实现边缘计算中能耗与延迟约束下的协同推理分割。

Bayes-Split-Edge: Bayesian Optimization for Constrained Collaborative Inference in Wireless Edge Systems

  • 基于贝叶斯优化联合调优传输功率与模型分割点。
  • 相比标准方法降低2.4倍评估成本,20次内完成优化。
  • 适合资源受限的移动边缘设备,如AR/VR头显。

移动边缘设备(如AR/VR头显)在计算和能源资源有限的情况下,需在给定截止时间内完成推理任务。本文研究无线边缘网络中的协同推理问题,通过在边缘设备与边缘服务器间分割神经网络,实现任务协作。提出一种名为Bayes-Split-Edge的新方法,以贝叶斯优化求解在能耗与延迟约束下的推理效用最大化问题。该框架采用新型混合采集函数,平衡任务效用、采样效率与约束违反惩罚。在VGG19+ImageNet-Mini与ResNet101+Tiny-ImageNet上,结合真实mMobile无线信道数据集进行评估。结果表明,该方法相比标准贝叶斯优化可减少2.4倍评估成本,接近线性收敛;优于CMA-ES、DIRECT、穷举搜索及近端策略优化(PPO),且在紧约束下性能接近穷举搜索。验证了其仅需最多20次函数评估即可实现高效、约束感知的边缘协同推理优化。

原文摘要 · Abstract (English)

Mobile edge devices (e.g., AR/VR headsets) typically need to complete timely inference tasks while operating with limited on-board computing and energy resources. In this paper, we investigate the problem of collaborative inference in wireless edge networks, where energy-constrained edge devices aim to complete inference tasks within given deadlines. These tasks are carried out using neural networks, and the edge device seeks to optimize inference performance under energy and delay constraints. The inference process can be split between the edge device and an edge server, thereby achieving collaborative inference over wireless networks. We formulate an inference utility optimization problem subject to energy and delay constraints, and propose a novel solution called Bayes-Split-Edge, which leverages Bayesian optimization for collaborative split inference over wireless edge networks. Our solution jointly optimizes the transmission power and the neural network split point. The Bayes-Split-Edge framework incorporates a novel hybrid acquisition function that balances inference task utility, sample efficiency, and constraint violation penalties. We evaluate our approach using the VGG19 model on the ImageNet-Mini dataset, and Resnet101 on Tiny-ImageNet, and real-world mMobile wireless channel datasets. Numerical results demonstrate that Bayes-Split-Edge achieves up to 2.4x reduction in evaluation cost compared to standard Bayesian optimization and achieves near-linear convergence. It also outperforms several baselines, including CMA-ES, DIRECT, exhaustive search, and Proximal Policy Optimization (PPO), while matching exhaustive search performance under tight constraints. These results confirm that the proposed framework provides a sample-efficient solution requiring maximum 20 function evaluations and constraint-aware optimization for wireless split inference in edge computing systems.

边缘计算贝叶斯优化协同推理资源约束

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