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

K-Forcing: Joint Next-K-Token Decoding via Push-Forward Language Modeling

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

मूल लेखक: Zhiwei Tang, Yuanyu He, Yizheng Han, Wangbo Zhao, Jiasheng Tang, Fan Wang, Bohan Zhuang

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

मूल लेखक: Zhiwei Tang, Yuanyu He, Yizheng Han, Wangbo Zhao, Jiasheng Tang, Fan Wang, Bohan Zhuang

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

यहाँ "K-Forcing" पेपर का सरल भाषा और रचनात्मक उपमाओं (analogies) का उपयोग करके किया गया स्पष्टीकरण दिया गया है।

समस्या: "एक-एक-कदम-पर-एक-समय" की बाधा (The "One-Step-at-a-Time" Bottleneck)

कल्पना कीजिए कि एक बहुत ही प्रतिभाशाली शेफ (AI) है जो हजारों लोगों के लिए एक विशाल दावत बना रहा है। यह शेफ अपनी बेहतरीन रेसिपी के लिए प्रसिद्ध है, लेकिन उसका एक बहुत सख्त नियम है: वह एक बार में केवल एक ही सामग्री (ingredient) बर्तन में डाल सकता है।

100 सामग्रियों वाला स्टू (stew) बनाने के लिए, शेफ को यह करना होगा:

  1. एक चम्मच उठाना।
  2. एक सामग्री डालना।
  3. चम्मच नीचे रखना।
  4. फिर से चम्मच उठाना।
  5. अगली सामग्री डालना।
    ...और इस प्रक्रिया को 100 बार दोहराना।

भले ही शेफ अगली सामग्री के बारे में सोचने में बहुत तेज़ हो, लेकिन प्रक्रिया धीमी है क्योंकि वह लगातार चम्मच उठाने और रखने के लिए रुकता रहता है। AI की दुनिया में, यह "चम्मच" कंप्यूटर की मेमोरी एक्सेस (memory access) है। AI को अपने दिमाग (weights) को मेमोरी से लोड करना होता है, एक शब्द सोचना होता है, उसे सेव करना होता है, और फिर अगला शब्द सोचने के लिए फिर से अपना दिमाग लोड करना होता है। यह लंबे टेक्स्ट जेनरेट करने को धीमा और महंगा बना देता है, खासकर जब एक साथ कई लोगों को सेवा देनी हो।

पुराने समाधान: वे पूरी तरह से क्यों काम नहीं आए?

वैज्ञानिकों ने इस "चम्मच वाली समस्या" को ठीक करने के लिए दो मुख्य तरीके आजमाए, लेकिन बड़ी भीड़ की सेवा करते समय दोनों में खामियां थीं:

  1. "ड्राफ्ट और वेरीफाई" विधि (Speculative Decoding):
    कल्पना कीजिए कि एक सहायक शेफ (sous-chef) जल्दी से अगली 5 सामग्रियों का अनुमान लगा लेता है। मुख्य शेफ फिर उनकी जांच करता है। यदि अनुमान सही है, तो बहुत अच्छा! यदि अनुमान गलत है, तो मुख्य शेफ को उसे फेंकना पड़ता है और फिर से शुरू करना पड़ता है।
  • खामी: कभी अनुमान 5 शब्दों के लिए सही होता है, कभी 2 के लिए, और कभी 0 के लिए। इससे एक अस्त-व्यस्त रसोई बन जाती है जहाँ कुछ बर्तन भरे होते हैं और कुछ खाली। जब हर कोई अलग-अलग समय पर काम पूरा कर रहा हो, तो रसोई को कुशलतापूर्वक व्यवस्थित करना कठिन होता है।
  1. "डिफ्यूजन" (Diffusion) विधि:
    कल्पना कीजिए कि आप एक खाली पन्ने से शुरुआत कर रहे हैं और धीरे-धीरे रैंडम शब्दों को भर रहे हैं जब तक कि वे समझ में न आने लगें, जैसे किसी धुंधले स्केच को मिटाना और फिर से लिखना।
  • खामी: एक अच्छा वाक्य पाने के लिए, आपको अक्सर उन्हीं शब्दों को कई बार फिर से लिखना पड़ता है। यह एक कैनवास पर पेंटिंग करने जैसा है जहाँ आप बार-बार रंगों को फैलाते और मिटाते रहते हैं। यह पैरेलल (parallel) है (आप एक साथ कई जगहों पर काम कर सकते हैं), लेकिन उच्च गुणवत्ता वाला परिणाम प्राप्त करने के लिए इसमें बहुत अधिक "पास" (passes) लेने पड़ते हैं।

नया समाधान: K-Forcing

इस पेपर के लेखक, K-Forcing, खाना पकाने का एक बिल्कुल नया तरीका प्रस्तावित करते हैं। एक-एक करके सामग्री डालने के बजाय, वे शेफ को सामग्रियों का एक पूर्व-मापा गया ब्लॉक (pre-measured block) पकड़ने और उसे एक ही बार में डालने के लिए सिखाते हैं।

यह कैसे काम करता है, इसके सरल चरण यहाँ दिए गए हैं:

1. "मैजिक नॉइज़" (Magic Noise) बॉक्स

कल्पना कीजिए कि शेफ के पास "मैजिक नॉइज़" (रैंडम नंबर) से भरा एक विशेष बॉक्स है। आमतौर पर, AI मॉडल को यह सीखने के लिए प्रशिक्षित किया जाता है कि उन्होंने अब तक जो टेक्स्ट लिखा है उसे देखकर अगला शब्द क्या होगा।
K-Forcing खेल बदल देता है। यह AI को सिखाता है: "यदि आप अब तक के टेक्स्ट को देखते हैं और इस विशिष्ट रैंडम नॉइज़ को देखते हैं, तो आप एक साथ अगले 4 शब्द प्रेडिक्ट (predict) कर सकते हैं।"

नॉइज़ को एक गुप्त कोड की तरह समझें।

  • यदि कोड 0.1 है, तो अगले 4 शब्द "The cat sat" हो सकते हैं।
  • यदि कोड 0.9 है, तो अगले 4 शब्द "The dog ran" हो सकते हैं।
    AI इस कोड और शब्दों के ब्लॉक के बीच के मानचित्र (map) को सीख जाता है।

2. "पुश-फॉरवर्ड" (Push-Forward) ट्रिक

पेपर इसे "पुश-फॉरवर्ड" मॉडल कहता है। एक ऐसी मशीन की कल्पना करें जो कच्चे आटे (रैंडम नॉइज़) के बैग को लेती है और उसे एक सांचे (mold) से गुजारती है ताकि तुरंत ब्रेड का एक आदर्श लोफ (loaf) बन सके।

  • पुराना तरीका: आटा गूंथना, इंतजार करना, आकार देना, इंतजार करना, एक लोफ बेक करना।
  • K-Forcing: आटे को सांचे से धकेलें, और पफ, आपके पास ब्रेड का एक लोफ तैयार है। इसे फिर से करें, और आपके पास एक और लोफ होगा।

3. "स्व-शिक्षण" प्रक्रिया (Progressive Distillation)

आप सिर्फ एक शेफ को तुरंत "एक शब्द" से "चार शब्द" प्रेडिक्ट करने के लिए नहीं कह सकते; वे गलती कर देंगे। इसलिए, लेखक एक चतुर प्रशिक्षण विधि का उपयोग करते हैं जिसे प्रोग्रेसिव सेल्फ-फोर्सिंग डिस्टिलेशन (Progressive Self-Forcing Distillation) कहा जाता है:

  • चरण 1: AI को मैजिक नॉइज़ का उपयोग करके 1 शब्द प्रेडिक्ट करना सिखाएं। (आसान!)
  • चरण 2: एक बार जब AI 1 शब्द में अच्छा हो जाए, तो उस AI का उपयोग करके एक नए AI को एक साथ 2 शब्द प्रेडिक्ट करना सिखाएं। नया AI पुराने AI को लगातार 2 शब्द जेनरेट करते हुए देखकर सीखता है।
  • चरण 3: इस प्रक्रिया को तब तक दोहराएं जब तक कि AI एक बार में 4 शब्द (या अधिक) प्रेडिक्ट न कर सके।

यह एक मास्टर शेफ द्वारा प्रशिक्षु (apprentice) को प्रशिक्षित करने जैसा है। पहले, प्रशिक्षु एक प्याज काटना सीखता है। फिर, प्रशिक्षु मास्टर को एक लय में दो प्याज काटते हुए देखता है, और खुद भी वैसा ही करना सीखता है। अंततः, वे एक ही झटके में सब्जियों का ढेर काट सकते हैं।

परिणाम: तेज़ और सुचारू (Faster and Smoother)

पेपर ने दो बड़े टेक्स्ट डेटासेट्स (जैसे किताबों और इंटरनेट का पुस्तकालय) पर इसका परीक्षण किया।

  • गति (Speed): जब AI को एक बार में 4 शब्द जेनरेट करने के लिए सेट किया गया, तो यह पुराने "एक-एक शब्द" वाले तरीके की तुलना में 2.4 से 3.5 गुना तेज़ था।
  • गुणवत्ता (Quality): टेक्स्ट मूल "एक-शब्द" वाले शेफ की तुलना में थोड़ा कम सटीक था, लेकिन फिर भी बहुत उच्च गुणवत्ता का था। आप गति में भारी वृद्धि के लिए थोड़ी सी गुणवत्ता का त्याग कर सकते हैं।
  • बैचिंग (Batching): चूंकि AI हमेशा प्रति स्टेप ठीक 4 शब्द उत्पन्न करता है, इसलिए रसोई पूरी तरह से व्यवस्थित रहती है। हर कोई अपना "ब्लॉक" एक ही समय में पूरा करता है, जिससे बिना किसी अराजकता के एक साथ हजारों ग्राहकों को सेवा देना आसान हो जाता है।

सारांश उपमा (Summary Analogy)

  • ऑटोरिग्रेसिव (Autoregressive - पुराना तरीका): एक टाइपिस्ट जो एक अक्षर टाइप करता है, "Enter" दबाता है, कंप्यूटर के प्रोसेस होने का इंतजार करता है, और फिर अगला अक्षर टाइप करता है।
  • स्पेक्युलेटिव डिकोडिंग (Speculative Decoding): एक टाइपिस्ट जो एक वाक्य टाइप करता है, फिर एक प्रूफरीडर उसकी जांच करता है। यदि प्रूफरीडर को कोई गलती मिलती है, तो टाइपिस्ट को उसे मिटाना और फिर से टाइप करना पड़ता है। यह कभी तेज़ होता है, तो कभी धीमा।
  • K-Forcing: एक टाइपिस्ट जिसके पास एक "स्टैम्पिंग मशीन" है। वे स्टैंप में 4 अक्षर लोड करते हैं, उसे नीचे दबाते हैं, और बूम, 4 अक्षर तुरंत दिखाई देते हैं। वे इसे बार-बार करते हैं। यह अविश्वसनीय रूप से तेज़ है, और क्योंकि हर स्टैंप का आकार एक समान है, मशीन कभी जाम नहीं होती।

संक्षेप में: K-Forcing AI को एक बार में एक शब्द सोचने के बजाय "चंक्स" (chunks/टुकड़ों) में सोचना सिखाता है, जिसमें रैंडम नॉइज़ का उपयोग उन टुकड़ों को तुरंत अनलॉक करने के लिए एक गुप्त कुंजी के रूप में किया जाता है। यह कंप्यूटर की मेमोरी को बाधित किए बिना टेक्स्ट जनरेशन को बहुत तेज़ बनाता है।

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

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

Digest आज़माएँ →