Efficient Hypergradient Descent for Inverse Reinforcement Learning
تقترح هذه الورقة طريقة فعالة للتعلم المعاكس للتعزيز تستفيد من التناسب بين مصفوفة هسيان الهدف الداخلي ومصفوفة معلومات فيشر للسياسة لاشتقاق تدرج فائق مهيكل، والذي يتم تقريبه عبر التخطيط الطيفي المتدفق للتغلب على اختناقات القابلية للتوسع المرتبطة بمصفوفات فيشر الكبيرة.
تخيل أنك تحاول تعليم روبوت كيف يرقص مثل راقص محترف. يمكنك عرض فيديو للرقصة على الروبوت وتقول له: "فقط قلد حركاتي بدقة". هذا ما يسمى بـ "التعلم بالتقليد" (Imitation Learning). ولكن ماذا لو تغيرت الأرضية، أو احتاج الروبوت للرقص على ترامبولين بدلاً من خشبة المسرح؟ إذا قام فقط بحفظ الحركات، فقد يسقط على وجهه. هناك نهج أذكى وهو معرفة "لماذا" تحرك الراقص بتلك الطوّلية. ما الذي كان الراقص يحاول تحقيقه؟ ما هي "النقاط" التي كان يحاول رفعها لأقصى حد؟ هذا هو هدف "التعلم المعزز العكسي" (Inverse Reinforcement Learning - IRL): فبدلاً من مجرد تقليد الرقصة، نحاول هندسة عكسية لنظام "المكافأة" الخفي الذي كان يتبعه الخبير. بمجرد معرفة قواعد اللعبة، يمكننا تعليم الروبوت الرقص على أي سطح، وليس فقط على السطح الذي رآه في الفيديو.
للقيام بذلك، يستخدم العلماء لعبة معقدة مكونة من خطوتين تسمى "التحسين ثنائي المستوى" (Bilevel Optimization). فكر في الأمر كمعلم وطالب. "المستوى الداخلي" هو الطالب الذي يحاول تعلم أفضل الحركات بناءً على مجموعة من القواعد (المكافأة) التي نعطيها له. أما "المستوى الخاري" فهو المعلم الذي يتحقق مما إذا كانت حركات الطالب تشبه حركات الخبير. إذا لم يتطابقا، يقوم المعلم بتعديل القواعد (المكافأة) ويرسل الطالب للتدريب مرة أخرى. المشكلة هي أن تحديد "كيفية" تعديل القواعد بدقة أمر صعب للغاية؛ فالأمر يشبه محاولة تخمين كيف ستؤثر تغييرات طفيفة في القواعد على عملية تعلم الطالب بأكملها. عادةً، يتطلب حساب ذلك قدرًا هائلاً من ذاكرة الكمبيوتر، كأنك تحاول حمل مكتبة في حقيبة ظهرك فقط لحل مسألة رياضية.
تقدم هذه الورقة البحثية اختصارًا ذكيًا لحل مشكلة الذاكرة هذه. اكتشف الباحثون، بقيادة نيكيتا سيفريوك وفريقه من جامعة HSE، أنه عندما يتعلم الطالب (الروبوت) القواعد بشكل مثالي، فإن "الشكل" الرياضي لعملية تعلمه يشبه تمامًا خريطة محددة تسمى "مصفوفة معلومات فيشر" (Fisher Information Matrix). وهذا أمر بالغ الأهمية لأن هذه الخريطة لها بنية خاصة تجعل التعامل معها أسهل. ومع ذلك، حتى هذه الخريطة قد تكون ضخمة جدًا بحيث لا يمكن تخزينها على جهاز كمبيوتر. لذا، ابتكر الفريق طريقة لاستخدام "رسم تخطيطي طيفي متدفق" (Streaming Spectral Sketch). تخيل بدلاً من كتابة كل تفصيل في الخريطة، أنك تأخذ لقطة سريعة وذكية تلتقط أهم الميزات مع التخلص من التفاصيل غير الضرورية. وقد أطلقوا على هذه الطريقة اسم "الاشتقاق المتدرد الفعال" (Efficient Hypergradient Descent).
اختبر الباحثون هذه الفكرة في عالمين مختلفين: لعبة موازنة العمود البسيطة (CartPole)، ومهمة تحكم مستمرة أكثر تعقيدًا (LQR). وقارنوا طريقتهم الجديدة في "التخطيط" (Sketching) بالطرق القد la والأبطأ في إجراء العمليات الحسابية. كانت النتائج واعدة؛ ففي بيئة LQR المعقدة، قللت طريقتهم من الذاكرة المطل उत् necesaria بنحو 1.31 مرة وكانت أسرع قليلاً. وفي لعبة CartPole الأبسط، كانت أسرع بنحو 1.3 مرة. وبينما لم تنتج طريقة "التخطيط" (Sketch) دائمًا خريطة مكافأة مثالية تمامًا مقارنة بالطرق الثقيلة والبطيئة، إلا أنها كانت قريبة جدًا منها. والأهم من ذلك، أنها سمحت للروبوت بتعلم أسلوب الخبير بنفس الكفاءة، ولكنها فعلت ذلك بفعالية أكبر بكثير. ويشير المؤلفون إلى أنه باستخدام هذه التقريبات الذكية وخفيفة الوزن، يمكننا تعليم الروبوتات التعلم من الخبراء دون الحاجة إلى حواسيب فائقة القدرة لتخزين كل تلك البيانات.
بيان المشكلة يسعى تعلم التعزيز العكسي (IRL) إلى استعادة دالة مكافأة من عروض الخبراء بحيث تعيد السياسة المثلى تحت هذه المكافأة إنتاج السلوك الملحوظ. يركز المؤلفون على تعلم التعزيز العكسي القائم على الاحتمالية القصوى (ML-IRL)، والذي يمكن صياغته كمسألة تحسين ثنائي المستوى. تعمل المستوى الداخلي على تحسين سياسة منظمة بالإنتروبيا تحت مكافأة مُتعلمة rϕ، بينما يعمل المستوى الخارجي على تقليل التباين (تحديداً تباعد KL) بين السياسة المستحثة وبيانات الخبير.
العقبة الحسابية الرئيسية في حل هذه المسألة ثنائية المستوى عبر التمايز الضمني هي الحاجة إلى حساب التدرج الفائق (hypergradient). يتضمن ذلك ضرب مصفوفة هيسيان معكوسة بمتجه (inverse-Hessian-vector product) للدالة الهدف الداخلية. يتطلب بناء الهسيان صراحةً ذاكرة تربيعية (O(dθ2))، بينما يمكن أن تكون التقريبات التكرارية (مثل التدرج المترافق) مكلفة حسابياً وحساسة لظروف المصفوفة. وبينما تم اقتراح "التدرج الفائق الطبيعي" (NHGD) لاستبدال الهسيان بمصفوفة معلومات فيشر كبديل، يفترض NHGD القياسي وجود توزيع بيانات ثابت. في ML-IRL، تتضمن الدالة الهدف الداخلية تباعد KL عكسي فوق توزيع مسار مستحث بواسطة السياسة، مما يجعل التطبيق المباشر لـ NHGD غير بديهي.
المنهجية يقترح الورقة طريقة لحساب التدرج الفائق بكفاءة لـ ML-IRL من خلال الاستفادة من البنية الهندسية للمسألة الداخلية واستخدام التخطيط الطيفي (matrix sketching).
تكافؤ فيشر-هسيان: يثبت المؤلفون أنه في ظل الأمثلية الداخلية الدقيقة والقابلية للتحقيق (حيث تكون فئة السياسة غنية بما يكفي لنمذجة توزيع الخبير تماماً)، فإن هسيان الدالة الهدف الداخلية يتناسب مع مصفوفة معلومات فيشر للمسار المخصم (Fθ). وتحديداً، ∇θ2Linner=αFθ. وهذا يسمح باستبدال ضرب الهسيان المعكوس بمتجه بضرب فيشر المعكوس بمتجه المطلوب لحساب التدرج الفائق.
تقدير التدرج الفائق: تقوم الطريقة بتفكيك حساب التدرج الفائق إلى ثلاثة مكونات:
التدرج الخارجي بالنسبة لمعلمات السياسة.
حل نظام خطي مخمد (damped linear system) يتضمن مصفوفة فيشر والتدرج الخارجي.
حد المشتق المختلط الذي يتضمن تدرج المكافأة وحل النظام الخطي.
التخطيط الطيفي المتدفق (SCFD): لتجنب تكلفة الذاكرة O(dθ2) لتخزين مصفوفة فيشر الكثيفة، تقرب المؤلفون ضرب فيشر-المعكوس المطلوب باستخدام مخطط "الاتجاهات المتكررة للتعويض الطيفي المتدفق" (SCDF). بدلاً من تجسيد مصفوفة فيشر، تقوم الخوارزمية بتدفق متجهات درجة السياسة (policy-score vectors) الموزونة من مسارات الوكيل إلى مخطط مدمج بحجم m. يقلل هذا من تعقيد التخزين إلى O(mdθ). ثم يُستخدم المخطط لحل النظام الخطي المخمد تقريبياً.
المساهمات الرئيسية
الاشتقاق النظري: اشتق الورقة مقدرات قائمة على العينات للتدرج الفائق الضمني لـ ML-IRL وأثبت أن هسيان الداخل يتناسب مع مصفوفة معلومات فيشر للمسار عند الوصول إلى أمثلية داخلية قابلة للتحقيق. ويُقدم هذا كأول تطبيق لهذه المتطابقة لحساب التدرج الفائق الضمني في ML-IRL.
حل قابل للتوسع: اقترح المؤلفون حلاً يعتمد على SCFD يقوم بتدفق متجهات درجة السياسة. تتجنب هذه الطريقة بناء مصفوفة الدرجة أو مصفوفة فيشر الكثيفة، مما يقلل بشكل كبير من متطلبات الذاكرة.
التحقق التجريبي: تم تقييم الطريقة مقابل خط أساس (single-loop ML-IRL) من الدرجة الأولى، وحل فيشر الصريح في بيئات التحكم المنفصلة (CartPole) والمستمرة (LQR).
النتائج أجريت التجارب تحت ميزانيات حسابية متطابقة (حد تدريب 24 ساعة) مع بنيات وضبط للمعلمات متشابهة.
أداء السياسة: حققت الطرق القائمة على فيشر جودة سياسة تنافسية. في CartPole، وصلت جميع الطرق إلى عوائد مستوى الخبير. في LQR، حقق ML-IRL أحادي الحلقة أفضل سالب لوغاريتم الاحتمالية (NLL) وأفضل عائد للبيئة، وإن كان الهامش مقارنة بطريقة "فيشر مع التخطيط" (Fisher with Sketching) صغيراً.
ترتيب المكافأة: أظهرت الطرق القائمة على فيشر جودة قوية في ترتيب المكافآت. في LQR، حققت طريقة "فيشر مع التخطيط" أعلى ارتباط رتبي (RankCorrelation) (0.972) بين عوائد المسار المستخلصة والحقيقية، متفوقة على كل من خط الأساس أحادي الحلقة وحل فيشر الصريح.
الكفاءة:
الذاكرة: وفر التخطيط (Sketching) أكبر فائدة في بيئة LQR ذات الأبعاد العالية، حيث قلل ذروة استخدام الذاكرة بنسبة تصل إلى 1.31× مقارنة بحل فيشر الصريح.
السرعة: في CartPole، حقق التخطيط تسريعاً قدره 1.29×. في LQR، قدمت أحجام التخطيط المتوسطة (مثل m=32) تسريعاً طفيفاً (1.04×)، بينما أصبحت الأحجام الأكبر (m=256) أبطأ من الحل الصريح بسبب العبء الإضافي لعملية التخطيط، رغم انخفاض قيم الخسارة النهائية.
الاستقرار: وجد المؤلفون أن الحل المخطط ظل مستقراً تحت معاملات تخميد (λ) أضعف بكثير من الصيغة الصريحة، مما يشير إلى أن عملية التخطيط قد تعمل كمنظم ضمني.
الأهمية والادعاءات يدعي الورق معالجة عقبة القابلية للتوسع في ML-بسبب التدرج الفائق عبر إقامة رابط مباشر بين هسيان الداخل ومصفوفة معلومات فيشر في سياق تحسين KL العكسي. من خلال الجمع بين هذه الرؤية النظرية وتخطيط المصفوفات المتدفق، يوضح المؤلفون أنه من الممكن حساب التدرجات الفائقة الضمنية دون تكلفة الذاكرة التربيعية لبناء مصفوفة الهسيان أو مصفوفة فيشر الصريحة.
يخلص المؤلفون بتواضع إلى أنه بينما تعد الطرق القائمة على فيشر تنافسية مع ML-IRL أحادي الحلقة من حيث جودة السياسة النهائية، إلا أنها تقدم ميزة متميزة في جودة ترتيب المكافأة والكفاءة الحسابية (تحديداً تقليل الذاكرة) للمسائل ذات الأبعاد العالية. تشير الدراسة إلى أن أحجام التخطيط المتوسطة توفر المقايضة المثلى بين جودة التحسين، واستخدام الذاكرة، ووقت التشغيل.