Reservoir of Importance: Learning Semi-Structured Sparsity with Differentiable Subset Sampling
تقترح الورقة البحثية "خزان الأهمية" (Reservoir of Importance - RoI)، وهو إطار عمل للتقليم شبه المهيكل خفيف الوزن يستخدم أخذ عينات المجموعات الفرعية القابلة للتفاضل لتعلم أقنعة التناثر مع تقليل كبير في عدد المعلمات وعبء الذاكرة، مما يتيح النشر القابل للتوسع والفعال للنماذج اللغوية الكبيرة.
المؤلفون الأصليون:Ha Dinh, Xuan Duy Ta, Khoat Than, Khac-Hoai Nam Bui
تُعد النماذج اللغوية الكبيرة هي المحركات الكامنة وراء العديد من أدوات الذكاء الاصطناعي الأكثر تقدمًا اليوم، وهي قادرة على كتابة القصص، وحل المشكلات، والإجابة على الأسئلة المعقدة. ومع ذلك، فإن هذه النماذج ضخمة، حيث تحتوي غالبًا على مليارات الأرقام التي تمثل معرفتها. وهذا الحجم الهائل يجعل من الصعب تشغيلها على أجهزة الكمبيوتر القياسية، مما يتطلب أجهزة باهظة الثمن وطاقة كبيرة. ولجعل هذه الأنظمة أكثر عملية، سعى الباحثون منذ فترة طويلة لإيجاد طرق لتقليص حجمها دون فقدان ذكائها. وتتضمن إحدى الاستراتيجيات الواعدة عملية "التقليم" (pruning)، وهي عملية تحديد وإزالة الأرقام الأقل أهمية داخل النموذج. وبينما يؤدي حذف الأرقام العشوائية غالبًا إلى تعطل النموذج، فإن هناك نهجًا أكثر دقة يسمى "التخلخل شبه المنظم" (semi-structured sparsity) يقوم بإزالة الأرقام في أنماط محددة ومنتظمة. يحافظ هذا الأسلوب على هيكل النموذج سليمًا بما يكفي للعمل مع شرائح الكمبيوتر الحالية، مما يوفر مسارًا لذكاء اصطناعي أسرع وأكثر كفاءة.
وعلى الرغم من الإمكانات التي توفرها تقنية التقليم هذه، إلا أن عقبة رئيسية ظلت قائمة: وهي أن تحديد الأرقام التي يجب إزالتها بدقة أمر صعب للغاية. حاولت الأساليب السابقة حل هذه المشكلة عبر معاملة كل نمط محتمل للإزالة كخيار منفصل، مما يعني مطالبة الكمبيوتر بتعلم قاعدة فريدة لكل مجموعة من الأرقام. ومع زيادة حجم هذه النماذج، أصبح هذا النهج غير قابل للإدارة، حيث تطلب قدرًا هائلاً من الذاكرة وقوة الحوسبة لدرجة جعلت من المستحيل تطبيقه على أكبر النماذج. وقد طور الباحثون وراء دراسة جديدة بعنوان "مستودع الأهمية" (Reservoir of Importance) طريقة مختلفة للتعامل مع هذه المشكلة؛ حيث اقترحوا طريقة تتعلم كيفية اختيار أفضل الأرقام للاحتفاظ بها عن طريق أخذ عينات منها بطريقة سلسة ومستمرة، بدلاً من محاولة حفظ كل تركيبة ممكنة.
اختبر الفريق نهجهم الجديد، الذي أطلقوا عليه اسم "مستودع الأهمية"، على عائلة من النماذج اللغوية الكبيرة تتراوح أحجامها من الصغيرة إلى الكبيرة جدًا. وبدلاً من بناء خريطة معقدة لكل نمط تقليم محتمل، تعاملت طريقتهم مع عملية الاختيار مثل سحب عدد محدد من العناصر من مجموعة دون إعادتها. تخيل أن لديك حقيبة من الكرات الزجاجية وتحتاج إلى اختيار اثنتين من كل أربع كرات للاحتفاظ بهما، ولكنك تريد اختيار الأفضل بينهما بناءً على مدى أهميتهما. كانت الأساليب القديمة ستحاول حساب الاحتمالات لكل طريقة ممكنة لاختيار هاتين الكرتين، وهي مهمة تزداد تعقيدًا بشكل هائل كلما كبر حجم الحقيبة. وفي المقابل، تقوم الطريقة الجديدة بتعيين درجة بسيطة لكل كرة زجاجية ثم تستخدم خدعة رياضية ذكية لاختيار أفضل اثنتين. وتسمح هذه الخدعة للكمبيوتر بتعلم الدرجات التي تعمل بشكل أفضل من خلال تعديلها قليًّا أثناء التدريب، تمامًا مثل ضبط جهاز الراديو للعثور على أوضح إشارة.
أدى هذا التحول في الاستراتيجية إلى فوائد فورية؛ فقد وجد الباحثون أن طريقتهم الجديدة تتطلب عددًا أقل بكثير من الإعدادات القابلة للضبط لتعلم أنماط التقليم. فبالنسبة لنمط شائع يتم فيه الاحتفاظ باثنين من كل أربعة أرقام، استخدمت الطريقة الجديدة حوالي ثلث عدد المعلمات (parameters) القابلة للتعلم أقل من النهج الرائد السابق. هذا الانخفاض يعني أن النظام احتاج إلى ذاكرة أقل بكثير للتشغيل، مما جعل من الممكن تدريبه على نماذج أكبر بكثير دون نفاد موارد الكمبيوتر. وعندما اختبروا النتائج، وجدت الطريقة أن النماذج التي تم تقليمها بهذه الطة الجديدة قدمت أداءً يضاهي، أو يتفوق قليلاً على، تلك التي تم تقليمها بالتقنيات القديمة. لقد حافظت على دقة عالية في مهام متنوعة، من الإجابة على أسئلة الاختيار من متعدد إلى التنبؤ بالكلمة التالية في الجملة، مع استخدام قدر أقل بكثير من القدرة الحوسبية للوصول إلى ذلك.
كما بحثت الدراسة فيما يحدث عندما يصبح التقليم أكثر عدوانية، مثل الاحتفاظ برقمين فقط من كل ثمانية. في هذه الحالات القصوى، غالبًا ما تفشل الأساليب القديمة التي تعتمد على قواعد بسيطة أو درجات أهمية ثابتة، مما يؤدي إلى فقدان النموذج لقدرته على فهم اللغة. ومع ذلك، استمرت الطريقة الجديدة في العمل بفعالية؛ فقد نجحت في تحديد الأرقام الصحيحة للاحتفاظ بها حتى في ظل هذه الظروف القاسية، مما أثبت أن تعلم نمط التقليم مباشرة هو أكثر قوة ومتانة من التخمين بناءً على قواعد ثابتة. ولاحظ الباحثون أنه كلما غدوا النموذج بمزيد من بيانات التدريب، استمر أداؤه في التحسن بثبات، بينما كانت الطرق الأخرى تميل إلى الوصول إلى سقف لا يساعد فيه إضافة المزيد من البيانات على تحسين الأداء.
وبينما تبدو النتائج واعدة، يشير الباحثون إلى أن الاستخدام العملي لهذه التكنولوجيا يعتمد بشكل كبير على جهاز الكمبيوتر الذي تعمل عليه. فالأنماط المحددة التي تستخدمها النماذج مصممة لتعمل بكفاءة على أنواع معينة من معالجات الرسوميات الحديثة، وهي شائعة في الحوسبة عالية الأداء ولكنها ليست متوفرة في جميع الأجهزة. وإذا كان الكمبيوتر يفتقر إلى هذا الدعم المحدد، فقد لا تتحقق فوائد السرعة، حتى لو كان النموذج أصغر حجمًا. ومع ذلك، فإن هذا العمل يوفر مسارًا واضحًا للمضي قدمًا في جعل نماذج الذكاء الاصطناعي الكبيرة أكثر كفاءة. فمن خلال تبسيط كيفية تعلم الكمبيوتر لتقليم نفسه، أظهر الباحثون أنه من الممكن تقليص حجم هذه الأنظمة الضخمة دون التضحية بذكائها، مما يمهد الطريق لأدوات ذكاء اصطناعي أكثر قوة وسهولة في الوصول إليها في المستقبل.
ملخص تقني: خزان الأهمية (Reservoir of Importance - RoI)
بيان المشكلة
برزت الندرة شبه المهيكلة بنمط N:M كتقنية بالغة الأهمية لتسريع النماذج اللغوية الكبيرة (LLMs) من خلال فرض أنماط ندرة صديقة للأجهزة (مثل 2:4 أو 4:8)، والتي تعمل على إزالة الأوزان الزائدة مع الحفاظ على الانتظام المتوافق مع المسرعات الحديثة. ومع ذلك، تواجه الأساليب القائمة على القناع القابل للتعلم (learnable-mask approaches) عقبة كبيرة في التوسع. فأساليب مثل MaskLLM وHyperPrune تقوم بنمذجة اختيار قناع الندرة كتوزيع فئوي (categorical distribution) عبر جميع الأنماط الممكنة ضمن مجموعة مكونة من M من المعلمات. يتطلب هذا توصيف توزيع متعدد الحدود (multinomial distribution) بحجم (NM)، مما يؤدي إلى تعقيد توافقي. ومع زيادة أحجام النماذج وشدة الندرة، تسبب هذه الطريقة عبئاً كبيراً في عدد المعلمات والذاكرة، مما يحد من قابليتها للتطبيق على النماذج واسعة النطاق وأنظمة الندرة الشديدة.
المنهجية: خزان الأهمية (RoI)
يقترح المؤلفون خزان الأهمية (RoI)، وهو إطار عمل خفيف الوزن لتعلم أقنعة الندرة شبه المهيكلة من خلال أخذ عينات فرعية قابلة للتفاضل (differentiable subset sampling). بدلاً من نمذجة التوزيع الفئوي الكامل لجميع الأنماط الممكنة، يعيد RoI صياغة المشكلة كعملية أخذ عينات لعدد N من الأوزان المهمة من بين M من المرشحين دون إرجاع.
الآليات الجوهرية
توصيف اللوجيت المدمج (Compact Logit Parameterization): يستبدل RoI المعلمات (NM) المطلوبة في الطرق السابقة بمتجه واحد من M من اللوجيتات القابلة للتعلم (ϕ). وهذا يقلل من تعقيد المعلمات القابلة للتعلم من O((NM)) إلى O(M) لكل مجموعة.
أخذ العينات الفرعية القابل للتفاضل (Differentiable Subset Sampling): لتمكين التحسين القائم على التدرج (gradient-based optimization)، يستخدم RoI عملية من خطوتين لأخذ عينات لـ N من العناصر من M من اللوجيتات دون إرجاع:
أخذ عينات الخزان الموزون (Weighted Reservoir Sampling - WRS): من الناحية النظرية، تتبع الطريقة توزيع WRS، حيث يتم أخذ عينات من العناصر باحتمالية تتناسب مع أوزانها.
خدعة Gumbel-Top-K: لجعل عملية أخذ العينات قابلة للتفاضل، يستخدم RoI استرخاء Gumbel-Top-K. حيث يقوم بتعديل اللوجيتات باستخدام ضجيج Gumbel لإنشاء مفاتيح عشوائية، ثم يختار العناصر الـ K العليا بشكل متكرر. يتم استرخاء عملية الاختيار باستخدام سلسلة من عمليات softmax مع معامل درجة حرارة τ، مما يسمح للتدرجات بالتدفق عائدًا إلى اللوجيتات.
التحسين التبايني العشوائي (Stochastic Variational Optimization): تتم صياغة مهمة التقليم (pruning) كتقليل لمتوسط خسارة الإنتروبيا المتقاطعة (cross-entropy loss) عبر التوزيع التبايني للأقنعة. ومن خلال أخذ عينات الأقنعة عبر عملية Gumbel-Top-K القابلة للتفاضل، يتعلم الإطار درجات الأهمية (exp(ϕ)) التي توجه اختيار الأوزان البارزة.
تخميد درجة الحرارة (Temperature Annealing): تستخدم الطريقة جداول تخميد لكل من درجة حرارة الـ softmax (τ) وعامل القياس للوجيتات (λ). تبدأ هذه الاستراتيجية بعشوائية عالية لتشجيع الاستكشاف الواسع لمساحة البحث، وتتجه تدريجيًا نحو اختيارات أقنعة حتمية ذات ثقة عالية مع تقدم التدريب.
المساهمات الرئيسية
تعلم الأقنعة القابل للتوسع: يقدم RoI توصيفاً يقلل عدد المعلمات القابلة للتعلم من التعقيد التوافقي إلى التعقيد الخطي (O(M))، مما يجعله قابلاً للتطبيق لتعلم الأقنعة للنماذج الكبيرة وأنماط الندرة الشديدة (مثل 2:8).
كفاءة الذاكرة: من خلال تجنب تخزين التوزيعات الفئوية الكاملة، يقلل RoI تكاليف الذاكرة بشكل كبير. بالنسبة لنمط ندرة 2:4، يتطلب MaskLLM 1.5 ضعف عدد معلمات النموذج الأصلي للأقنعة القابلة للتعلم، مقارنة بـ 1 ضعف لـ RoI. وبالنسبة لندرة 2:8، يتطلب MaskLLM 3.5 ضعف عدد معلمات النموذج الأصلي، مقارنة بـ 1 ضعف لـ RoI.
أخذ العينات الفرعية القابل للتفاضل: تثبت الورقة أن أخذ عينات من المجموعات الفرعية دون إرجاع عبر Gumbel-Top-K هو آلية كافية وفعالة لاكتشاف أنماط ندرة عالية الجودة، متفوقة على الأساليب الحتمية القائمة على التنظيم (regularization-based approaches) في الإعدادات واسعة النط scale.
النتائج التجريبية
تم تقييم الطريقة على عائلة النماذج اللغوية الكبيرة Qwen2.5 (من 0.5 مليار إلى 7 مليارات معلمة) باستخدام نمط ندرة 2:4، مع إجراء اختبارات جهد على ندرة 2:8.
الأداء: يحقق RoS باستمرار دقة (zero-shot accuracy) ودرجة تعقيد (perplexity - PPL) تنافسية أو متفوقة مقارنة بالنماذج المرجعية بما في ذلك Magnitude وWanda وSparseGPT وProxSparse وMaskLLM.
في نموذج 7B مع ندرة 2:4، حقق RoI متوسط دقة قدره 64.54% وPPL لـ WikiText-2 بلغ 14.89، متفوقًا على MaskLLM (64.44% / 14.91) ومتجاوزًا بشكل كبير الأساليب القائمة على البروز (saliency-based methods).
في نظام الندرة الشديدة 2:8، حافظ RoI على درجة تعقيد معقولة (32.33 لنموذج 7B)، بينما انهارت النماذج المرجعية القائمة على البروز (PPL > 100).
القابلية للتوسع: على عكس ProxSparse، الذي فشل في التقارب على نموذج 3B، أظهر RoI استقرارًا في التحسين عبر جميع أحجام النماذج.
كفاءة التدريب: يتطلب RoI فقط 1 ضعف عدد معلمات النموذج الأصلي لتعلم القناع (الأوزان مجمدة)، بينما يتطلب MaskLLM 1.5 ضعف لندرة 2:4 و3.5 ضعف لندرة 2:8. وينتج عن ذلك توفير كبير في تخصيص الذاكرة، وحالات المحسن (optimizer states)، وتكالوات الانتشار العكسي (backpropagation).
كفاءة البيانات: مع زيادة عدد التوكينات (tokens) أثناء التدريب، يظهر كل من RoI وMaskLLM تحسنًا رتيبًا، بينما تصل الأساليب القائمة على التنظيم (مثل ProxSparse) إلى مرحلة الثبات مبكرًا، مما يسلط الض الضوء على فائدة تعلم القناع التبايني المباشر.
الأهمية والادعاءات
تضع الورقة RoI كـ حل عملي وقابل للتوسع للتقليم شبه المهيكل للنماذج اللغوية الكبيرة. تكمن أهميته الأساسية في التوفيق بين الكفاءة والقابلية للتوسع:
فهو يتيح نشر أنماط ندرة شديدة (مثل 2:8) على النماذج الكبيرة دون العبء المرهق للذاكرة الذي تفرضه طرق تعلم الأقمة السابقة.
يوفر "صياغة الحد الأدنى للقناع القابل للتعلم" التي تقلل من فقدان الدقة مع تقليل العبء الحسابي لعملية التقليم نفسها بشكل جذري.
يزعم المؤلفون أن RoI يوفر مسارًا قابلًا للتطبيق نحو نشر النماذج اللغوية الكبيرة بكفاءة، لا سيما في البيئات محدودة الموارد حيث يكون التوازن بين كفاءة الضغط والحفاظ على الأداء أمرًا بالغ الأهمية.
القيود
يقر المؤلفون بوجود قيود محددة:
الاعتماد على الأجهزة: الفائدة العملية للندرة شبه المهيكلة محدودة حاليًا بالمنصات التي تدعم نوى (kernels) أصلية (مثل وحدات معالجة الرسوميات NVIDIA Ampere/Hopper وAMD ROCm المحددة). على الأجهزة التي تفتقر إلى دعم 2:4، تكون سرعة الاستدلال هامشية.
نطاق التقييم: تقتصر التجارب حاليًا على النماذج الكثيفة المتمحورة حول اللغة الإنجليزية ومعايير معالجة اللغات الطبيعية القياسية. تظل القدرة على التعميم للنماذج متعددة اللغات، وبنيات خليط الخبراء (MoE)، والمهام المتخصصة مجالًا مفتوحًا للبحث المستقبلي.