तकनीकी सारांश: MADA-RL
समस्या विवरण
लार्ज लैंग्वेज मॉडल्स (LLMs) ने मजबूत तर्क क्षमताएं प्रदर्शित की हैं, फिर भी मल्टी-स्टेप लॉजिकल रीजनिंग में उच्च प्रदर्शन प्राप्त करने के लिए अक्सर अत्यधिक प्रशिक्षण लागत की आवश्यकता होती है, विशेष रूप से कॉम्पैक्ट मॉडल्स (≤4 बिलियन पैरामीटर्स) के लिए। जबकि मॉडल का स्केल बढ़ाना रीजनिंग में सुधार करता है, यह गणनात्मक रूप से महंगा है। मौजूदा रणनीतियों में RL फाइन-ट्यूनिंग और टेस्ट-टाइम स्केलिंग मेथड्स (जैसे, चेन-ऑफ-थॉट, मल्टी-एजेंट डिबेट) शामिल हैं। हालांकि, RL को डिबेट के साथ एकीकृत करने के पिछले प्रयास अक्सर महत्वपूर्ण कम्प्यूटेशनल ओवरहेड, आर्किटेक्चरल जटिलता, अस्थिर क्रेडिट असाइनमेंट और फुल-मॉडल फाइन-ट्यूनिंग पर निर्भरता से ग्रस्त होते हैं। एक ऐसे लाइटवेट, पैरामीटर-एफिशिएंट फ्रेमवर्क की आवश्यकता है जो कॉम्पैक्ट मॉडल्स की क्षमताओं को फुल फाइन-ट्यूनिंग की लागत के बिना बढ़ाने के लिए स्ट्रक्चर्ड मल्टी-एजेंट रीजनिंग को RL के साथ प्रभावी ढंग से जोड़ सके।
कार्यप्रणाली: MADA-RL
लेखक MADA-RL (मल्टी-एजेंट डिबेट-अवेयर रीइन्फोर्समेंट लर्निंग) का प्रस्ताव करते हैं, जो एक पोस्ट-ट्रेनिंग फ्रेमवर्क है जिसे कॉम्पैक्ट मॉडल्स को विशिष्ट जेनरेटर (generator) और क्रिटिक (critic) भूमिकाओं में विशेषज्ञ बनाने के लिए डिज़ाइन किया गया है। यह फ्रेमवर्क लो-रैंक एडेप्टेशन (LoRA) एडेप्टर्स और ग्रुप रिलेटिव पॉलिसी ऑप्टिमाइजेशन (GRPO) का उपयोग करते हुए एक लाइटवेट पाइपलाइन पर संचालित होता है।
1. टेस्ट-टाइम डिबेट प्रोटोकॉल
इन्फरेंस प्रक्रिया एक मल्टी-एजेंट, मल्टी-राउंड, मल्टी-रोल प्रोटोकॉल का पालन करती है:
- जेनरेटर्स (Generators): कई जेनरेटर एजेंट (Gi) प्रश्न x के लिए स्वतंत्र रूप से प्रारंभिक उत्तर उत्पन्न करते हैं।
- क्रिटिक्स (Critics): कई क्रिटिक एजेंट मूल प्रश्न को जेनरेटर्स के रिस्पॉन्स के साथ जोड़कर प्राप्त करते हैं। वे इस संदर्भ के आधार पर अपडेटेड अनुमान (guesses) उत्पन्न करते हैं।
- इटरेशन (Iteration): यह प्रक्रिया R राउंड तक दोहराई जाती है। अंतिम सटीकता अंतिम राउंड के आउटपुट के आधार पर गणना की जाती है।
- डिज़ाइन चॉइस: प्रोटोकॉल मेमोरी एफिशिएंसी और सरलता बनाए रखने के लिए समराइजेशन (summarization) के बजाय डायरेक्ट रिस्पॉन्स कॉन्कैटिनेशन (direct response concatenation) का उपयोग करता है, क्योंकि एजेंटों की संख्या इतनी कम है कि वे कॉन्टेक्स्ट विंडो के भीतर फिट हो सकें।
2. ट्रेनिंग फ्रेमवर्क
ट्रेनिंग GRPO का उपयोग करते हुए एक दो-चरणीय प्रक्रिया है, जो एक वैल्यू-मॉडल-फ्री पॉलिसी ऑप्टिमाइजेशन मेथड है:
- चरण 1: जेनरेटर ट्रेनिंग: जेनरेटर एजेंटों को डेटासेट के डिस्क जॉइंट सबसेट्स पर स्वतंत्र रूप से प्रशिक्षित किया जाता है। उन्हें एक कंपोजिट रिवॉर्ड फंक्शन का उपयोग करके अनुकूलित किया जाता है जो सटीकता (ग्राउंड ट्रुथ के साथ बाइनरी सिम्बोलिक इक्विवेलेंस) और संक्षिप्तता (अनावश्यक टोकन लेंथ को दंडित करना) के बीच संतुलन बनाता है।
- चरण 2: क्रिटिक ट्रेनिंग: क्रिटिक एजेंटों को मूल प्रश्न और जेनरेटर रिस्पॉन्स के सेट के साथ समृद्ध डेटासेट पर प्रशिक्षित किया जाता है। उन्हें एक नवीन काउंटरफैक्चुअल क्रिटिक एडवांटेज (Counterfactual Critic Advantage) का उपयोग करके अनुकूलित किया जाता है।
3. काउंटरफैक्चुअल क्रिटिक एडवांटेज
MADA-RL का मुख्य नवाचार काउंटरफैक्चुअल एडवांटेज सिग्नल है जिसका उपयोग क्रिटिक्स को प्रशिक्षित करने के लिए किया जाता है। मानक RL के विपरीत, जहाँ एक क्रिटिक केवल सही उत्तर को दोहराना सीख सकता है, यह सिग्नल स्पष्ट रूप से क्रिटिक को जेनरेटर एन्सेम्बल के कंसेंसस (consensus) में सुधार करने के लिए अनुकूलित करता है।
- मैकेनिज्म: एक क्रिटिक के लिए एडवांटेज की गणना क्रिटिक के कुल रिवॉर्ड को एक डायनेमिक, रोल-कंडीशन्ड बेसलाइन से घटाकर की जाती है।
- बेसलाइन परिभाषा: बेसलाइन डेटा क्रिएशन के समय जेनरेटर एन्सेमबल की प्रति-इंस्टेंस औसत सटीकता (accG) है, जिसे रिवॉर्ड वेटिंग से मेल खाने के लिए दो के कारक (factor of two) से स्केल किया गया है।
- फॉर्मूला: AC(y^,y,accG)=RC(y^,y)−2⋅accG।
- प्रभाव: यह एक ऐसा लर्निंग सिग्नल बनाता है जहाँ क्रिटिक्स को विशेष रूप से जेनरेटर्स द्वारा की गई गलतियों को सुधारने के लिए पुरस्कृत किया जाता है (अर्थात, जब क्रिटिक सही होता है और जेनरेटर कंसेंसस गलत होता है) और कंसेंसस से कम प्रदर्शन करने के लिए दंडित किया जाता है। यह बिना किसी बाहरी वेरीफायर या वैल्यू मॉडल के क्रेडिट असाइनमेंट को तेज करता है।
मुख्य योगदान
- काउंटरफैक्चुअल क्रिटिक एडवांटेज: रोल-स्पेशलाइज्ड RL के लिए एक डायनेमिक बेसलाइन जो जेनरेटर एन्सेम्बल की सटीकता से बेहतर प्रदर्शन करने के लिए क्रिटिक्स को पुरस्कृत करता है, जिससे लक्षित क्रेडिट असाइनमेंट संभव होता है।
- MADA-RL फ्रेमवर्क: एक पैरामीटर-एफिशिएंट पोस्ट-ट्रेनिंग मेथड जो LoRA और GRPO का उपयोग करके कॉम्पैक्ट LLMs (विशेष रूप से DeepSeek-R1-Distill-Qwen-1.5B) पर इस सिग्नल को लागू करता है, जिसमें फुल फाइन-ट्यूनिंग की तुलना में बहुत कम ट्रेन करने योग्य पैरामीटर्स की आवश्यकता होती है।
- नियंत्रित विश्लेषण: प्रदर्शन लाभ के स्रोत को अलग करने वाला एक अध्ययन, जो यह प्रदर्शित करता है कि सुधार केवल टेस्ट-टाइम डेलिब्रेशन वॉल्यूम के बजाय सीखे गए सुधारात्मक व्यवहार (corrective behavior) से आते हैं।
- एम्पिरिकल बेंचमार्किंग: एक सामान्य रोल-स्पेशलाइज्ड प्रोटोकॉल के तहत मौजूदा 1.5B फाइन-ट्यूनिंग्स का व्यापक मूल्यांकन, जिसमें ट्रेनिंग-पैरामीटर और इन्फरेंस-टोकन ट्रेड-ऑफ का विस्तृत विवरण शामिल है।
प्रयोगात्मक परिणाम
लेखकों ने DeepSeek-R1-Distill-Qwen-1.5B मॉडल का उपयोग करके पांच गणितीय रीजनिंग बेंचमार्क (Math-500, AIME 2024/2025, AMC-23, Minerva-Math) पर MADA-RL का मूल्यांकन किया।
- प्रदर्शन: MADA-RL ने बेस मॉडल की औसत सटीकता को 39.9% से बढ़ाकर 41.9% (+2.0 पॉइंट्स, p<0.001) कर दिया।
- पैरामीटर एफिशिएंसी: इस विधि ने DeepScaleR और Still-3 जैसे फुलली फाइन-ट्यून्ड बेसलाइन्स (1.78B बनाम 110M ट्रेन करने योग्य पैरामीटर्स) की तुलना में 16 गुना कम ट्रेन करने योग्य पैरामीटर्स का उपयोग करके यह लाभ प्राप्त किया।
- मजबूत बेसलाइन्स के साथ तुलना: हालांकि MADA-RL डेटा-हैवी, फुलली फाइन-ट्यून्ड बेसलाइन्स (DeepScaleR 44.3%, Still-3 43.1%) की रॉ एक्यूरेसी को पार नहीं करता है, लेकिन यह उन्हें काफी पीछे छोड़ देता है जब उन बेसलाइन्स को उसी डेटा पर LoRA के साथ फिर से प्रशिक्षित किया जाता है (DeepScaleR-LoRA: 40.5%, Still-3-LoRA: 41.3%)। यह सुझाव देता है कि मजबूत बेसलाइन्स का लाभ डेटा स्केल में निहित है न कि उस मैकेनिज्म में जिसे LoRA द्वारा संरक्षित किया गया है।
- क्रिटिक इम्प्रूवमेंट रेट: MADA-RL ने उच्चतम क्रिटिक इम्प्रूवमेंट रेट (19.6%) हासिल किया, जो दर्शाता है कि इसके प्रशिक्षित क्रिटिक्स अन्य मूल्यांकित मॉडलों की तुलना में जेनरेटर त्रुटियों को अधिक बार सुधारते हैं।
- एब्लेशन स्टडीज (Ablation Studies):
- डिबेट स्ट्रक्चर को हटाने (सिंगल-एजेंट सेटिंग) से प्रदर्शन में महत्वपूर्ण गिरावट (-5.2 पॉइंट्स) आई, जो मल्टी-एजेंट प्रोटोकॉल पर इसकी निर्भरता की पुष्टि करता है।
- राउंड या एजेंटों की संख्या कम करने से प्रदर्शन काफी गिर गया।
- काउंटरफैक्चुअल एडवांटेज को हटाने से दिशात्मक गिरावट (0.8 पॉइंट्स) हुई, हालांकि यह परीक्षण किए गए सैंपल साइज पर सांख्यिकीय रूप से महत्वपूर्ण नहीं था; फिर भी, मैकेनिज्म-लेवल साक्ष्य (उच्च सुधार दर) इसकी प्रभावकारिता का समर्थन करते हैं।
महत्व और दावे
यह पेपर MADA-RL को सख्त ट्रेनिंग बजट के तहत कॉम्पैक्ट मॉडल्स में रीजनिंग को बढ़ाने के एक व्यावहारिक नुस्खे के रूप में प्रस्तुत करता है। इसका प्राथमिक महत्व यह प्रदर्शित करने में है कि पैरामीटर-एफिशिएंट स्पेशलाइजेशन (जेनरेटर्स और क्रिटिक्स को अलग करना) को काउंटरफैक्चुअल लर्निंग सिग्नल के साथ जोड़ने से फुल फाइन-ट्यूनिंग की लागत के बिना पर्याप्त रीजनिंग लाभ मिल सकता है।
लेखक विनम्रतापूर्वक दावा करते हैं कि MADA-RL कॉम्पैक्ट मॉडल्स को एक्यूरेसी–ट्रेन करने योग्य पैरामीटर पारेटो फ्रंट (Pareto front) पर रखता है, जो मूल्यांकित मॉडलों के बीच प्रति ट्रेन करने योग्य पैरामीटर उच्चतम एक्यूरेसी गेन प्रदान करता है। वे स्पष्ट रूप से कहते हैं कि जबकि यह विधि बड़े पैमाने के, डेटा-हैवी मॉडल्स के प्रदर्शन के करीब पहुँचती है, यह उनसे आगे नहीं निकलती है, और इस अंतर का कारण प्रशिक्षण डेटा का पैमाना है न कि स्वयं मैकेनिज्म। इसके अलावा, लेखक ट्रेड-ऑफ को स्वीकार करते हैं: यह विधि मल्टी-राउंड, मल्टी-एजेंट प्रोटोकॉल के कारण उच्च इन्फरेंस लेटेंसी (latency) लाती है, जिससे यह लेटेंसी-सेंसिटिव डिप्लॉयमेंट के लिए कम उपयुक्त हो जाती है, लेकिन उन परिदृश्यों के लिए अत्यधिक प्रभावी है जहाँ प्रशिक्षण संसाधन सीमित हैं। यह कार्य लाभ के स्रोत को केवल डेलिब्रेशन वॉल्यूम के बजाय सीखे गए सुधारात्मक व्यवहार (learned corrective behavior) के रूप में अलग करता है, जो यह समझने का एक स्पष्ट तरीका प्रदान करता है कि मल्टी-एजेंट डायनेमिक्स को RL के माध्यम से कैसे अनुकूलित किया जा सकता है।