Poly-OPD: Heterogeneous Multi-Teacher On-Policy Distillation for Capability-Selectable Flow Models
يُعد Poly-OPD إطار عمل للتقطير القائم على السياسة متعدد المعلمين غير المتجانس، والذي يدمج نقاط القوة المتكاملة من نماذج تحويل النص إلى صورة غير المتوافقة في نموذج طالب واحد مدمج يعتمد على مطابقة التدفق، وذلك عبر جسر الفضاءات الكامنة بواسطة إعادة الترميز على مستوى البكسل والإشراف باستخدام DINOv2، مع توظيف محولات متوافقة مع التدرج ومنهج تعليمي مدرك للفجوات لتحقيق أداء فائق في كل من اتباع التعليمات التركيبية والمواءمة الجمالية.
المؤلفون الأصليون:Siming Fu, Haojun Xu, Ruizhe He, Zheming Fu, Hualiang Wang, Jie Huang, Xiaoxiao Ma, Mingchen Zhong, Weihu Huang, Xiaoxuan He, Linjiang Huang, Si Liu
تخيل عالماً يُصنع فيه الفن بواسطة آلات تتعلم من خلال التخمين والتصحيح، وهو مجال يُعرف باسم توليد الصور من النصوص. في هذا المرسم الرقمي، "المعلمون" هم نماذج ضخمة ومعقدة تحول الكلمات إلى صور. ولكن هنا تكمن العقبة: تماماً مثل الخبراء البشر، تمتلك هذه الآلات تخصصات مختلفة. فبعضها رسامون بارعون يبدعون مشاهد واقعية مذهلة الجمال، لكنهم قد يغفلون عن تفاصيل صغيرة مثل "ثلاث قطط" أو "كرة حمراء على اليسار". والبعض الآخر محاسبون صارمون يتبعون التعليمات بدقة، لكنهم قد ينتجون صوراً تبدو مسطحة أو غريبة بعض الشيء. المشكلة الكبرى هي أن هؤلاء الخبراء يتحدثون "لغات" داخلية مختلفة؛ فهم يستخدمون مخططات رياضية وحسابية مختلفة لبناء صورهم، لذا لا يمكنك ببساطة أن تطلب منهم تبادل الملاحظات أو دمج عقولهم بسهولة. إذا كنت تريد صورة جميلة ودقيقة تماماً في آن واحد، فعادة ما يتعين عليك تشغيل حاسوبين ضخمين ومكلفين في نفس الوقت، وهو أمر بطيء ومبدد للموارد. وقد حاول العلماء اكتشاف كيفية دمج هؤلاء الخبراء المختلفين في نموذج واحد أصغر وأذكى يمكنه القيام بالوظيفتين معاً دون أن يصاب بالارتباك.
تقدم هذه الورقة البحثية طريقة مبتكرة جديدة تسمى Poly-OPD، والتي تعمل كمترجم ومدرب بارع، حيث نجحت في دمج "معلمين" مختلفين تماماً في نموذج طالب واحد مدمج. لقد أخذ الباحثون معلماً "جميلاً" (FLUX.1-dev) ومعلماً "دقيقاً" (Z-Image)، وعلموا نموذجاً طالباً أصغر (نسخة بـ 2.5 مليار معلمة من SD3.5) كيف يكون كلاهما معاً. وتكمن الخدعة السحرية في أنه بدلاً من إجبار الطالب على نسخ الأفكار الداخلية للمعلمين (والتي تكون بلغات غير متوافقة)، يسمح النظام للطالب برسم صورة، ثم يحول تلك الرسمة إلى لغة "بكسل" مشتركة يفهمها الجميع، وبعد ذلك يطلب من المعلم تصحيحها. المعلم لا يكتفي بالقول "عمل جيد" فحسب، بل يقوم فعلياً بإعادة رسم الصورة لتصحيح الأخطاء، ويتعلم الطالب من هذا التصحيح. ولضمان عدم ارتباك الطالب عند محاولة تعلم أسلوبين مختلفين في وقت واحد، يستخدم النظام أجزاءً "قابلة للتبديل" خاصة في دماغه؛ فهو يتشارك الأجزاء التي تتعامل مع الانتباه العام (مثل النظر إلى المشهد ككل) ولكنه يحتفظ بأجزاء منفصلة للمهام المحددة (مثل العد أو التلوين). علاوة على ذلك، فإن النظام ذكي فيما يتعلق بما يتدرب عليه: فهو يلاحظ المهارات التي لا يزال الطالب سيئاً فيها، ويقضي وقتاً أطول في التدريب عليها، بدلاً من إضاعة الوقت في أشياء يعرفها الطالب بالفعل.
النتائج مثيرة للإعجاب حقاً. فباستخدام هذه الطريقة، لم يتعلم النموذج الطالب الصغير أن يكون متوسط المستوى فحسب، بل أصبح في الواقع أفضل من كلا معلميه الضخمين في مهام محددة. في اختبار يسمى GenEval، والذي يتحقق مما إذا كان النموذج يتبع تعليمات معقدة مثل "قطة على يمين مضرب تنس"، قفزت درجة الطالب من 67.3 إلى 73.3، متفوقاً على معلمه الذي يمتلك 12 مليار معلمة ومعلمه الذي يمتلك 6 مليارات معلمة. وفي اختبار للجمال البصي والتفضيل البشري (HPSv3)، ارتفعت الدرجة من 9.34 إلى 11.35. وتشير الورقة البحثية إلى أن هذا النجاح يعود إلى نهج "السياسة المباشرة" (on-policy)، حيث يتعلم الطالب من أخطائه الخاصة التي يصححها المعلم، بدلاً من مجرد حفظ رسومات المعلم المثالية. وقد وجد الباحثون أنه إذا حاولوا إجبار الطالب على التعلم من البيانات الداخلية للمعلم مباشرة، فإن العملية تفشل بسبب عدم تطابق "اللغات". كما اكتشفوا أنه إذا لم يستخدموا الأجزاء القابلة للتبديل الخاصة، فإن المهارات ستتضارب، وسيسوء أداء النموذج. في النهاية، يظهر Poly-OPD أنه يمكنك بناء نموذج واحد صغير وقابل للتبديل، يمكنه التنقل بين كونه فناناً جميلاً ومهندساً دقيقاً، جامعاً بين أفضل ما في العالمين دون الحاجة إلى تشغيل حاسوبين ضخمين.
ملخص تقني: Poly-OPD
بيان المشكلة غالبًا ما تُظهر نماذج تحويل النص إلى صورة المفتوحة نقاط قوة متكاملة: فبعضها يتفوق في الجماليات المتوافقة مع التفضيلات البشرية (مثل FLUX.1-dev)، بينما يتفوق البعض الآخر في اتباع التعليمات التركيبية بدقة مثل العد، والعلاقات المكانية، وربط الصفات (مثل Z-Image). ومع ذلك، فإن دمج هذه القدرات في نموذج واحد يعوقه التباين الهيكلي؛ حيث تستخدم هذه النما actually المتخصصة مشفرات تلقائية (autoencoders)، ومزيلات ضجيج (denoisers)، وجداول زمنية للضجيج (noise schedules) مختلفة، مما يجعل مساحاتها الكامنة (latent spaces) غير قابقة للفهم المتبادل. كما أن طرق التقطير الحالية (مثل مطابقة الدرجة أو السرعة أو المسار) تتطلب نظام إحداثيات مشترك، وبالتالي فهي غير قابلة للتطبيق. علاوة على ذلك، تعاني عملية التدريب التقليدية خارج السياسة (off-policy) على الصور المولدة من المعلم من عدم تطابق بين التدريب والاستدلال، بينما يؤدي التدريب المشترك على قدرات متعددة إلى تداخل مدمر بين اتجاهات التدرج المتعارضة.
المنهجية: Poly-OPD يقترح المؤلفون إطار عمل Poly-OPD (التقطير على السياسة متعدد الأهداف)، وهو إطار مصمم لتقطير معلمين متباينين وغير متوافقين في المساحة الكامنة إلى نموذج طالب واحد مدمج (تحديدًا نموذج 2.5B SD3.5-Medium). يعالج هذا الإطار ثلاثة تحديات أساسية:
التقطير على السياسة المتباين عبر جسر بكسلي (Pixel Bridge): لجسر المساحات الكامنة غير المتوافقة، يستخدم Poly-OPD "جسراً بكسلياً". يقوم الطالب بتوليد عينة، يتم فك تشفيرها إلى مساحة البكسل (وهي نظام الإحداثيات الوحيد المشترك بين النماذج المتباينة). بعد ذلك، يتم إعادة ترميز هذه الصورة بواسطة مشفر الـ VAE الخاص بالمعلم النشط. يقوم المعلم بتحسين هذا الكامن المبرمج من مستوى ضجيج يتطابق من حيث المقدار وليس مؤشر الخطوة الزمنية، مما ينتج صورة مستهدفة مصححة. لا يتم تطبيق الإشراف في المساحة الكامنة للمعلم، بل في مساحة سمات DINOv2 المجمدة (تضمين CLS). هذه المساحة السماتية (semantic feature space) ثابتة تجاه الإحصائيات البكسلية الخاصة بكل نموذج، مما يسمح للطالب بتعلم المحتوى الدلالي لتصحيح المعلم دون الحاجة إلى كامنات متوافقة. وتعتبر هذه العملية "على السياسة" (on-policy) لأن الهدف مشتق من العينة التي ولدها الطالب نفسه، مما يخفف من عدم التطابق بين التدريب والاستدلال.
محولات قابلة لاختيار القدرات مع توافق التدرج: لمنع التداخل المدمر بين القدرات المتميزة (مثل الجماليات مقابل التركيب)، يستخدم Poly-OPD تصميم محولات معياري. قدم المؤلفون تشخيص توافق التدرج لتحديد المعلمات التي يجب مشاركتها والتي يجب أن تظل معزولة تجريبيًا. كشفت القياسات أن تدرجات الانتباه (attention gradients) متوافقة عبر الأنماط، بينما غالبًا ما تتعارض تدرجات الشبكات الأمامية (FFN gradients). بناءً على ذلك، يشارك Poly-OPD نموذج LoRA للانتباه واحد عبر جميع المعلمين، ولكنه يعزل محولات FFN لكل قدرة محددة. يسم un هذا لنموذج أساسي واحد مجمد باستضافة قدرات متعددة، مع اشتراط تبديل المحول فقط عند الاستدلال.
أخذ عينات تكيفي مدرك للفجوة: إدراكًا بأن المهارات الفرعية التركيبية (مثل العد، والموقع) تصل إلى مرحلة التشبع بمعدلات مختلفة، يستخدم إطار العمل منهجًا درسيًا (curriculum) يخصص ميزانية التدريب بناءً على الفجوة بين المعلم والطالب المتبقية بدلاً من صعوبة المطالبة (prompt) الخام. تُستخدم مجموعة اختبار لتقدير فجوة الأداء لكل فئة، ويتم تعديل توزيع أخذ العينات ديناميكيًا للتركيز على الفئات التي لا يزال الطالب يتخلف فيها بشكل كبير عن المعلم، مع التوجه نحو أخذ العينات الموحد مع انغلاق الفجوات.
المساهمات الرئيسية
الصياغة: يصيغ البحث مشكلة دمج القدرات المتعددة من معلمين متباينين هيكليًا وغير متوافقين في المساحة الكامنة، وهو وضع تفشل فيه طرق تقطير المساحة الكامنة الحالية.
إطار العمل: يقدم Poly-OPD آلية الجسر البكسلي للإشراف على السياسة، واستراتيجية مشاركة المحولات القائمة على قياس التدرج، ومنهج أخذ عينات مدرك للفجوة.
الأداء: ينجح نموذج الطالب المستخلص من Poly-OPD في دمج نقاط قوة FLUX.1-dev (12B) و Z-Image (6B) في نموذج طالب واحد 2.5B SD3.5-Medium.
النتائج التجريبية عند التقييم باستخدام المعايير القياسية، يتفوق نموذج Poly-OPD على كلا النموذجين المعلمين الأكبر حجمًا في مجالاتهما الخاصة:
التركيب (GenEval): يحقق الطالب 73.3، متفوقًا على Z-Image (69.4) و FLUX.1-dev (65.2). لوحظت مكاسب كبيرة في الفئات المهيكلة مثل تركيب جسمين (+14.93) وربط الصفات (+8.25).
الجماليات (DrawBench): يحقق الطالب درجة HPSv3 تبلغ 11.35 و ImageReward تبلغ 1.168، متفوقًا على نموذج SD3.5-Medium الأساسي واستعادة جزء كبير من الفجوة مع معلم FLUX.1-dev (12B).
دراسات الاستئصال (Ablation Studies): يؤدي إزالة المحولات القابلة لاختيار القدرات (باستخدام FFN مشترك) إلى انخفاض حاد في الأداء (على سبيل المثال، -11.0 نقطة في GenEval)، مما يؤكد ضرورة عزل التدرجات المتعارضة. وبالمثل، فإن إزالة مرحلة البدء الدافئ (warm-start) أو أخذ العينات المدرك للفجوة يؤدي إلى تدهور النتائج بشكل كبير، مما يثبت صحة خط التدريب المقترح.
الأهمية والادعاءات يزعم البحث أن Poly-OPD يتيح دمج القدرات المتكاملة في نموذج واحد قابل للتبديل دون تكاليف الذاكرة وزمن الوصول المرتبطة بنشر عدة نماذج كبيرة. ومن النتائج الرئيسية التي سلط المؤلفون الضوء عليها هي أن الطالب المستخلص يتفوق على معلميه في مقاييس محددة (مثل GenEval و ImageReward). ويعزو المؤلفون قدرة "التفوق" هذه بشكل فريد إلى طبيعة التدريب على السياسة (on-policy)؛ فخلافًا لطرق "خارج السياسة" (off-policy) التي تتقيد بتوزيع العينات المولدة من المعلم، يسمح التقطير "على السياسة" للطالب بتحسين مساره الخاص بناءً على تصحيحات المعلم، مما قد يتجاوز جودة مخرجات المعلم الخام. يثبت إطار العمل أن التقطير المتباين أمر ممكن دون مساحات كامنة مشتركة، بشرًا بتثبيت الإشراف في مساحة سمات دلالية مشتركة وإدارة التدريب من خلال بنية مدركة للتدرج ومنهج تعليمي.