Accelerating Visual On-Policy Distillation with Batched Speculative Jacobi Rollouts
تقدم هذه الورقة HB-SJD، وهو خلفية برمجية لفك التشفير من نوع (Speculative Jacobi Decoding) بنظام الدفعات، تعمل على تسريع عملية التقطير البصري في الوقت الفعلي (visual on-policy distillation) عبر تمكين المعالجة المتوازية لعدة رموز وأنماط تنفيذ تكيفية، مما يقلل زمن التدريب بشكل كبير مع الحفاظ على جودة التوليد دون تغيير إطار عمل التقطير.
في عالم الذكاء الاصطناعي، هناك صراع مستمر بين إنشاء نماذج ذكية بما يكفي لإنتاج صور مذهلة وبين إبقائها صغيرة بما يكفي لتعمل بسرعة. فالأنظمة الضخمة، التي تُسمى غالباً "المعلمين"، يمكنها إنتاج صور دقيقة للغاية من خلال التنبؤ بكل قطعة صغيرة من الصورة واحدة تلو الأخرى، وبناء الصورة بأكملها من نقطة بداية واحدة. ومع ذلك، فإن هذه العملية التي تتم خطوة بخطوة بطيئة ومكلفة. ولجعل هذه الأنظمة أسرع، يستخدم الباحثون تقنية تُسمى "تقطير المعرفة"، حيث يتعلم نموذج "طالب" أصغر وأبسط محاكاة "المعلم". وهناك نسخة أحدث وأكثر فعالية من هذه العملية، تُعرف باسم "التقطير في السياسة المتبعة" (on-policy distillation)، والتي تطلب من الطالب التدرب عبر توليد أمثلة تدريبية خاصة به في الوقت الفعلي. يضمن ذلك أن يتعلم الطالب بالضبط كيف سيتصرف عندما يقوم بإنشاء الفن فعلياً، بدلاً من مجرد حفظ الأمثلة القديمة. والمشكلة هي أن جلسة التدريب هذه بطيئة بشكل مؤلم؛ إذ لا يزال يتعين على الطالب بناء صوره قطعة صغيرة تلو الأخرى، مما يخلق عنق زجاجة يبطئ عملية التعلم بأكملها.
لقد وجد فريق من الباحثين من شنزن طريقة لكسر عنق الزجاجة هذا دون تغيير القواعد الأساسية لكيفية تعلم الطالب. فقد قدموا طريقة جديدة تسمى HB-SJD، والتي تعمل كمحرك أسرع لجلسات تدريب الطالب. فبدلاً من إجبار الطالب على بناء صورة واحدة "توكن" (أو وحدة بصرية صغيرة) في كل مرة، يسمح هذا النظام الجديد للنموذج باقتراح عدة قطع مستقبلية من الصورة في وقت واحد ثم التحقق مما إذا كانت تلك التخمينات صحيحة. فكر في الأمر مثل طالب يؤدي اختباراً: بدلاً من كتابة إجابة واحدة، وانتظار المعلم لتصحيحها، ثم كتابة الإجابة التالية، يكتب الطالب صفحة كاملة من الإجابات دفعة واحدة، ويقوم المعلم بسرعة بتحديد أي منها صحيح. بعد ذلك، يحتفظ الطالب بالإجابات الصحيحة ويعيد فقط كتابة الإجابات التي كانت خاطئة، مما يجعله يتقدم بسرعة أكبر بكثير.
اكتشف الباحثون أن مجرد تشغيل هذه الطريقة الأسرع على مجموعة من الصور في نفس الوقت لم يكن كافياً. ففي الإعداد القياسي، إذا انتهى توليد صورة واحدة بسرعة، فسيتعين عليها الانتظار حتى تنتهي أبطأ صورة في المجموعة قبل أن يتمكن النظام من الانتقال إلى الخطوة التالية. وكانت فترة الانتظار هذه تهدر طاقة حوسبية قيمة. ولحل هذه المشكلة، صمم الفريق نظامهم بحيث يمكن لكل صورة أن تتقدم بسرعتها الخاصة. فإذا انتهت صورة ما، فإنها تتنحى جانباً، ويستمر النظام في العمل على الصور المتبقية دون توقف. علاوة على على ذلك، وجدوا أن أفضل طريقة لتشغيل النظام تتغير مع تناقص عدد الصور النشطة. فعندما يتم معالجة العديد من الصور، يعمل النظام بشكل أفضل من خلال إبقاء المجموعة كاملة نشطة. ولكن مع انتهاء الصور وتصاغر المجموعة، ينتقل النظام إلى وضع أكثر رشاقة يعالج فقط الصور النشطة المتبقية، متجنباً الأعباء الإضافية لإدارة الخانات الفارغة.
وعند اختبار هذا النهج الجديد على نماذج توليد الصور، ثبتت فعاليته العالية. فقد وجد الباحثون أنه يمكنه تقليل الوقت المطلوب للطالب لتوليد صور التدريب الخاصة به إلى النصف تقريباً، مع سرعات تتراوح بين 1.4 إلى 1.6 مرة أسرع من الطريقة التقليدية. والأهم من ذلك، أن هذه السرعة لم تأتِ على حساب الجودة؛ فقد كانت الصور التي أنتجتها نماذج الطلاب التي تم تدريبها بهذه الطة الجديدة حادة وواقعية تماماً مثل تلك التي تم تدريبها بالطريقة القديمة الأبطأ. وقد عمل النظام باستمرار وبشكل جيد سواء كانت الصور قصيرة أو طويلة، وسواء كانت مجموعة الصور التي يتم معالجتها صغيرة أو كبيرة، وفي مراحل مختلفة من عملية التدريب. ومن خلال استبدال المحرك الذي يولد بيانات التدريب فقط، نجح الباحثون في جعل عملية التعلم بأكملها أسرع بشكل ملحوظ، مما أثبت أنه يمكنك تسريع تدريب الذكاء الاصطنا الاصطناعي دون المساس بجودة النتيجة النهائية.
ملخص تقني: تسريع التقطير البصري في السياسة المباشرة باستخدام عمليات "جاكوبي" التخمينية المجمعة
1. بيان المشكلة
أظهرت النماذج التوليدية ذاتية الانحدار (AR) البصرية قدرة قوية على التوسع في توليد الصور. ويعد "التقطير المعرفي" (Knowledge Distillation) نهجًا قياسيًا لنقل القدرات من النماذج الضخمة (المعلم) إلى النماذج الأصغر (الطالب). وتعمل التطورات الأخيرة، مثل "التقطير في السياسة المباشرة" (On-Policy Distillation - OPD)، على تحسين التدريب من خلال جعل الطالب يولد مساراته الخاصة أثناء التدريب، والتي يتم الإشراف عليها لاحقًا بواسطة المعلم. وهذا يقلل من عدم التطابق بين سلوكيات التدريب وسلوكيات الاستنتاج.
ومع ذلك، يفرض الـ OPD عائقًا حسابيًا كبيرًا: يجب على الطالب توليد لواحق المسارات (rollout suffixes) عبر الإنترنت قبل أن يتمكن المعلم من تقديم الإشراف. عملية فك التشفير التقليدية لـ AR تتم بتوكن تلو الآخر، مما يتطلب L من خطوات فك التشفير المتتالية لـ L-توكن من اللواحق. هذه العملية تقع مباشرة في المسار الحرج للتدريب، مما يضيف تكلفة باهظة لكل تحديث في السياسة المباشرة. وبينما يوفر "فك تشفير جاكوبي التخميني" (Speculative Jacobi Decoding - SJD) طريقة لمعالجة عدة توكنات بالتوازي دون الحاجة إلى نموذج مسودة مساعد، فإن SJD الأصلي مصمم للاستنتاج أحادي التسلسل. وتطبيق SJD مباشرة على إعدادات الدفعات الكبيرة (large-batch) المستخدمة في الـ OPD البصري يطرح تحديين:
فقدان التوازي: تشغيل SJD بشكل منفصل لكل صورة يكسر التوازي في دفعات وحدة معالجة الرسومات (GPU batch parallelism).
أعباء المزامنة: معالجة جميع الصور بشكل متزامن (Batch-Synchronous) تجبر الصور الأسرع على الانتظار حتى تنتهي الصور الأبطأ عند حدود النوافذ المشتركة، مما يلغي مكاسب الكفاءة.
2. المنهجية: HB-SJD
يقترح المؤلفون "فك تشفير جاكوبي التخميني الهجين المجمع" (Hybrid Batched Speculative Jacobi Decoding - HB-SJD)، وهو محرك خلفي للـ (rollout) مصمم خصيصًا للـ OPD البصري. يستبدل HB-SJD محرك توليد المسارات القياسي للطالب مع ترك المعلم، وهدف التقطير، وإجراء التحسين دون تغيير.
المكونات الأساسية
تقدم مستقل لكل صورة: على عكس SJD المتزامن للدفعة، يسمح HB-SJD لكل صورة في الدفعة بالتقدم وفقًا لتقدم فك التشفير الخاص بها. لا تنتظر الصور العضو الأبطأ في الدفعة لإكمال نافذة التحقق.
التحقق المجمع: رغم التقدم المستقل، يحافظ HB-SJD على كفاءة وحدة معالجة الرسومات. فهو يبني مدخلات التحقق للصور النشطة باستخدام آخر التوكنات الملتزم بها ونوافذ المسودة. تسمكن مؤشرات الموقع (position indices) وأماكن ذاكرة التخزين المؤقت (KV-cache) لكل صورة من معالجة هذه المدخلات غير المتجانسة معًا في تمريرة أمامية واحدة مجمعة للطالب.
استراتيجية التنفيذ الهجين: مع انتهاء الصور من توليد مساراتها، يتناقص عدد الصور النشطة. يقوم HB-SJD بالتبديل ديناميكيًا بين نمطين من التنفيذ بناءً على عتبة مُعايرة عتاديًا (γ):
التنفيذ الكامل (Full Execution): يحافظ على شكل الدفعة الفيزيائية الأصلية. تُعامل الصور المنتهية كصور غير نشطة منطقيًا (لا يوجد تحديث للحالة) ولكنها تشغل صفوفًا في الذاكرة. هذا فعال عندما تكون الدفعة النشطة كبيرة.
التنفيذ المدمج (Compact Execution): يشغل المحقق فقط على الصور النشطة المتبقية، مع الوصول فقط إلى صفوف ذاكرة التخزين المؤقت (KV-cache) المقابلة لها. هذا يتجنب عبء معالجة الصفوف غير النشطة عندما تكون الدفعة صغيرة.
التهيئة والتحقق:
التهيئة القائمة على التاريخ: يستغل HB-SJD المحلية المكانية عن طريق ملء نوافذ المسودة بالتوكنات الملتزم بها سابقًا عند إزاحة تاريخية ثابتة، بدلاً من التهيئة العشوائية.
استراتيجيات التحقق: يدعم النظام كلاً من التحقق الجشع (Greedy Verification) (قبول المسودات التي تطابق توقع الـ argmax الخاص بالطالب) والتحقق الاحتمالي (Probabilistic Verification) (قبول المسودات بناءً على نسبة الاحتمال، مما يسمح بمقترحات غير الـ argmax).
3. المساهمات الرئيسية
SJD للـ OPD البصري: يقدم البحث SJD كمحرك لتوليد المسارات في التقطير البصري في السياسة المباشرة. ومن خلال إعادة استخدام التوقعات من تكرارات "جاكوبي" الحالية للطالب، فإنه يتيح توليد مسارات متعددة التوكنات دون الحاجة إلى نموذج مسودة مساعد يتم تدريبه بشكل منفصل.
بنية HB-SJD: يقترح المؤلفون HB-SJD لتوسيع نطاق SJD من استنتاج التسلسل الواحد إلى الـ OPD البصري ذي الدفعات الكبيرة. فهو يجمع بشكل فريد بين التقدم المستقل لكل صورة والتنفيذ المجمع الفعال، مما يسمح للصور بالتقدم بسرعات مختلفة مع الاحتفاظ بتوازي وحدة معالجة الرسومات.
التكامل المباشر (Drop-in Integration): تم تصميم HB-SJD ليكون بديلاً مباشرًا لمحرك توليد المسارات للطالب. وهو لا يغير أهداف التقطير أو إجراءات التدريب للأساليب الموجودة (مثل GKD و VarKD).
التنفيذ الهجين: يقدم النهج استراتيجية تنفيذ هجينة (كاملة/مدمجة) تتكيف مع حجم الدفعة المتناقص أثناء توليد المسارات، مما يحسن زمن الاستجابة عبر مراحل التوليد المختلفة.
4. النتائج التجريبية
أُجريت التجارب باستخدام LlamaGen (نسختي B و L) على مجموعة بيانات ImageNet، مع مقارنة HB-SJD مقابل عمليات التخزين المؤقت التلقائية (AR) القياسية ضمن أطر عمل GKD و VarKD.
الكفاءة: يقلل HB-SJD بشكل كبير من وقت توليد المسارات ووقت التدريب الإجمالي.
تسريع السرعة: حقق تسريعًا في توليد المسارات بنسبة 1.48× إلى 1.65× اعتمادًا على حجم النموذج وطريقة التقطير (على سبيل المثال، 1.58× لـ VarKD على LlamaGen-B مع التحقق الجشع).
وقت التدريب: قلل إجمالي وقت التدريب بنحو 1.18×.
الدراسة الاستقصائية (Ablation): أظهرت الدراسة أن التقدم المستقل أمر بالغ الأهمية؛ حيث كان SJD المتزامن للدفعة (Batch-Synchronous SJD) في الواقع أبطأ من الـ AR المخزن مؤقتًا (تسريع 0.939×) بسبب انتظار المزامنة. كما وفرت استراتيجية التنفيذ الهجين تسريعًا إضافيًا قدره 1.182× مقارنة بالتنفيذ الكامل وحده.
الجودة: ظلت جودة التوليد قابلة للمقارنة مع نماذج AR المرجعية.
المقاييس: أظهرت مقاييس FID، وInception Score (IS)، والدقة (Precision)، والاستدعاء (Recall) فروقًا ضئيلة جدًا. على سبيل المثال، حقق LlamaGen-B مع GKD+HB-SJD قيمة FID قدرها 4.95 مقارنة بـ 4.96 لنموذج AR المرجعي.
أطوال المسارات المختلفة (زاد التسريع مع المسارات الأطول).
أحجام الدفعات المختلفة (من 8 إلى 64).
مراحل التدريب المختلفة (من 2K إلى 22K خطوة).
أحجام نافذة "جاكوبي" المختلفة (حيث تم تحديد W=16 كإعداد متوازن).
5. الأهمية والادعاءات
يزعم البحث أن HB-SJD يعالج التوتر الرئيسي في الـ OPD البصري: الحاجة إلى مسارات يولدها الطالب مقابل التكلفة العالية لتوليدها. ومن خلال فصل محرك توليد المسارات عن عملية التعلم، يسمح HB-SJD بتسريع كبير دون المساس بالفوائد النظرية للتقطير في السياسة المباشرة أو جودة النموذج المقطر.
يؤكد المؤلفون أن نهجهم متميز عن التشفير التخميني في نماذج اللغة المعززة بالتعلم التعزيزي (RL) (الذي يتطلب غالبًا التوافق مع سياسة متغيرة) لأن SJD يعيد استخدام توقعات نفس الطالب، مما يتجنب الحاجة إلى نموذج مسودة منفصل. علاوة على ذلك، يُقدم النهج كحل هندسي عملي يمكن دمجه في خطوط أنابيب التقطير التوليدي ذات الانحدار الذاتي البصري الحالية لتقليل تكاليف التدريب مع الحفاظ على دقة التوليد.