Efficiently Executing High-throughput Lightweight LLM Inference Applications on Heterogeneous Opportunistic GPU Clusters with Pervasive Context Management
Phung, Thain
The rise of Generative AI introduces a new class of HPC workloads that integrates lightweight LLMs with traditional high-throughput applications to accelerate scientific discovery. The current design of HPC clusters is inadequate to support this new class however, either incurring long wait times on static batch queues or repeatedly paying expensive LLM startup costs upon resource preemption. To circumvent both the long queues and high startup costs, we propose to "decouple" the LLM initialization context from the actual LLM inferences, and retain the context in GPUs until it is no longer needed, a technique we term "Pervasive Context Management". We transform a fact verification application to enable this technique, allowing it to reduce its execution time by 72.1% (from 3 hours to 48 minutes) using the same amount of GPUs, and scale opportunistically on 32.8% of all GPUs in the cluster and further reduce the execution time to 13 minutes.
academic
বিষমজাতীয় সুযোগবাদী GPU ক্লাস্টারে ব্যাপক প্রসঙ্গ ব্যবস্থাপনা সহ উচ্চ-থ্রুপুট লাইটওয়েট LLM অনুমান অ্যাপ্লিকেশন দক্ষতার সাথে সম্পাদন
উৎপাদনশীল AI-এর উত্থান একটি নতুন শ্রেণীর HPC কর্মভার প্রবর্তন করেছে যা বৈজ্ঞানিক আবিষ্কার ত্বরান্বিত করার জন্য লাইটওয়েট LLM-কে ঐতিহ্যবাহী উচ্চ-থ্রুপুট অ্যাপ্লিকেশনের সাথে একীভূত করে। তবে বর্তমান HPC ক্লাস্টার ডিজাইন এই নতুন কর্মভারগুলিকে পর্যাপ্তভাবে সমর্থন করতে পারে না, হয় স্ট্যাটিক ব্যাচ প্রসেসিং সারিতে দীর্ঘ অপেক্ষার সময় তৈরি করে বা সম্পদ প্রিয়েম্পশনের সময় বারবার ব্যয়বহুল LLM স্টার্টআপ খরচ বহন করে। দীর্ঘ সারি এবং উচ্চ স্টার্টআপ খরচ এড়াতে, এই পেপারটি LLM প্রাথমিক প্রসঙ্গকে প্রকৃত LLM অনুমান থেকে "বিচ্ছিন্ন" করার এবং প্রসঙ্গটি GPU-তে রাখার প্রস্তাব দেয় যতক্ষণ না এটি আর প্রয়োজন হয়, এই কৌশলটিকে "ব্যাপক প্রসঙ্গ ব্যবস্থাপনা" (Pervasive Context Management) বলা হয়। তথ্য যাচাইকরণ অ্যাপ্লিকেশনের পুনর্নির্মাণের মাধ্যমে, এই কৌশলটি সম্পাদন সময় ৭২.১% হ্রাস করেছে (৩ ঘন্টা থেকে ৪৮ মিনিটে), এবং ক্লাস্টারের ৩২.৮% GPU-তে সুযোগবাদীভাবে স্কেল করতে পারে, সম্পাদন সময় আরও ১৩ মিনিটে হ্রাস করে।
বৃহৎ ভাষা মডেল (LLM) প্রযুক্তির দ্রুত বিকাশের সাথে, একটি নতুন শ্রেণীর HPC কর্মভার উদীয়মান হচ্ছে যা লাইটওয়েট LLM অনুমান (সাধারণত বিলিয়ন পরামিতি সহ) ঐতিহ্যবাহী উচ্চ-থ্রুপুট অ্যাপ্লিকেশনে একীভূত করে। এই ধরনের অ্যাপ্লিকেশনগুলি প্রোটিন ফোল্ডিং, বিতরণকৃত AI-চালিত বৈজ্ঞানিক কম্পিউটিং এবং অন্যান্য ক্ষেত্রে বিশাল সম্ভাবনা প্রদর্শন করে।
স্ট্যাটিক বরাদ্দ মডেলের সীমাবদ্ধতা: ঐতিহ্যবাহী স্ট্যাটিক GPU বরাদ্দ মডেলের জন্য নির্দিষ্ট আকারের GPU ব্যাচ একচেটিয়া দখল প্রয়োজন, যা গুরুতর সারি অপেক্ষার সময় এবং ক্লাস্টার সম্পদ ব্যবহারের অপ্রতুলতার দিকে পরিচালিত করে
সুযোগবাদী বরাদ্দের স্টার্টআপ খরচ: যদিও সুযোগবাদী সম্পদ বরাদ্দ গতিশীলভাবে উপলব্ধ GPU সম্পদ ব্যবহার করতে পারে, LLM-এর স্টার্টআপ প্রক্রিয়া (বিলিয়ন পরামিতি মডেল বিতরণকৃত ফাইল সিস্টেম থেকে স্থানীয় ডিস্ক, হোস্ট মেমরি এবং অবশেষে GPU মেমরিতে লোড করা) I/O-নিবিড় এবং কয়েক মিনিট সময় নিতে পারে
সম্পদ প্রিয়েম্পশনের খরচ: যখন একটি কাজ প্রিয়েম্ট করা হয়, সম্পূর্ণ ব্যয়বহুল স্টার্টআপ প্রক্রিয়া নতুন সম্পদে পুনরায় সম্পাদন করতে হবে, প্রায়শই স্টার্টআপ খরচ প্রকৃত গণনার সময়কে অতিক্রম করে
ব্যাপক প্রসঙ্গ ব্যবস্থাপনা কৌশল প্রস্তাব: LLM প্রাথমিক প্রসঙ্গকে ক্লাস্টারে একটি প্রথম-শ্রেণীর স্থায়ী সত্তায় উন্নীত করা, একাধিক কাজ জুড়ে পুনর্ব্যবহার সক্ষম করা
Parsl-TaskVine ফ্রেমওয়ার্ক-ভিত্তিক উচ্চ-থ্রুপুট তথ্য যাচাইকরণ অ্যাপ্লিকেশন বাস্তবায়ন: বিতরণকৃত ডেটা-নিবিড় ফ্রেমওয়ার্কে লাইটওয়েট LLM-এর প্রয়োগ প্রদর্শন করা
দ্রুত অ্যাপ্লিকেশন রূপান্তর পদ্ধতি ডিজাইন: সহজ কোড পুনর্গঠনের মাধ্যমে অ্যাপ্লিকেশনগুলিকে প্রসঙ্গ-সচেতন সমর্থন করা
উল্লেখযোগ্য কর্মক্ষমতা উন্নতি যাচাই: একই GPU সংখ্যায় সম্পাদন সময় ৭২.১% হ্রাস, এবং ক্লাস্টারের ৩২.৮% GPU-তে সুযোগবাদীভাবে স্কেল করা
এই গবেষণা উচ্চ-থ্রুপুট লাইটওয়েট LLM অনুমান অ্যাপ্লিকেশনগুলিকে লক্ষ্য করে, বিশেষত যেখানে বিষমজাতীয় সুযোগবাদী GPU ক্লাস্টারে বড় সংখ্যক স্বাধীন অনুমান কাজ সম্পাদন করা প্রয়োজন। ইনপুট হল অসংখ্য অনুমান অনুরোধ, আউটপুট হল অনুমান ফলাফল, সীমাবদ্ধতাগুলির মধ্যে রয়েছে GPU সম্পদের গতিশীল উপলব্ধতা এবং অপ্রত্যাশিত প্রিয়েম্পশন।
ব্যাপক প্রসঙ্গ ব্যবস্থাপনার মূল ধারণা হল ব্যয়বহুল LLM প্রসঙ্গ প্রাথমিকীকরণকে প্রকৃত অনুমান সম্পাদন থেকে বিচ্ছিন্ন করা, প্রসঙ্গটিকে ক্লাস্টার নোড জুড়ে স্থায়ী এবং পুনর্ব্যবহারযোগ্য একটি প্রথম-শ্রেণীর সত্তা করা।
Full-context সংস্করণ বিভিন্ন ব্যাচ আকারে সম্পাদন সময়ের পরিবর্তন মাত্র ১৩.৬%, যখন Partial-context সংস্করণ ব্যাচ আকার ১ হলে সম্পাদন সময় ১৪১,১০০ সেকেন্ডে বৃদ্ধি পায়, অত্যন্ত উচ্চ সংবেদনশীলতা প্রদর্শন করে।
ক্লাউড কম্পিউটিংয়ে Spot ইনস্ট্যান্সগুলি অনুরূপ সুযোগবাদী কম্পিউটিং মডেল প্রদান করে, তবে সাধারণত ৩০-১২০ সেকেন্ডের প্রিয়েম্পশন সতর্কতা প্রদান করে, যখন HPC পরিবেশে প্রিয়েম্পশন প্রায়শই তাৎক্ষণিক, ঐতিহ্যবাহী অবস্থা সংরক্ষণ প্রক্রিয়া অকার্যকর।
ঐতিহ্যবাহী সম্পদ ব্যবস্থাপক থেকে আধুনিক Python-নেটিভ ওয়ার্কফ্লো সিস্টেমে বিবর্তন, এই পেপারের Parsl-TaskVine একীকরণ গণনা প্রসঙ্গ ভাগাভাগি সমর্থন করার নতুন দিক প্রতিনিধিত্ব করে।
পেপারটি ৬১টি সম্পর্কিত সংদর্ভ উদ্ধৃত করে, যা LLM প্রযুক্তি, HPC সময়সূচী, ওয়ার্কফ্লো সিস্টেম এবং অন্যান্য ক্ষেত্রের গুরুত্বপূর্ণ কাজ অন্তর্ভুক্ত করে, গবেষণার জন্য একটি দৃঢ় তাত্ত্বিক ভিত্তি প্রদান করে।
সামগ্রিক মূল্যায়ন: এটি HPC পরিবেশে উদীয়মান AI কর্মভারের জন্য একটি উচ্চ-মানের গবেষণা পেপার। লেখকরা প্রকৃত সমস্যা সঠিকভাবে সনাক্ত করেছেন, একটি উদ্ভাবনী সমাধান প্রস্তাব করেছেন, এবং ব্যাপক পরীক্ষার মাধ্যমে পদ্ধতির কার্যকারিতা যাচাই করেছেন। যদিও প্রযোজ্য পরিসীমা এবং তাত্ত্বিক বিশ্লেষণে নির্দিষ্ট সীমাবদ্ধতা রয়েছে, এটি সম্পর্কিত ক্ষেত্রের গবেষণা এবং অনুশীলনে মূল্যবান অবদান প্রদান করে।