Large Language Model (LLM) inference heavily relies on cloud systems, resulting in inherent limitations such as prolonged delays, escalated bandwidth expenditure, and potential data security risks. Edge computing emerges as a solution by enabling LLM execution on edge devices, closer to data sources. However, challenges such as device heterogeneity and limited memory capacity pose significant barriers to conventional tensor parallelism of LLM. Therefore, we propose EdgeInfer-TP, a resource-aware collaborative tensor parallelism inference system tailored for heterogeneous edge devices. It integrates MILP-based model allocation optimization and multi-level dynamic offloading of KV cache to enable efficient and stable collaborative inference on resource-constrained devices. The experimental results across multiple real-world heterogeneous devices and diverse network environments demonstrate that our approach achieves up to 22.23% latency reduction and 27.87% throughput improvement over baseline approaches.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

EdgeInfer-TP: A Collaborative Tensor Parallelism Inference System for Heterogeneous Edge Devices

  • Yutao Zhang,
  • Wentao Zhong,
  • Xuerui Liu,
  • Fengyi Huang,
  • Wenhua Wang,
  • Tian Wang,
  • Weijia Jia

摘要

Large Language Model (LLM) inference heavily relies on cloud systems, resulting in inherent limitations such as prolonged delays, escalated bandwidth expenditure, and potential data security risks. Edge computing emerges as a solution by enabling LLM execution on edge devices, closer to data sources. However, challenges such as device heterogeneity and limited memory capacity pose significant barriers to conventional tensor parallelism of LLM. Therefore, we propose EdgeInfer-TP, a resource-aware collaborative tensor parallelism inference system tailored for heterogeneous edge devices. It integrates MILP-based model allocation optimization and multi-level dynamic offloading of KV cache to enable efficient and stable collaborative inference on resource-constrained devices. The experimental results across multiple real-world heterogeneous devices and diverse network environments demonstrate that our approach achieves up to 22.23% latency reduction and 27.87% throughput improvement over baseline approaches.