← नवीनतम पेपर
💬 NLP

Vectorizing the Trie: Efficient Constrained Decoding for LLM-based Generative Retrieval on Accelerators

यह शोध पत्र STATIC को प्रस्तुत करता है, जो एक नवीन कंस्ट्रेंड डिकोडिंग तकनीक है जो हार्डवेयर एक्सेलेरेटर्स पर कुशल, उच्च-थ्रूपुट, कड़ाई से प्रतिबंधित जनरेटिव रिट्रीवल को सक्षम करने के लिए ट्राइ-आधारित प्रीफिक्स ट्रीज़ को स्पार्स मैट्रिक्स ऑपरेशन्स में वेक्टरइज़ करता है, जिससे भारी गति वृद्धि प्राप्त होती है और एक प्रोडक्शन वीडियो रिकमेंडेशन सिस्टम में सफलतापूर्वक बड़े पैमाने पर तैनात किया जाता है।

मूल लेखक: Zhengyang Su, Isay Katsman, Yueqi Wang, Ruining He, Lukasz Heldt, Raghunandan Keshavan, Shao-Chuan Wang, Xinyang Yi, Mingyan Gao, Onkar Dalal, Lichan Hong, Ed Chi, Ningren Han

प्रकाशित 2026-02-27
📖 6 मिनट में पढ़ें🧠 गहराई से पढ़ें

मूल लेखक: Zhengyang Su, Isay Katsman, Yueqi Wang, Ruining He, Lukasz Heldt, Raghunandan Keshavan, Shao-Chuan Wang, Xinyang Yi, Mingyan Gao, Onkar Dalal, Lichan Hong, Ed Chi, Ningren Han

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

🎬 बड़ी तस्वीर: "जादुई जिन्न" की समस्या

कल्पना कीजिए कि आपके पास एक जादुई जिन्न (AI मॉडल) है जो आपके फोन या कंप्यूटर के अंदर रहता है। आप उससे पूछते हैं, "मुझे देखने के लिए एक वीडियो सुझाएं!"

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

लेकिन यहाँ एक समस्या है: जिन्न थोड़ा ख्याली पुलाव पकाने वाला (daydreamer) है।

  • यदि आप कहते हैं, "पिछले 24 घंटों के ताज़ा वीडियो दिखाओ," तो जिन्न आत्मविश्वास के साथ एक ऐसे वीडियो का शीर्षक बना सकता है जो अस्तित्व में ही नहीं है, या वह जो तीन साल पहले अपलोड किया गया था।
  • यदि आप "खाना पकाने के बारे में वीडियो" मांगते हैं, तो यह गलती से "कार रिपेयर" के बारे में वीडियो का सुझाव दे सकता है क्योंकि यह बस अगले शब्द का अनुमान लगा रहा है।

वास्तविक दुनिया में (जैसे YouTube), आप जिन्न को कल्पना (hallucination) करने की अनुमति नहीं दे सकते। आपको बिजनेस रूल्स (Business Rules) लागू करने होंगे:

  • "केवल वे वीडियो दिखाएं जो पिछले एक सप्ताह में अपलोड किए गए हैं।"
  • "केवल वे वीडियो दिखाएं जो अमेरिका में उपलब्ध हैं।"
  • "केवल वे वीडियो दिखाएं जो स्टॉक में हैं।"

🚧 पुराना तरीका: "उंगली दिखाने वाला" ट्रैफिक पुलिस

जिन्न को गलतियाँ करने से रोकने के लिए, इंजीनियरों ने एक प्रिफिक्स ट्री (Prefix Tree या Trie) का उपयोग किया। इसे एक विशाल, शाखाओं वाले फ्लोचार्ट की तरह समझें जिसमें सभी वैध वीडियो शीर्षक मौजूद हैं।

  • यह कैसे काम करता था: हर बार जब जिन्न अगला शब्द लिखना चाहता था, तो उसे रुकना पड़ता था, फ्लोचार्ट को देखना पड़ता था, और एक ट्रैफिक पुलिस (CPU प्रोसेसर) से पूछना पड़ता था: "क्या 'Cooking' अगला वैध शब्द है?"
  • बॉटलनेक (रुकावट): यह एक बहुत तेज़ रेस कार (AI चिप, या TPU/GPU) के लिए ऐसा है जैसे उसे हर चौराहे पर रुककर एक इंसान (पुलिस वाले) से मैप चेक करने का इंतज़ार करना पड़े।
    • जिन्न तेज़ है।
    • पुलिस वाला धीमा है।
    • जिन्न अपना 99% समय पुलिस वाले का इंतज़ार करने में बिता देता है।
    • परिणाम: सिस्टम अविश्वसनीय रूप से धीमा और महंगा हो जाता है।

⚡ नया तरीका: STATIC (द "सुपर-हाईवे")

इस पेपर के लेखकों ने STATIC पेश किया। उन्होंने महसूस किया कि हर बार मैप चेक करने के लिए पुलिस वाले से पूछना बहुत धीमा है। इसलिए, उन्होंने पूरे फ्लोचार्ट को एक एकल, सपाट, डिजिटल मैप में बदल दिया जिसे सुपर-कंप्यूटर तुरंत पढ़ सकता है।

उन्होंने इसे तीन सरल चरणों का उपयोग करके किया:

1. ट्री को चपटा करना (The "Spreadsheet" Trick)

जटिल शाखाओं और पॉइंटर्स वाले ट्री के बजाय, उन्होंने पूरी संरचना को एक विशाल स्पार्स मैट्रिक्स (Sparse Matrix) (सोचिए एक विशाल, ज्यादातर खाली स्प्रेडशीट) में बदल दिया।

  • उपमा: कल्पना कीजिए कि एक लाइब्रेरी है जहाँ किताबें एक जटिल भूलभुलैया (maze) में रखी हैं। अगली किताब खोजने के लिए भूलभुलैया में चलने के बजाय, उन्होंने एक सिंगल इंडेक्स कार्ड प्रिंट कर दिया जो आपको बताता है कि हर किताब ठीक कहाँ है।
  • जादू: यह एक "पीछा करने" (पॉइंटर्स का पीछा करना) को "लुकअप" (स्प्रेडशीट की एक पंक्ति पढ़ना) में बदल देता है। कंप्यूटर स्प्रेडशीट पढ़ने में माहिर होते हैं।

2. प्रक्रिया का वेक्टराइज़ेशन (The "Group Hug")

पुराने सिस्टम एक बार में एक वीडियो पाथ (path) को चेक करते थे। STATIC एक ही समय में हजारों पाथ को एक साथ चेक करता है।

  • उपमा: कल्पना कीजिए कि एक शिक्षक 1,000 छात्रों से पूछता है, "क्या आपका उत्तर सही है?"
    • पुराना तरीका: शिक्षक एक-एक करके प्रत्येक छात्र को डेस्क तक बुलाता है। (बहुत समय लगता है)।
    • STATIC तरीका: शिक्षक एक विशाल उत्तर कुंजी (answer key) बांट देता है। सभी 1,000 छात्र एक ही पल में उस शीट के खिलाफ अपने उत्तरों को एक साथ चेक करते हैं।
  • इसे वेक्टराइज़ेशन (Vectorization) कहा जाता है। यह हार्डवेयर (TPUs/GPUs) की पूरी शक्ति को अनलॉक करता है।

3. "ब्रांच-फ्री" कर्नेल (The "No-Stop" Highway)

कंप्यूटर "if/else" निर्णयों (जैसे, "यदि पथ वैध है, तो X करें; अन्यथा, Y करें") से नफरत करते हैं। ये निर्णय धीमे होते हैं क्योंकि कंप्यूटर को सोचने के लिए रुकना पड़ता है।

  • STATIC का ट्रिक: उन्होंने एक ऐसा सिस्टम बनाया जहाँ कंप्यूटर को निर्णय लेने की आवश्यकता नहीं होती। वह बस हर संभावना के लिए गणित करता है, और फिर एक "मास्क" (फ़िल्टर) का उपयोग करके खराब विकल्पों को तुरंत शून्य कर देता है।
  • उपमा: यह पूछने के बजाय कि "क्या यह कार इस लेन में जा सकती है?", आप बस उन सभी लेन पर एक बड़ा लाल "X" पेंट कर देते हैं जो बंद हैं। कारें बस सीधे चलती रहती हैं, और जिन पर "X" होता है, वे स्वाभाविक रूप से रुक जाते हैं। किसी सोच-विचार की आवश्यकता नहीं।

🏆 परिणाम: गति और सफलता

लेखकों ने इसका परीक्षण YouTube पर किया, जहाँ अरबों उपयोगकर्ता और लाखों वीडियो हैं।

  • गति: STATIC पिछले सबसे अच्छे तरीकों की तुलना में 47 से 1,000 गुना तेज़ है।
    • उपमा: यदि पुराने तरीके को एक लिस्ट चेक करने में 30 सेकंड लगते, तो STATIC इसे पलक झपकने के समय (0.03 मिलीसेकंड) में कर देता है।
  • दक्षता (Efficiency): यह उपयोगकर्ता के अनुभव में लगभग शून्य देरी (zero delay) जोड़ता है। आपको पता भी नहीं चलेगा कि यह वहाँ है।
  • वास्तविक प्रभाव: जब उन्होंने "फ्रेशनेस" (केवल नए वीडियो दिखाना) चालू किया, तो सिस्टम पूरी तरह से काम कर गया।
    • उपयोगकर्ताओं ने 5% अधिक नए वीडियो देखे।
    • क्लिक बढ़ गए।
    • उपयोगकर्ता खुश थे।

🧊 बोनस: "कोल्ड स्टार्ट" समस्या का समाधान

पेपर ने यह भी दिखाया कि यह ट्रिक कोल्ड स्टार्ट (Cold Start) आइटम्स (नए उत्पाद या वीडियो जिन्हें AI ने पहले कभी नहीं देखा है) के साथ भी मदद करती है।

  • समस्या: आमतौर पर, यदि AI ने किसी नए वीडियो को नहीं देखा है, तो वह उसे अनदेखा कर देता है।
  • समाधान: AI को केवल "नए आइटम्स" की सूची देखने के लिए मजबूर करके (उनके नए मैप मेथड का उपयोग करके), AI को तुरंत इन नई चीजों के बारे में सीखने के लिए मजबूर किया जाता है। यह एक "न्यू अराइवल्स" (New Arrivals) का साइन शेल्फ पर लगाने और जिन्न से कहने जैसा है, "तुम केवल यहीं से चुन सकते हो।"

📝 एक वाक्य में सारांश

STATIC एक धीमी, स्टेप-बाय-स्टेप "मैप चेक करने" की प्रक्रिया को एक बिजली जैसी तेज़, "एक साथ पूरी शीट पढ़ने" की प्रक्रिया में बदल देता है, जिससे AI को इंटरनेट को धीमा किए बिना ताज़ा, प्रासंगिक और वैध वीडियो की सिफारिश करने की अनुमति मिलती है।

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

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

Digest आज़माएँ →