Quantize-Sample-and-Verify: LLM Acceleration via Adaptive Edge-Cloud Speculative Decoding
Zhang, Cai, Yu et al.
In edge-cloud speculative decoding (SD), edge devices equipped with small language models (SLMs) generate draft tokens that are verified by large language models (LLMs) in the cloud. A key bottleneck in such systems is the limited communication bandwidth between edge and cloud, which necessitates quantization of the information transmitted about generated tokens. In this work, we introduce a novel quantize-sample (Q-S) strategy that provably preserves the output distribution of the cloud-based model, ensuring that the verified tokens match the distribution of those that would have been generated directly by the LLM. We develop a throughput model for edge-cloud SD that explicitly accounts for communication latency. Leveraging this model, we propose an adaptive mechanism that optimizes token throughput by dynamically adjusting the draft length and quantization precision in response to both semantic uncertainty and channel conditions. Simulations demonstrate that the proposed Q-S approach significantly improves decoding efficiency in realistic edge-cloud deployment scenarios.
academic
কোয়ান্টাইজ-স্যাম্পল-এন্ড-ভেরিফাই: অভিযোজিত এজ-ক্লাউড স্পেকুলেটিভ ডিকোডিং এর মাধ্যমে এলএলএম ত্বরণ
এজ-ক্লাউড স্পেকুলেটিভ ডিকোডিং (এসডি) সিস্টেমে, ছোট ভাষা মডেল (এসএলএম) সহ এজ ডিভাইসগুলি খসড়া টোকেন তৈরি করে, যা পরে ক্লাউডে বড় ভাষা মডেল (এলএলএম) দ্বারা যাচাই করা হয়। এই ধরনের সিস্টেমের মূল বাধা হল এজ এবং ক্লাউডের মধ্যে সীমিত যোগাযোগ ব্যান্ডউইথ, যা প্রেরিত টোকেন তথ্যের পরিমাণকরণকে প্রয়োজনীয় করে তোলে। এই কাজটি একটি উপন্যাস কোয়ান্টাইজ-স্যাম্পল (কিউ-এস) কৌশল উপস্থাপন করে যা প্রমাণযোগ্যভাবে ক্লাউড মডেলের আউটপুট বিতরণ বজায় রাখে, নিশ্চিত করে যে যাচাইকৃত টোকেনগুলি সরাসরি এলএলএম দ্বারা উত্পাদিত টোকেনের বিতরণের সাথে মেলে। আমরা এজ-ক্লাউড এসডির জন্য একটি স্পষ্ট থ্রুপুট মডেল তৈরি করেছি যা যোগাযোগ বিলম্ব বিবেচনা করে। এই মডেলের উপর ভিত্তি করে, আমরা একটি অভিযোজিত প্রক্রিয়া প্রস্তাব করি যা শব্দার্থিক অনিশ্চয়তা এবং চ্যানেল অবস্থার প্রতিক্রিয়ায় খসড়া দৈর্ঘ্য এবং পরিমাণকরণ নির্ভুলতা গতিশীলভাবে সামঞ্জস্য করে, যার ফলে টোকেন থ্রুপুট অপ্টিমাইজ করা হয়। সিমুলেশন ফলাফল দেখায় যে প্রস্তাবিত কিউ-এস পদ্ধতি বাস্তবসম্মত এজ-ক্লাউড স্থাপনার পরিস্থিতিতে ডিকোডিং দক্ষতা উল্লেখযোগ্যভাবে উন্নত করে।
এই গবেষণা যা মূল সমস্যা সমাধান করে তা হল এজ-ক্লাউড স্পেকুলেটিভ ডিকোডিং সিস্টেমে যোগাযোগ ব্যান্ডউইথ সীমাবদ্ধতা। ঐতিহ্যবাহী স্পেকুলেটিভ ডিকোডিংয়ে, এজ ডিভাইসগুলিকে ক্লাউডে প্রচুর পরিমাণে সম্ভাব্যতা বিতরণ তথ্য প্রেরণ করতে হয়, যা ব্যান্ডউইথ-সীমিত পরিবেশে সিস্টেম কর্মক্ষমতাকে গুরুতরভাবে প্রভাবিত করে।
এজ-ক্লাউড স্পেকুলেটিভ ডিকোডিং কাজটি সংজ্ঞায়িত করা হয়েছে: ইনপুট প্রিফিক্স s¹ দেওয়া হলে, সিস্টেমকে এজ এসএলএম এর মাধ্যমে খসড়া টোকেন তৈরি করতে হবে, ক্লাউড এলএলএম যাচাই করবে, চূড়ান্তভাবে সরাসরি এলএলএম ব্যবহার করে একই বিতরণের টোকেন সিকোয়েন্স তৈরি করবে।
প্রস্তাবনা ১: কিউ-এস এজ-ক্লাউড এসডি নিশ্চিত করে যে উত্পাদিত টোকেন xᵗₗ এর সম্ভাব্যতা P(X = xᵗₗ) এলএলএমের সংশ্লিষ্ট সম্ভাব্যতা pᵗₗ,xᵗₗ এর সমান।
এই সম্পত্তির মূল চাবিকাঠি হল নমুনা এবং যাচাইকরণ একই পরিমাণকৃত বিতরণ ব্যবহার করে, যখন এস-কিউ পদ্ধতি বিভিন্ন বিতরণ ব্যবহার করে বিতরণ পরিবর্তন সৃষ্টি করে।
পত্রটি বিস্তারিত পরীক্ষামূলক সেটআপ এবং ওপেন-সোর্স কোড লিঙ্ক প্রদান করে, ভাল পুনরুৎপাদনযোগ্যতা রয়েছে। তবে বাস্তব এজ ডিভাইসের স্থাপনা যাচাইকরণ আরও কাজের প্রয়োজন।
Chen, C., et al. "Accelerating large language model decoding with speculative sampling." arXiv:2302.01318, 2023.
Hao, Z., et al. "Hybrid SLM and LLM for edge-cloud collaborative inference." Proc. Worksh. Edge Mobil. Found. Models, 2024.
Leviathan, Y., et al. "Fast inference from transformers via speculative decoding." Proc. Int. Conf. Mach. Learn. (ICML), 2023.
Reznik, Y. A. "An algorithm for quantization of discrete probability distributions." Data Compress. Conf. (DCC), 2011.
সামগ্রিক মূল্যায়ন: এটি এজ-ক্লাউড স্পেকুলেটিভ ডিকোডিং ক্ষেত্রে গুরুত্বপূর্ণ অবদান সহ একটি উচ্চ-মানের পত্র। তাত্ত্বিক বিশ্লেষণ কঠোর, পরীক্ষামূলক যাচাইকরণ ব্যাপক, বাস্তব প্রয়োগে মূল সমস্যা সমাধান করা হয়েছে। যদিও কিছু সীমাবদ্ধতা রয়েছে, তবে এর উদ্ভাবনী এবং ব্যবহারিক মূল্য এটিকে এই ক্ষেত্রের একটি গুরুত্বপূর্ণ কাজ করে তোলে।