Mechanistic Interpretability of Discrete Logarithm Grokking: How Representation Determines the Internal Algorithm
تُثبت هذه الورقة أن تمثيل المدخلات لمسألة اللوغاريتم المنفصل يحدد بشكل جوهري الخوارزمية الداخلية التي تكتشفها الشبكات العصبية، مما يكشف أن التوافق الجبري مع التحليل الحقيقي للمسألة يعد شرطًا مسبقًا للتعميم السريع (grokking)، وأن البنى الحسابية الغنية بالتفاعلات اللازمة تتشكل خلال مرحلة الحفظ قبل وقت طويل من تحسن الأداء.
في الزوايا الهادئة لعلوم الحاسوب، يحاول الباحثون فهم كيف يفكر الذكاء الاصطناعي حقاً. لسنوات، عرفنا أن هذه الأنظمة يمكنها تعلم حل الألغاز المعقدة، لكن العملية غالباً ما تبدو كالسحر: قد يفشل النموذج مراراً وتكرال لفترة طويلة، ثم يحفظ الإجابات فجأة، وبعد توقف غريب، يبدأ في التعميم وحل مشكلات جديدة لم يرها من قبل. هذه الظاهرة، حيث يبدو أن الآلة قد "استوعبت" (grok) مفهوماً ما بعد فترة طويلة من مجرد حفظ البيانات، أصبحت لغزاً مركزياً. ولحل هذا الشفرة، غالباً ما يستخدم العلماء مشكلات رياضية ذات بنية واضحة ومخفية، تماماً مثل قفل له مفتاح محدد. إحدى هذه المشكلات هي اللوغاريتم المنفصل (discrete logarithm)، وهي مهمة تطلب من الحاسوب عكس عملية ضرب لإيجاد رقم مخفي. ورغم أن هذا يبدو مجرداً، إلا أنه نوع من الألغاز المنطقية التي تكشف عن الآلية الداخلية للشبكات العصبية. ومن خلال مراقبة كيفية تعلم الحاسوب لحل هذا اللغز تحديداً، يستطيع الباحثون رؤية الاستراتيجيات الداخلية التي ابتكرتها الآلة بالضبط، وما إذا كانت الطريقة التي نعرض بها المشكلة تغير الحل الذي تجده.
تأخذ دراسة جديدة أجراها ديفيد فيسترولوند هذا التحقيق خطوة أبعد عبر طرح سؤال بسيط ولكنه عميق: هل تغير الطريقة التي نقدم بها المعلومات للكمبيوتر طريقة تفكيره؟ لمعرفة ذلك، قام الباحث بتدريب نماذج الذكاء الاصطناعي لحل مشكلة اللوغاريتم المنفصل باستخدام طريقتين مختلفتين تماماً في عرض البيانات. في الطريقة الأولى، أُعطي النموذج الأرقام الخام، تماماً كما قد يراها الإنسان على صفحة، دون أي تلميحات خاصة حول البنية الأساسية. وفي الطريقة الثانية، قام الباحث بتفكيك المشكلة إلى أجزائها الرياضية الطبيعية قبل عرضها على النموذج، مما يعني عملياً تقديم "مجموعة أدوات" مرتبة مسبقاً للحاسوب. وكشفت النتائج أن هذين النهجين أدت بهما النماذج إلى اكتشاف خوارزميات داخلية مختلفة تماماً. فقد طور النموذج المدرب على البيانات الخام شبكة معقدة ومتشابكة من الروابط حيث كانت الأجزاء المختلفة من المشكلة متداخلة بعمق. وفي المقابل، طور النموذج الذي أُعطي بيانات منظمة ومرتبة مسبقاً تنظيماً نظيفاً ومنفصلاً حيث يتم التعامل مع الأجزاء المختلفة من المشكلة بشكل مستقل.
ربما كان الاكتشاف الأكثر إثارة للدهشة هو توقيت هذا التعلم. فقد وجد الباحثون أن البنية المعقدة والمتشابكة في نموذج البيانات الخام بدأت في التشكل قبل فترة طويلة من بدء النموذج في الحصول على الإجابات الصحيحة. فلآلاف الخطوات، كانت الروابط الداخلية للنموذج تعيد ترتيب نفسها في هذا النمط المعقد بينما ظلت درجات الاختبار الخاصة به عالقة عند مستوى منخفض. ولم يحدث ذلك إلا في وقت لاحق بكثير، بعد أن اكتمل بناء هذه البنية الداخلية، حيث بدأ النموذج فجأة في "الاستيعاب" وبدأ يحل المشكلة بشكل صحيح. وهذا يشير إلى أن الآلة كانت تبني الآليات اللازمة للفهم قبل وقت طويل من قدرتها على إظهار هذا الفهم. كما أظهرت الدراسة أن هذه البنية الداخلية ليست مجرد أثر جانبي بل هي ضرورية للمهمة؛ فعندما قام الباحثون بإزالة أجزاء محددة من الرياضيات الداخلية للنموذج جراحياً، فشل النظام تماماً، مما أثبت أن هذه الأنماط المحددة كانت هي المحرك الفعلي للحل.
كما صححت الدراسة سوء فهم من الأبحاث السابقة فيما يتعلق بكيفية اختبار هذه النماذج. تضمنت تقنية شائعة خلط تسميات البيانات لمعرفة ما إذا كان النموذج يعتمد على الترتيب المحدد للأرقام. وقد أثبت التحليل الجديد أن طريقة الخلط هذه كانت في الواقع خدعة؛ لأن النموذج يتعلم قاموسه الخاص من الرموز، فإن مجرد إعادة تسمية الأرقام لم يغير في الواقع البنية الرياضية الأساسية التي يراها. كان النموذج لا يزال يحل نفس المشكلة المهيكلة، ولكن بأسماء مختلفة. كان هذا الاكتشاف حاسماً لأنه مهد الطريق لاختبار أكثر صدقاً. ثم أنشأ الباحثون تسلسلاً هرمياً من الاختبارات الجديدة، بعضها حافظ على البنية الرياضية سليمة والبعض الآخر كسرها عمداً. ووجدوا أن النماذج لا يمكنها تعلم حل المشكلة بسرعة إلا إذا حافظ المدخل على "التقسيم" (partition) الرياضي الحقيقي للمشكلة. فإذا قام المدخل بخلط العلاقة بين أجزاء المعادلة، فإن النموذج سيحفظ بيانات التدريب بشكل مثالي، لكنه لن يتعلم أبداً حل مشكلات جديدة، حتى بعد التدريب لفترة طويلة جداً.
في النهاية، تشير هذه العمل إلى أن تمثيل المدخلات لا يسرع أو يبطئ التعلم فحسب، بل يحدد بشكل أساسي الخوارزمية التي يكتشفها الجهاز. فعندما تُقدم المشكلة بطريقة تحترم تقسيماتها الرياضية الطبيعية، يجد الجهاز مساراً بسيطاً وفعالاً للحل. وعندما تُقدم المشكلة بطريقة خام وغير مهيكلة، يُجبر الجهاز على بناء عالم داخلي أكثر تعقيداً لسد الفجوات، ويستغرق وقتاً أطول بكثير لإدراك كيفية استخدام هذا التعقيد. تؤكد الدراسة أنه بالنسبة لهذه الأنظمة الاصطناعية، فإن الطريق إلى الفهم ممهد ببنية المعلومات التي تُعطى لها. الأمر ليس مجرد مسألة كمية البيانات التي يراها، بل بكيفية ترتيب تلك البيانات. فالآلة لا تتعلم الإجابة فحسب، بل تتعلم طريقة تفكير محددة تمليها عليها بنية السؤال نفسه.
يستقصي هذا البحث كيف تحدد تمثيلات المدخلات مسألة جبرية ما الخوارزمية الداخلية التي تكتشفها الشبكة العصبية. تركز الدراسة على مسألة اللوغاريتم المنفصل (DLP) في سياق ظاهرة الاستيعاب المفاجئ (Grokking) — وهي الظاهرة التي يحفظ فيها النموذج بيانات التدريب لفترة طويلة قبل أن يتعمم فجأة على بيانات الاختبار.
السؤال الجوهمري ليس مجرد هل يحدث الاستيعاب المفاجئ، بل ما هو الآلية الداخلية التي تبنيها الشبكة وكيف يتشكل هذا الهيكل بناءً على ما إذا كان تمثيل المدخلات يكشف أو يعطل البنية الجبرية الأساسية (تحديداً تحليل نظرية الباقي الصينية (CRT)، أو تفكيك الزمرة الدورية).
2. المنهجية والإعداد التجريبي
2.1 المهمة والبيانات
المسألة: حل h=gx(modp) لإيجاد x، حيث p=113 (عدد أولي) ورتبة الزمرة هي N=p−1=112=7×16.
تفكيك نظرية الباقي الصينية (CRT): تسمح المسألة بتفكيك طبيعي إلى x(mod7) و x(mod16).
مجموعة البيانات: جميع الثلاثيات (5,376) من نوع (g,gx,x) الناتجة عن 48 مولداً. التقسيم: 30% للتدريب، 70% للاختبار.
النموذج: محول (Transformer) ذو طبقتين (427 ألف معلمة، بُعد خفي 128، 4 رؤوس).
2.2 تمثيلات المدخلات
تقارن الدراسة عدة ترميزات للمدخلات لاختبار ضرورة المحاذاة الجبرية:
الخام (R0): تسلسل قياسي [BOS,g,SEP,h,EOS]. لا يوجد تفكيك صريح.
نظرية الباقي الصينية الحقيقية (R1): مدخلات مفككة صراحة [BOS,gb,SEP,ga,SEP,hb,SEP,ha,EOS]، مما يكشف عن إسقاطات المجموعات الفرعية.
المُشوش (SCRAMBLED): يطبق تبديلاً (permutation) ثابتاً واحداً لجميع مواضع الـ CRT.
ضوابط سلبية (تسلسل هرمي للتدمير الجبري):
الـ CRT المُبدل (R2): تبديلات مستقلة لكل عامل (يحافظ على التقسيم، يدمر الترتيب).
الـ CRT منفصل الأدوار: تبديلات مختلفة لمواضع المولد مقابل مواضع النتيجة (يحافظ على التقسيم، يدمر الترافق المشترك).
التحويل العشوائي العالمي (R3): تحويل متباين (bijection) عشوائي يدمر قانون الزمرة وتقسيم العوامل.
المطابقة الرتبوية (R3O) والأساس المختلط (R4): تنويعات تحافظ على المعلومات ولكنها تدمر البنية الجبرية.
2.3 تقنيات التحليل
تفكيك فوريه لـ 2D CRT: يتم تفكيك تنشيطات الطبقات الخفية إلى أنماط فوريه: EA (العامل A فقط)، EB (العامل B فقط)، و EAB (التفاعل بين العاملين). تقيس نسبة التفاعل Rint=EAB/(EA+EB+EAB) مدى قابلية الفصل في التمثيل.
التحليل الزمني: تتبع طاقة فوريه ودقة الاختبار عبر 32 نقطة تفتيش (0–65 ألف حقبة/epoch) لتحديد متى تتشكل الهياكل الداخلية بالنسبة للتعميم.
الاستئصال السببي (Causal Ablation): استئصال أعلى k من الفضاءات الفرعية لفوريه (UA,UB,UAB) باستخدام 100 ضابط عشوائي مطابق في الرتبة والتباين للتحقق من الضرورة السببية.
درجات البنية الجبرية: مقاييس تشمل درجة التشاكل (homomorphism score Shom) ومؤشر راند المعدل (ARI) لقياس مدى حفاظ التمثيل على قانون الزمرة وتقسيم العوامل.
3. المساهمات الرئيسية
خوارزميات داخلية متباينة: أثبتت أن تمثيلات RAW و CRT تؤدي إلى تنظيمات داخلية مختلفة نوعياً. يطور نموذج RAW بنية غير قابلة للفصل وغنية بالتفاعلات (EAB≈0.65)، بينما يطور نموذج CRT بنية قابلة للفصل (EAB≈0.01)، رغم حلهما لنفس المهمة.
التشكّل الزمني للتفاعل: أظهرت أن البنية الغنية بالتفاعلات في نماذج RAW تتشكل خلال هضبة الحفظ (memorization plateau) (الحقب 15 ألف–30 ألف)، أي قبل حوالي 20 ألف حقبة من التعميم المرئي (الاستيعاب المفاجئ عند ~50 ألف).
تدقيق الـ "SCRAMBLED": أثبتت أن تمثيل "SCRAMBLED" الذي تمت دراسته سابقاً هو مرافق تماماً (exactly conjugate) لـ TRUE CRT عند التهيئة. نظرًا لأن النموذج يستخدم تضمينات فئوية متعلمة، فإن التبديل العالمي للرموز (tokens) يتم امتصاصه بواسطة جدول التضمين، مما يعني أن SCRAMBLED يحافظ على جميع البنى الجبرية (Shom=1.0) ولا يختبر التدمير الجبري.
هرم الضوابط السلبية: أثبتت أن الحفاظ على تقسيم عامل الـ CRT الحقيقي هو المتطلب الحرج للتعميم السريع. التمثيلات التي تحافظ على التقسيم (TRUE CRT، PERMUTED CRT، ROLE-DECOUPLED CRT) جميعها تحقق الاستيعاب المفاجئ، بينما تلك التي تدمر التقسيم (R3، R3O، R4) تفشل في تحقيق الاستيعاب حتى بعد 100 ألف حقبة.
الضرورة السببية لأنماط فوريه: أكدت عبر استئصال أعلى k من الفضاءات الفرعية لفوريه أن أنماط A، وأنماط B، وأنماط التفاعل AB هي جميعها ضرورية سببياً للحساب (p<0.01).
4. النتائج الرئيسية
4.1 التمثيل يحدد الخوارزمية
نموذج RAW: يطور خوارزمية داخلية معقدة تعتمد على التفاعلات بين العوامل. ترتفع طاقة التفاعل (EAB) من 35% إلى 52% خلال الهضبة بينما تظل دقة الاختبار ثابتة (~30%).
نموذج CRT: يطور خوارزمية قابلة للفصل. يكتشف النموذج معلومات المكونات (x(mod7) و x(mod16)) مبكراً ولكنه يتعلم كيفية إعادة دمجها لاحقاً.
الاستنتاج: تمثيل المدخلات لا يغير سرعة التعلم فحسب؛ بل يغير بشكل جوهري الخوارزمية الداخلية (قابلة للفصل مقابل غنية بالتفاعلات).
4.2 ضرورة الحفاظ على تقسيم العوامل
الاستيعاب المفاجئ السريع: TRUE CRT (20 ألف حقبة)، PERMUTED CRT (8 آلاف)، و ROLE-DECOUPLED CRT (~26 ألف) جميعها تحقق تعميماً شبه مثالي.
الفشل في الاستيعاب: GLOBAL RANDOM BIJECTION (R3)، و ORDER-MATCHED (R3O)، و MIXED RADIX (R4) تفشل في التعميم (Tgrok>100K) رغم الحفظ الكامل وأقصى تضاؤل للأوزان (weight decay).
الدلالة: مجرد تفكيك المدخلات إلى إحداثيين ليس كافياً. يجب أن يحافظ التمثيل على التقسيم الفعلي للعوامل (الربط المحدد لعناصر الزمرة بالـ CRT).
4.3 سوء فهم الـ "SCRAMBLED"
تصحح الدراسة تفسيراً محتملاً خاطئاً في الأدبيات السابقة. بما أن SCRAMBLED هو مرافق لـ CRT، فإن استيعابه السريع لا يعني أن المحاذاة الجبرية غير ضرورية؛ بل يؤكد ببساطة أن النموذج يمكنه تعلم الخوارزمية عندما يتم الحفاظ على البنية الجبرية عبر إعادة تسمية المفردات (vocabulary relabeling).
4.4 التحقق السببي
أدى استئصال أعلى k من الفضاءات الفرعية لفوريه إلى انخفاضات كارثية في الأداء (Δ≈−0.89 إلى $-0.96$)، بينما أظهرت الضوابط العشوائية تأثيراً ضئيلاً. وهذا يؤكد أن أنماط فوريه المحددة ليست مجرد سمات مرتبطة، بل هي مطلوبة سببياً للحساب.
5. الأهمية والادعاءات
تدعي الورقة أن تمثيل المدخلات هو المحدد الأساسي للخوارزمية الداخلية التي تكتشفها الشبكة العصبية.
رؤية ميكانيكية: الاستيعاب المفاجئ ليس ظاهرة متجانسة؛ فالمسار نحو التعميم يعتمد على ما إذا كان على الشبكة بناء أنماط تفاعل معقدة (RAW) أو الاعتماد على هياكل مفصولة مكشوفة مسبقاً (CRT).
المحاذاة الجبرية: يتطلب التعميم السريع في المهام الجبرية أن يحافظ تمثيل المدخلات على التقسيم الفعلي للعوامل لبنية الزمرة الأساسية. مجرد الحفاظ على المعلومات أو التفكيك العشوائي ليس كافياً.
تصحيح منهجي: تسلط الدراسة الضوء على فخ حرج في التفسير الآلي: "تشويش" المدخلات عبر تبديل الرموز قد يكون عديم الفائدة إذا كان النموذج يستخدم تضمينات متعلمة، حيث يمكن امتصاص التبديل دون تدمير المنطق الجبري الأساسي.
الديناميكيات الزمنية: تتشكل البنية الداخلية المطلوبة للتعميم (وتحديداً تفاعلات العوامل في RAW) قبل فترة طويلة من إظهار النموذج لقدرات التعميم، مما يشير إلى أن "الهضبة" هي مرحلة بناء هيكلي وليست مجرد مرحلة حفظ.
يحافظ المؤلف على نطاق متواضع، مشيراً إلى قيود مثل استخدام بذرة (seed) واحدة للتحليل الزمني والتقيد الحالي بـ p=113، واضعاً إياها كاتجاهات للعمل المستقبلي بدلاً من كونها استنتاجات حالية.