Scaling with Collapse: Efficient and Predictable Training of LLM Families
यह शोध पत्र प्रदर्शित करता है कि इष्टतम स्केलिंग रेसिपी के तहत LLM परिवारों के लिए ट्रेनिंग लॉस कर्व्स एक सार्वभौमिक प्रक्षेपवक्र (यूनिवर्सल ट्रजेक्टरी) पर सिमट जाते हैं, जो इस घटना को कंप्यूट-कुशल प्रशिक्षण के एक हस्ताक्षर के रूप में स्थापित करता है जो विकृतियों की शीघ्र पहचान और हाइपरपैरामीटर ट्यूनिंग को सक्षम बनाता है, जिसका उपयोग लेखक प्रतिस्पर्धी *Celerity* मॉडल परिवार विकसित करने के लिए करते हैं।
मूल पेपर CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/) के तहत लाइसेंस किया गया है। नीचे दिए गए पेपर की यह व्याख्या AI से तैयार की गई है। इसे लेखकों ने न तो लिखा है, न इसका समर्थन किया है। तकनीकी सटीकता के लिए मूल पेपर देखें। पूरा डिस्क्लेमर पढ़ें
कल्पना कीजिए कि आप छात्रों के एक समूह को एक ही जटिल विषय पढ़ा रहे हैं, जिसमें उनकी उम्र और क्षमता का स्तर अलग-अलग है—एक किंडरगार्टनर से लेकर एक पीएचडी उम्मीदवार तक।
आर्टिफिशियल इंटेलिजेंस की दुनिया में, ये "छात्र" लार्ज लैंग्वेज मॉडल्स (LLMs) हैं। आमतौर पर, जब शोधकर्ता इन मॉडल्स को प्रशिक्षित (train) करते हैं, तो वे एक दीवार से टकरा जाते हैं: स्केल (scale) अप्रत्याशित होता है। एक रणनीति जो एक छोटे मॉडल के लिए पूरी तरह से काम करती है, वह एक विशाल मॉडल के लिए बुरी तरह विफल हो सकती है। यह एक अकेले कुकी (cookie) के लिए रेसिपी का उपयोग करके हज़ार लोगों के लिए केक बनाने की कोशिश करने जैसा है; गणित मेल नहीं खाता, और अक्सर अंत में या तो सब कुछ जल जाता है या आटा कच्चा रह जाता है।
यह पेपर, "स्केलिंग विद कोलैप्स" (Scaling with Collapse), इन AI मॉडल्स को प्रशिक्षित करने का एक क्रांतिकारी तरीका पेश करता है जो इस प्रक्रिया को पूर्वानुमानित (predictable), कुशल और यहाँ तक कि हमें समस्याओं को पूरे बैच को बर्बाद करने से पहले पहचानने की अनुमति भी देता है।
यहाँ सरल उपमाओं (analogies) का उपयोग करके इसका विवरण दिया गया है:
1. समस्या: "लामा-2" (Llama-2) का बिखराव
लेखकों ने Llama-2 जैसे प्रसिद्ध मॉडल्स का अध्ययन किया। उन्होंने देखा कि जब आप एक 7-बिलियन-पैरामीटर वाले मॉडल और एक 70-बिलियन-पैरामीटर वाले मॉडल को प्रशिक्षित करते हैं, तो उनके "लर्निंग कर्व्स" (ग्राफ जो दिखाते हैं कि वे समय के साथ कितनी अच्छी तरह सीख रहे हैं) पूरी तरह से अलग दिखते हैं।
- उपमा: कल्पना कीजिए कि दो धावक हैं। एक स्प्रिंटर (तेज दौड़ने वाला) है, दूसरा मैराथन धावक। यदि आप उनकी प्रगति को देखते हैं, तो उनके रास्ते अराजक होते हैं और आपस में मेल नहीं खाते। आप केवल स्प्रिंटर की गति को देखकर यह नहीं बता सकते कि मैराथन धावक अच्छा कर रहा है या नहीं। यह बड़े मॉडल्स को प्रशिक्षित करना "अनुमान लगाने और जांचने" का एक खेल बना देता है, जो अविश्वसनीय रूप से महंगा और धीमा है।
2. खोज: "द ग्रेट कोलैप्स" (The Great Collapse)
शोधकर्ताओं को एक गुप्त नुस्खा मिला। यदि आप प्रशिक्षण सेटिंग्स को बिल्कुल सही तरीके से ट्यून करते हैं, तो सभी अलग-अलग आकार के मॉडल अचानक एक ही ट्रैक पर दौड़ने लगते हैं।
वे इसे "कोलैप्स" (Collapse) कहते हैं।
- उपमा: कल्पना कीजिए कि आपके पास एक पहाड़ का नक्शा है। यदि आप एक छोटी पहाड़ी पर ज़ूम इन करते हैं या एक विशाल पर्वत श्रृंखला पर ज़ूम आउट करते हैं, तो ढलान का आकार अलग दिखता है। लेकिन, यदि आप नक्शे को सामान्य (normalize) कर देते हैं (स्केल को समायोजित करते हैं), तो आप महसूस करते हैं कि हर पहाड़ का ढलान प्रोफाइल बिल्कुल एक जैसा है।
- पेपर में, वे दिखाते हैं कि यदि आप तीन विशिष्ट "नॉब्स" (knobs - नियंत्रण बटन) को ठीक कर देते हैं (कितना डेटा प्रत्येक पैरामीटर देखता है, ऑप्टिमाइज़र कितनी जल्दी पुरानी गलतियों को "भूलता" है, और लर्निंग रेट शेड्यूल), तो एक 300-मिलियन-पैरामीटर मॉडल और एक 3.9-बिलियन-पैरामीटर मॉडल के लर्निंग कर्व्स एक ही सार्वभौमिक रेखा पर कोलैप्स हो जाते हैं।
3. तीन जादुई नॉब्स (The Three Magic Knobs)
इस "कोलैप्स" को प्राप्त करने के लिए, आपको तीन चीजों को पूरी तरह से ट्यून करना होता है:
- TPP (टोकन्स पर पैरामीटर): मस्तिष्क के प्रत्येक हिस्से को अध्ययन करने के लिए कितना "पढ़ने का साहित्य" मिलता है।
- (टाऊ - मेमोरी का नॉब): यह नियंत्रित करता है कि मॉडल अपनी पिछली गलतियों को कितनी देर तक याद रखता है। इसे एक छात्र की अल्पकालिक स्मृति (short-term memory) की तरह समझें। यदि मेमोरी बहुत कम है, तो वे अभी जो सीखा है उसे भूल जाएंगे। यदि यह बहुत अधिक है, तो वे पुरानी त्रुटियों में फंस जाएंगे। पेपर ने पाया कि इसके लिए डेटा के उपयोग के आधार पर एक "गोल्डिलॉक्स" (Goldilocks - एकदम सही) सेटिंग मौजूद है।
- लर्निंग रेट शेड्यूल (Learning Rate Schedule): शुरुआत में और अंत में मॉडल कितनी तेजी से सीखता है।
जब ये तीनों सही ढंग से सेट होते हैं, तो सभी आकार के मॉडल समान व्यवहार करते हैं।
4. यह क्यों मायने रखता है: दो महाशक्तियाँ (Two Superpowers)
एक बार जब आपके पास यह "कोलैप्स" आ जाता है, तो आपको दो महाशक्तियाँ मिलती हैं:
A. "कोयले की खान में कैनरी" (Canary in the Coal Mine - शुरुआती चेतावनी प्रणाली)
आमतौर पर, जब कोई ट्रेनिंग रन गलत हो जाता है (कंप्यूटर की खराबी या खराब डेटा के कारण), तो लॉस कर्व (त्रुटि दर) लंबे समय तक ठीक दिख सकता है और फिर अचानक बढ़ सकता है। जिस समय तक आप उस उछाल को देखते हैं, तब तक आपने कंप्यूटिंग पावर के लाखों डॉलर बर्बाद कर दिए होते हैं।
- उपमा: कल्पना कीजिए कि आप कारों का एक बेड़ा चला रहे हैं। यदि एक कार थोड़ा सा भी अपने यूनिवर्सल रोड से भटकने लगती है, तो आप तुरंत जान जाते हैं कि उस विशिष्ट कार के इंजन के साथ कुछ गलत है, भले ही स्पीडोमीटर ने अभी तक बदलाव न दिखाया हो।
- परिणाम: लेखकों ने इसका उपयोग अपने 1.8-बिलियन मॉडल में एक छिपी हुई कंप्यूटर त्रुटि को पहचानने के लिए किया, जो कि हफ्तों पहले हुआ जब वह त्रुटि नग्न आंखों से दिखाई देने वाली होती। उन्होंने इसे ठीक किया और अपने ट्रेनिंग रन को बचा लिया।
B. "क्रिस्टल बॉल" (Crystal Ball - भविष्य देखने की शक्ति)
एक विशाल AI मॉडल को प्रशिक्षित करने में महीनों लगते हैं और इसमें भारी खर्च आता है। आमतौर पर, आपको यह देखने के लिए अंत तक प्रतीक्षा करनी पड़ती है कि आपकी सेटिंग्स (जैसे बैच साइज या लर्निंग रेट) अच्छी थीं या नहीं।
- उपमा: कल्पना कीजिए कि आप एक परफेक्ट रेसिपी खोजने के लिए कुकीज़ के 100 अलग-अलग बैच बना रहे हैं। सामान्यतः, आपको उन्हें चखने के लिए पूरी तरह से बेक होने और ठंडा होने तक इंतजार करना पड़ता है। इसमें बहुत समय लगता है।
- परिणाम: क्योंकि सभी कर्व्स एक अनुमानित पथ पर कोलैप्स होते हैं, इसलिए लेखक केवल 10% से 30% बेकिंग समय के बाद ही एक कुकी को देख सकते हैं। अपनी "यूनिवर्सल रेसिपी" के साथ उस आंशिक कुकी की तुलना करके, वे सटीक रूप से भविष्यवाणी कर सकते हैं कि अंतिम कुकी का स्वाद कैसा होगा। वे खराब बैचों को जल्दी रोक सकते हैं और भारी मात्रा में समय और पैसा बचा सकते हैं।
5. परिणाम: "सेलेरिटी" (Celerity)
इन जानकारियों का उपयोग करते हुए, टीम ने सेलेरिटी (Celerity) नामक मॉडल्स का एक नया परिवार बनाया।
- उन्होंने इन मॉडल्स को अत्यधिक कुशल (कम कंप्यूटिंग पावर का उपयोग करके समान बुद्धिमत्ता प्राप्त करना) बनाने के लिए प्रशिक्षित किया।
- उन्होंने साबित किया कि "कोलैप्स" नियमों का पालन करके, वे सबसे बड़े और सबसे महंगे मॉडल्स के साथ प्रतिस्पर्धा करने वाले मॉडल्स बना सकते हैं, लेकिन बहुत कम बर्बादी के साथ।
सारांश
यह पेपर AI के सीखने के लिए "यूनिवर्सल ग्रामर" खोजने जैसा है।
- पहले: AI को प्रशिक्षित करना एक ऐसी क्लास को पढ़ाने की कोशिश करने जैसा था जहाँ हर छात्र को एक अलग, अप्रत्याशित पाठ योजना की आवश्यकता होती थी।
- अब: हमने पाया कि यदि हम सही "मेमोरी" और "पढ़ने की गति" सेट करते हैं, तो सबसे छोटे से लेकर सबसे बड़े तक, हर छात्र बिल्कुल एक ही लय में सीखता है।
- लाभ: अब हम एक ट्रेनिंग रन के भविष्य की भविष्यवाणी कर सकते हैं, त्रुटियों को तुरंत पहचान सकते हैं, और खराब प्रयोगों पर पैसा बर्बाद करना बंद कर सकते हैं। यह AI प्रशिक्षण को एक अराजक जुए से बदलकर एक सटीक, पूर्वानुमानित विज्ञान में बदल देता है।
अपने क्षेत्र के पेपरों की भीड़ में उलझे हुए हैं?
आपके रिसर्च कीवर्ड से मेल खाने वाले सबसे नए और अलग सोच वाले पेपरों का रोज़ाना Digest पाएँ—तकनीकी सारांश के साथ, आपकी भाषा में।