OPTIMA: Optimal One-shot Pruning for LLMs via Quadratic Programming Reconstruction
تُعد OPTIMA طريقة تقليم قابلة للتوسع وذات دفعة واحدة لما بعد التدريب للنماذج اللغوية الكبيرة، حيث تصيغ إعادة بناء أوزان الطبقات كبرامج تربيعية مستقلة لكل عمود مع مصفوفة هسيان مشتركة، مما يحقق دقة متفوقة على النهج الاستدلالي مع الحفاظ على الكفاءة الحسابية على المسرعات الحديثة.
المؤلفون الأصليون:Mohammad Mozaffari, Samuel Kushnir, Maryam Mehri Dehnavi, Amir Yazdanbakhsh
تعد النماذج اللغوية الكبيرة برامج حاسوبية قوية تعلمت فهم وتوليد اللغة البشرية من خلال دراسة كميات هائلة من النصوص. يمكنها كتابة القصص، والإجابة على أسئلة معقدة، وحل المشكلات، ولكنها تأتي مع عيب كبير: وهي ضخامتها. تحتوي هذه النماذج على مليارات المعلمات (parameters)، وهي الأرقام الداخلية التي يعدلها النظام لفهم اللغة. وبسبب حجمها، يتطلب تشغيلها أجهزة قوية ومكلفة، مما يجعل استخدامها صعباً في المهام اليومية أو على الأجهزة الصغيرة. ولجعل هذه النماذج أكثر عملية، يبحث الباحثون عن طرق لتقليص حجمها دون فقدان قدرتها على التفكير بوضوح. إحدى الطرق الشائعة تسمى "التقليم" (pruning)، والتي تتضمن إزالة الأرقام الأقل أهمية من النموذج لتقليل حجمه. ومع ذلك، فإن هذه العملية معقدة؛ فإذا قمت ببساطة بحذف الأرقام، فغالباً ما يصبح النموذج أسوأ بكثير في أداء مهمته. وإذا حاولت إصلاح النموذج عن طريق إعادة حساب جميع الأرقام دفعة واحدة للتعويض عن الخسارة، فإن العمليات الحسابية ستصبح ثقيلة جداً لدرجة يستح مستحيل تنفيذها على الحواسيب الحديثة.
لقد طور فريق من الباحثين نهجاً جديداً يسمى OPTIMA لحل هذه المشكلة تحديداً. كان هدفهم هو إيجاد منطقة وسطى بين طريقة الحذف المباشر للأرقام التي تتسم بالسرعة ولكنها غير دقيقة، وطريقة محاولة إصلاح كل شيء دفعة واحدة التي تتسم بالبطء الشديد. لقد أدركوا أنه بعد اتخاذ القرار بشأن الأرقام المراد إزالتها، يمكن تقسيم مهمة إصلاح الأرقام المتبقية إلى العديد من المشكلات الصغيرة والمستقلة. فبدلاً من محاولة حل رياضيات النموذج بأكملها في خطوة واحدة عملاقة ومستحيلة، قاموا بمعاملة كل عمود من أعمدة الأرقام في النموذج بشكل منفصل. واكتشفوا أنه بينما كان لكل عمود قيوده الخاصة بناءً على الأرقام التي تمت إزالتها، إلا أنها جميعاً تشترك في بنية أساسية واحدة. سمح هذا باستخدام أداة رياضية متخصصة، تُعرف باسم "البرمجة التربيعية" (quadratic program)، لإيجاد القيم البديلة المثالية لكل عمود. ولأن البنية كانت هي نفسها لكل عمود، فقد تمكنوا من حل آلاف هذه المشكلات الصغيرة في وقت واحد على شريحة حاسوبية واحدة، بدلاً من حلها واحدة تلو الأخرى.
اختبر الباحثون هذه الطريقة على عدة عائلات مختلفة من النماذج اللغوية الكبيرة، بما في ذلك نماذج شهيرة مثل LLaMA وGemma. وقد طبقوا تقنيتهم على نماذج تم تقليمها بالفعل بواسطة طرق موجودة مسبقاً، حيث عملت تقنيتهم بمثابة "لمسة نهائية" لتحسين النتائج. وفي معظم الحالات، جعلت هذه الخطوة الإضافية النماذج أكثر دقة بشكل ملحوظ. فعلى سبيل المثال، عندما اختبروا النماذج في مجموعة متنوعة من المهام القياسية، حسنت الطريقة الجديدة متوسط الدرجات بنحو أربع نقاط في بعض السيناريوهات الصعبة، لا سيما عندما تكون النماذج صغيرة جداً. وعملت الطريقة بشكل جيد عبر مختلف أنماط التقليم وأحجام النماذج، بدءاً من النماذج الأصغر ذات المليار معلمة وصولاً إلى النماذج الكبيرة جداً التي تحتوي على مليارات المعلمات. ومن الجدير بالذكر أنه بينما حسنت هذه الطريقة الدقة، إلا أنها استغرقت وقتاً أطول في التشغيل من أسرع الطرق الموجودة، حيث تطلبت حوالي أربعين ساعة لمعالجة نموذج مكون من ثمانية مليارات معلمة على بطاقة رسوميات واحدة عالية الأداء. ومع ذلك، أكد الباحثون أن هذه تكلفة تُدفع لمرة واحدة قبل استخدام النموذج، وأن النموذج الناتج يعمل بنفس سرعة النسخة الأصلية المقلمة.
كما سلطت الدراسة الضوء على أن هذا النهج ليس حلاً سحرياً لكل موقف. ففي حالات محددة قليلة، لم تؤدِ الطة الجديدة إلى تحسين أداء النموذج بل جعلته أحياناً أسوأ قليلاً، مما يشير إلى أن التفاعل بين طريقة التقليم وخطوة التصحيح يمكن أن يكون معقداً. علاوة على ذلك، وجد الباحثون أن الطريقة تعمل بشكل أفضل عندما يترك التقليم الأولي مجالاً كبيراً للتحسين، كما هو الحال عندما يصبح النموذج "متناثراً" (sparse) للغاية. كما لاحظوا أن نجاح الطريقة يعتمد على نوع النموذج المحدد وكيفية إجراء عملية التقليم في الأصل. ورغم هذه القيود، يوضح العمل أنه من الممكن استخدام التحسين الرياضي الصارم لإصلاح النماذج المقلمة بطريقة عملية وقابلة للتوسع. ومن خلال تقسيم مشكلة ضخمة ومستحيلة إلى آلاف المهام المتوازية الصغيرة، أظهر الباحثون أنه يمكننا الحفاظ على النماذج اللغوية الكبيرة صغيرة وفعالة دون التضحية بذكائها. وهذا يفتح الباب أمام ذكاء اصطناعي أكثر قدرة للعمل على أجهزة ضعيفة حالياً لا تدعم تشغيله، مما يقرب أدوات اللغة القوية من الاستخدام اليومي.
ملخص تقني: OPTIMA: التقليم الأمثل بضربة واحدة للنماذج اللغوية الكبيرة عبر إعادة البناء بالبرمجة التربيعية
بيان المشكلة
يهدف التقليم بضربة واحدة (one-shot pruning) بعد مرحلة التدريب إلى تقليل تكاليف الاستدلال للنماذج اللغوية الكبيرة (LLMs) عن طريق إزالة المعلمات باستخدام مجموعة بيانات معايرة صغيرة فقط، وتجنب تكلفة الضبط الدقيق الكامل. ومع ذلك، يواجه هذا المجال مقايضة جوهرية:
الطرق الاستدلالية (Heuristic methods) (مثل Wanda) منخفضة التكلفة الحسابية ولكنها غالبًا ما تؤدي إلى تدهور كبير في الدقة لأنها تقوم ببساطة بتصفير الأوز、 دون تحديث الأوزان المتبقية.
طرق التحسين المشترك المبدئية (Principled joint optimization methods) (مثل Optimal Brain Surgeon) يمكنها استعادة الدقة عبر حل مشكلة تحديث الأوزان المثلى، لكنها غير مجدية حسابيًا على نطاق النماذج اللغوية الكبيرة الحديثة بسبب تعقيد عكس مصفوفات هسيان (Hessian) الضخمة.
التقريبات الحالية (Existing approximations) (مثل SparseGPT وThanos) تحاول سد هذه الفجوة، لكنها تضحي بالكمال في إعادة البناء، مما يترك "هامشًا" من الدقة يمكن استرداده.
التحدي الأساسي هو تطوير طريقة توازن بين الدقة (عبر حل مشكلة إعادة البناء بشكل أمثل) والقابلية للتوسع (عبر كونها مجدية حسابيًا على المسرعات الحديثة).
المنهجية
يقدم البحث إطار عمل OPTIMA، الذي يعيد صياغة خطوة إعادة بناء الأوزان على مستوى الطبقة (layer-wise) بعد اختيار القناع (mask) كمسائل برمجة تربيعية (Quadratic Programs - QPs) مستقلة على مستوى الأعمدة.
1. إعادة الصياغة الرياضية: بعد اختيار قناع ثنائي Ml لمصفوفة أوزان Wl، يكون الهدف هو إيجاد أوزان محدثة W^l تقلل خطأ إعادة بناء المخرجات ∥XlWl−Xl(Ml⊙W^l)∥F2.
يقوم OPTIMA بتفكيك هذه المشكلة على مستوى الأعمدة. لكل عمود j، يتم صياغة التحديث Δwj كبرنامج تربيعي (QP): ΔwjminΔwjTHΔwjsubject to(Δwj)i=−(wj)i,∀i∈Sj حيث أن H=XlTXl هو مصفوفة الهسيان (تباين مدخلات التنشيط)، و Sj تمثل مؤشرات الأوزان التي تم تقليمها.
الرؤية الجوهرية: تشترك جميع أعمدة نفس مصفوفة الأوزان في نفس مصفوفة الهسيانH. وتختلف فقط القيود الخطية (المحددة بالقناع) لكل عمود. تتيح هذه البنية المشتركة إمكانية المعالجة المتوازية (batching) بكفاءة.
2. الحل والتنفيذ:
الخوارزمية: يستخدم OPTIMA حلاً من الدرجة الأولى (first-order primal-dual solver)، وتحديدًا rAPDHG (الخوارزمية الهجينة المتسارعة لثنائية المبدأ مع إعادة التشغيل)، وهي مناسبة للمسائل المقيدة وتعتمد على عمليات ضرب المصفوفات في متجه مع مصفوفة الهسيان المشتركة.
تحسين المسرعات:
المعالجة المتوازية (Batching): يتم حل آلاف المسائل التربيعية الصغيرة بالتوازي على مسرع واحد (GPU/TPU).
كفاءة الذاكرة: بدلاً من تخزين مصفوفة التنشيط الكثيفة بالكامل، يتم تجميع مصفوفة الهسيان تدريجيًا من تسلسلات المعايرة.
التعامل مع القيود: يتم فرض المدخلات المقلمة عبر حدود متغيرات ضيقة بدلاً من مصفوفات مساواة صريحة، مما يتجنب العبء الناتج عن مصفوفات القيود الكبيرة.
آلية التراجع (Fallback Mechanism): إذا لم تتقارب المسألة التربيعية لأحد الأعمدة ضمن الميزانية المحددة من التكرارات (عبر شهادة تقارب)، يتم تجاهل التحديث والاحتفاظ بالأوزان الأصلية (مع تصفير المدخلات المقلمة). يضمن هذا أن الطريقة لن تؤدي أبدًا إلى نتائج أسوأ من تحديث مختار القناع الأساسي.
المساهمات الرئيسية
إعادة صياغة البرمجة التربيعية (QP Reformulation): صياغة إعادة البناء بعد التدريب كمسائل تربيعية متوازية على مستوى الأعمدة تشترك في مصفوفة هسيان واحدة لكل مصفوفة أوزان، مما يتيح الوصول إلى الأمثلية العالمية لإعادة البناء لكل عمود.
خط معالجة قابل للتوسع (Scalable Solver Pipeline): تصميم خط معالجة صديق للمسرعات يقوم بتجميع مصفوفات الهسيان تدريجيًا، ويفرض القيود عبر الحدود، ويحل آلاف المسائل التربيعية بالتوازي باستخدام rAPDHG.
النمطية (Modularity): إثبات أن OPTIMA يعمل كخطوة "إضافة" (drop-in) لتحديث الأوزان متوافقة مع مختاري الأقنعة الحاليين (Wanda, SparseGPT, Thanos, ProxSparse, ADDM).
التحقق التجريبي: تقديم نتائج واسعة النطاق عبر عائلات نماذج LLaMA وGemma وQwen، مما يظهر تحسينات مستمرة في الدقة دون الحاجة لضبط دقيق.
النتائج
قام المؤلفون بتقييم OPTIMA على نماذج LLaMA 3.1/3.2، وGemma 2/3، وQwen 2.5 عبر أنظمة تقليم غير مهيكلة (50%، 60%) وشبه مهيكلة (2:4).
مكاسب الدقة: حقق OPTIMA تحسنًا في متوسط دقة "الصفر" (zero-shot accuracy) في 40 حالة من أصل 49 (نموذج، مختار قناع، درجة تقليم).
متوسط المكاسب: +0.61 نقطة عند تقليم 50% غير مهيكل، و +0.56 عند 2:4، و +1.01 عند تقليم 60% غير مهيكل.
أقصى مكسب: +3.97 نقطة (LLaMA 3.2 3B عند تقليم 60%).
في نماذج Qwen 2.5، وصلت المكاسب إلى +4.19 نقطة.
المقارنة مع النماذج المرجعية:
تفوق OPTIMA باستمرار على نموذج "عدم التحديث" المرجعي (Wanda).
تفوق على تحديثات SparseGPT وThanos في معظم الإعدادات، وإن كانت المكاسب أقل عندما كان النموذج المرجعي يقوم بالفعل بتحديثات الأوزان.
على أقنعتها الخاصة، طابق OPTIMA أو تجاوز قليلاً إعادة البناء الدقيقة القائمة على ADMM (Boža, 2024).
الارتباك (Perplexity) مقابل الدقة: يشير البحث إلى أن الارتباك ودقة المهام النهائية لا يترابطان دائمًا؛ حيث يعطي OPTIMA الأولوية لدقة المهام النهائية، وهي المقياس الأساسي لجودة التقليم في هذا السياق.
الكفاءة:
تقليم نموذج بـ 8 مليار معلمة على وحدة NVIDIA H100 واحدة يستغرق 40 ساعة (بذروة ذاكرة 60 جيجابايت).
تقليم نموذج بـ 70 مليار معلمة يستغرق حوالي 350 ساعة.
رغم أنه أبطأ من الطرق الاستدلالية (Wanda/SparseGPT) لكل طبقة، إلا أنه يظل تكلفة لمرة واحدة خارج الإنترنت (offline cost).
الأهمية والادعاءات
يدعي البحث أن إعادة البناء بالبرمجة التربيعية (QP) على مستوى الأعمدة هي بديل عملي لتحديثات الأوزان الاستدلالية في عملية التقليم بضربة واحدة بعد التدريب.
سد الفجوة: يغلق OPTIMA الفجوة بين الاستدلالات السريعة وغير الدقيقة والطرق الدقيقة غير الممكنة، وذلك عبر استغلال بنية مصفوفة الهسيان المشتركة لتمكين الحل المتوازي المعتمد على المسرعات.
لا يتطلب ضبطًا دقيقًا: تحقق هذه الطة هذه المكاسب دون الحاجة لضبط دقيق بعد التقليم، مما يجعلها تقنية بحتة لتقليل تكلفة الاستدلال.
القابلية للتوسع: من خلال حل المشكلة على مسرع واحد دون الحاجة لبنية تحتية للتدريب الموزع، يثبت OPTIMA أن إعادة البناء الدقيقة أمر ممكن للنماذج الكبيرة.
النمطية: منهج OPTIMA مستقل عن اختيار القناع، مما يعني أنه يمكنه تعزيز أي استراتيجية تقليم تعتمد على قناع ثابت.
يقر المؤلفون بوجود قيود، مشيرين إلى أن OPTA أبطأ من التحديثات ذات الصيغة المغلقة (closed-form updates)، وأن تقليل خطأ إعادة البناء على مستوى الطبقة لا يتوافق دائمًا بشكل مثالي مع أداء المهام النهائي. ومع ذلك، يجادلون بأن الطريقة توفر نهجًا قويًا ومؤسسًا نظريًا لتعظيم فائدة النماذج المقلمة.