Feature Identification via the Empirical NTK
यह लेख प्रदर्शित करता है कि एम्पेरिकल न्यूरल टेंगेंट कर्नेल (eNTK) का आइगन-एनालिसिस प्रशिक्षित न्यूरल नेटवर्क में ग्राउंड-ट्रुथ और व्याख्या योग्य विशेषताओं की प्रभावी ढंग से पहचान करता है, जो सिंथेटिक अंकगणितीय कार्यों के साथ-साथ एक प्री-ट्रेन्ड लैंग्वेज मॉडल पर PCA की तुलना में ज्ञात संरचनाओं के साथ बेहतर संरेखण प्रदर्शित करता है।
मूल पेपर CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/) के तहत लाइसेंस किया गया है। नीचे दिए गए पेपर की यह व्याख्या AI से तैयार की गई है। इसे लेखकों ने न तो लिखा है, न इसका समर्थन किया है। तकनीकी सटीकता के लिए मूल पेपर देखें। पूरा डिस्क्लेमर पढ़ें
मुख्य विचार: AI में "छिपे हुए स्विच" खोजना
कल्पना कीजिए कि एक विशाल, जटिल मशीन (जैसे कि एक न्यूरल नेटवर्क) है जिसने कोई कार्य करना सीख लिया है, जैसे कि संख्याएँ जोड़ना या कहानियाँ लिखना। आप मशीन को काम करते हुए देख सकते हैं, लेकिन आप यह नहीं देख सकते कि वह कैसे सोचती है। यह एक ब्लैक बॉक्स को देखने जैसा है: आप एक नंबर डालते हैं, और दूसरा नंबर बाहर आता है, फिर भी उसके अंदर के गियर छिपे रहते हैं।
वैज्ञानिक उस बॉक्स को खोलना चाहते हैं और उन विशिष्ट "स्विचों" या "नॉब्स" (knobs) को खोजना चाहते हैं जो मशीन "व्याकरण" (grammar), "जोड़" (addition), या "भावना" (sentiment) जैसी अवधारणाओं को समझने के लिए उपयोग करती है। इसे मैकेनिस्टिक इंटरप्रिटेबिलिटी (mechanistic interpretability) कहा जाता है।
समस्या यह है कि मशीन में लाखों नॉब्स हैं, जो आपस में उलझे हुए हैं। किसी एक को रैंडम तरीके से चुनना घास के ढेर में सुई खोजने जैसा है।
जेनिफर लिन का पेपर इन सुइयों को खोजने का एक नया, चतुर तरीका प्रस्तावित करता है। अनुमान लगाने के बजाय, लेखिका एक गणितीय उपकरण का उपयोग करती हैं जिसे एम्पीरिकल न्यूरल टेंगेंट कर्नेल (eNTK) कहा जाता है।
उपमा: "इको चैंबर" (गूँज कक्ष) परीक्षण
कल्पना कीजिए कि न्यूरल नेटवर्क एक विशाल इको चैंबर (गूँज कक्ष) है। यदि आप एक विशिष्ट शब्द चिल्लाते हैं (जैसे कोई फीचर जैसे "संज्ञा" या "5 जोड़ें"), तो ध्वनि कमरे में चारों ओर गूँजती है और दीवारों (मॉडल के पैरामीटर्स) से टकराती है, जिससे एक बहुत ही विशिष्ट पैटर्न बनता है।
eNTK एक अत्यधिक संवेदनशील माइक्रोफ़ोन की तरह है जो रिकॉर्ड करता है कि जब आप चिल्लाते हैं तो पूरा कमरा कैसे कंपन करता है।
- यदि आप "संज्ञा" (noun) चिल्लाते हैं, तो कमरा एक विशिष्ट लय में कंपन करता है।
- यदि आप "क्रिया" (verb) चिल्लाते हैं, तो कमरा एक अलग लय में कंपन करता है।
लेखिका की परिकल्पना है: यदि हम इस इको चैंबर में सबसे मजबूत कंपनों (principal eigen-directions) का विश्लेषण करें, तो हम सटीक रूप से पहचान सकते हैं कि कौन से शब्द चिल्लाए गए थे।
तकनीकी शब्दों में, पेपर का दावा है कि मॉडल के आंतरिक गियर कैसे चलते हैं, इसके "सबसे मजबूत पैटर्न" की जांच करके, हम उन विशिष्ट दिशाओं की पहचान कर सकते हैं जिनका उपयोग मॉडल फीचर्स का पता लगाने के लिए करता है।
तीन प्रयोग: सरल गणित से लेकर लार्ज लैंग्वेज मॉडल्स तक
लेखिका ने इस "इको चैंबर" विचार का परीक्षण तीन अलग-अलग प्रकार की मशीनों पर किया, जिनमें से प्रत्येक अधिक जटिल होती जा रही थी।
1. सरल गणित मशीन (MLP)
- कार्य: एक सरल मशीन ने एक प्राइम नंबर (एक विशिष्ट गणितीय पहेली) के मोड्यूलो में संख्याएँ जोड़ना सीखा।
- "सत्य": हम पहले से ही जानते थे कि मशीन जिस गुप्त रेसिपी का उपयोग करती है, वह क्या है: वह संख्याओं को तरंगों (फूरियर फीचर्स) में बदल देती है, उदाहरण के लिए, एक संख्या को साइन वेव (sine wave) में बदलकर।
- परिणाम: लेखिका ने मशीन को "सुनने" के लिए eNTK का उपयोग किया। eNTK द्वारा पाए गए सबसे मजबूत कंपन उस "साइन वेव" रेसिपी से पूरी तरह मेल खाते थे।
- "ग्रोकिंग" (Grokking) क्षण: एक घटना है जिसे "ग्रोकिंग" कहा जाता है, जहाँ एक मॉडल केवल याद रखने (memorization) के लंबे समय के बाद अचानक एक टेस्ट को पास करने से लेकर उसे पूरी तरह से हल करने की ओर बढ़ जाता है। पेपर ने पाया कि जिस क्षण मशीन ने "ग्रोक" किया (गणित को समझा), उस समय eNTK कंपनों और गणितीय फीचर्स के बीच का संरेखण (alignment) तेजी से बढ़ गया। यह ऐसा है जैसे, ठीक उसी क्षण जब मशीन ने अंततः इसे "समझ" लिया, इको चैंबर अचानक सही गाना गाने लगा।
2. थोड़ी स्मार्ट गणित मशीन (Transformer)
- कार्य: एक थोड़ा अधिक जटिल मशीन (एक ट्रांसफॉर्मर) ने उसी गणितीय पहेली को सीखा।
- अंतर: इस मशीन ने हर संभव वेव का उपयोग नहीं किया; इसने समस्या को हल करने के लिए कुछ रैंडम, विशिष्ट फ्रीक्वेंसी चुनीं।
- परिणाम: भले ही मशीन ने रैंडम फ्रीक्वेंसी चुनीं, eNTK ने उन्हें फिर भी ढूंढ लिया। इसने सफलतापूर्वक उन विशिष्ट "नोट्स" की पहचान की जिनका उपयोग मशीन ने गणित के लिए किया था।
3. लार्ज लैंग्वेज मॉडल (Gemma-3-270M)
- कार्य: यह एक वास्तविक, प्री-ट्रेन्ड लैंग्वेज मॉडल है (जैसे कि आपके साथ चैट करने वाला एक मिनी-AI) जो कहानियाँ पढ़ता है।
- चुनौती: यहाँ, हमें "गुप्त रेसिपी" नहीं पता है। हम केवल यह देखना चाहते हैं कि क्या मशीन व्याकरण (जैसे संज्ञा, क्रिया, या भूतकाल) को पहचान सकती है।
- परीक्षण: लेखिका ने कहानियों का एक छोटा सेट लिया और पूछा: "क्या eNTK कंपनों से हमें पता चल सकता है कि कौन से शब्द संज्ञा हैं?"
- तुलना: उन्होंने eNTK पद्धति की तुलना PCA (एक मानक, पुराना तरीका जो केवल मशीन के सबसे सक्रिय हिस्सों को देखता है) से की।
- परिणाम: eNTK पद्धति बेहतर थी। इसने पुराने तरीके की तुलना में "ग्रामर स्विच" को अधिक सटीकता से पाया। उदाहरण के लिए, यह पुराने तरीके की तुलना में "क्रियाओं" या "भूतकाल" को पहचानने में बेहतर था।
मुख्य निष्कर्ष
पेपर का दावा है कि मॉडल की सीखने की प्रक्रिया के "कंपनों" का विश्लेषण करना (eNTK के माध्यम से) एक शक्तिशाली नई टॉर्च है।
- यह सरल गणितीय मॉडलों पर काम करता है जहाँ हम उत्तर जानते हैं।
- यह जटिल भाषा मॉडलों पर भी काम करता है जहाँ हमें उत्तर नहीं पता है, और यह मौजूदा मानक उपकरणों की तुलना में व्याकरण संबंधी फीचर्स को बेहतर ढंग से पाता है।
- यह ठीक उसी समय सबसे अधिक चमकता है जब कोई मॉडल अचानक किसी अवधारणा को समझ लेता है ("ग्रोकिंग" क्षण)।
पेपर क्या दावा नहीं करता है
यह महत्वपूर्ण है कि हम केवल वही कहें जो पेपर वास्तव में कहता है:
- यह कोई रामबाण इलाज नहीं है: पेपर स्वीकार करता है कि ये "सहसंबंधी" (correlative) परिणाम हैं। केवल इसलिए कि eNTK एक ऐसी दिशा पाता है जो "व्याकरण" जैसी दिखती है, इसका मतलब यह साबित नहीं होता कि उस दिशा को बदलने से मॉडल को ठीक किया जा सकता है। यह एक खोज उपकरण (discovery tool) है, न कि अनिवार्य रूप से एक कंट्रोल पैनल।
- यह भविष्य की AI सुरक्षा के बारे में नहीं है: पेपर में उल्लेख है कि यह भविष्य में सुरक्षा के लिए उपयोगी हो सकता है, लेकिन यह सुरक्षा अनुप्रयोगों या क्लिनिकल उपयोगों को प्रस्तुत नहीं करता है। यह पूरी तरह से यह समझने का एक तरीका है कि मॉडल अभी कैसे काम करते हैं।
- यह पूर्ण (perfect) नहीं है: लैंग्वेज मॉडल वाले प्रयोग में अपेक्षाकृत छोटा डेटासेट और एक विशिष्ट मॉडल इस्तेमाल किया गया था। लेखिका ने सुझाव दिया है कि निश्चित होने के लिए बड़े मॉडलों और डेटासेट्स पर इसका परीक्षण किया जाना चाहिए।
एक वाक्य में सारांश
यह पेपर प्रस्तावित करता है कि न्यूरल नेटवर्क के सीखने के "प्रतिध्वनियों" (echoes) को सुनकर (eNTK नामक टूल का उपयोग करके), हम उन छिपे हुए "स्विचों" को सफलतापूर्वक पहचान सकते हैं जिनका उपयोग मॉडल गणित और व्याकरण को समझने के लिए करता है, और अक्सर इन्हें पिछले तरीकों की तुलना में अधिक स्पष्ट रूप से पाता है।
अपने क्षेत्र के पेपरों की भीड़ में उलझे हुए हैं?
आपके रिसर्च कीवर्ड से मेल खाने वाले सबसे नए और अलग सोच वाले पेपरों का रोज़ाना Digest पाएँ—तकनीकी सारांश के साथ, आपकी भाषा में।