← नवीनतम पेपर
🤖 machine learning

COBS: Cumulant Order Block Sparse Attention

यह शोध पत्र COBS को प्रस्तुत करता है, जो एक ब्लॉक स्पार्स अटेंशन विधि है जो अटेंशन मास (attention mass) का बेहतर अनुमान लगाने के लिए संकुचित द्वितीय-क्रम सांख्यिकी (compressed second-order statistics) वाले एक नवीन सेलेक्टर का उपयोग करके लॉन्ग-कॉन्टेक्स्ट रिट्रीवल प्रदर्शन में सुधार करता है, जिससे हार्डवेयर दक्षता बनाए रखते हुए डेंस अटेंशन के साथ गुणवत्ता अंतराल को काफी कम किया जा सकता है।

मूल लेखक: Alexander Tian, Aditya Ghai, Sanjit Neelam, Zaal Vasania, Akshay Mishra

प्रकाशित 2026-07-13
📖 7 मिनट में पढ़ें🧠 गहराई से पढ़ें

मूल लेखक: Alexander Tian, Aditya Ghai, Sanjit Neelam, Zaal Vasania, Akshay Mishra

मूल पेपर CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/) के तहत लाइसेंस किया गया है। नीचे दिए गए पेपर की यह व्याख्या AI से तैयार की गई है। इसे लेखकों ने न तो लिखा है, न इसका समर्थन किया है। तकनीकी सटीकता के लिए मूल पेपर देखें। पूरा डिस्क्लेमर पढ़ें

कल्पना कीजिए कि आप घास के एक विशाल ढेर में एक विशिष्ट सुई खोजने की कोशिश कर रहे हैं, लेकिन घास का ढेर इतना बड़ा है कि आप हर एक तिनके को नहीं देख सकते, अन्यथा आपका दिमाग (या कंप्यूटर) थक कर रुक जाएगा। यह उस समस्या का रूप है जिसका सामना आधुनिक AI मॉडल तब करते हैं जब वे बहुत लंबे दस्तावेज़ पढ़ने की कोशिश करते हैं। उन्हें अब तक पढ़े गए हर एक शब्द को याद रखना पड़ता है, और हर एक मेमोरी टोकन की जाँच करना धीमा और महंगा होता है।

इस समस्या को हल करने के लिए, शोधकर्ताओं ने ब्लॉक स्पार्स अटेंशन (Block Sparse Attention) नामक एक शॉर्टकट आज़माया। हर एक तिनके को देखने के बजाय, उन्होंने तिनकों के छोटे समूहों (जिन्हें "ब्लॉक्स" कहा जाता है) को देखने का फैसला किया और केवल उन्हीं को चुनने का निर्णय लिया जो सबसे दिलचस्प लग रहे हों। यह एक स्काउट (जासूस) को नियुक्त करने जैसा है जो कुछ बंडलों को स्कैन करे और आपको बताए कि किनमें सुई होने की संभावना है।

समस्या: स्काउट बहुत साधारण था

यह शोध पत्र NSA (Native Sparse Attention) नामक एक लोकप्रिय पद्धति का अध्ययन करता है। इस प्रणाली में, स्काउट घास के एक बंडल को देखता है और इस बारेği एक त्वरित अनुमान लगाता है कि क्या वह महत्वपूर्ण है। शोध पत्र ने पाया कि यह स्काउट एक बहुत ही सरल ट्रिक का उपयोग कर रहा था: वह बंडल में तिनकों की औसत (average) स्थिति को देख रहा था।

इसे ऐसे समझें: कल्पना कीजिए कि आपके पास तिनकों के दो बंडल हैं।

  • बंडल A में तिनके बीच में बहुत कसकर एक साथ रखे गए हैं।
  • बंडल B में तिनके बिखरे हुए हैं, कुछ बहुत बाईं ओर और कुछ बहुत दाईं ओर।

यदि आप केवल औसत स्थिति देखते हैं, तो दोनों बंडल बिल्कुल एक जैसे दिखते हैं! लेकिन वास्तव में, बंडल B में सुई मिलने की संभावना बहुत अधिक है क्योंकि यह अधिक क्षेत्र को कवर करता है। पुराने स्काउट्स (प्रथम-क्रम या first-order तरीके) इस "फैलाव" या "वक्रता" (curvature) को देखने में अंधे थे। वे उस व्यक्ति की तरह थे जो केवल अपने केंद्र बिंदु को देखकर किसी बादल के आकार का अनुमान लगाने की कोशिश करता है; वे उन किनारों को भूल गए जो वास्तव में मायने रखते थे।

समाधान: COBS (एक स्मार्ट स्काउट)

लेखक COBS (Cumulant Order Block Sparse Attention) नामक एक नई विधि प्रस्तावित करते हैं। केवल औसत स्थिति बताने के बजाय, COBS का स्काउट एक छोटा, संकुचित (compressed) मानचित्र लेकर चलता है जो न केवल यह दिखाता है कि तिनके औसतन कहाँ हैं, बल्कि यह भी दिखाता है कि वे कितने फैले हुए हैं

गणितीय शब्दों में, शोध पत्र इसे "सेकंड-ऑर्डर स्टैटिस्टिक" या "कोवेरिएंस" (covariance) कहता है। हमारी उपमा में, यह स्काउट के यह महसूस करने जैसा है कि, "हे, यह बंडल चौड़ा और अस्त-व्यस्त है, इसलिए इसमें सुई होने की संभावना अधिक है!" इस अतिरिक्त जानकारी को रखने से (लेकिन इसे संकुचित करके ताकि यह बहुत अधिक जगह न ले), COBS बेहतर अनुमान लगा सकता है।

परिणाम: एक बड़ी छलांग

टीम ने इसे 32k RULER बेंचमार्क (लंबे संदर्भ वाले रिट्रीवल कार्यों के 11 अलग-अलग परीक्षणों का एक टेस्ट) पर परखा। उन्हें यह पता चला:

  • पुराना तरीका (NSA MLP): साधारण स्काउट का स्कोर 0.2999 था। वह सुइयों को खोजने के लिए संघर्ष कर रहा था।
  • परफेक्ट तरीका (OSA): यदि आप जादू से बिना किसी शॉर्टकट के सटीक उत्तर जान लेते (जिसे "ओरेकल" कहा जाता है), तो आपको 0.9040 का स्कोर मिलता।
  • नया तरीका (COBS): स्मार्ट स्काउट, जिसके पास फैलाव का मैप था, उसे 0.8195 का स्कोर मिला।

इसका मतलब है कि COBS ने संघर्ष कर रहे पुराने तरीके और परफेक्ट तरीके के बीच के अंतर को लगभग 86% तक कम कर दिया। यह एक बहुत बड़ा सुधार है!

लागत: क्या यह सार्थक है?

आमतौर पर, स्मार्ट बनने का मतलब अधिक काम करना होता है। लेकिन COBS कुशल है।

  • पुराने तरीके ने एक निश्चित मात्रा में डेटा पढ़ा।
  • परफेक्ट तरीका (जो सब कुछ पढ़ता है) COBS की तुलना में 15.15 गुना अधिक डेटा पढ़ता है।
  • COBS केवल पुराने, साधारण तरीके की तुलना में 1.21 गुना अधिक डेटा पढ़ता है।

तो, COBS आपको लगभग पूर्णता के करीब ले जाता है जबकि यह पुराने, सरल तरीके की तुलना में बहुत कम अतिरिक्त काम मांगता है।

शोध पत्र किन चीज़ों को "ना" कहता है

लेखक उन विचारों को खारिज करने में बहुत सावधान रहे जो अच्छे शॉर्टकट लग सकते हैं लेकिन वास्तव में काम नहीं करते:

  1. केवल औसत में अधिक जटिलता जोड़ना: उन्होंने "औसत" के अनुमान को स्मार्ट बनाने के लिए एक फैंसी न्यूरल नेटवर्क (MLP) का उपयोग करने की कोशिश की, लेकिन इससे ज्यादा मदद नहीं मिली। समस्या औसत की जटिलता नहीं थी; समस्या यह थी कि औसत स्वयं गलत उपकरण था। आपको औसत के बजाय "फैलाव" की जानकारी चाहिए थी।
  2. एक साधारण बॉक्स में "फैलाव" देखना: एक अन्य पद्धति ने न्यूनतम और अधिकतम तिनकों (एक बॉक्स) को देखकर फैलाव का अनुमान लगाने की कोशिश की। इससे थोड़ी मदद मिली, लेकिन यह COBS के फैलाव वाले मैप जितना सटीक नहीं था।
  3. अधिक जटिल गणित (Third-order) जोड़ना: लेखकों ने "स्क्यू" (skew - यह मापने का तरीका कि फैलाव कितना एकतरफा है) जोड़ने का परीक्षण किया। आश्चर्यजनक रूप से, इसने कम जटिलता वाले स्तरों पर चीज़ों को और खराब कर दिया, जिससे मॉडल भ्रमित हो गया। यह केवल तभी मदद करता था जब मॉडल पहले से ही बहुत जटिल हो और विफल हो रहा हो, जो कि एक समाधान के बजाय एक 'बैंड-एड' की तरह था। उन्होंने "फैलाव" (second-order) को ही सही संतुलन के रूप में चुना।

वे कितने आश्वस्त हैं?

शोध पत्र इन आंकड़ों के बारे में बहुत आश्वस्त है क्योंकि उन्होंने नियंत्रित प्रयोग किए। उन्होंने केवल अनुमान नहीं लगाया; उन्होंने 32k RULER टेस्ट पर प्रदर्शन को मापा और पाया कि COBS लगातार पुराने तरीकों से बेहतर प्रदर्शन करता है। उन्होंने यह भी जांचा कि क्या इससे मॉडल की छोटी वाक्यों को समझने की क्षमता प्रभावित होती है (ऐसा नहीं हुआ) और यह भी कि क्या यह वास्तव में पुराने तरीकों की तुलना में लंबे टेक्स्ट में अगले शब्द की भविष्यवाणी करने में मॉडल की मदद करता है (हाँ, इसने मदद की)।

हालाँकि, लेखक अपनी सीमाओं के बारे में ईमानदार हैं:

  • उन्होंने इसे लगभग 1.2 बिलियन पैरामीटर्स वाले मॉडल पर परखा। उन्हें यकीन नहीं है कि क्या यह बड़ी टेक कंपनियों द्वारा उपयोग किए जाने वाले विशाल मॉडलों पर बिल्कुल वैसा ही काम करेगा, हालांकि गणित सुझाव देता है कि ऐसा होना चाहिए।
  • उन्होंने इस तकनीक को एक विशिष्ट प्रकार के सिंथेटिक डेटा (RULER-शैली) पर प्रशिक्षित किया। हालांकि यह लंबे संदर्भ (long-context) की क्षमता का परीक्षण करने का एक मानक तरीका है, वास्तविक दुनिया का डेटा थोड़ा अलग व्यवहार कर सकता है।

मुख्य निष्कर्ष

यह शोध पत्र दिखाता है कि घास के ढेर में सुई कुशलतापूर्वक खोजने के लिए, आप केवल बंडल के केंद्र को नहीं देख सकते। आपको यह जानने की आवश्यकता है कि बंडल कितना फैला हुआ है। फैलाव के एक छोटे, संकुचित मानचित्र को जोड़कर, COBS AI मॉडल्स को बिना धीमा किए बहुत अधिक सटीकता के साथ लंबे दस्तावेज़ पढ़ने की अनुमति देता है, जो बहुत कम अतिरिक्त प्रयास के साथ "काफी अच्छे" और "परफेक्ट" के बीच के अंतर को पाट देता है।

अपने क्षेत्र के पेपरों की भीड़ में उलझे हुए हैं?

आपके रिसर्च कीवर्ड से मेल खाने वाले सबसे नए और अलग सोच वाले पेपरों का रोज़ाना Digest पाएँ—तकनीकी सारांश के साथ, आपकी भाषा में।

Digest आज़माएँ →