Hidden Decoding at Scale: Latent Computation Scaling for Large Language Models
यह शोध पत्र हिडन डिकोडिंग (Hidden Decoding) को प्रस्तुत करता है, जो एक नवीन स्केलिंग विधि है जो स्ट्रीम फैक्टराइजेशन के माध्यम से अनुक्रम-लंबाई (sequence-length) आयाम के साथ टोकन गणना का विस्तार करके लार्ज लैंग्वेज मॉडल्स को उन्नत करती है, जिससे फिक्स्ड ट्रांसफार्मर बैकबोन को संशोधित किए बिना या अत्यधिक प्रशिक्षण लागतों के बिना फ्रंटियर स्केल्स पर महत्वपूर्ण प्रदर्शन लाभ सक्षम होता है।
मूल पेपर CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/) के तहत लाइसेंस किया गया है। नीचे दिए गए पेपर की यह व्याख्या AI से तैयार की गई है। इसे लेखकों ने न तो लिखा है, न इसका समर्थन किया है। तकनीकी सटीकता के लिए मूल पेपर देखें। पूरा डिस्क्लेमर पढ़ें
कल्पना कीजिए कि आपके पास एक सुपर-स्मार्ट रोबोट दिमाग (एक लार्ज लैंग्वेज मॉडल) है जो पहले से ही काफी अद्भुत है। आमतौर पर, इसे और भी स्मार्ट बनाने के लिए, वैज्ञानिक इसके दिमाग को बड़ा करने की कोशिश करते हैं—अधिक परतें (layers), अधिक न्यूरॉन्स, और सब कुछ जोड़कर। लेकिन यह एक गगनचुंबी इमारत के ऊपर दूसरी गगनचुंबी इमारत बनाने जैसा है: इसमें बहुत पैसा खर्च होता है, इसमें बहुत समय लगता है, और कभी-कभी निर्माण दल (ट्रेनिंग कंप्यूटर्स) इसका आकार संभालने में सक्षम नहीं होते।
वीचैट (WeChat) एआई टीम ने एक अलग सवाल पूछा: क्या होगा अगर हम दिमाग का आकार बिल्कुल वही रखें, लेकिन उसे प्रत्येक शब्द के बारे में सोचने के लिए अधिक समय दें?
"लूपिंग" (Looping) के साथ समस्या
कुछ शोधकर्ताओं ने इसे रोबोट के सोचने के तरीके को "लूप" बनाकर हल करने की कोशिश की। कल्पना कीजिए कि एक छात्र एक वाक्य पढ़ता है, फिर उसे फिर से पढ़ता है, फिर से पढ़ता है, एक ही ब्रेन सेल्स का बार-बार उपयोग करके एक बेहतर उत्तर प्राप्त करता है। इसे "लूप्ड" या "रिकरेंट" (recurrent) मॉडल कहा जाता है।
लेकिन यहाँ एक पेंच है: जब आप सबसे बड़े रोबोट्स (वे जिनमें सैकड़ों अरब पैरामीटर्स होते हैं) को प्रशिक्षित करने की कोशिश करते हैं, तो यह लूपिंग वाला विचार असेंबली लाइन को तोड़ देता है। जो कंप्यूटर इन मॉडल्स को प्रशिक्षित करते हैं, वे एक पाइपलाइन में काम करते हैं, जहाँ दिमाग का प्रत्येक हिस्सा अपना काम एक बार करता है और फिर अगली बारी सौंप देता है। यदि आप रोबोट को एक ही वाक्य को बार-बार पढ़ने के लिए रोक देते हैं, तो पूरी लाइन रुक जाती है, और महंगे कंप्यूटर खाली बैठे रहते हैं। यह एक फैक्ट्री कन्वेयर बेल्ट की तरह है जो एक कर्मचारी को एक ही पेंच को दोबारा करने देने के लिए बार-बार रुक जाती है; इससे फैक्ट्री का काम ठप हो जाता है।
समाधान: "हिडन डिकोडिंग" (Hidden Decoding - गुप्त असेंबली लाइन)
वीचैट टीम ने हिडन डिकोडिंग नामक एक चतुर तकनीक विकसित की। रोबोट को एक ही शब्द को बार-बार लूप में पढ़ने के बजाय, वे रोबोट को उस शब्द को एक साथ कई बार पढ़ने के लिए कहते हैं, लेकिन गुप्त समानांतर लेन (parallel lanes) में।
इसे एक रेस कार टीम की तरह समझें। एक कार को सबसे अच्छा लैप टाइम पाने के लिए ट्रैक के चक्कर लगाने के बजाय, वे एक साथ चार समान कारें (streams) ट्रैक पर भेजते हैं।
- सेटअप: जब रोबमा को एक शब्द (जैसे "apple") दिखता है, तो वह केवल उसे एक कोड में नहीं बदलता। वह उसे चार अलग-अलग कोड में बदल देता है, जिनमें से प्रत्येक अपनी गुप्त लेन में जाता है।
- गुप्त कार्य: ये चार लेन दिमाग की परतों से होकर गुजरती हैं। पहली तीन लेन "छिपी हुई" (hidden) होती हैं। वे सारा भारी काम, सोच-विचार और तर्क करती हैं, लेकिन उन्हें अभी उत्तर बोलने की अनुमति नहीं है। वे बैक रूम में काम करने वाली विचार-मंथन टीम (brainstorming team) की तरह हैं।
- अंतिम निर्णय: केवल चौथी लेन (अंतिम स्ट्रीम) को अगला शब्द बोलने की अनुमति होती है। दिमाग को केवल इस अंतिम उत्तर के लिए ही ग्रेड दिया जाता है।
- स्मृति (Memory): महत्वपूर्ण बात यह है कि दिमाग याद रखता है कि पहली तीन लेन ने क्या सोचा था। वे अपने नोट्स (जिन्हें "की-वैल्यू कैश" कहा जाता है) रखते हैं ताकि अगला शब्द पिछले शब्द के विचार-मंथन वाले नोट्स को पढ़ सके।
इस तरह, रोबोट को हर शब्द के लिए चार गुना अधिक सोचने की शक्ति मिलती है, लेकिन इसे एक बड़ा दिमाग बनाने या असेंबली लाइन को रोकने की आवश्यकता नहीं होती। यह बस एक बार में डेटा के एक लंबे अनुक्रम (sequence) को प्रोसेस करता है।
किफायती बनाना: "स्ट्रीम-फैक्टराइज्ड" (Stream-Factorized) ट्रिक
आप सोच सकते हैं, "रुको, अगर मेरी चार लेन आपस में बात कर रही हैं, तो क्या गणित बहुत जटिल नहीं हो जाएगा?" यदि प्रत्येक लेन हर चरण में एक-दूसरे से बात करने की कोशिश करती, तो लागत बहुत बढ़ जाती (यह 16 गुना तक बढ़ जाती!)।
इसे ठीक करने के लिए, टीम ने स्ट्रीम-फैक्टराइज्ड अटेंशन (Stream-Factorized Attention) नामक एक विधि का उपयोग किया।
- ज्यादातर समय: लेन अपने स्वयं के बुलबुलों (bubbles) में रहती हैं। लेन 1, लेन 1 से बात करती है, लेन 2, लेन 2 से। वे एक-दूसरे को परेशान नहीं करतीं।
- कभी-कभी: केवल कुछ विशिष्ट परतें ही लेन को नोट्स बदलने और अपने विचारों को मिलाने की अनुमति देती हैं।
यह लागत को कम रखता है। ऐसा नहीं होता कि लागत 16 गुना बढ़ जाए, बल्कि यह केवल लगभग 4.4 से 5.1 गुना तक बढ़ती है (जो कि 4x लेन की वृद्धि के करीब है)। यह इसे संभव बनाता है कि बिना बजट बिगाड़े इन सुपर-लार्ज मॉडल्स को प्रशिक्षित किया जा सके।
क्या यह काम आया?
टीम ने दो विशाल मॉडल्स पर इसका परीक्षण किया: एक 80 बिलियन पैरामीटर्स वाला और एक विशाल 617 बिलियन वाला मॉडल। उन्होंने दिमाग का आकार नहीं बदला; उन्होंने बस "4-लेन" मोड चालू कर दिया।
- परिणाम: मॉडल्स अधिक स्मार्ट हो गए। कठिन गणित और विज्ञान परीक्षणों पर, 617-बिलियन मॉडल ने "GPQA डायमंड" टेस्ट पर अपने स्कोर को 89.1 से बढ़ाकर 91.2 कर दिया। भौतिकी के "PHYBench" टेस्ट पर, यह 75.3 से बढ़कर 76.3 हो गया।
- ट्रेंड (Trend): उन्होंने 2 लेन, 4 लेन और 8 लेन के साथ भी परीक्षण किया। जैसे-जैसे उन्होंने अधिक लेन जोड़ीं, स्कोर बढ़ता गया। यह सुझाव देता है कि अधिक "सोचने वाली लेन" जोड़ना, नया बड़ा दिमाग बनाए बिना मॉडल्स को स्मार्ट बनाने का एक वास्तविक तरीका है।
उन्होंने क्या खारिज किया
टीम ने सावधानीपूर्वक जांच की कि क्या यह केवल एक इत्तेफाक था। उन्होंने अतिरिक्त लेन का उपयोग करने के अन्य तरीकों को भी आजमाया:
- हर लेन को अनुमान लगाने के लिए मजबूर करना: यदि उन्होंने पहली तीन लेन को भी उत्तर का अनुमान लगाने के लिए मजबूर किया (केवल अंतिम लेन के बजाय), तो मॉडल वास्तव में खराब हो गया।
- उत्तरों को मिलाना: यदि उन्होंने केवल चारों लेन के विचारों का औसत निकाला, तो यह अंतिम लेन द्वारा निर्णय लेने जितना प्रभावी नहीं रहा।
- नोट्स साझा करना: यदि उन्होंने लेन को अपने स्वयं के नोट्स रखने के बजाय एक ही मेमोरी नोट्स साझा करने के लिए कहा, तो मॉडल थोड़ा कम सटीक हो गया।
यह साबित करता है कि असली सफलता (secret sauce) पहली लेन को शांत, निजी सोच करने देने और उनके नोट्स को अलग रखने में है, ताकि अंतिम लेन उस गहरे, संचित ज्ञान का उपयोग करके सबसे अच्छा अनुमान लगा सके।
निचोड़ (The Bottom Line)
हिडन डिकोडिंग दिखाती है कि स्मार्ट होने के लिए आपको हमेशा एक बड़े दिमाग की आवश्यकता नहीं होती। कभी-कभी, आपको बस दिमाग को समानांतर (parallel) रूप से सोचने के लिए अधिक समय देने की आवश्यकता होती है। एक शब्द को विचारों की एक गुप्त असेंबली लाइन में बदलकर, वीचैट टीम ने मौजूदा मॉडल्स की बुद्धिमत्ता को बढ़ाने का एक तरीका खोजा है, बिना एक नया, बड़ा दिमाग बनाए। यह हमारे पास मौजूद मॉडल्स से अधिक बुद्धिमत्ता निकालने का एक व्यावहारिक और इंजीनियरिंग-अनुकूल तरीका है।
अपने क्षेत्र के पेपरों की भीड़ में उलझे हुए हैं?
आपके रिसर्च कीवर्ड से मेल खाने वाले सबसे नए और अलग सोच वाले पेपरों का रोज़ाना Digest पाएँ—तकनीकी सारांश के साथ, आपकी भाषा में।