Looped Latent Attention: Cross-Loop KV Compression for Looped Transformers
यह शोध पत्र Looped Latent Attention (LLA) प्रस्तुत करता है, जो एक पोस्ट-ट्रेनिंग कोडेक है जो वेट-टाइड ट्रांसफॉर्मर्स में लूप-इंडेक्स्ड KV कैश की लो-रैंक संरचना का लाभ उठाकर SVD इनिशियलाइजेशन और डिस्टिलेशन के माध्यम से नियर-लॉसलेस प्रदर्शन बनाए रखते हुए अत्यधिक संपीड़न (32x तक) प्राप्त करता है और बैच क्षमता को महत्वपूर्ण रूप से बढ़ाता है।
मूल पेपर CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/) के तहत लाइसेंस किया गया है। नीचे दिए गए पेपर की यह व्याख्या AI से तैयार की गई है। इसे लेखकों ने न तो लिखा है, न इसका समर्थन किया है। तकनीकी सटीकता के लिए मूल पेपर देखें। पूरा डिस्क्लेमर पढ़ें
AI दिमागों में मेमोरी की समस्या
कल्पना कीजिए कि आप एक रोबोट को कहानी लिखना सिखाने की कोशिश कर रहे हैं। रोबोट के पास लाखों छोटे स्विचों (पैरामीटर्स) से बना एक दिमाग है जो जानते हैं कि शब्द आपस में कैसे जुड़ते हैं। लेकिन यहाँ एक पेंच है: एक लंबी कहानी लिखने के लिए, रोबोट केवल अपने दिमाग को नहीं देख सकता; उसे एक रफ-नोटबुक (स्क्रैचपैड) की आवश्यकता होती है। हर बार जब वह एक नया शब्द लिखता है, तो उसे यह याद रखना पड़ता है कि उसने अभी तक क्या लिखा है ताकि वह खुद को दोहरा न दे या कहानी का संदर्भ न खो दे। आर्टिफिशियल इंटेलिजेंस की दुनिया में, इस स्क्रैचपैड को KV कैश (KV cache) कहा जाता है। यह हर एक बातचीत के लिए एक निजी नोटबुक है, और जैसे-जैसे रोबोट बोलता जाता है, यह बड़ा होता जाता है।
अब, एक विशेष प्रकार के रोबोट की कल्पना करें जिसे बेहद कुशल बनने के लिए डिज़ाइन किया गया है। अलग-अलग कार्यों के लिए कई अलग-अलग कमरों (लेयर्स) वाले विशाल दिमाग के बजाय, इस रोबोट के पास एक छोटा, चतुर कमरा है जहाँ वह बार-बार जाता है। वह निर्देशों के एक ही सेट का पुन: उपयोग करता है, गहराई से सोचने के लिए उन्हें बार-बार दोहराता है। इसे लूपड ट्रांसफार्मर (Looped Transformer) कहा जाता है। यह दिमाग के स्थान को बचाता है, जो कि बहुत अच्छी बात है। लेकिन एक छिपी हुई समस्या है: भले ही रोबोट एक ही कमरे का पुन: उपयोग करता है, फिर भी वह हर बार लूप करने पर अपने स्क्रैचपैड में एक नया पन्ना लिखता है। यदि रोबोट गणित की समस्या हल करने के लिए चार बार लूप करता है, तो उसके पास कहानी के उसी क्षण के लिए नोट्स के चार अलग-अलग पन्ने हो जाते हैं। इसका मतलब है कि "लूप वाला" रोबोट चलाने के लिए अभी भी बहुत अधिक मेमोरी की आवश्यकता रखता है, जिससे कुशलता का उद्देश्य ही समाप्त हो जाता है। वैज्ञानिकों के सामने बड़ा सवाल यह है: क्या हम उस स्क्रैचपैड को छोटा कर सकते हैं बिना यह भुलाए कि रोबोट को कैसे सोचना है?
शोध का निष्कर्ष: नोट्स में एक गुप्त शॉर्टकट
यह पेपर उस मेमोरी समस्या को हल करने के लिए लूपड लेटेंट अटेंशन (Looped Latent Attention - LLA) नामक एक चतुर तकनीक पेश करता है। शोधकर्ताओं ने पाया कि रोबोट के नोट्स उतने अस्त-व्यस्त नहीं हैं जितने वे दिखते हैं। जब रोबोट अपनी सोचने की प्रक्रिया के माध्यम से लूप करता है, तो एक ही शब्द के लिए उसके द्वारा लिखे गए नोट्स बेतरतीब ढंग से नहीं बदलते; वे एक सुचारू, अनुमानित पथ का पालन करते हैं, जैसे एक कंचा एक ढलान वाली पहाड़ी से नीचे लुढ़क रहा हो। रोबोट के स्क्रैचपैड के हर एक पन्ने को बचाने के बजाय, यह नई विधि उस पथ का केवल एक छोटा सा "सारांश" और उन निर्देशों का एक सेट सहेजती है कि जब रोबोट को वास्तव में उन्हें पढ़ने की आवश्यकता हो, तो विशिष्ट पन्नों को कैसे फिर से बनाया जाए।
इसे ऐसे समझें: कल्पना कीजिए कि आप एक फिल्म देख रहे हैं जहाँ मुख्य पात्र एक गलियारे में चल रहा है। एक सामान्य कंप्यूटर में, आप हर एक कदम पर पात्र की एक पूर्ण, हाई-डेफिनिशन फोटो सहेजेंगे। इसमें बहुत अधिक स्टोरेज लगता है। लेकिन LLA के साथ, आप महसूस करते हैं कि पात्र बस एक सीधी रेखा में चल रहा है। इसलिए, हजारों फोटो सहेजने के बजाय, आप एक फोटो और एक सरल नोट सहेजते हैं: "चरण 2 के लिए 1 इंच आगे बढ़ें, चरण 3 के लिए 2 इंच आगे बढ़ें।" जब कंप्यूटर को चरण 3 पर पात्र को देखने की आवश्यकता होती है, तो वह उस नोट का उपयोग करके जल्दी से वह चित्र बना लेता है। AI के लिए यह पेपर बिल्कुल यही करता है: यह मेमोरी के "लूप" वाले हिस्से को संकुचित (compress) करता है, चार पन्नों के ढेर को एक एकल, छोटे सारांश में बदल देता है जिसे तुरंत पूर्ण पन्नों में वापस विस्तारित किया जा सकता है।
शोधकर्ताओं ने इसे Ouro-1.4B जैसे कई AI मॉडलों पर परखा। उन्होंने पाया कि यह तरीका अविश्वसनीय रूप से अच्छा काम करता है। वास्तव में, H200 नामक एक शक्तिशाली कंप्यूटर चिप पर, वे एक साथ 768 अलग-अलग बातचीत को मेमोरी में फिट करने में सक्षम थे, जबकि पहले वे केवल 32 ही फिट कर सकते थे। यह इस बात में एक बड़ी छलांग है कि कितने लोग एक ही समय में AI का उपयोग कर सकते हैं। उन्होंने यह भी दिखाया कि यह ट्रिक तब भी काम करती है जब AI बहुत लंबी, कठिन गणितीय समस्याओं को हल कर रहा होता है, हालांकि इसे यह सुनिश्चित करने के लिए थोड़े अतिरिक्त प्रशिक्षण की आवश्यकता होती है कि यह भ्रमित न हो जाए जब यह केवल शिक्षक की नकल करने के बजाय अपनी खुद की कहानी लिख रहा हो।
यह पेपर किस बात को "ना" कहता है
यह जानना महत्वपूर्ण है कि यह विधि क्या नहीं है। शोधकर्ताओं ने पहले एक बहुत ही सरल विचार का परीक्षण किया: "क्या होगा यदि हम नोट्स के पहले तीन पन्नों को फेंक दें और केवल अंतिम पन्ने को रखें?" उन्होंने सोचा कि शायद रोब de कुछ लूप्स के बाद एक अंतिम उत्तर पर स्थिर हो जाता है, इसलिए शुरुआती नोट्स मायने नहीं रखते। उन्होंने इसे आजमाया, और यह एक आपदा साबित हुआ। AI ने गणित करना पूरी तरह से भूल गया, और परीक्षणों में शून्य स्कोर किया। यह साबित करता है कि नोट्स जो "पथ" लेते हैं वह महत्वपूर्ण है; आप केवल अंत बिंदु को नहीं पकड़ सकते। आपको यात्रा को पकड़ना होगा।
उन्होंने अन्य तरीकों की भी तुलना की जिनसे मेमोरी को छोटा किया जाता है, जैसे नोट्स को दबाना (क्वांटाइजेशन) या मस्तिष्क के विभिन्न हिस्सों के बीच नोट्स साझा करना (हेड-एक्सिस कंप्रेशन)। हालांकि वे तरीके थोड़ा मदद करते हैं, लेकिन पेपर दिखाता है कि लूप अक्ष (loop axis) को संकुचित करना ही असली सफलता है। यदि आप लूप के बजाय अन्य हिस्सों को संकुचित करने की कोशिश करते हैं, तो AI तर्क करने की अपनी क्षमता खो देता है। पेपर बहुत स्पष्ट है: लूप मेमोरी का सबसे अनावश्यक हिस्सा है, और यही एकमात्र जगह है जहाँ आपको दबाव डालना चाहिए।
वे कितने आश्वस्त हैं?
लेखक इन परिणामों को लेकर काफी आश्वस्त हैं क्योंकि उन्होंने उन्हें सीधे मापा है। उन्होंने केवल अनुमान नहीं लगाया; उन्होंने वास्तविक कार्यों जैसे गणित की समस्याओं (GSM8K) को हल करने और कोड लिखने (HumanEval) पर AI चलाया। उन्होंने दिखाया कि 4x संपीड़न (मेमोरी को 4 गुना छोटा करना) के साथ, AI अभी भी मूल, असंकुचित संस्करण के लगभग समान प्रदर्शन करता है। जब उन्होंने इसे अत्यधिक संपीड़न (21.3x) तक धकेला, तो AI अभी भी कई कार्य संभाल सकता था, हालांकि यह बहुत लंबे, जटिल तर्क चरणों के साथ संघर्ष करने लगा।
उन्होंने इसे Huginn-3.5B नामक एक अलग प्रकार के AI मॉडल पर भी परखा, जिसका मस्तिष्क संरचना अलग है और जो 32 बार लूप करता है। वहां भी, यह तरीका काम कर गया, जिससे साबित हुआ कि यह केवल एक विशिष्ट रोबोट के लिए कोई इत्तेफाक नहीं है। पेपर सुझाव देता है कि यह "लो-रैंक ट्राजेक्टरी" (नोट्स का सुचारू पथ) कि कैसे ये लूपिंग रोबोट सोचते हैं, इसकी एक मौलिक विशेषता है। हालांकि उन्होंने यह सिद्ध किया है कि यह मेमोरी बचाने और क्षमता बढ़ाने के लिए काम करता है, वे नोट करते हैं कि बहुत लंबे कार्यों पर सर्वोत्तम प्रदर्शन के लिए, AI को प्रशिक्षण के दूसरे दौर की आवश्यकता होती है जहाँ वह केवल एक शिक्षक की नकल करने के बजाय अपने स्वयं के संकुचित नोट्स के साथ अभ्यास करता है। यह "ऑन-पॉलिसी" प्रशिक्षण उन छोटी त्रुटियों को ठीक करता है जो तब होती हैं जब रोबोट अपनी खुद की कहानी लिखना शुरू करता है।
संक्षेप में, यह पेपर दिखाता है कि हम लूपिंग AI मॉडलों को बहुत अधिक कुशल बना सकते हैं क्योंकि उनकी मेमोरी एक विशिष्ट तरीके से दोहराव वाली होती है। इस दोहराव के एक छोटे से सारांश को संग्रहीत करके, हम एक ही कंप्यूटर चिप पर सैकड़ों गुना अधिक बातचीत को फिट कर सकते हैं, जिससे महंगे हार्डवेयर की आवश्यकता के बिना शक्तिशाली AI अधिक लोगों के लिए उपलब्ध हो जाता है।
अपने क्षेत्र के पेपरों की भीड़ में उलझे हुए हैं?
आपके रिसर्च कीवर्ड से मेल खाने वाले सबसे नए और अलग सोच वाले पेपरों का रोज़ाना Digest पाएँ—तकनीकी सारांश के साथ, आपकी भाषा में।