Entropy Meets Importance: A Unified Head Importance-Entropy Score for Stable and Efficient Transformer Pruning
تقدم هذه الورقة البحثية HIES، وهو معيار تقليم موحد يجمع بين درجات أهمية الرأس (Head Importance Scores) وبين اعتلاج الانتباه (attention entropy) للتغلب على قيود الطرق القائمة على التدرج، محققاً تحسينات ملحوظة في جودة النموذج واستقراره من أجل ضغط نماذج المحولات (Transformer) بكفاءة.
المؤلفون الأصليون:Minsik Choi, Hyegang Son, Changhoon Kim, Young Geun Kim
تخيل نموذج "ترانسفورمر" (Transformer) -وهو العقل الكامن وراء الذكاء الاصطناعي الحديث- كأوركسترا ضخمة وعالية القدرة. تتكون هذه الأوركسترا من طبقات عديدة، وفي كل طبقة، يوجد عشرات الموسيقيين الذين يُطلق عليهم اسم "رؤوس الانتباه" (Attention Heads). يستمع كل رأس منها إلى المدخلات (مثل جملة معينة) ويقرر أي الكلمات هي الأهم للتركيز عليها.
الهدف من هذه الورقة البحثية هو جعل هذه الأوركسترا أصغر وأسرع (عملية تُسمى التقليم - Pruning) دون إفساد الموسيقى. التحدي يكمن في: أي الموسيقيين يمكننا الاستغناء عنهم، ومن يجب أن نحتفظ بهم؟
الطريقة القديمة: طريقة "مقبض الصوت" (HIS)
في السابق، استخدم الباحثون طريقة تسمى "درجة أهمية الرأس" (Head Importance Score - HIS). فكر في هذا كـ "مقبض صوت" أو "مقياس حساسية".
كيف كانت تعمل: كانت تقيس مدى ازدياد "علو" الموسيقى (أو سوئها) إذا توقف موسيقي معين عن العزف. إذا لم يغير غياب الموسيقي من مستوى الصوت كثيراً، يتم الاستغناء عنه.
المشكلة: كانت هذه الطريقة تقيس فقط مقدار مساهمة الرأس في النتيجة النهائية، وليس كيفية مساهمته.
التشبيه: تخيل موسيقياً يعزف "سولو" (منفرداً) محدداً للغاية على نوتة واحدة (مركز جداً). وموسيقياً آخر يعزف طنيناً خلفياً غامضاً ومشتتاً يملأ الغرفة بأكملها (منتشر). قد يقول "مقبض الصوت" إن كلاهما مهم بنفس القدر لأن كلاهما يؤثر على إجمالي الصوت. ولكن إذا استغنيت عن عازف السولو المركز، ستنهار الأغنية. أما إذا استغنيت عن عازف الطنين المشتت، فستبقى الأغنية جيدة. الطريقة القديمة لم تستطع التمييز بينهما، مما أدى إلى "هبوط حاد" في الجودة عند إزالة الكثير من الرؤوس.
الطالطريقة الجديدة: طريقة "التركيز والتنوع" (HIES)
يقترح المؤلفون درجة جديدة تسمى HIES (درجة أهمية وانتروبيا الرأس). وهي بمثابة إعطاء قائد الأوركسترا أداة ثانية: "مقياس التركيز" (الانتروبيا - Entropy).
الأداتان معاً:
الأهمية (HIS): ما مدى أهمية هذا الرأس للإجابة النهائية؟ (هذا هو "مستوى الصوت").
الانتروبيا (AE): ما مدى "تركيز" هذا الرأس؟ هل يركز بدقة على كلمات محددة وحاسمة، أم يوزع انتباهه في كل مكان؟ (هذا هو "التركيز").
التشبيه:
انتروبيا منخفضة (تركيز عالٍ): مثل القناص. إنه يصوب نحو هدف واحد محدد وحاسم (كلمة مفتاحية في الجملة). حتى لو لم يكن "صوته" عالياً، فهو حيوي لأنه يصيب الهدف بدقة.
انتروبيا عالية (انتشار): مثل الرشاش المائي (المرشة). هو يرش الماء في كل اتجاه. قد يغطي مساحة واس كبيرة، لكنه لا يصيب شيئاً محدداً.
تقول طريقة HIES: "لا تستغني عن الموسيقيين الهادئين فحسب. بل استغنِ عن الموسيقيين المشتتين (المرشات) إذا لم يضيفوا قيمة فريدة، ولكن احتفظ بالموسيقيين المركزين (القناصة)، حتى لو بدا صوتهم منخفضاً، لأنهم هم من يحافظون على تماسك البنية."
لماذا هذا مهم: الاستقرار
تزعم الورقة البحثية أنه باستخدام هذا المزيج الجديد من "التركيز + الصوت"، يصبح الذكاء الاصطناعي أكثر استقراراً.
مشكلة "الهبوط الحاد": مع الطريقة القديمة، إذا حاولت تقليص حجم الأوركسترا بنسبة 50%، فستصبح الموسيقى سيئة فجأة (هبوط حاد).
حل HIES: مع الطريقة الجديدة، يمكنك تقليص حجم الأوركسترا بنسبة 50% وتظل الموسيقى رائعة.
النتيجة: تظهر الورقة أن HIES يحسن جودة النموذج بنسبة تصل إلى 15.2% مقارنة بالطريقة القديمة، ويجعل النموذج أكثر استقراراً بمرتين (أقل عرضة للانهيار أو إعطاء إجابات غريبة عند تغير المدخلات قليلاً).
ملخص موجز
المشكلة: طرق التقليم القديمة كانت تستغني عن الموسيقيين الخطأ لأنها كانت تقيس "علو الصوت" فقط، وتتجاهل أهمية "التركيز".
الحل: الطريقة الجديدة (HIES) تقيس كلاً من "علو الصوت" و"التركيز".
النتيجة: يمكنك تصغير نموذج الذكاء الاصطناعي بشكل كبير (مما يجعله أسرع وأرخص في التشغيل) دون فقدان الدقة أو جعل النموذج غير مستقر. الأمر يشبه تقليم الشجرة دون قطع الأغصان التي تحمل الثمار.
ملخص تقني: التقاء الإنتروبيا بالأهمية (HIES)
1. بيان المشكلة
حققت النماذج القائمة على المحولات (Transformers)، وخاصة النماذج اللغوية الكبيرة (LLMs)، أداءً فائقاً، لكنها تعاني من تكاليف حوسبة وذاكرة كبيرة أثناء الاستدلال بسبب هياكلها العميقة وكثرة رؤوس الانتباه لديها. وبينما تقدم طرق التقليم الهيكلي، وتحديداً تقليم الرؤوس (Head Pruning)، حلاً عبر إزالة رؤوس الانتباه الزائدة، تواجه الأساليب الحالية قيوداً حرجة:
الاعتماد على درجات التدرج (HIS): تعتمد الطرق الحالية بشكل أساسى على درجات أهمية الرأس (HIS)، والتي تقيس مساهمة الرأس في دالة الخسارة عبر حساسية التدرج. ورغم فعاليتها عند مستويات التخلخل المتوسطة، إلا أن الطرق القائمة على HIS تظهر "هبوطاً حاداً" في الدقة عندما تصبح نسب التقليم عدوانية.
عدم الاستقرار والافتقار إلى التنوع: يفشل HIS في التقاط تنوع أنماط الانتباه. فهو يعامل الرأس الذي يمتلك انتباهاً مركزاً (يركز على الرموز الحاسمة) والرأس ذو الانتباه المنتشر معاملة متماثلة إذا كانت قيم التدرج لديهما متقاربة. يؤدي هذا إلى إزالة الرؤوس ذات الأهمية الوظيفية التي تعمل على استقرار النموذج، مما ينتج عنه ضعف في المتانة تجاه اضطرابات المدخلات وتحولات التوزيع.
القيود المستقلة عن الطبقات: غالباً ما تطبق المعايير الحالية قواعد تقليم موحدة عبر جميع الطبقات، متجاهلة الأدلة التي تشير إلى أن الطبقات المختلفة تتطلب سلوكيات انتباه متميزة، مما يؤدي إلى تقليم غير متوازن وتقلبات في الأداء.
2. المنهجية: درجة أهمية الرأس–الإنتروبيا (HIES)
يقترح المؤلفون HIES (درجة أهمية الرأس–الإنتروبيا)، وهو معيار تقليم موحد يدمج الأهمية القائمة على التدرج مع التوزيع الهيكلي للانتباه.
المكونات الأساسية
درجة أهمية الرأس (HIS):
تقيس الزيادة في الخسارة من الدرجة الأولى في حال تم حجب الرأس.
الصيغة: HISh=Ex∼D∂mh∂L(x)، حيث mh هو متغير الحجب للرأس h.
تلتقط مقدار مساهمة الرأس في الخسارة ولكنها تتجاهل أنماط توزيع الانتباه.
إنتروبيا الانتباه (AE):
تقيس مدى تركيز أو تشتت أوزان انتباه الرأس عبر رموز المدخلات.
الصيغة: AEh=−∑i=1npi(h)logpi(h).
تشير الإنتروبيا المنخفضة إلى انتباه مركز (غالباً على الرموز الحاسمة)، بينما تشير الإنتروبيا العالية إلى انتباه منتشر. ويفترض المؤلفون أن الرؤوس ذات الإنتروبيا المنخفضة ضرورية للاستقرار ومنع "انهيار الإنتروبيا".
مقياس HIES:
يجمع HIES بين HIS وAE المعايرين في درجة واحدة: HIESh=α⋅HISh+(1−α)⋅(1−AEh)
المعايرة: يتم معايرة كل من HIS وAE باستخدام (min-max) لتكون في النطاق [0,1] لضمان قابلية المقارنة عبر المقاييس المختلفة.
الأوزان:α هو معلمة فائقة قابلة للضبط. يضمن المصطلح (1−AEh) أن الرؤوس ذات الإنتروبيا الأقل (الأكثر تركيزاً) تحصل على درجات أعلى، مما يعطي الأولوية للاحتفاظ بها.
التبرير النظري
يقدم البحث تحليلاً لتفكيك المخاطر يوضح أن:
HIS يتحكم في حد زيادة الخسارة (تقليل تدهور الدقة الفوري).
AE يضع حدوداً لـ فجوة التعميم (تقليل عدم الاستقرار والحساسية لتحولات التوزيع).
التعامد: ثبت أن تدرجات HIS وAE متعامدة في التوقع، مما يعني أنهما يلتقطان إشارات متكاملة: مقدار المساهمة مقابل تشتت الانتباه. إن تقليل حد المخاطر المركب عبر HIES يضمن نظرياً استقراراً أفضل وحفاظاً على الدقة.
3. المساهمات الرئيسية
معيار تقليم مبتكر: تقديم HIES، الذي يوحد الأهمية القائمة على التدرج مع إنتروبيا الانتباه لمعالجة ظاهرة "الهبوط الحاد" في حالات التقليم العدوانية.
التركيز على الاستقرار: صياغة إنتروبيا الانتباه صراحة كوكيل لاستقرار الاستدلال والمتانة، مع الإشارة إلى أن الحفاظ على رؤوس ذات إنتروبيا منخفضة يمنع انهيار التنوع الهيكلي.
إطار نظري: يوفر حداً نظرياً للمخاطر يربط تقليم الرؤوس بكل من زيادة الخسارة وفجوات التعميم، مما يبرر تعامد وتكامل HIS وAE.
التقليم المتكيف مع الطبقات: يوضح أن HIES يتيح تقليماً أكثر توازناً عبر الطبقات مقارنة بالأنماط الصاعدة (bottom-up) التي تظهر غالباً في طرق HIS فقط.
4. النتائج التجريبية
قام المؤلفون بتقييم HIES على نماذج BERT-base، و LLaMA-2-7B، و ViT-Large، و LLaVA-1.5-7B عبر مهام متنوعة في معالجة اللغات الطبيعية، والرؤية، والمهام متعددة الوسائط (مثل GLUE، HellaSwag، ImageNet، VizWiz-VQA).
جودة النموذج:
في BERT-base، حسن HIES متوسط جودة النموذج بنسبة 8.21% مقارنة بـ HIS عبر نسب التقليم.
عند نسبة تقليم 50%، حقق HIES تحسناً يصل إلى 15.2% في جودة النموذج مقارم بأفضل خط أساس (baseline) أداءً.
في LLaMA-2-7B، حسن HIES الدقة بما يصل إلى +10.54% مقارنة بـ HIS.
في ViT-Large، عند تخلخل بنسبة 20%، حقق HIES دقة أعلى بنسبة 49.82% من HIS، مما منع انهيار الدقة الملاحظ في الطرق القائمة على التدرج فقط.
الاستقرار:
أظهر HIES تحسناً في الاستقرار يصل إلى 2.04 ضعفاً (مقارنة بالنموذج غير المقلم) مقارنة بطرق HIS فقط.
عند تقليم بنسبة 50% في BERT، أظهر HIES زيادة في الاستقرار بنسبة 8.65% عن HIS.
القدرة على التعميم:
تفوق HIES باستمرار على خطوط الأساس (Random, L2-Norm, HIS, Attention Deficit, LLM-Pruner, SliceGPT) عبر مهام معالجة اللغات الطبيعية، والرؤية، والاستدلال المعقد متعدد الوسائط.
نجح HIES في تأخير نظام "الهبوط الحاد" إلى مستويات تخلخل أكثر عدوانية، محافظاً على الأداء حيث فشل HIS.
5. الأهمية والادعاءات
يزعم البحث أن HIES يقدم حلاً أكثر عملية ومتانة لنشر نماذج المحولات في البيئات محدودة الموارد. من خلال الحفاظ على كل من الدقة والاستقرار حتى في حالات الضغط العدوانية، يعالج HIES العائق الحرج المتمثل في تقلبات الأداء غير المتوقعة في عمليات النشر الواقعية.
يؤكد المؤلفون أن عملهم يوفر اتجاهاً مبدئياً للتقليم المستقر والفعال، متجاوزاً مجرد حساسية التدرج لدمج التنوع الهيكلي (الإنتروبيا) كمكون أساسي لضغط النموذج. ويؤكدون أن هذا النهج قابل للتطبيق على نطاق واسع عبر بنيات مختلفة (BERT، LLaMA، ViT، LLaVA) وأنماط مختلفة، مما يوفر مساراً موثوقاً لنشر نماذج LLM فعالة دون التضحية بالمتانة.