Safety Cost of Steering Vectors Is Separable and Reducible
تقترح هذه الورقة طريقة تحسين لاحقة (post-hoc) تحدد وتزيل مكوناً قابلاً للفصل ومسبباً لتدهور السلامة من نواقل توجيه النماذج اللغوية الكبيرة، مما يخفف من مخاطر السلامة مع الحفاظ على التوجيه السلوكي المنشود وتقليل حالات الرفض الخاطئ.
تخيل أن لديك صديقاً آلياً ذكياً جداً تعلم كيف يكون متعاوناً، ولكنه أيضاً يتعلم قول "لا" عندما يُطلب منه القيام بشيء خطير، مثل صنع قنبلة أو اختراق بنك. هذا الروبوت يشبه "النماذج اللغوية الكبيرة" (LLM)، وهي نوع من أنواع الذكاء الاصطناعي التي تقرأ وتكتب النصوص. لقد اكتشف العلماء حيلة ذكية لتغيير طريقة تفكير هذا الروبوت دون الحاجة إلى إعادة بنائه من الصفر. يسمون هذه الحيلة "التوجيه" (steering). تخيل عقل الروبوت كخريطة عملاقة متعددة الأبعاد؛ فمن خلال إضافة دفعة صغيرة غير مرئية في اتجاه معين على هذه الخريطة، يمكننا جعل الروبوت يتصرف بثقة أكبر، أو برغبة أكبر في المساعدة، أو بوعي أكبر بذاته. الأمر يشبه إعطاء الروبوت دفعة لطيفة لتميل شخصيته في اتجاه جديد.
ومع ذلك، هناك عقبة. أحياناً، عندما تدفع الروبوت ليكون أكثر "تعاوناً" أو "وعياً بذاته"، قد تدفعه بالخطأ بقوة في الاتجاه الخاطئ. الأمر يشبه محاولة توجيه سيارة نحو اليسار، لكن عجلة القيادة لزجة جداً بحيث تضغط بالخطأ أيضاً على دواسة الفرامل، مما يجعل السيارة تتوقف عندما لا ينبغي لها ذلك، أو والأسوأ من ذلك، تضغط على دواسة البنزين بينما تريد أنت التوقف. في عالم الذكاء الاصطناائي، يعني هذا أن محاولة جعل الروبوت يتصرف بطريقة معينة يمكن أن تجعله يتجاهل قواعد السلامة الخاصة به ويوافق على القيام بأشياء سيئة. وهذه مشكلة كبيرة لأننا نريد للذكاء الاصطناعي أن يكون مفيداً وآمناً في آن واحد.
هذه الورقة البحثية، التي قُدمت في مؤتمر رئيسي لعلوم الحاسوب، تبحث في مشكلة "عجلة القيادة اللزجة" هذه تحديداً. أراد الباحثان، يو شياو لي وجيرجي كاسنيسي، معرفة ما إذا كان بإمكاننا إصلاح عجلة القيادة لتوجه الروبوت حيث نريد دون أن تضغط بالخطأ على مكابح السلامة. وقد وجدا أن الجزء "السيئ" من دفعة التوجيه هو في الواقع منفصل عن الجزء "الجيد". الأمر يشبه اكتشاف أن الشحم الذي يسبب التصاق عجلة القيادة هو في الواقع ذرة صغيرة من الأوسធ يمكن إزالتها، وليس ترساً مكسوراً.
لقد طور الفريق طريقة جديدة تسمى CAST (التجريد المقيد للتوجيه الآمن - Constrained Ablation for Safe STeering). تخيل أن لديك أثراً لطبعة حذاء طينية على أرضية نظيفة (متجه التوجيه). بدلاً من فرك الأرضية بأكملها والمخاطرة بإتلاف السجاد الجميل (سلوك الروبوت المفيد)، يعمل CAST مثل مكنسة كهربائية فائقة الدقة؛ فهو يحدد البقعة المحددة من الطين التي تسبب انزلاق السلامة ويشفطها، تاركاً بقية طبعة الحذاء سليمة تماماً. لقد اختبروا هذه الطريقة على ثلاثة نماذج مختلفة للذكاء الاصطناعي ووجدوا أن طريقتهم نجحت في إزالة مخاطر السلامة. في الواقع، بعد استخدام CAST، كانت الروبوتات جيدة في اتباع التعليمات تماماً كما كانت من قبل، لكنها توقفت عن الموافقة على الطلبات الضارة، حتى عندما كانت تلك الطلبات مموهة بطرق مخادعة.
يشير الباحثون إلى أن هذا ينجح لأن جزء دماغ الذكاء الاصطناعي الذي يتعامل مع "السلامة" والجزء الذي يتعامل مع "التوجيه" متميزان هندسياً، مثل لونين مختلفين من الطلاء ممزوجين معاً. يمكنك فصلهما دون إفساد اللوحة النهائية. وبينما لم يثبتوا أن هذا يعمل مع كل ذكاء اصطناعي أو حالة ممكنة، فقد أظهرت تجاربهم أن هذا النهج "الجراحي" قلل باستمرار من خطر أن يصبح الذكاء الاصطناعي خطيراً، مع الحفاظ على كونه مفيداً في الوقت نفسه. إنها خطوة واعدة نحو ضمان أنه عندما نعدل شخصية الذكاء الاصطناعي لدينا، فإننا لا نطفئ ضميره بالخطأ.
ملخص تقني: "هل يمكن للتدخلات أثناء الاستدلال الحفاظ على التوافق؟ تكلفة السلامة لمتجهات التوجيه قابلة للفصل والتقليل"
بيان المشكلة توفر متجهات التوجيه (Steering vectors) آلية خفيفة الوزن للتحكم في سلوك النماذج اللغوية الكبيرة (LLMs) أثناء الاستدلال، وذلك عبر إضافة اتجاه محدد إلى تنشيطات مسار المتبقي (residual stream) للنموذج. ومع ذلك، تشير أدلة حديثة إلى أن متجهات التوجيه التي تبدو حميدة يمكن أن تضعف سلامة النموذج بشكل غير مقصود، مما يزيد من الامتثال للطلبات الضارة (الاختراقات/jailbreaks) ويقلل من آليات الرفض. وبينما اقترح عمل سابق أن تدهور السلامة ينشأ من التداخل بين متجهات التوجيه واتجاهات "الرفض" المعروفة، فإن عملية الاستئصال البسيطة لهذه الاتجاهات المعروفة تفشل في استعادة السلامة بشكل موثوق. ويعود ذلك إلى أن سلوك الرفض يتم بوساطة فضاء فرعي عالي الأبعاد يتكون من عدة اتجاهات مستقلة ميكانيكيًا، كما أن المكون المحدد لمخالفة السلامة في متجه التوجيه ليس بالضرورة متوافقًا مع أي اتجاه محدد مسبقًا للرفض. علاوة على على ذلك، فإن تعزيز الرفض لمواجهة تدهور السلامة غالبًا ما يؤدي إلى زيادة غير مقبولة في حالات الرفض الخاطئ للمدخلات الحميدة.
المنهجية: CAST (الاستئصال المقيد للتوجيه الآمن) يقترح المؤلفون CAST، وهي طريقة لاحقة (post-hoc) لتطهير متجهات التوجيه دون إعادة تدريب النموذج. الفرضية الأساسية هي أن المكون المسبب لتدهور السلامة في متجه التوجيه يمكن فصله هندسيًا عن مكونه المؤثر سلوكيًا. تصيغ CAST عملية تحديد وإزالة هذا المكون المسبب لتدهور السلامة كمسألة تحسين مقيدة.
البارامترية (Parameterization): يتم اشتقاق المتجه المطهر v∗ عن طريق إسقاط فضاء فرعي S متعلم من متجه التوجيه الأصلي v. في التجارب الرئيسية، هذا عبارة عن تقريب من الرتبة 1 (rank-1 approximation) حيث يلتقط متجه وحدة واحد متعلم r^ اتجاه تدهور السلامة: v∗=v−r^r^⊤v.
دالة الهدف (Objective Function): تعمل الطريقة على تقليل خسارة السلامة (Lsafety) التي تقيس التباعد في احتمالات رموز الرفض (refusal-token probabilities) بين النموذج الموجه والنموذج المرجعي غير الموجه عند استخدام مطالبات ضارة.
القيود (Constraints): لضمان المنفعة ومنع الإفراط في الرفض، تخضع عملية التحسين لقيود صريحة يتم فرضها عبر نهج "لاجرانج" الأولي-الثنائي (Lagrangian primal-dual approach):
الحفاظ على التأثير (Leffect): يجب أن يظل توزيع المخرجات لمتجه التوجيه المطهر على المطالبات المصممة لاستحضار السلوك المستهدف قريبًا من متجه التوجيه الأصلي.
التحكم في الرفض الخاطئ (Lfrr): يجب ألا تزداد احتمالية الرفض في المطالبات الحميدة بالنسبة للنموذج المرجعي غير الموجه.
التحسين (Optimization): يتم حل المسألة باستخدام تحديثات أولية-ثنائية (primal-dual updates). يتم تحديث المتغير الأولي (r^) عبر الانحدار الاشتقاقي (gradient descent)، بينما تقوم المتغيرات الثنائية (λe,λf) بتعديل أوزان الجزاء بشكل تكيفي لفرض القيود. تستخدم خسارة السلامة بديلًا قابلًا للاشتقاق يعتمد على التباعد في الرموز الدالة على الرفض (مثل "لا يمكنني"، "آسف") بدلاً من الأحكام غير القابلة للاشتقاق.
النتائج الرئيسية تم تقييم الطريقة عبر ثلاثة نماذج (Llama-3.1-8B, Qwen2.5-7B, Qwen2.5-14B)، وثلاثة مجالات سلوكية (القابلية للإصلاح Corrigibility، السعي وراء القوة Power-Seeking، الوعي بالذات Self-Awareness)، وسبعة سيناريوهات لهجمات الاختراق (بما في ذلك القوالب الثابتة والهجمات التكيفية مثل GCG وAutoDAN).
استعادة السلامة: تقلل CAST بشكل كبير من معدل نجاح الهجوم (ASR) الناجم عن متجهات التوجيه. عبر جميع النماذج والسلوكيات، خفضت المتجهات المطهرة متوسط معدل نجاح الهجوم (ASR) إلى مستوى أو أقل من مستويات النموذج غير الموجه. في المقابل، قدم استئصال اتجاهات الرفض المعروفة تحسينات طفيفة فقط.
التعميم (Generalization): تعمم اتجاه تدهور السلامة المتعلم بفعالية على أنواع الهجمات غير المرئية (مثل الهجمات التكيفية التي لم تُستخدم أثناء التدريب) وعبر مختلف معاملات التوجيه (α).
الحفاظ على التأثير: تم الحفاظ على التأثيرات السلوكية لمتجهات التوجيع إلى حد كبير. في العديد من الإعدادات، حافظت المتجهات المطهرة على التحول السلوكي المقصود أو حتى عززته مقارنة بالمتجهات الأصلية، مما يشير إلى أن مكونات السلامة والمنفعة ليست فقط قابلة للفصل، بل هي متنافسة جزئيًا.
الرفض الخاطئ: نجحت CAST في منع تضخم حالات الرفض الخاطئ. وفي مجموعات البيانات الحميدة القياسية (Alpaca)، ظل معدل الرفض الخاطئ دون تغيير أو انخفض. وفي المطالبات الحدية الصعبة (XSTest)، كانت الزيادات في الرفض الخاطئ ضئيلة (عادةً أقل من 1-2%)، مما يمثل مقايضة موفقة مقابل التخفيض الكبير في مخاطر السلامة.
تحليل الرتبة (Rank Analysis): أظهرت التجارب مع فضاءات الاستئصال ذات الأبعاد الأعلى (k>1) عوائد متناقصة. حقق الاستئصال من الرتبة 1 أفضل مقايضة، بينما مالَت الرتب الأعلى إلى إزالة المكونات المتداخلة مع التأثير السلوكي المقصود دون تقديم مكاسب تناسبية في السلامة.
الأهمية والادعاءات يزعم البحث أن "ضريبة السلامة" لتدخلات وقت الاستدلال ليست نتيجة حتمية لتوجيه التنشيط. بدلاً من ذلك، ينشأ تدهور السلامة من مكون محدد وقابل للإزالة داخل متجه التوجيه، وهو قابل للفصل هندسيًا عن مكون المنفعة. من خلال صياغة المشكلة كتحسين مقيد على المتجه نفسه، توفر CAST وصفة عامة لتطبيق التدخلات على مستوى التنشيط دون المساس بالسلامة أو التسبب في رفض خاطئ. يؤكد المؤلفون أن هذا النهج يقدم تصحيحًا جراحيًا وقابلًا للتفسير يستعيد سلامة النموذج مع الحفاظ على التحكم السلوكي المقصود، مما يجعل توجيه التنشيط أداة أكثر موثوقية للنشر.