ضغط النماذج2015متوسط10 دقيقة قراءة

تقطير المعرفة في الشبكات العصبية

Distilling the Knowledge in a Neural Network

Hinton, G. · Vinyals, O. · Dean, J. — NeurIPS Deep Learning Workshop

المشكلة

تجميع عدة شبكات عصبية عميقة معاً يحقق أعلى دقة ممكنة، لكن نشر كل هذه النماذج الضخمة في بيئة إنتاجية أمر غير عملي — الاستجابة بطيئة، والتكلفة مرتفعة، واستهلاك الذاكرة يتجاوز قدرة الأجهزة المحمولة والأنظمة الآنية. في المقابل، نموذج صغير واحد من الصفر على البيانات نفسها يُنتج تعميماً أضعف بوضوح. فكيف نستخلص المعرفة المتراكمة في مجموعة نماذج ضخمة ونضغطها في نموذج واحد صغير يصلح للنشر الفعلي؟

الإسهام

: ندرّب شبكة صغيرة (الطالب) على محاكاة المخرجات الاحتمالية المرنة لنموذج كبير (المعلم) أو مجموعة نماذج، وذلك برفع حرارة دالة softmax. عند الحرارة المرتفعة تكشف مخرجات المعلم عن «» — أي الاحتمالات النسبية للفئات الخاطئة — وهي تحمل بنية التشابه بين الفئات التي تضيع تماماً في التسميات الصارمة. يتدرّب الطالب على مجموع موزون من هدفين: مطابقة أهداف المعلم المرنة عبر ، ومطابقة التسميات الحقيقية عبر الإنتروبيا التقاطعية. تقدّم الورقة أيضاً فكرة النماذج المتخصصة — مجموعات فرعية تُعنى بالفئات التي يسهل الخلط بينها — وتُثبت أنه يمكن تقطير هذه المجموعة المتخصصة في طالب عام واحد.

الأثر

أصبح تقطير المعرفة الأسلوب المعياري . استخدمه DistilBERT لضغط نموذج BERT إلى 60% من حجمه مع الاحتفاظ بـ97% من أدائه، وطبّقه DeiT لتدريب محوِّلات الرؤية دون الحاجة لمجموعات بيانات ضخمة. ، والتقطير الذاتي، وفك الترميز التخميني — كلها تعود فكرياً إلى هذه الورقة. أثبتت الورقة أن معرفة الشبكة المُدرَّبة ليست في أوزانها — بل في التوزيع الاحتمالي على مخرجاتها، بما في ذلك الإجابات «الخاطئة».

تخيّل أستاذاً خبيراً يُهيّئ مُحاضراً جديداً لتولّي مادته. بدل أن يُعطيه الكتاب المقرر فقط، يُسلّمه أوراق اختبارات مُصحَّحة مع ملاحظاته: «هذا الطالب كتب 7، لكن لاحظ أن الخط يميل نحو 1، والانحناءة العلوية تُشبه 9.»

هذه التفاصيل الدقيقة — الدرجات الجزئية وحالات الشبه — تحمل فهماً أعمق بكثير من مجرد مفتاح إجابات. تقطير المعرفة يعمل بنفس المبدأ: الكبير المُدرَّب يمرّر أحكامه المرنة إلى الصغير، فيتعلّم ليس الإجابات فحسب، بل الحدس الكامن وراء حالات الغموض.

المشكلة: النماذج الكبيرة أثقل من أن تُنشر

بحلول عام 2015، كانت أفضل الأنظمة أداءً على مقاييس مثل ImageNet عبارة عن — عادةً من 5 إلى 10 شبكات عميقة تُدرَّب بشكل مستقل ثم تُحسب المتوسط بين تنبؤاتها. هذا الأسلوب القائم على القوة الحوسبية المحضة كان يتفوق باستمرار على أي نموذج منفرد.

لكن هناك فجوة حادة بين مرحلة التدريب ومرحلة النشر. في التدريب نريد استخلاص أقصى قدر من المعرفة من البيانات، ولا تهمّنا التكلفة الحوسبية ولا زمن الاستجابة. أما في النشر، فنحتاج نموذجاً يستجيب خلال أجزاء من الثانية على هاتف محمول أو يخدم ملايين الطلبات في الثانية. عشر شبكات ضخمة مجتمعة لا تستطيع تحقيق أيٍّ من ذلك.

الحل البديهي — تدريب نموذج صغير مباشرةً على البيانات — يُعطي أضعف دائماً. النموذج الصغير لا يرى إلا التسميات الصارمة («هذا 3»)، ويفوّت البنية الغنية التي اكتشفتها المجموعة التجميعية: أن هذا الـ3 تحديداً فيه شبه بالـ8 ولا يُشبه الـ7 إطلاقاً.

افتح في المختبر
قارن بين التسميات الصارمة (أحادية القيمة) والتسميات المرنة من المعلم. لاحظ كيف تحتفظ التسميات المرنة ببنية التشابه بين الفئات.
تستيقظ التجربة عند وصولك…

المعرفة الخفية: ما تُخبرنا به الإجابات الخاطئة

حين يرى نموذج مُدرَّب على تمييز الأرقام صورةً للرقم 2، قد يُخرج شيئاً مثل: «احتمال أنه 2 = 0.93، واحتمال أنه 3 = 0.05، واحتمال أنه 7 = 0.01، والباقي قريب من الصفر.» التسمية الصارمة تقول فقط «هذا 2». لكن المخرج المرن يحمل معلومات أغنى بكثير — يقول إن النموذج يلمح شبهاً بين 2 و3 (كلاهما فيه انحناءات)، وشبهاً خافتاً بالـ7 (بسبب الخط المفتوح في الأعلى)، ولا يكاد يرى أي علاقة بالـ0 أو الـ1.

يُسمّي Hinton هذه النسب الاحتمالية الدقيقة بين الفئات غير الصحيحة المعرفة الخفية. هذه النسب تُرمّز علاقات بنيوية بين الفئات — علاقات تُضيّعها التسميات الصارمة بالكامل. فالرقم 2 الذي يكاد يُشبه الـ3 يحمل معلومات أكثر من رقم 2 لا يُشبه أي شيء آخر.

المشكلة أن مخرجات لنموذج مُدرَّب جيداً تكون حادّة للغاية — الفئة الصحيحة تأخذ احتمالاً يقترب من 1.0، وكل تلك المعلومات العلائقية الغنية تنضغط في احتمالات ذيلية ضئيلة يكاد يتجاهلها الحساب. فكيف نستخرج هذه البنية المخفية؟

الحرارة: تليين توزيع softmax

الآلية المحورية هي TT الذي يُطبَّق على دالة softmax. عند القيمة الافتراضية (T=1T = 1) يكون التوزيع الناتج حاداً — أكبر يُهيمن على المخرج. كلما رفعنا TT يلين التوزيع: كل الفئات تحصل على حصص احتمالية أكثر تقارباً، وتبدأ الفروق النسبية بين احتمالات الفئات الخاطئة بالظهور.

تخيّل TT كعدسة مكبّرة: عند T=1T = 1 ترى الصورة بالعين المجردة فلا تُميّز إلا الشكل الأبرز (الفئة الصحيحة). ارفع TT وستبدأ التفاصيل الدقيقة بالظهور — تلك الفروق الخفية بين الفئات (المعرفة الخفية) التي كانت غير مرئية في المقياس العادي.

qi=exp(zi/T)jexp(zj/T)q_i = \frac{\exp(z_i / T)}{\sum_j \exp(z_j / T)}
Softmax مع الحرارةتتحكّم الحرارة في مقدار حدّة التوزيع الاحتمالي الناتج أو انبساطه. الحرارة المرتفعة توزّع الاحتمال بشكل أكثر تساوياً بين الفئات، فتكشف أوجه التشابه النسبية بينها. أما الحرارة المنخفضة فتُركّز الاحتمال على الفئة الأرجح وتُنتج تنبؤات أكثر حسماً. دالة softmax القياسية هي ببساطة الحالة الخاصة عند T = 1.
افتح في المختبر
حرّك شريط الحرارة وراقب كيف يتغيّر شكل التوزيع الاحتمالي. لاحظ كيف تكشف الحرارة المرتفعة عن البنية المخفية بين الفئات.
تستيقظ التجربة عند وصولك…

هدف التقطير: فقدان مزدوج وطالب واحد

يُدرَّب نموذج الطالب على مزيج موزون من حدَّي :

الفقد المرن (فقد التقطير): وهو تباعد KL بين المخرج المرن لـالمعلم (عند حرارة TT) والمخرج المرن للطالب (عند الحرارة نفسها). هذا الحد ينقل المعرفة الخفية — أي لا يُخبر الطالب بالإجابة الصحيحة فحسب، بل يُخبره أيضاً بمدى شبه كل عينة بالفئات الأخرى.

الفقد الصارم (الفقد المعياري): وهو بين مخرج الطالب (عند T=1T=1) والتسميات الحقيقية. هذا الحد يُثبّت الطالب على الإجابات الصحيحة فعلاً.

الفقد الكلي هو:

L=αT2KL ⁣(σ(zT/T)    σ(zS/T))+(1α)CE ⁣(y,  σ(zS))\mathcal{L} = \alpha \cdot T^2 \cdot \text{KL}\!\left(\sigma(z_T / T) \;\|\; \sigma(z_S / T)\right) + (1 - \alpha) \cdot \text{CE}\!\left(y,\; \sigma(z_S)\right)
فقد التقطير المُركَّبيجمع هذا الهدف بين إشارتَي تعلُّم. الأولى تدفع نموذج الطالب لمحاكاة توزيع مخرجات المعلم، فيتعلّم العلاقات الدقيقة بين الفئات بدلاً من الاكتفاء بالإجابة الصحيحة. والثانية تُدرّبه على التسميات الحقيقية الموجودة في البيانات. يتحكّم معامل الموازنة في مقدار اعتماد الطالب على كل مصدر، بينما تجعل الحرارة تنبؤات المعلم أكثر انبساطاً فتكشف معلومات أغنى عن درجة ثقته وعلاقات الفئات ببعضها.
افتح في المختبر
تتبّع مسار البيانات من المعلم إلى الطالب. انقر على كل مرحلة لمعرفة ما يحدث فيها.
تستيقظ التجربة عند وصولك…

لماذا ينجح التقطير: إشراف غني بالمعلومات

الفكرة الجوهرية تتعلّق بـكثافة المعلومات في كل عينة تدريب. التسمية الصارمة تُعطي بتّة معلومات واحدة: أي فئة هي الصحيحة. أما المتجه الاحتمالي المرن من المعلم فيُعطي بتّات عديدة: مدى قرب المُدخَل من كل فئة، كما اكتشفه نموذج سبق له أن عمّم جيداً.

هذا يعني أن الطالب يحصل على إشارة تدرج أغنى بكثير لكل عينة. أظهر Hinton أنه مع الأهداف المرنة يمكن التدريب بعدد عينات أقل بكثير أو أقل مما يتطلّبه التدريب على التسميات الصارمة وحدها. على مجموعة MNIST مثلاً، تفوّق نموذج مُقطَّر استخدم 3% فقط من بيانات التدريب (منقولة من المعلم) على نموذج تدرّب على 100% من البيانات بتسميات صارمة.

هناك أيضاً أثر تنظيمي. التسميات المرنة تمنع الطالب من الإفراط في الثقة بفئة واحدة، لأن المعلم حتى لرقم 3 واضح يُعطي احتمالاً صغيراً لكنه غير صفري لأرقام متشابهة بصرياً. هذا يعمل عملياً كـ ضمني — يُعاقب التنبؤات المفرطة في الثقة — وهو شبيه جداً بالأسلوب الذي صِيغ لاحقاً تحت اسم تنعيم التسميات.

افتح في المختبر
اختر صوراً مختلفة لأرقام وشاهد ما تكشفه تنبؤات المعلم المرنة من تشابهات بين الفئات لا تظهر في التسميات الصارمة.
تستيقظ التجربة عند وصولك…

الوصفة: خطوة بخطوة

تمرّ عملية تقطير المعرفة بمراحل واضحة:

الخطوة 1 — تدريب المعلم. نبني نموذجاً كبيراً (أو مجموعة نماذج) وندرّبه حتى على كامل البيانات. هذا هو الجزء المُكلف حسابياً، ويُنفَّذ مرة واحدة فقط.

الخطوة 2 — توليد الأهداف المرنة. نمرّر بيانات التدريب عبر المعلم عند حرارة TT (عادةً بين 3 و20)، ونحفظ المتجهات الاحتمالية المرنة الناتجة.

الخطوة 3 — تدريب الطالب. ندرّب بنية أصغر على الفقد المُركَّب: مطابقة أهداف المعلم المرنة (تباعد KL عند حرارة TT) إلى جانب مطابقة التسميات الحقيقية (الإنتروبيا التقاطعية عند T=1T = 1).

الخطوة 4 — نشر الطالب عند T=1T = 1 لمرحلة . أصبح الطالب الآن يحمل قدرة المعلم على التعميم في عدد أقل بكثير من .

افتح في المختبر
شاهد عملية تدريب الطالب. منحنى الفقد يوضّح كيف يتقارب النموذج أسرع مع الأهداف المرنة مقارنةً بالتسميات الصارمة وحدها.
تستيقظ التجربة عند وصولك…

النماذج المتخصصة: توزيع مناطق الالتباس

مهام التصنيف الكبيرة (مثل JFT التي تضم 15,000 فئة) تحتوي على عناقيد من الفئات التي يسهل الخلط بينها — سلالات كلاب مختلفة، أو طرازات سيارات، أو أنواع فطر. المعلم العام يُهدر جزءاً من سعته في التفريق بين الراعي الألماني والهاسكي السيبيري، بينما نموذج فرعي متخصص يستطيع ذلك بكفاءة أعلى.

يقترح Hinton ما يسمّيه : شبكات صغيرة تُدرَّب كل منها على عنقود من الفئات المتشابهة (مثل «جميع سلالات الكلاب» أو «جميع أنواع الفطر»)، مع إضافة عينة عشوائية من فئات أخرى لمنع . عند الاستدلال، يُحدّد النموذج العام أولاً أي متخصص يُناسب المُدخَل، ثم يتولّى المتخصص صقل التنبؤ.

النتيجة الأبرز: مجموعة مكوّنة من معلم عام واحد مع عدة متخصصين قُطّرت في طالب واحد تفوّق على مجموعة مماثلة من نماذج عامة فقط — بحوسبة إجمالية أقل بكثير.

افتح في المختبر
استكشف كيف تتعامل النماذج المتخصصة مع عناقيد الفئات المتشابهة، وكيف تُدمج معارفها أثناء التقطير.
تستيقظ التجربة عند وصولك…

النتائج: نموذج صغير بأداء كبير

على مجموعة MNIST، طابق نموذج مُقطَّر واحد المجموعة التجميعية الأصلية. والأكثر إثارة أن النموذج المُقطَّر تعرّف على الرقم «3» بنجاح حتى حين أُزيلت جميع صور الـ3 من مجموعة النقل — فقد كانت المعرفة الخفية عن شبه الـ3 بأرقام أخرى (من خلال أهداف المعلم المرنة على أمثلة لا تحتوي الـ3) كافيةً وحدها.

وعلى مهمة التعرّف على الكلام واسعة النطاق داخل Google (نظام )، قُطّرت مجموعة من 10 نماذج في نموذج واحد بالبنية نفسها. النموذج المُقطَّر اقترب من معدل خطأ الكلمات الذي حقّقته المجموعة، مع تكلفة تشغيل أقل بعشر مرات.

وأظهرت تجارب النماذج المتخصصة على JFT أن مزيجاً من المعرفة المتخصصة والعامة، بعد تقطيره، يُضاهي النماذج المرجعية المُدرَّبة بحوسبة أكبر أو يتفوّق عليها.

افتح في المختبر
قارن بين دقة النماذج المُقطَّرة وكفاءتها مقابل المجموعات التجميعية والنماذج المرجعية المُدرَّبة بشكل منفرد.
تستيقظ التجربة عند وصولك…

ما الذي فتحه تقطير المعرفة

  1. 2015

    تقطير المعرفة (هذه الورقة)

    أثبتت أن الأهداف المرنة من المعلم تحمل معرفة خفية — بنية التشابه بين الفئات — وأن طالباً صغيراً يمكنه وراثتها عبر softmax مُعدَّلة بالحرارة.

  2. 2017

    تنعيم التسميات

    استبدل التسميات الصارمة بمزيج من الترميز الأحادي وتوزيع منتظم. من حيث المفهوم، هذا حالة خاصة من التقطير يكون فيها «المعلم» توزيعاً منتظماً.

  3. 2019

    DistilBERT

    طبّق التقطير لضغط BERT إلى 60% من حجمه مع الاحتفاظ بـ97% من أدائه واستدلال أسرع بـ60%. أتاح نشر المحوِّلات في بيئات محدودة الموارد.

  4. 2021

    DeiT — محوِّلات صور كفوءة بالبيانات

    استخدم شبكة التفافية معلماً لتقطير المعرفة في محوِّل بصري، مستغنياً عن التدريب المسبق بحجم JFT. وقدّم رمز تقطير يعمل إلى جانب رمز التصنيف.

  5. 2021

    DINO — تقطير ذاتي بلا تسميات

    تقطير ذاتي — الطالب والمعلم بالبنية نفسها، والمعلم متوسط متحرك أُسّي للطالب. لا حاجة لتسميات. تعلّم سمات تُقسّم الأجسام في الصورة تلقائياً.

  6. 2023

    فك الترميز التخميني

    نموذج مسوّدة صغير يقترح الرموز بسرعة، ونموذج مُحقّق كبير يتحقق منها بالتوازي. عدم التماثل نفسه بين المعلم والطالب كما في التقطير، لكنه مُوظَّف هنا لتسريع الاستدلال.

أعمق ما يُعلّمنا إياه التقطير هو درسٌ فلسفي: معرفة ليست في أوزانها ولا في بنيتها — بل في الدالة التي تعلّمتها، أي التوزيع الاحتمالي على المخرجات لأي مُدخَل. مجرد وعاء، والتوزيع الاحتمالي هو المحتوى. حين نستوعب هذه الفكرة، نُدرك أن المعرفة يمكن نقلها من وعاء لآخر — من نموذج كبير إلى صغير، ومن مجموعة نماذج إلى نموذج واحد، ومن معلم إلى طالب.

المرجعHinton, Vinyals, Dean. Distilling the Knowledge in a Neural Network. NeurIPS Deep Learning Workshop, 2015.

مصطلحات هذه الورقة