Predictive Divergence Masks for LLM RL
यह शोध पत्र प्रेडिक्टिव डायवर्जेंस मास्क (Predictive Divergence Masks) का प्रस्ताव करता है, जो एलएलएम (LLM) सुदृढीकरण शिक्षण (reinforcement learning) के लिए एक नवीन विधि है जो प्रॉक्सिमिटी मानदंड (proximity criterion) के रूप में उपयोग किए जाने वाले संभाव्यता विचलन (probability divergence) को अगला ग्रेडिएंट स्टेप बढ़ाएगा या घटाएगा, इसके लिए एक क्लोज्ड-फॉर्म भविष्यवाणी के माध्यम से PPO के अनुपात-आधारित दिशा मानदंड को प्रतिस्थापित करता है, जिससे मॉडल के पैमानों (scales) में प्रशिक्षण स्थिरता और प्रदर्शन में सुधार होता है।
मूल पेपर CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/) के तहत लाइसेंस किया गया है। नीचे दिए गए पेपर की यह व्याख्या AI से तैयार की गई है। इसे लेखकों ने न तो लिखा है, न इसका समर्थन किया है। तकनीकी सटीकता के लिए मूल पेपर देखें। पूरा डिस्क्लेमर पढ़ें
कल्पना कीजिए कि आप एक सुपर-स्मार्ट रोबोट को कहानियाँ लिखना सिखा रहे हैं। आप चाहते हैं कि वह अपनी गलतियों से सीखे, लेकिन आपका एक सख्त नियम है: "एक बार में अपनी लिखने की शैली को बहुत अधिक नाटकीय रूप से न बदलें।" यदि रोबोट बहुत उत्साहित हो जाता है और पूरी तरह से अलग आवाज़ में लिखने लगता है, तो आपको ब्रेक लगाना होगा। यह रीइन्फोर्समेंट लर्निंग (RL) का काम है जो लार्ज लैंग्वेज मॉडल्स (LLMs) के लिए किया जाता है।
लंबे समय तक, उन ब्रेक्स को लगाने का मानक तरीका PPO नामक एक विधि थी। यह एक एकल शब्द पर आधारित एक साधारण ट्रैफिक लाइट की तरह काम करता था। हर बार जब रोबट एक शब्द चुनता था, तो सिस्टम जाँचता था: "क्या यह शब्द उस शब्द से बहुत अधिक संभावित है जिसे रोबोट आमतौर पर चुनता है? और क्या रिवॉर्ड हमें इसे और भी अधिक संभावित बनाने के लिए कह रहा है?" यदि उत्तर दोनों के लिए "हाँ" था, तो सिस्टम कहता, "रुकिए! यह बहुत दूर है!" और उस शब्द के सबक को अनदेखा कर देता।
लेकिन यहाँ समस्या यह है: रोबोट केवल एक शब्द नहीं चुनता; वह उस शब्द को चुनने के लिए अपने पूरे व्यक्तित्व (अपनी प्रोबेबिलिटी डिस्ट्रीब्यूशन) को पुनर्व्यवस्थित करता है। पुराना तरीका केवल उसके द्वारा चुने गए एक शब्द को देखता था, जैसे केवल एक वायलिन को सुनकर पूरे ऑर्केस्ट्रा का न्याय करना। यह इस बात को चूक जाता था कि बाकी संगीत कैसे बदल रहा है।
नया विचार: "प्रेडिक्टिव डायवर्जेंस मास्क" (Predictive Divergence Mask)
टेनसेंट (Tencent), UIUC और NUS के शोधकर्ताओं की एक टीम ने महसूस किया कि यह सिंगल-वर्ड व्यू भ्रामक था। उन्होंने एक नया नियम प्रस्तावित किया जिसे प्रेडिक्टिव डायवर्जेंस मास्क कहा गया।
केवल उस एक शब्द को देखने के बजाय जिसे रोबोट ने चुना है, यह नई विधि एक बड़ा सवाल पूछती है: "यदि हम यह कदम उठाते हैं, तो क्या रोबोट का पूरा व्यक्तित्व उसके मूल स्वरूप से और दूर चला जाएगा?"
इसे इस तरह सोचें:
- पुराना तरीका (PPO): आप एक बच्चे को खाई की ओर दौड़ते हुए देखते हैं। आप जाँचते हैं कि क्या वह विशिष्ट कदम खतरनाक है। यदि हाँ, तो आप चिल्लाते हैं "रुक जाओ!"
- नया तरीका (प्रेडिक्टिव डायवर्जेंस): आप बच्चे को दौड़ते हुए देखते हैं। आप पूरे रास्ते को देखते हैं। भले ही वह विशिष्ट कदम ठीक लग रहा हो, शायद बच्चे का मोमेंटम उसे घुमाने वाला है और उसे एक अलग खाई में भेज देगा। नया तरीका उस पूरे 'स्पिन' (घूमने) को कदम उठाने से पहले ही प्रेडिक्ट (पूर्वानुमान) कर लेता है।
यह कैसे काम करता है (जादुв गणित)
शोधकर्ताओं ने इस "पूरे स्पिन" की भविष्यवाणी करने के लिए एक चतुर गणितीय ट्रिक खोजी, जिसके लिए उन्हें पहले से सिमुलेशन चलाने की आवश्यकता नहीं है।
- लोकल टर्म (Local Term): यह वह हिस्सा है जिसे पुराना तरीका पहले से जानता था। यह रोबोट द्वारा चुने गए विशिष्ट शब्द में परिवर्तन है।
- ग्लोबल टर्म (Global Term): यह नया सीक्रेट सॉस है। यह इस बात का हिसाब रखता है कि एक शब्द को बदलने से रोबोट को उन अन्य सभी शब्दों की संभावनाओं को कैसे समायोजित करना पड़ता है जिन्हें उसने नहीं चुना (क्योंकि कुल संभावना हमेशा 100% होनी चाहिए)।
पुराने तरीके ने ग्लोबल टर्म को अनदेखा कर दिया था। नया तरीका इसे जोड़ देता है। यदि गणित दिखाता है कि रोबोट का पूरा व्यक्तित्व बहुत दूर जा रहा है, तो मास्क अपडेट को ब्लॉक कर देता है। यदि गणित दिखाता है कि रोबोट वास्तव में सुरक्षा की ओर वापस आ रहा है, तो अपडेट की अनुमति दी जाती है।
"टॉप-K" समस्या
असल दुनिया में, रोबोट का इंजन (रोलआउट इंजन) समय और मेमोरी बचाने के लिए केवल सबसे संभावित शीर्ष 20 शब्दों ("Top-K") को दिखाता है। यह हजारों अन्य शब्दों को एक "टेल" (tail) बकेट में छिपा देता है।
इस सीमित दृश्य के साथ अपनी भविष्यवाणी को काम करने योग्य बनाने के लिए, शोधकर्ताओं ने दो हल्के एस्टिमेटर्स (estimators) का आविष्कार किया:
- एग्रीगेटेड-टेल एस्टिमेटर (Aggregated-Tail Estimator): यह मान लेता है कि सभी छिपे हुए शब्द एक विशाल ढेर की तरह हैं।
- यूनिफॉर्म-टेल एस्टिमेटर (Uniform-Tail Estimator): यह मान लेता है कि छिपे हुए शब्द समान रूप से संभावित हैं।
उन्होंने पाया कि क्योंकि शीर्ष 20 शब्द आमतौर पर लगभग सारी संभावना (99% से अधिक) रखते हैं, इसलिए दोनों तरीके लगभग एक जैसा काम करते हैं। यह मौसम का अनुमान लगाने जैसा है; आपको यह जानने के लिए कि बारिश होने वाली है या नहीं, हर एक बादल को गिनने की आवश्यकता नहीं है, बस आसमान को देखना काफी है।
प्रयोग क्या दिखाते हैं
टीम ने विभिन्न आकार के रोबोटों (4 बिलियन से 30 बिलियन पैरामीटर्स वाले मॉडल) पर इसका परीक्षण किया और यहाँ तक कि तब भी परीक्षण किया जब वे "FP8" प्रिसिजन (एक सुपर-फास्ट, थोड़ी कम सटीक गणित मोड जो ट्रेनिंग और राइटिंग इंजन को अलग तरह से कार्य करने पर मजबूर करती है) पर चल रहे थे।
- परिणाम: नए मास्क ने ट्रेनिंग को अधिक स्थिर (stable) बनाया।
- प्रमाण: उन्होंने 61 अलग-अलग प्रयोग (seeds) चलाए। उन्होंने पाया कि पुराना तरीका कभी-कभी उन अपडेट्स को रख लेता था जो वास्तव में रोबोट को और दूर ले जाते थे (एक "अनसेफ कीप" रेट 36.9%)। नया तरीका इसे घटाकर 34.2% कर देता है।
- सुधार: हालांकि संख्याएँ छोटी लग सकती हैं, लेकिन विशाल AI मॉडल को प्रशिक्षित करने की दुनिया में, कुछ बुरे कदमों को रोकना भी रोबोट को बेहतर और तेजी से सीखने में मदद करता है। नया तरीका सभी मॉडल आकारों और प्रिसिजन सेटिंग्स में लगातार काम करता रहा।
उन्होंने स्पष्ट रूप से किसे खारिज किया
पेपर बहुत स्पष्ट है कि क्या उतना अच्छा काम नहीं करता:
- वे तर्क देते हैं कि अपडेट की दिशा तय करने के लिए सैंपलड-टोकन इम्पॉर्टेंस रेशियो (पुराना "सिंगल वायलिन" चेक) का उपयोग करने के खिलाफ हैं। उन्होंने दिखाया कि यह एकल-शब्द चेक अक्सर रोबोट के पूरे व्यक्तित्व में वास्तविक परिवर्तन से असहमत होता है।
- उन्होंने यह भी दिखाया कि केवल "सेफ्टी ज़ोन" (ट्रस्ट रीजन) को कड़ा करने से (0.15 के बजाय 0.05 के थ्रेशोल्ड का उपयोग करके) सब कुछ बदतर हो जाता है, जो यह सुझाव देता है कि केवल अधिक प्रतिबंधात्मक होना ही समाधान नहीं है; बल्कि दिशा के बारे में अधिक स्मार्ट होना ज़रूरी है।
वे कितने आश्वस्त हैं?
लेखक अपने निष्कर्षों को लेकर अपने प्रयोगों से प्राप्त मापे गए डेटा के आधार पर आश्वस्त हैं। उन्होंने केवल शून्य में सिमुलेशन नहीं चलाया; उन्होंने वास्तविक गणितीय समस्याओं (DAPO-Math-17k डेटासेट का उपयोग करके) पर वास्तविक मॉडल को प्रशिक्षित किया और AIME24 और AIME25 जैसे बेंचमार्क पर सटीकता को मापा।
वे इस दृष्टिकोण का सुझाव देते हैं कि यह नियमों का एक बेहतर संरेखण है: यदि आप पूरे व्यक्तित्व (डायवर्जेंस) को देखकर सुरक्षा क्षेत्र को परिभाषित करते हैं, तो आपको अपडेट की दिशा को भी पूरे व्यक्तित्व को देखकर ही जांचना चाहिए, न कि केवल एक शब्द को। परिणाम सुझाव देते हैं कि इससे ट्रेनिंग अधिक स्थिर होती है, लेकिन वे स्वीकार करते हैं कि यह अभी भी एक "लोकल फर्स्ट-ऑर्डर एप्रोक्सिमेशन" है—जिसका अर्थ है कि यह अगले कदम के आधार पर एक बहुत अच्छा अनुमान है, भविष्य के लिए कोई पूर्ण क्रिस्टल बॉल नहीं।
संक्षेप में, नया तरीका रोबोट को एक ऐसे GPS देने जैसा है जो केवल वर्तमान कदम को नहीं, बल्कि पूरे मानचित्र को देखता है। यह रोबोट को दुनिया के किनारे से भटकने दिए बिना तेजी से सीखने में मदद करता है।
अपने क्षेत्र के पेपरों की भीड़ में उलझे हुए हैं?
आपके रिसर्च कीवर्ड से मेल खाने वाले सबसे नए और अलग सोच वाले पेपरों का रोज़ाना Digest पाएँ—तकनीकी सारांश के साथ, आपकी भाषा में।