artificial intelligence

Efficient and Privacy Aware Edge Cloud Collaborative Inference for Large Language Models

arXiv:2607.13093

summary

The paper proposes a privacy‑focused edge‑cloud collaborative framework for large language model inference that authenticates and encrypts KV cache data, allowing lightweight edge devices to handle preprocessing and partial decoding while the cloud performs the heavy decoder work, achieving lower latency and bandwidth use.

Abstract

On-device LLM inference faces a trilemma of response latency, limited hardware resources and user privacy. Full cloud inference delivers strong computing power but exposes user prompts and dialogue data, while standalone on-device inference is unfeasible for most consumer and embedded edge devices. This paper presents a privacy-centric edge-cloud collaborative LLM inference framework built on endpoint-authenticated KV cache. Local endpoints handle input preprocessing, embedding computation, adaptive feature optimization, KV cache authentication, speculative decoding and low-dimensional model head calculation, while the cloud conducts authenticated decoder inference, KV cache management, token verification and high-dimensional vocabulary projection. Endpoints fuse partial outputs, apply language-adaptive masking and sample target tokens. All transmitted data and truncated logits are quantized and AES-GCM encrypted for privacy, with core lightweight modules, draft parameters and cache access policies kept local to avoid leakage. The framework supports heterogeneous devices including CPU-only, GPU-equipped and embedded devices via optimized streaming, batching and quantized ONNX deployment. Evaluations demonstrate that the framework reduces per-token latency by up to 46.1\% and downlink payloads by up to 67.4\% over baseline split inference, retaining comparable performance to full cloud inference.

Topics & keywords

#edge computing#collaborative inference#large language models#privacy#kv cache authenticationKV cacheAES-GCM encryptionspeculative decodingquantized ONNXtoken verification