Optimizing What Policies Learn From: Recoverability-aware Rollout Intervention Learning
यह शोध पत्र रिकवरेबिलिटी-अवेयर इंटरवेंशन लर्निंग (RAIL) का प्रस्ताव करता है, जो एक प्रशिक्षण-समय ढांचा है जो रोलआउट जनरेशन को एक अनुकूली ऑनलाइन कॉन्टेक्स्टुअल-बैंडिट समस्या के रूप में मानता है ताकि सुधार संकेतों के आधार पर हस्तक्षेप रणनीतियों को गतिशील रूप से अनुकूलित किया जा सके, जिससे सीमित रोलआउट बजट के तहत लार्ज लैंग्वेज मॉडल्स के लिए पोस्ट-ट्रेनिंग दक्षता और प्रदर्शन को बढ़ाया जा सके।
मूल पेपर CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/) के तहत लाइसेंस किया गया है। नीचे दिए गए पेपर की यह व्याख्या AI से तैयार की गई है। इसे लेखकों ने न तो लिखा है, न इसका समर्थन किया है। तकनीकी सटीकता के लिए मूल पेपर देखें। पूरा डिस्क्लेमर पढ़ें
कल्पना कीजिए कि आप एक रोबोट को भूलभुलैया (maze) सुलझाना सिखा रहे हैं। पुराने दिनों में, आप शायद सिर्फ रोबोट को कह देते, "100 बार आगे बढ़ने की कोशिश करो, और अगर तुम दीवार से टकरा जाओ, तो बाईं ओर मुड़ने की कोशिश करो।" यह काम तो करता है, लेकिन यह बर्बादी भरा है। कभी-कभी रोबोट एक ऐसे डेड एंड (बंद रास्ते) पर पहुँच जाता है जिससे निकलना असंभव होता है; वहाँ 100 कोशिशें बर्बाद करना एक बड़ी गलती है। दूसरी ओर, कभी-कभी रोबोट एक शानदार शॉर्टकट से बस एक कदम की दूरी पर होता, लेकिन आप उसे केवल एक बार ही मौका देते हैं, इसलिए वह उस खोज को चूक जाता है। यही समस्या है कि हम वर्तमान में लार्ज लैंग्वेज मॉडल्स (LLMs) को स्मार्ट एजेंट बनाने के लिए कैसे प्रशिक्षित करते हैं: हम उन्हें हर एक समस्या के लिए प्रयासों की एक निश्चित संख्या (जिसे "रोलआउट्स" कहा जाता है) देते हैं, चाहे वे प्रयास उपयोगी हों या समय की बर्बादी।
यह शोध पत्र इस अक्षमता को दूर करता है। यह एक नई विधि पेश करता है जिसे RAIL (रिकवरेबिलिटी-अवेयर इंटरवेंशन लर्निंग) कहा जाता है। RAIL को एक सुपर-स्मार्ट कोच की तरह समझें जो रोबोट के बगल में खड़ा है। अंधे होकर रोबोट को 100 बार दौड़ने देने के बजाय, कोच रोबोट की प्रगति को देखता है। यदि रोबोट किसी ऐसे जाल में फँस गया है जहाँ से वह कभी नहीं निकल सकता, तो कोच कहता है, "रुको! यहाँ और कोशिश करने में समय बर्बाद मत करो।" लेकिन यदि रोबोट एक ऐसी कठिन स्थिति में है जहाँ थोड़ी अतिरिक्त खोज एक छिपा हुआ रास्ता प्रकट कर सकती है, तो कोच चिल्लाता है, "जाओ! अभी तीन और विविधताओं को आज़माओ!" कोच यह सीखता है कि कब हस्तक्षेप करना है और कैसे हस्तक्षेप करना है, न कि कठोर नियमों के आधार पर अनुमान लगाकर।
समस्या: "एक ही आकार सबके लिए" वाला जाल (The "One-Size-Fits-All" Trap)
AI की दुनिया में, विशेष रूप से मॉडल को तर्क करने और टूल्स (जैसे डेटाबेस या शॉपिंग वेबसाइट का उपयोग करना) का उपयोग करने के लिए सिखाते समय, हम एक तकनीक का उपयोग करते हैं जिसे रीइन्फोर्समेंट लर्निंग कहा जाता है। मॉडल एक कार्य को हल करने का प्रयास करता है, यदि वह सफल होता है तो उसे इनाम मिलता है, और वह अपनी गलतियों से सीखता है। एक लोकप्रिय विधि जिसे GRPO (ग्रुप रिलेटिव पॉलिसी ऑप्टिमाइज़ेशन) कहा जाता है, एक ही प्रश्न के लिए अलग-अलग उत्तरों का एक समूह (एक "रोलआउट") उत्पन्न करती है और देखती है कि उनमें से सबसे अच्छा कौन सा है।
समस्या यह है कि GRPO आमतौर पर हर प्रश्न के साथ एक जैसा व्यवहार करता है। यह कहता है, "हर प्रश्न के लिए, 16 उत्तर उत्पन्न करें।" लेकिन वास्तव में, कुछ प्रश्न आसान होते हैं (मॉडल पहले से ही उत्तर जानता है), और कुछ असंभव होते हैं (मॉडल एक तार्किक चक्र में फँसा हुआ है)। एक असंभव प्रश्न के लिए 16 उत्तर उत्पन्न करना उस घास के ढेर में सुई खोजने जैसा है जो वहाँ है ही नहीं; यह केवल कंप्यूटर पावर को जलाता है। इसके विपरीत, एक कठिन प्रश्न के लिए, 16 उत्तर एक चतुर समाधान खोजने के लिए पर्याप्त नहीं हो सकते हैं।
इसे ठीक करने के पिछले प्रयासों ने "ह्यूरिस्टिक" नियमों का उपयोग किया—सरल, निश्चित निर्देश जैसे "यदि मॉडल भ्रमित लग रहा है, तो अधिक प्रयास करें।" लेकिन ये नियम स्थिर हैं। वे AI के स्मार्ट होने के साथ नहीं बदलते। एक नियम जो AI के शुरुआती दौर में काम आया था, वह AI के विशेषज्ञ बनने पर बेकार हो सकता है। यह 1990 के मानचित्र का उपयोग करके एक ऐसे शहर में नेविगेट करने जैसा है जो पूरी तरह से पुनर्निर्मित किया जा चुका है।
समाधान: एक कोच जो हस्तक्षेप करना सीखता है
इस शोध पत्र के लेखक RAIL का प्रस्ताव करते हैं, जो एक ऐसी प्रणाली है जो "कब और कैसे अधिक प्रयास करना है" के निर्णय को स्वयं एक सीखने की प्रक्रिया में बदल देती है। एक निश्चित नियम के बजाय, RAIL एक "रिकवरेबिलिटी कंट्रोलर" (Recoverability Controller) को प्रशिक्षित करता है।
यह कैसे काम करता है, एक सरल उदाहरण के माध्यम से समझते हैं:
कल्पना कीजिए कि आप एक वीडियो गेम खेल रहे हैं जहाँ आप अपना रास्ता "ब्रांच" (शाखा बनाना) चुन सकते हैं। कभी-कभी, आप सड़क के मोड़ पर पहुँचते हैं।
- शैडो फेज (प्रशिक्षण शिविर): वास्तविक खेल खेलने से पहले, यह एक "शैडो" प्रशिक्षण मोड से गुजरता है। यहाँ, यह विभिन्न रणनीतियाँ आज़माता है: "क्या होगा अगर मैं 4 अतिरिक्त पथ आज़माऊँ?" "क्या होगा अगर मैं 8 आज़माऊँ?" "क्या होगा अगर मैं सोचने का एक अलग तरीका अपनाऊँ?" यह रिकॉर्ड करता है कि इनमें से किस विकल्प ने वास्तव में एक बेहतर समाधान खोजने में मदद की। यह एक कोच द्वारा ड्रिल चलाने जैसा है यह देखने के लिए कि कौन से व्यायाम वास्तव में खिलाड़ी के प्रदर्शन में सुधार करते हैं।
- कंट्रोलर (स्मार्ट कोच): इन ड्रिल्स के आधार पर, AI एक "रिकवरेबिलिटी कंट्रोलर" बनाता है। यह एक छोटा, तेज़ मस्तिष्क है जो वर्तमान स्थिति को देखता है और पूछता है: "यदि मैं अभी हस्तक्षेप करता हूँ, तो क्या इससे मदद मिलेगी?" यह रिकवरेबिलिटी (recoverability) को मापता है—एक फैंसी शब्द जिसका अर्थ है "यदि हम अभी अधिक प्रयास करें तो हम कितना बेहतर हो सकते हैं?"
- लाइव फेज (असली खेल): अब, AI असली खेल खेलता है। कंट्रोलर हर कदम पर नज़र रखता है। यदि AI ऐसी जगह पर है जहाँ अतिरिक्त प्रयास निश्चित रूप से मदद करेंगे (उच्च रिकवरेबिलिटी), तो कंट्रोलर कहता है, "हस्तक्षेप करो! शाखा बनाओ!" यदि AI एक ऐसे डेड एंड में है जहाँ अधिक प्रयास मदद नहीं करेंगे (कम रिकवरेबिलिटी), तो कंट्रोलर कहता है, "छोड़ दो। अपनी ऊर्जा बचाओ।"
उन्होंने क्या पाया
शोधकर्ताओं ने चार चुनौतीपूर्ण कार्यों पर RAIL का परीक्षण किया जहाँ AI एजेंटों को ऑपरेटिंग सिस्टम, डेटाबेस, वेब शॉप और डेटा विश्लेषण टूल के साथ इंटरैक्ट करना था।
- यह बेहतर काम करता है: RAIL ने लगातार सभी अन्य तरीकों को पछाड़ दिया, जिसमें मानक "एक ही आकार सबके लिए" वाला दृष्टिकोण और अन्य "स्मार्ट" तरीके शामिल हैं जो निश्चित नियमों का उपयोग करते हैं। इसने कार्यों को हल करने में उच्च सफलता दर प्राप्त की।
- यह पैसे बचाता है: शायद सबसे प्रभावशाली बात यह है कि RAIL ने इन बेहतर परिणामों को प्राप्त करने के लिए अन्य तरीकों की तुलना में कम कुल प्रयासों (रोलआउट्स) का उपयोग किया। यह न केवल स्मार्ट बना; यह अधिक कुशल भी बना। इसने उन निराशाजनक स्थितियों पर समय बर्बाद करना बंद कर दिया जहाँ कोई उम्मीद नहीं थी और अपनी ऊर्जा वहीं केंद्रित की जहाँ इसकी आवश्यकता थी।
- यह अनुकूलित होता है: शोध पत्र दिखाता है कि हस्तक्षेप करने का "सबसे अच्छा" तरीका AI के सीखने के साथ बदल जाता है। एक रणनीति जो शुरुआत में काम करती है, वह बाद में काम नहीं कर सकती। क्योंकि RAIL का कंट्रोलर नए परिणामों से लगातार सीखता रहता है, यह AI के बढ़ते कौशल के अनुकूल होता है, जबकि पुराने निश्चित नियम पीछे छूट जाते हैं।
बड़ी तस्वीर
यह शोध पत्र सुझाव देता है कि AI के लिए प्रशिक्षण डेटा उत्पन्न करने का तरीका एक निश्चित, उबाऊ प्रक्रिया नहीं होना चाहिए। इसके बजाय, "चीजों को आज़माने" की प्रक्रिया स्वयं एक गतिशील, सीखने योग्य हिस्सा होनी चाहिए। AI को यह पहचानने के सिखाकर कि उसे कब गहराई तक जाने की आवश्यकता है और कब आगे बढ़ना चाहिए, हम AI प्रशिक्षण को तेज़, सस्ता और अधिक प्रभावी बना सकते हैं। यह किसी समस्या पर अंधाधुंध अधिक कंप्यूटर पावर लगाने के बजाय, उस शक्ति को ठीक वहीं निर्देशित करने की ओर एक बदलाव है जहाँ इसकी आवश्यकता है।
अपने क्षेत्र के पेपरों की भीड़ में उलझे हुए हैं?
आपके रिसर्च कीवर्ड से मेल खाने वाले सबसे नए और अलग सोच वाले पेपरों का रोज़ाना Digest पाएँ—तकनीकी सारांश के साथ, आपकी भाषा में।