Numerical stability analysis of large language models
यह शोध पत्र ट्रांसफॉर्मर इन्फरेंस (inference) का एक मिश्रित-परिशुद्धता (mixed-precision) विश्लेषण प्रस्तुत करता है जो लेयरनॉर्म (LayerNorm), RMSNorm और सेल्फ-अटेंशन (self-attention) जैसे प्रमुख घटकों के लिए नवीन त्रुटि सीमाएं (error bounds) और स्थिरता स्थितियां व्युत्पन्न करता है, जो अंततः यह प्रकट करता है कि संख्यात्मक स्थिरता भार के परिमाण (weight magnitudes) और रेसिडुअल स्ट्रीम की वृद्धि (residual stream growth) के बीच परस्पर क्रिया द्वारा नियंत्रित होती है, एक ऐसा निष्कर्ष जिसे GPT-2 पर प्रयोगों द्वारा सत्यापित किया गया है।
मूल पेपर CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/) के तहत लाइसेंस किया गया है। नीचे दिए गए पेपर की यह व्याख्या AI से तैयार की गई है। इसे लेखकों ने न तो लिखा है, न इसका समर्थन किया है। तकनीकी सटीकता के लिए मूल पेपर देखें। पूरा डिस्क्लेमर पढ़ें
एक लार्ज लैंग्वेज मॉडल (जैसे कि चैटबॉट्स को चलाने वाले मॉडल) की कल्पना एक विशाल, बहु-मंजिला कारखाने के रूप में करें। इस कारखाने के अंदर, जानकारी ज़मीन से लेकर ऊपरी मंजिल तक प्रवाहित होती है, जो सैकड़ों कमरों (परतों) से होकर गुजरती है। प्रत्येक कमरे में, जानकारी को संसाधित (process) किया जाता है, मिलाया जाता है और अगले कमरे में भेजने से पहले उसका आकार बदला जाता है।
प्रस्तुत शोध पत्र इस कारखाने के लिए एक सुरक्षा निरीक्षण रिपोर्ट (safety inspection report) है। लेखक पूछ रहे हैं: "यदि हम पहले कमरे में धूल के एक नन्हे, लगभग अदृश्य कण (एक छोटा सा कंप्यूटर राउंडिंग एरर) से शुरुआत करते हैं, तो जब तक वह ऊपर की मंजिल तक पहुँचेगा, वह कण कितना बड़ा हो जाएगा?"
यहाँ उनके निष्कर्षों का सरल उपमाओं (analogies) का उपयोग करके विवरण दिया गया है:
1. "लो-प्रिसिजन" (कम-सटीकता वाला) शॉर्टकट
इन कारखानों को तेज़ और सस्ता बनाने के लिए, इंजीनियर अक्सर "लो-प्रिसिजन" गणित का उपयोग करते हैं। इसे ऐसे समझें जैसे सामग्री को मापने के लिए मिलीमीटर के निशानों के बजाय केवल इंच के निशानों वाले रूलर का उपयोग करना। यह तेज़ है, लेकिन आप थोड़ी सटीकता खो देते हैं।
- समस्या: जब आप क्रमवार हजारों गणनाएँ करते हैं, तो वे छोटे "इंच मार्क" वाले एरर जमा होते जा सकते हैं। लेखक जानना चाहते थे: क्या इन छोटे एरर्स के भार से कारखाना ढह जाएगा, या यह स्थिर रहेगा?
2. "नॉर्मलाइजिंग" दरवाजे (LayerNorm बनाम RMSNorm)
एक नए कमरे में प्रवेश करने से पहले, जानकारी को एक "नॉर्मलाइजिंग दरवाजे" से गुजरना पड़ता है जो डेटा के आकार को समायोजित करता है।
- पुराना दरवाजा (LayerNorm): यह दरवाजा डेटा के औसत आकार को घटा देता है। लेखकों ने पाया कि यदि डेटा में एक विशाल आउटलायर (एक बहुत बड़ी संख्या जो बाकी सब से बहुत बड़ी है) है, तो यह दरवाजा डगमगा सकता है। यह एक सी-सॉ (seesaw) को संतुलित करने जैसा है जहाँ एक तरफ हाथी है और दूसरी तरफ चूहा; गणित यहाँ संवेदनशील हो जाता है।
- नया दरवाजा (RMSNorm): यह आधुनिक मॉडलों में उपयोग किया जाने वाला दरवाजा है। यह औसत को घटाता नहीं है; यह केवल कुल आकार के आधार पर स्केल करता है। लेखकों ने पाया कि यह दरवाजा पूर्णतः स्थिर (unconditionally stable) है। उस सी-सॉ पर हाथी होने पर भी, यह दरवाजा मजबूती से टिका रहता है। यह एक अधिक मजबूत डिज़ाइन है।
3. "अटेंशन" (Attention) स्पॉटलाइट
इन मॉडलों का सबसे प्रसिद्ध हिस्सा "सेल्फ-अटेंशन" (Self-Attention) है, जो यह तय करता है कि एक वाक्य के कौन से शब्द एक-दूसरे के लिए महत्वपूर्ण हैं।
- "सॉफ्टमैक्स" (Softmax) स्विच: यह वह तंत्र है जो कच्चे स्कोर को संभावनाओं (प्रतिशत) में बदल देता है। लेखकों ने "शिफ्टिंग" (सबसे बड़ी संख्या को घटाने) नामक एक सामान्य ट्रिक का विश्लेषण किया ताकि गणित को फटने (overflow होने) से रोका जा सके।
- निष्कर्ष: उन्होंने सिद्ध किया कि यह "शिफ्टिंग" ट्रिक सुरक्षित है। यह सिस्टम को शोर (noise) के प्रति थोड़ा अधिक संवेदनशील बना सकता है (अधिकतम 4 गुना), लेकिन यह गणित को तोड़ता नहीं है। यह सूरज को देखने के लिए चश्मा पहनने जैसा है; यह आपके देखने के तरीके को थोड़ा बदल देता है, लेकिन आपको अंधा नहीं करता।
- "अटेंशन सिंक" (Attention Sink): उन्होंने यह भी देखा कि लंबी बातचीत में, मॉडल अक्सर शुरुआती कुछ शब्दों (सिंक) पर बहुत अधिक ध्यान केंद्रित करता है। उनका गणित दिखाता है कि इन लंबी अनुक्रमों (sequences) के साथ भी, त्रुटियाँ पुराने सिद्धांतों के सुझाव के मुकाबले उतनी बुरी तरह से नियंत्रण से बाहर नहीं होती हैं।
4. "रेसिड्यूअल स्ट्रीम" (कन्वेयर बेल्ट)
डेटा एक "रेसिड्यूअल स्ट्रीम" के माध्यम से कारखाने में बहता है—एक कन्वेयर बेल्ट जो मुख्य संदेश को आगे ले जाती है।
- वृद्धि: जैसे-जैसे डेटा ऊपरी मंजिलों की ओर बढ़ता है, कन्वेयर बेल्ट स्वाभाविक रूप से "भारी" (संख्याएँ बड़ी) होती जाती है।
- वेट्स (Weights): संतुलन बनाए रखने के लिए कारखाने के ऊंचे होने के साथ-साथ मशीनों (weights) को कम (scale down) किया जाता है।
- बड़ी खोज: लेखकों ने एक स्वर्णिम नियम पाया: यदि आप कन्वेयर बेल्ट को ऊपर या नीचे स्केल करते हैं, तो सापेक्ष त्रुटि (relative error) वही रहती है।
- उपमा: कल्पना कीजिए कि आप एक ट्रेडमिल पर चल रहे हैं। यदि आप ट्रेडमिल की गति को दोगुना कर देते हैं और साथ ही अपने जूतों का आकार भी दोगुना कर देते हैं, तो आपकी चलने की शैली (आपके आकार के सापेक्ष) नहीं बदलती है।
- शर्त: यह तभी काम करता है जब कन्वेयर बेल्ट "रैखिक" (linear/predictable) व्यवहार करती है। यदि आप वेट्स को बहुत अधिक स्केल करते हैं, तो कन्वेयर बेल्ट अचानक पूरी तरह से अलग मोड में जा सकती है (एक "क्वालिटेटिव ट्रांज़िशन")। उस विशिष्ट मामले में, स्थिरता टूट जाती है। लेकिन जब तक बेल्ट अपने सामान्य लय में रहती है, वेट्स को स्केल करने से सटीकता को नुकसान नहीं पहुँचता है।
5. "GPT-2" प्रयोग
यह साबित करने के लिए कि उनका गणित केवल सिद्धांत नहीं था, उन्होंने GPT-2 नामक एक वास्तविक मॉडल पर इसका परीक्षण किया।
- रैंडम वेट्स (Random Weights): जब उन्होंने रैंडम, अनट्रेन्ड वेट्स वाला मॉडल इस्तेमाल किया, तो त्रुटियाँ धीरे-धीरे और अनुमानित रूप से बढ़ीं।
- ट्रेन्ड वेट्स (Trained Weights): जब उन्होंने एक वास्तविक, प्रशिक्षित मॉडल का उपयोग किया, तो त्रुटियाँ थोड़ी तेज़ी से (एक्सपोनेंशियल रूप से) बढ़ीं, लेकिन फिर भी प्रबंधनीय सीमाओं के भीतर रहीं।
- निर्णय: गणित सही साबित हुआ। "सापेक्ष त्रुटि" (डेटा के आकार के मुकाबले गलती) सुसंगत रही, जब तक कि उन्होंने मॉडल को बहुत अधिक वजन बदलकर एक अजीब, अस्थिर स्थिति में नहीं धकेल दिया।
सारांश
शोध पत्र यह निष्कर्ष निकालता है कि लार्ज लैंग्वेज मॉडल्स संख्यात्मक रूप से स्थिर (numerically stable) हैं।
भले ही वे "रफ" गणित (लो-प्रिसिजन) का उपयोग करते हैं और उन्हें विशाल संख्याओं को संभालना पड़ता है, लेकिन आर्किटेक्चर इस तरह से डिज़ाइन किया गया है कि त्रुटियाँ आपदा में नहीं बदलतीं। इसकी स्थिरता की कुंजी वेट्स के आकार और डेटा स्ट्रीम की वृद्धि के बीच का तालमेल है। जब तक मॉडल को किसी अजीब, अस्थिर स्थिति में नहीं धकेला जाता, तब तक लो-प्रिसिजन गणित द्वारा उत्पन्न "शोर" (noise) एक छोटी, प्रबंधनीय धूल के कण जैसा रहता है, न कि किसी भूस्खलन जैसा।
अपने क्षेत्र के पेपरों की भीड़ में उलझे हुए हैं?
आपके रिसर्च कीवर्ड से मेल खाने वाले सबसे नए और अलग सोच वाले पेपरों का रोज़ाना Digest पाएँ—तकनीकी सारांश के साथ, आपकी भाषा में।