Elicitation without Backpropagation: Steering Model Behavior by Optimizing the Latent Posterior
यह शोध पत्र पोस्टीरियर प्रीफिक्स ट्यूनिंग (PPT) प्रस्तुत करता है, जो एक नवीन विधि है जो अपेक्षित उपयोगिता को अधिकतम करने के लिए प्रॉम्प्ट वितरणों को अनुकूलित करने के माध्यम से बेयस-फ़िल्टर्ड ट्रांसफॉर्मर से विशिष्ट व्यवहार उत्पन्न करती है, जिससे अनुकूलन के दौरान बैकप्रोपैगेशन या अतिरिक्त ट्रांसफॉर्मर फॉरवर्ड पासेस की आवश्यकता समाप्त हो जाती है।
मूल पेपर CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/) के तहत लाइसेंस किया गया है। नीचे दिए गए पेपर की यह व्याख्या AI से तैयार की गई है। इसे लेखकों ने न तो लिखा है, न इसका समर्थन किया है। तकनीकी सटीकता के लिए मूल पेपर देखें। पूरा डिस्क्लेमर पढ़ें
कल्पना कीजिए कि आप एक सुपर-स्मार्ट रोबोट को कहानी सुनाना सिखाने की कोशिश कर रहे हैं। आप उसे केवल तथ्य नहीं देते; आप उसे एक "प्रॉम्ट" (prompt) देते हैं, एक छोटी शुरुआती वाक्य जैसे "एक समय की बात है..." और फिर देखते हैं कि आगे क्या होता है। यह लार्ज लैंग्वेज मॉडल्स (LLMs) की दुनिया है, जो वे AI चैटबॉट्स हैं जो कविताएँ, कोड और चुटकुले लिख सकते हैं। लेकिन यहाँ पेचीदा बात यह है कि ये रोबोट केवल चीज़ों को "जानते" नहीं हैं; वे वास्तव में एक विशाल अनुमान लगाने वाला खेल (guessing game) खेल रहे होते हैं। वे आपके द्वारा दिए गए शब्दों को देखते हैं और उन छिपे हुए नियमों के आधार पर अगले शब्द का अनुमान लगाने की कोशिश करते हैं जो उन्होंने प्रशिक्षण के दौरान सीखे हैं।
वैज्ञानिकों ने पाया है कि ये रोबोट एक रहस्य सुलझाने वाले जासूस की तरह काम करते हैं। जैसे ही रोबोट आपका प्रॉम्ट पढ़ता है, वह इस बारे में अपने "विश्वास" (belief) को अपडेट करता है कि आप किस तरह की कहानी सुनाना चाहते हैं। यह ऐसा है जैसे रोबोट के पास हजारों अलग-अलग "कहानीकारों" (कुछ जिन्हें सुखद अंत पसंद है, कुछ जिन्हें हॉरर पसंद है, कुछ जिन्हें गणित पसंद है) की एक मानसिक लाइब्रेरी है, और आपका प्रॉम्ट यह तय करने में मदद करता है कि अभी किस कहानीकार को सुनना है। इसे लेटेंट पोस्टीरियर मॉडल (latent posterior model) कहा जाता है। बड़ा सवाल जो शोधकर्ता पूछ रहे हैं वह यह है: क्या हम रोबोट को एक विशिष्ट कहानीकार चुनने के लिए मजबूर कर सकते हैं ताकि वह बिल्कुल वैसा ही कहे जैसा हम चाहते हैं, भले ही वह कोई अजीब या खतरनाक चीज़ हो? इसे एलिसिटेशन प्रॉब्लम (elicitation problem) कहा जाता है। यह एक जादुई मंत्र (प्रॉम्ट) खोजने की कोशिश करने जैसा है जिससे जिनी (genie) आपकी एक विशिष्ट इच्छा पूरी करे, लेकिन जिनी एक 'ब्लैक बॉक्स' है, और संभावित मंत्रों का स्थान इतना विशाल है कि उन्हें एक-एक करके जांचना संभव नहीं है।
भारी मेहनत के बिना जादुई मंत्र
इस शोध पत्र में, गैरेट बेकर, टिमेअस, विनायक पाठक, डैनियल मर्फ़ेट और सुसान वे नाम की शोधकर्ताओं की एक टीम एक चतुर नया तरीका पेश करती है जिससे वे उन जादुic मंत्रों को खोज सकें। वे अपने इस तरीके को पोस्टीरियर प्रीफिक्स ट्यूनिंग (Posterior Prefix Tuning - PPT) कहते हैं।
आमतौर पर, जब लोग किसी AI को एक निश्चित तरीके से व्यवहार करने के लिए प्रेरित करने हेतु सही प्रॉम्ट खोजने की कोशिश करते हैं, तो वे ग्रीडी कोऑर्डिनेट ग्रेडिएंट (Greedy Coordinate Gradient - GCF) नामक एक ब्रूट-फोर्स (brute-force) विधि का उपयोग करते हैं। कल्पना कीजिए कि आप एक विशिष्ट गाना खोजने के लिए रेडियो को ट्यून करने की कोशिश कर रहे हैं। GCG विधि रेडियो के डायल को एक बार में एक बहुत छोटे कदम से घुमाने, शोर सुनने, वापस घुमाने, दूसरी दिशा में घुमाने, फिर से सुनने और इसे हजारों बार दोहराने जैसी है। यह काम तो करता है, लेकिन यह धीमा है और इसके लिए हर बार थोड़ा सा बदलाव करने पर रोबोट को "सोचना" (एक गणना चलाना) पड़ता है।
इस शोध पत्र के लेखक कहते हैं, "ठहरिए! हमें हर बार रेडियो सुनने की ज़रूरत नहीं है।" उन्होंने महसूस किया कि क्योंकि रोबोट अनिवार्य रूप से अपने विश्वास को अपडेट करने वाला एक जासूस है, हम रेडियो को पूरी तरह से छोड़ सकते हैं। रोबोट को एक कहानी उत्पन्न करने के लिए कहने और फिर यह जाँचने के बजाय कि क्या वह अच्छी है, उन्होंने सीधे रोबोट की मानसिक लाइब्रेरी (latent posterior) को देखने का निर्णय लिया।
"एक-बार" की लाइब्रेरी यात्रा
यहाँ जादुई ट्रिक है: शोधकर्ता पहले रोबोट को बिना किसी विशिष्ट प्रॉम्ट के कई रैंडम कहानियाँ सुनाने के लिए कहते हैं। इन रैंडम कहानियों से, वे रोबोट की पूरी मानसिक लाइब्रेरी का एक मानचित्र (map) बनाते हैं। वे पता लगाते हैं कि लाइब्रेरी में कौन से "कहानीकार" (latent models) हैं और रोबोट उनके चुने जाने की कितनी संभावना रखता है। यह रोबोट के मस्तिष्क का एक बार लिया गया स्नैपशॉट लेने जैसा है, जिसमें लाइब्रेरी का नक्शा बनाने के लिए रोबोट को 5,000 लंबी कहानियाँ (rollouts) उत्पन्न करने के लिए कहना शामिल है।
एक बार जब उनके पास यह स्नैपशॉट आ जाता है, तो उन्हें ऑप्टिमाइज़ेशन प्रक्रिया के दौरान रोबोट को फिर से सोचने के लिए कहने की आवश्यकता नहीं होती है। वे गणित का उपयोग करके यह सिम्युलेट (simulate) कर सकते हैं: "यदि हमने रोबोट को यह विशिष्ट प्रॉम्ट दिया, तो वह किस कहानीकार को चुनेगा? और क्या वह कहानीकार वांछित परिणाम देगा?"
वे ऐसा रोबोट के विश्वासों का एक "झुका हुआ" (tilted) संस्करण बनाकर करते हैं। कल्पना कीजिए कि आपके पास कंचों (marbles) का एक थैला है, जिसमें से प्रत्येक एक अलग कहानीकार का प्रतिनिधित्व करता है। कुछ कंचे लाल (अच्छे कहानीकार) हैं, कुछ नीले (बुरे वाले) हैं। रोबोट आमतौर पर पासे के निष्पक्ष रोल के आधार पर कंचों को चुनता है। PPT एक ऐसी विधि है जो आपको जादुчески लाल कंचों पर वजन बढ़ाने की अनुमति देती है ताकि रोबोट उन्हें चुनने के लिए मजबूर हो जाए, बिना वास्तव में दोबारा पासा फेंके।
परिणाम: जीत और हार का मिश्रण
टीम ने इस परीक्षण को दो प्रकार के सरलीकृत रोबोट दिमागों (बेयस-फिल्टर्ड ट्रांसफॉर्मर) पर टेस्ट किया। एक सरल सिक्का उछालने वाला (Beta–Bernoulli) था, और दूसरा थोड़ा अधिक जटिल पैटर्न-फॉलोअर (Reinforced Urn) था। उन्होंने इन रोबोटों से तीन चीजें करने की कोशिश की:
- रिवर्स क्रॉस-एन्ट्रॉपी (Reverse Cross-Entropy): रोबole के आउटपुट को एक विशिष्ट लक्ष्य पैटर्न जैसा बनाना।
- फ्रीक्वेंसी मैचिंग (Frequency Matching): रोबोट को एक विशिष्ट अनुपात में "0" और "1" कहना सिखाना।
- डिक वैलिडिटी (Dyck Validity): रोबोट को कोष्ठकों (parentheses) का एक ऐसा क्रम उत्पन्न करना सिखाना जो पूरी तरह से संतुलित हो (जैसे
()()या(()))।
उन्होंने अपने नए "वन-शॉट" तरीके (PPT) की तुलना पुराने "डायल-घुमाने" वाले तरीके (GCG) से की।
- अच्छी खबर: अधिक जटिल "Reinforced Urn" रोबोट पर, PPT ने विशिष्ट परिदृश्यों में प्रभावशाली परिणाम दिखाए। उदाहरण के लिए, जब प्रॉम्ट छोटे (6 अक्षर) थे और लक्ष्य संतुलित कोष्ठक (Dyck validity) उत्पन्न करना था, तो PPT-RB ने हर एक प्रयास में सही प्रॉम्ट खोजा, जबकि पुराना तरीका संघर्ष करता रहा। यह ऐसा था जैसे PPT मील दूर से ही समाधान देख सकता था, जबकि GCG अंधेरे में लड़खड़ा रहा था।
- मिली-जुली खबर: सरल "Coin-flipper" रोबोट पर, दोनों विधियाँ काफी अच्छी थीं, लेकिन पुराना तरीका (GCG) कभी-कभी बेहतर प्रदर्शन करता था, विशेष रूप से जब प्रॉम्ट लंबे (50 अक्षर) होते थे।
- आश्चर्य: प्रदर्शन कार्य और प्रॉम्ट की लंबाई के आधार पर बदल गया। जटिल रोबोट के लिए "Dyck Validity" परीक्षण पर, लंबे प्रॉम्ट (50 अक्षर) के साथ, पुराने तरीके (GCG) ने वास्तव में जीत हासिल की, जिसने वह समाधान खोजा जिसे PPT मिस कर गया। इसी तरह, सरल रोबोट पर, जब प्रॉम्ट लंबे थे तो GCG अक्सर PPT से बेहतर रहा।
यह क्यों मायने रखता है
PPT की सबसे बड़ी जीत केवल यह नहीं है कि यह कभी-कभी बेहतर प्रॉम्ट खोजता है; बल्कि इसकी दक्षता (efficiency) है। पुराने तरीके को प्रॉम्ट में हर एक छोटे बदलाव के लिए रोबोट को "सोचने" (एक फॉरवर्ड पास चलाना) और फिर "पीछे की ओर सोचने" (बैकप्रोपैगेट करना) की आवश्यकता होती थी। यह महंगा और धीमा है।
हालाँकि, PPT को वास्तविक ऑप्टिमाइज़ेशन चरणों के दौरान रोबोट को शून्य बार सोचने के लिए कहता है। यह सारा भारी काम शुरुआत में ली गई लाइब्रेरी के "स्नैपशॉट" (प्रारंभिक 5,000 रोलआउट्स) का उपयोग करके करता है। यह पूरे भूलभुलैया के मानचित्र को शुरू करने से पहले ही प्राप्त करने जैसा है। आप अपने दिमाग में तुरंत लाखों रास्तों को आजमा सकते हैं, बिना कभी एक भी कदम चले।
लेखक नोट करते हैं कि यह विधि तब सबसे अच्छा काम करती है जब रोबोट का व्यवहार अच्छी तरह से समझा गया हो और विशिष्ट गणितीय नियमों का पालन करता हो (जैसे कि परीक्षण किए गए सरलीकृत मॉडल)। वे स्वीकार करते हैं कि वास्तविक दुनिया के, जटिल AI मॉडल के लिए, यह अभी तक एक पूर्ण समाधान नहीं हो सकता है। लेकिन "रोबोट के छिपे हुए विश्वासों को नियंत्रित करने के लिए हम एक विशिष्ट परिणाम कैसे प्राप्त करें" की विशिष्ट समस्या के लिए, उन्होंने एक ऐसा रास्ता दिखाया है जो तेज़, सस्ता और कई मामलों में आश्चर्यजनक रूप से प्रभावी है।
संक्षेप में, उन्होंने रोबोट के दिमाग को उसके आंतरिक मानचित्र को देखकर नियंत्रित करने का एक तरीका खोजा है, न कि केवल दरवाजा खटखटाकर उम्मीद करने का। यह जिनी से इच्छा मांगने का एक स्मार्ट तरीका है, और यह हमें यह समझने में मदद कर सकता है कि इन शक्तिशाली उपकरणों को हमारे इच्छित तरीके से व्यवहार करने के लिए कैसे बनाए रखा जाए।
अपने क्षेत्र के पेपरों की भीड़ में उलझे हुए हैं?
आपके रिसर्च कीवर्ड से मेल खाने वाले सबसे नए और अलग सोच वाले पेपरों का रोज़ाना Digest पाएँ—तकनीकी सारांश के साथ, आपकी भाषा में।