Learn What's Left, Not What's Mastered: Saturation Aware Advantage Reweighting for Multi-Reward Policy Optimization
تقدم هذه الورقة البحثية طريقة "إعادة وزن الميزة الواعية بالتشبع لتحسين السياسات متعددة المكافآت" (SA-MRPO)، وهي طريقة مبتكرة تعمل على توحيد معايير أهداف مكافآت متعددة بشكل مستقل وإعادة وزنها تكيفياً بناءً على مستويات تشبعها لنقل تركيز التحسين ديناميكياً نحو الأهداف غير المحسنة كفاية، مما يؤدي إلى تحسين الأداء بشكل كبير في الاختبارات المعيارية الصعبة مع الحفاظ على الكفاءة في المهام التي تم حلها بالفعل.
المؤلفون الأصليون:Yixuan Wang, Yifei Chen, Haichao Zhang, Haozheng Luo, Xander Wu, Jie Ni, Yun Fu, Nuno Vasconcelos, Yijiang Li
في عالم الذكاء الاصطناي الذي يتطور بسرعة، يقوم الباحثون بتعليم البرامج الحاسوبية التفكير بشكل يشبه البشر، لا سيما عند حل المشكلات المعقدة التي تتطلب تفكيراً خطوة بخطوة. ولتعليم هذه الأنظمة، غالباً ما يستخدم العلماء طريقة تسمى "التعلم التعزيزي"، حيث يجرب البرنامج طرقاً مختلفة ويتلقى ملاحظات في شكل مكافآت. فإذا حل البرنامج مسألة رياضية بشكل صحيح، يحصل على نقطة؛ وإذا اتبع تنسيقاً معيناً، يحصل على نقطة أخرى. والهدف هو تعظيم هذه المكافآت حتى يتعلم البرنامج تقديم إجابات أفضل. ومع ذلك، فإن المهام في العالم الحقيقي نادراً ما تقتصر على هدف واحد؛ فالمساعد النافع يجب أن يكون دقيقاً، ولكن عليه أيضاً أن يكون موجزاً، وآمناً، وملتزماً بقواعد تنسيق صارمة. وتنشأ التحديات عندما يحاول البرنامج الموازنة بين كل هذه الأهداف في وقت واحد؛ فإذا أصبح البرنامج بارعاً جداً في مهمة واحدة، مثل إبقاء الإجابات قصيرة، فقد يتوقف عن محاولة التحسن في هذا المجال، ومع ذلك قد يستمر نظام التدريب في دفعه ليكون أكثر إيجازاً، مما يهدر وقتاً قيماً للتعلم كان من الممكن استغلاله في إصلاح مشكلة أخرى أكثر صعوبة، مثل حل المسألة الرياضية بشكل صحيح.
لقد حدد فريق من الباحثين خللاً في كيفية تعامل الأنظمة الحالية مع هذه الأهداف المتعددة، واقترحوا طريقة جديدة لتدريبها تركز على ما لا يزال يتعين تعلمه بدلاً مما تم إتقانه بالفعل. في دراستهم، لاحظوا أن طرق التدريب القياسية غالباً ما تعامل جميع الأهداف كأنها متساوية الأهمية من البداية إلى النهاية، بغض النظر عن مدى جودة أداء البرنامج فيها. ويؤدي هذا إلى وضع يستمر فيه النظام في صقل مهارة أتقنها بالفعل، بينما يهمل مهارة أصعب لا تزال هناك مساحة للتحسين فيها. ولإصلاح ذلك، طور الباحثون تقنية أطلقوا عليها اسم "إعادة وزن الميزة المدركة للتشبع" (Saturation Aware Advantage Reweighting). تعمل هذه الطريقة مثل مدير ذكي يتحقق باستمرار من التقدم المحرز في كل هدف؛ فعندما يقترب هدف ما من الكمال، يقوم النظام تلقائياً بتقليل الضغط عليه، محولاً التركيز والطاقة نحو الأهداف التي لا تزال تعاني.
اختبر الباحثون هذا النهج على النماذج اللغوية المكلفة بحل مشكلات رياضية صعبة وكتابة أكواد برمجية. وفي هذه الاختبارات، كان على النماذج الموازنة بين الوصول إلى الإجابة الصحيحة وبين اتباع القواعد المتعلقة بطول الإجابة أو كيفية تنسيقها. وتحت طرق التدريب القديمة، كانت النماذج غالباً ما تواجه صعوبة في تحسين دقتها بمجرد تعلمها قواعد الطول بشكل مثالي. ومع ذلك، أدركت الطريقة الجديدة أن قاعدة الطول لم تعد تشكل تحدياً، فتوقفت عن إضاعة الجهد فيها، وبدلاً من ذلك، وجهت انتباه النموذج إلى المهمة الأصعب وهي حل الرياضيات بشكل صحيح. وكانت النتائج واضحة: عبر خمس عشرة مقارنة مختلفة شملت مختلف المسابقات الرياضية والمعايير، حسنت الطريقة الجديدة دقة المهام الأصعب في اثنتي عشرة منها. وفي اختبار محدد يتعلق بـ "الامتحان الأمريكي للمسائل الاستدلالية" (American Invitational Mathematics Examination)، وصل التحسن إلى خمسة بالمائة. والأهم من ذلك، أن هذه الزيادة في الدقة لم تأتِ على حساب المهام الأسهل؛ إذ استمرت النماذج في اتباع قواعد الطول والتنسيق بنفس الكفاءة السابقة.
عمل هذا النهج أيضاً عندما اختبره الباحثون على "الاستدلال التكيفي"، حيث كان الهدف هو إيجاد التوازن الصحيح بين أن يكون النموذج صحيحاً وفعالاً. لقد أنشأوا سيناريو كان فيه هدف "الطول" له حد واضح: فبمجرد أن تصبح الإجابة قصيرة بما يكفي، فإن جعلها أقصر لا يوفر أي فائدة إضافية. لاحظت طريقة التدريب الجديدة أن النموذج قد وصل بالفعل إلى هذا الحد وتوقفت عن محاولة تقصير الإجابات أكثر من اللازم. وبدلاً من ذلك، استخدمت ذلك الجهد الموفر لجعل الإجابات أكثر دقة. وأدى ذلك في المتوسط إلى زيادة في الدقة بنسبة تقارب أربعة بالمائة عبر خمسة معايير مختلفة، مع أكبر قفزة تجاوزت تسعة بالمائة في مسابقة رياضية محددة. لم تصبح النماذج مهملة في مسألة الطول؛ بل اكتفت فقط بالتوقف عن محاولة أن تكون أقصر مما ينبغي والتركيز على أن تكون أكثر ذكاءً.
كما استكشفت الدراسة كيف يقرر النظام متى يتوقف عن الضغط على هدف ما. فقد قدم الباحثون "مقبض تحكم"، وهو رقم واحد يحدد مدى قوة تجاهل النظام للأهداف التي تم استيفاؤها بالفعل. ووجدوا أن رفع هذا المقبض يجعل النظام يركز بشكل أكبر على المهام الصعبة، مما يحسن الدقة ولكنه يؤدي أحياناً إلى إجابات أطول قليلاً. أما خفضه فيحافظ على الإجابات أقصر ولكنه لا يحسن الدقة بنفس القدر. وهذا يظهر أن الطريقة مرنة ويمكن ضبطها لإيجاد أفضل توازن لحالة معينة. كما اختبر الباحثون الطريقة في مهام البرمجة الحاسوبية، حيث كان على النموذج كتابة برامج تعمل دون أخطاء وتجتاز اختبارات محددة. ومرة أخرى، ساعدت الطريقة الجديدة النموذج على تحسين قدرته على اجتياز الاختبارات مع الحفاظ على قدرته على كتابة أكواد تعمل بشكل صحيح.
إن الاكتشاف الجوهري هو أن التدريب الفعال يتطلب الانتباه إلى الإمكانات المتبقية للتحسين في كل مجال، بدلاً من معاملة جميع الأهداف كأهداف ثابتة. فمن خلال الضبط الديناميكي لمقدار الاهتمام الذي يُبذل لكل هدف بناءً على مدى قربه من الكمال، يتعلم النظام بكفاءة أكبر. وقد أظهر الباحثون أن هذه ليست مجرد فكرة نظرية، بل هي تحسن عملي يعمل عبر أنواع مختلفة من الاستدلال وأحجام مختلفة من النماذج الحاسوبية. لقد أثبتوا أنه من خلال التخلي عن "الانتصارات السهلة"، يمكن للنظام تحقيق نتائج أفضل بكثير في المهام الصعبة، مما يؤدي إلى ذكاء اصطناعي أكثر ذكاءً وقدرة دون التضحية بالقواعد الأساسية التي يجب أن يتبعها. وهذا يشير إلى أن مستقبل تدريب هذه الأنظمة يكمن في فهم ليس فقط ما تعلمته، بل ما لا يزال يتعين عليها تعلمه.
ملخص تقني: إعادة وزن الميزة الواعية بالتشبع لتحسين السياسة متعددة المكافآت (SA-MRPO)
1. بيان المشكلة
أصبح التعلم المعزز بالمكافآت القابلة للتحقق (RLVR) هو المعيار لتدريب النماذج اللغوية الكبيرة (LLMs) بعد مرحلة ما بعد التدريب لتحسين قدرات الاستدلال. وبينما قامت خوارزمية تحسين السياسة النسبية للمجموعة (GRPO) ومتغيراتها بتبسيط هذه العملية من خلال إلغاء الحاجة إلى دالات القيمة المتعلمة، إلا أنها تفترض عادةً وجود مكافأة قياسية واحدة (scalar reward). وفي الممارسة العملية، يجب على نماذج الاستدلال تلبية أهداف متعددة في آن واحد، مثل الصحة (correctness)، والالتزام بالتنسيق (format compliance)، والسلامة، وقابلية التنفيذ.
يعتمد النهج القياسي لتحسين الأهداف المتعددة على تحويل هذه المكافآت إلى مجموع موزون ثابت قبل إجراء التقييس على مستوى المجموعة. وقد حدد المؤلفون قصورين جوهريين في هذا التصميم:
فقدان دقة المكافأة: يمكن لملفات تعريف المكافآت المتميزة أن تنهار لتصبح ذات قيمة قياسية واحدة. على سبيل المثال، قد تحصل عملية (rollout) ذات صحة عالية ولكن تنسيق سيئ على نفس المكافأة القياسية (وبالتالي نفس الميزة/advantage) كعملية ذات صحة ضعيفة ولكن تنسيق مثالي، طالما أن المجموع الموزون متطابق.
تجاهل تشبع الأهداف: تعامل الأوزان الثابتة جميع الأهداف على أنها متساوية الأهمية طوال فترة التدريب، بغض النظر عن حالتها الراهنة. فإذا كان هدف مساعد (مثل الالتزام بالتنسيق) قد وصل بالفعل إلى حالة التشبع (أي تم حله)، فإن عملية التحسين تستمر في تخصيص ميزانية التدرج (gradient budget) له. وهذا يحول الجهد بعيدًا عن الأهداف الأصعب والأقل تحسينًا (مثل الصحة الرياضية) التي لا تزال تمتلك "مساحة" كبيرة للتحسين.
تعالج الطرق الحالية مثل تحسين السياسة منزوع الارتباط بمكافآت المجموعة (GDPO) المشكلة الأولى عبر تطبيع المكافآت بشكل مستقل، لكنها تفشل في معالجة المشكلة الثانية، لأنها لا تزال تعتمد على أوزان ثابتة لا تتكيف مع مستوى التشبع لكل هدف.
2. المنهجية: SA-MRPO
يقترح المؤلفون إعادة وزن الميزة الواعية بالتشبع لتحسين السياسة متعددة المكافآت (SA-MRPO). تقوم هذه الطريقة بالحفاظ على التطبيع لكل هدف على حدة، مع إعادة وزن كل هدف بشكل تكيفي بناءً على درجة تشبعه الحالية.
الآلية الجوهرية
لكل دفعة من العمليات (rollouts) وعدد n من أهداف المكافأة، تحسب SA-MRPO ميزة واعية بالتشبع كما يلي:
تقدير التشبع: لكل هدف k، يتم تقدير نسبة التشبع s(k) باستخدام متوسط دفعة المكافأة rˉ(k) بالنسبة للنطاق المتاح للهدف [rmin(k),rmax(k)]: s(k)=rmax(k)−rmin(k)rˉ(k)−rmin(k)∈[0,1] تشير قيمة s(k) القريبة من 1 إلى أن الهدف مشبع (قريب من الحد الأقصى للمكافأة)، بينما تشير القيمة القريبة من 0 إلى وجود مساحة كبيرة للتحسين.
إعادة الوزن التكيفي: يتم تعديل الوزن المقرر wk لكل هدف بواسطة عامل تشبع يتم التحكم فيه بواسطة أس γ≥0: w~k=wk(1−s(k))γ عندما يقترب هدف ما من التشبع (s(k)→1)، ينخفض وزنه الفعال w~k. وعلى العكس من ذلك، تحتفظ الأهداف ذات التشبع المنخفض بأوزان أعلى.
بناء الميزة (Advantage): يتم حساب الميزة الإجمالية للعملية (rollout) عن طريق جمع الميزات المطبعة للأهداف الفردية، والموزونة بأوزانها المعدلة حسب التشبع: A~(i,j)=k=1∑nw~kAk(i,j) حيث Ak(i,j) هي الميزة النسبية للمجموعة للهدف k. ثم يتم تطبيع هذا المجموع الإجمالي عبر الدفعة قبل استخدامه في هدف البديل المقطوع (clipped surrogate objective) القياسي.
الخصائص النظرية
التعميم: تعمم SA-MRPO بشكل صارم طريقة GDPO (عندما يكون γ=0) وطريقة GRPO (في حالة الهدف الواحد).
التغيير الاتجاهي: على عكس إعادة القياس البسيطة، يمكن لإعادة الوزن الواعية بالتشبع أن تعكس إشارة التحديث. إذا تعارض هدف مشبع مع هدف آخر لم يتم تحسينه بعد، فإن تقليل الوزن على الهدف المشبع يمكن أن يؤدي إلى قلب الميزة الإجمالية، مما يغير فعليًا اتجاه تحديث السياسة لصالح الهدف غير المشبع.
التعامل مع الصراع: تقر الطريقة بأن إعطاء الأولوية لهدف غير مشبع قد يؤدي إلى تدهور هدف تم تحسينه سابقًا إذا تعارضت تدرجاتهما. ومع ذلك، يجادل المؤلفون بأن هذه مقايضة ضرورية لتعظيم الأداء العام، حيث أن الربح من تحسين الهدف الأقل تحسينًا يفوق الخسارة في الهدف المشبع.
3. المساهمات الرئيسية
تحديد المشكلة: يحدد البحث أن تحسين المكافآت المتعددة عبر التجميع القياسي يعاني من فقدان دقة المكافأة، والأهم من ذلك، تجاهل تشبع الأهداف، مما يؤدي إلى تخصيص غير فعال لجهود التحسين.
اقتراح الخوارزمية: تقدم SA-MRPO آلية لإعادة وزن أهداف المكافأة المطبعة ديناميكيًا بناءً على نطاق المكافأة المتاح المتبقي. وتثبت أن هذا يمكن أن يغير اتجاه تحديثات السياسة، وليس فقط حجمها.
التحقق التجريبي: تم التحقق من صحة الطريقة عبر مجالات الاستدلال الرياضي، والاستدلال التكيفي، وتوليد الكود. وهي تنقل جهود التحسين باستمرار من الأهداف المشبعة إلى الأهداف الأقل تحسينًا، مما يحسن الأخيرة مع الحفاظ على أداء الأولى.
4. النتائج التجريبية
قيم المؤلفون SA-MRPO مقابل GDPO والنماذج المرجعية عبر إعدادات متنوعة:
الاستدلال الرياضي: باستخدام نماذج Qwen2.5 (3B و 7B) مع هدفين وثلاثة أهداف (الصحة، الطول، التنسيق)، حققت SA-MRPO تحسنًا في الدقة مقارنة بـ GDPO في 12 من أصل 15 مقارنة مرجعية. تشمل المكاسب الملحوظة تحسنًا بنسبة 5.0% في AIME24 و 3.5% في MATH500 للنموذج 7B. والأهم من ذلك، تم تحقيق هذه المكاسب دون تدهور كبير في الأداء في هدف الطول المساعد.
الاستدلال التكيفي: في إعداد به هدف طول مشبع صراحة (حيث تحصل الاستجابات التي تقل عن حد معين من الرموز/tokens على أقصى مكافأة)، حسنت SA-MRPO الدقة في جميع الاختبارات الخمسة بمتوسط 3.8%، مع ذروة مكسب بلغت 9.2% في AMC23. سمحت الطريقة للنموذج باستخدام رموز إضافية للاستدلال بمجرد استيفاء قيد الطول، مما أظهر إعادة تخصيص فعالة للقدرة.
توليد الكود: في اختبارات مثل APPS و Codeforces، ومن خلال التحسين لكل من قابلية التنفيذ (الأسهل تحقيقًا) ومعدل النجاح (الأصعب)، حسنت SA-MRPO معدلات النجاح في 3 من أصل 4 اختبارات (بزيادة تصل إلى 2.3%) مع الحفاظ على مستويات قابلية تنفيذ مماثلة لـ GDPO.
حساسية المعلمات الفائقة (Hyperparameter Sensitivity): أظهر أن الأس الخاص بالتشبع γ يتحكم مباشرة في المقايضة بين تحسين الأهداف غير المحسنة والحفاظ على الأهداف المشبعة. وفرت القيم المتوسطة (مثل γ=0.5) أفضل توازن، مما حقق أعلى متوسط دقة.
5. الأهمية والادعاءات
يدعي البحث أن التحسين الفعال للسياسة متعددة المكافآت يتطلب أكثر من مجرد موازنة مقاييس المكافأة؛ بل يجب أن يأخذ في الاعسبار التحسين المفيد المتبقي في كل هدف.
التخصيص الديناميكي: توفر SA-MRPO طريقة مبدئية لإعادة تخصيص ميزانية التدرج ديناميكيًا بعيدًا عن المشكلات "المحلولة" نحو تلك التي لا تزال تمتلك مساحة، دون تعديل آلية تحديث سياسة GRPO الأساسية.
عكس التحديثات: الطريقة قادرة على عكس إشارة التحديث، وهي قدرة تفتقر إليها عمليات التجميع القياسية أو فك الارتباط ذو الأوزان الثابتة، مما يسمح للسياسة بإلغاء أولوية الأهداف المشبعة عندما تتعارض مع الأهداف الأكثر صعوبة.
قابلية التطبيق العامة: أظهر النهج فعالية عبر مجالات مختلفة (الرياضيات، الاستدلال التكيفي، الكود) وأحجام نماذج مختلفة، مما يشير إلى أن إعادة الوزن الواعية بالتشبع هي استراتيجية قوية للتعلم المعزز متعدد الأهداف (RLVR).
خلص المؤلفون إلى أن استراتيجيات تحسين المكافآت المتعددة في المستقبل يجب أن تنظر صراحةً في حالة تشبع الأهداف لتجنب إضاعة قدرة التحسين على قيود تم استيفاؤها بالفعل.