Endpoint Replay: Compressing the Recency Buffer in Deep Reinforcement Learning
यह शोध पत्र "एंडपॉइंट रिप्ले" (Endpoint Replay) को प्रस्तुत करता है, जो एक ऐसी विधि है जो जुड़े हुए n-स्टेप अनुक्रमों के एंडपॉइंट्स से प्रतिनिधि ट्रांज़िशनों को संग्रहीत करके एक्सपीरियंस रिप्ले बफ़र को संकुचित करती है, जिससे भंडारण आवश्यकताओं को एक क्रम (order of magnitude) तक कम करते हुए पारंपरिक बड़े बफ़र्स के तुलनीय प्रदर्शन प्राप्त होता है।
मूल पेपर CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/) के तहत लाइसेंस किया गया है। नीचे दिए गए पेपर की यह व्याख्या AI से तैयार की गई है। इसे लेखकों ने न तो लिखा है, न इसका समर्थन किया है। तकनीकी सटीकता के लिए मूल पेपर देखें। पूरा डिस्क्लेमर पढ़ें
कल्पना कीजिए कि एक रोबोट वीडियो गेम खेलना सीख रहा है। बेहतर बनने के लिए, उसे अभ्यास करने की आवश्यकता है, लेकिन वह केवल पिछले कुछ सेकंड में जो हुआ उस पर निर्भर नहीं रह सकता; उसे घंटों पहले सीखे गए सबक याद रखने होंगे। आर्टिफिशियल इंटेलिजेंस की दुनिया में, इस मेमोरी बैंक को "एक्सपीरियंस रिप्ले बफर" (experience replay buffer) कहा जाता है। इसे एक छात्र की नोटबुक की तरह समझें जहाँ वह अपनी हर चाल, अपना स्कोर और उसके बाद क्या हुआ, यह सब लिखता है। अधिकांश आधुनिक AI रोबोट एक विशाल नोटबुक का उपयोग करते हैं—अक्सर दस लाख प्रविष्टियों (entries) तक—यह सुनिश्चित करने के लिए कि वे महत्वपूर्ण चीजों को न भूलें। वे अपने अतीत की गलतियों और जीत से सीखने के लिए इस नोटबुक के पन्नों को बेतरतीब ढंग से पलटते हैं। लेकिन यहाँ एक पेंच है: दस लाख प्रविष्टियों को स्टोर करने के लिए बहुत अधिक कंप्यूटर मेमोरी और ऊर्जा की आवश्यकता होती है। वैज्ञानिक एक सरल प्रश्न पूछ रहे हैं: क्या हम इस नोटबुक को बिना रोबोट को खेलना भुलाए, एक बहुत छोटे आकार में सिकोड़ सकते हैं?
यह शोध पत्र (paper) ठीक इसी समस्या से निपटता है। यह इस मेमोरी को कंप्रेस करने का एक चतुर नया तरीका पेश करता है, और यह सिद्ध करता है कि एक AI मानक बुफर्स की तुलना में 10 से 50 गुना छोटी नोटबुक के साथ भी उतना ही अच्छा सीख सकता है। शोधकर्ताओं ने पाया कि यदि आप एक विशाल नोटबुक से केवल कुछ "महत्वपूर्ण" पन्ने चुनने की कोशिश करते हैं, तो रोबोट भ्रमित हो जाता है क्योंकि उन पन्नों का आपस में कोई संबंध नहीं होता। उनका समाधान? अलग-थलग पन्नों को रखने के बजाय, वे घटनाओं की "शृंखलाएं" (chains) रखते हैं जो आपस में जुड़ी होती हैं, जिससे यह सुनिश्चित होता है कि हर सबक तार्किक रूप से अगले सबक की ओर ले जाए। समय के अंतराल को संभालने के लिए एक विशेष गणितीय ट्रिक का उपयोग करके, वे एक विशाल मेमोरी को एक बहुत छोटे स्थान में समेटने में सफल रहे, जिससे रोबोट अपने कौशल को खोए बिना तेजी से और कम खर्च में सीख सका।
समस्या: बिखरे हुए पन्नों वाली एक नोटबुक
वर्षों से, AI एजेंटों (जैसे कि जो अटाari गेम्स खेल रहे हैं) को सिखाने का मानक तरीका एक विशाल "FIFO" (First-In, First-Out) बफर रखना रहा है। एक कन्वेयर बेल्ट की कल्पना करें जहाँ नए अनुभव सामने से अंदर आते हैं और पुराने पीछे से बाहर गिर जाते हैं। यह बेल्ट आमतौर पर दस लाख स्टेप्स लंबी होती है। AI इस बेल्ट से रैंडम पन्ने चुनकर सीखता है। यह काम करता है, लेकिन यह बहुत भारी (bulky) है।
शोधकर्ताओं ने पहले भी इस बेल्ट को सिकोड़ने की कोशिश की है। कुछ ने केवल "सबसे महत्वपूर्ण" पन्ने रखने की कोशिश की, जिसे कोरेसेट (coreset) तकनीक कहा जाता है। विचार यह था कि सबसे अच्छे, सबसे प्रतिनिधि अनुभवों को चुनें और बाकी को फेंक दें। हालाँकि, इस पेपर के लेखकों ने इस दृष्टिकोण में एक छिपे हुए दोष को पाया।
जब आप एक विशाल इतिहास से रैंडम, अलग-थलग पन्ने चुनते हैं, तो अक्सर आपके पास एक ऐसा पन्ना होता है जो कहता है, "मैंने यह क्रिया की, और फिर मैं इस स्थिति (state) में पहुँचा।" लेकिन यदि आप बीच के सभी पन्नों को हटा देते हैं, तो AI वास्तव में यह कभी नहीं सीख पाता कि उस लैंडिंग स्टेट के अंदर क्या हुआ। यह एक कॉमिक बुक पढ़ने जैसा है जहाँ आप केवल एक दृश्य के पहले और अंतिम पैनल को रखते हैं। आप नायक को कूदते हुए देखते हैं, और फिर आप उन्हें उतरते हुए देखते हैं, लेकिन आपको पता ही नहीं चलता कि वे वहाँ कैसे पहुँचे या उतरने का अनुभव कैसा था। AI की भाषा में, "बूटस्ट्रैप टारगेट" (वह मान जिसे AI अगले कदम के लिए अनुमान लगाने की कोशिश करता है) अनएंकर्ड (unanchored) यानी बिना किसी आधार के हो जाता है। AI उस स्थिति के मान का अनुमान लगा रहा होता है जिसे उसने अपनी वर्तमान, छोटी मेमोरी में कभी अपडेट या देखा ही नहीं है। इससे गलत अनुमान और खराब प्रदर्शन होता है।
समाधान: एंडपॉइंट रिप्ले (Endpoint Replay)
लेखक एंडपॉइंट रिप्ले नामक एक नया तरीका प्रस्तावित करते हैं। अलग-थलग पन्ने रखने के बजाय, वे शृंखलाएं (chains) रखते हैं।
कल्पना कीजिए कि आप एक लंबी फिल्म को एक 'हाइलाइट रील' में कंप्रेस कर रहे हैं। रैंडम फ्रेम चुनने के बजाय, आप एक शुरुआती फ्रेम चुनते हैं, 10 फ्रेम आगे बढ़ते हैं, और फिर उस बीच मिले सभी पुरस्कारों (rewards) के पूरे क्रम को एक एकल "सारांश" के रूप में रिकॉर्ड करते हैं। आप शुरुआती बिंदु और अंतिम बिंदु को रखते हैं, लेकिन उन्हें उस जंप के दौरान अर्जित कुल स्कोर के साथ जोड़ देते हैं।
यह उनके सिस्टम में इस प्रकार काम करता है:
- दो बफर्स: AI तत्काल सीखने के लिए एक छोटा "रिसेंसी बफर" (पिछले 10,000 स्टेप्स) रखता है, बिल्कुल एक मानक नोटबुक की तरह।
- लैग बफर (The Lag Buffer): जैसे ही पुराना डेटा रिसेंसी बफर से बाहर निकलता है, उसे केवल हटाया नहीं जाता। वह एक छोटे "लैग बफर" में जाता है जो लगभग 10 स्टेप्स को रखता है।
- शृंखला (The Chain): एक बार जब लैग बफर भर जाता है (या एपिसोड समाप्त हो जाता है), तो AI उन 10 स्टेप्स के लिए कुल रिवॉर्ड की गणना करता है और एक एकल "n-स्टेप ट्रांजिशन" बनाता है। यह शुरुआती स्थिति, अंतिम स्थिति और बीच में अर्जित कुल रिवॉर्ड को स्टोर करता है।
- कोरेसेट (The Coreset): ये "चेन किए गए" सारांश एक दूसरे, थोड़े बड़े बफर में रखे जाते हैं जिसे कोरेसेट कहा जाता है।
जादू यह है कि कोरेसेट में प्रत्येक अंतिम स्थिति (ending state) एक अन्य शृंखला के लिए शुरुआती स्थिति (starting state) भी होती है। इसका मतलब है कि AI को कभी भी ऐसी स्थिति के मान का अनुमान नहीं लगाना पड़ता जिसे उसने देखा ही नहीं है; हर "लैंडिंग" एक "टेकऑफ" द्वारा एंकर की जाती है जो स्वयं मेमोरी में मौजूद है। भले ही बीच के स्टेप्स गायब हों, लेकिन कनेक्शन सुरक्षित रहते हैं।
"पुरानी खबर" के पूर्वाग्रह (Bias) को ठीक करना
एक और बाधा थी। क्योंकि ये शृंखलाएं AI के एक पुराने संस्करण द्वारा एकत्र किए गए डेटा से बनाई गई हैं (चूंकि एजेंट लगातार बेहतर होता जा रहा है), इसलिए शृंखला में रिवॉर्ड्स वर्तमान, स्मार्ट AI द्वारा प्राप्त किए जाable स्तर की तुलना में "निराशावादी" या बहुत कम लग सकते हैं।
इसे ठीक करने के लिए, लेखकों ने एक्सपेक्टाइल सारसा (Expectile Sarsa) नामक एक गणितीय उपकरण का उपयोग किया।
- मानक लर्निंग औसत रिवॉर्ड खोजने की कोशिश करती है।
- एक्सपेक्टाइल (Expectile) लर्निंग एक "निराशावादी फिल्टर" की तरह है। यह उन रिवॉर्ड्स पर ध्यान केंद्रित करती है जो औसत से बेहतर हैं, जो प्रभावी रूप से AI को बताता है, "सिर्फ बीच के स्तर पर न रुकें; उच्च स्कोर का लक्ष्य रखें।"
इस विशिष्ट प्रकार के गणित का उपयोग करके, AI उन पुराने, 10-स्टेप की शृंखलाओं को देख सकता है और अभी भी सही मान सीख सकता है, बिना अपने अतीत के गलत निर्णयों से नीचे गिरे।
उन्होंने क्या पाया
शोधकर्ताओं ने इस विचार का परीक्षण दो बहुत अलग दुनियाओं में किया:
- पिनबॉल (Pinball): एक भौतिकी-आधारित गेम जहाँ एक गेंद मेज के चारों ओर उछलती है।
- अटाari 2600 (Atari 2600): 12 क्लासिक वीडियो गेम्स (जैसे Breakout, Space Invaders, और Pong) का एक संग्रह।
उन्होंने अपने एंडपॉइंट रिप्ले तरीके की तुलना निम्नलिखित से की:
- एक मानक विशाल बफर (1 मिलियन स्टेप्स)।
- एक छोटा बफर (100k स्टेप्स) बिना किसी चेन के।
- रैंडम सैंपलिंग वाला छोटा बफर (पुराना तरीका)।
- MeDQN जैसे अन्य कंप्रेशन तरीके।
परिणाम:
- भारी कंप्रेशन: एंडपॉइंट रिप्ले ने 1-मिलियन-स्टेप के विशाल बफर के समान ही प्रदर्शन किया, भले ही इसमें 10 से 50 गुना कम स्टोरेज का उपयोग किया गया हो।
- बुनियादी बातों को पछाड़ना: 10x और 50x छोटे सेटिंग्स में, एंडपॉइंट रिप्ले ने रैंडम सैंपलिंग या मानक 1-स्टेप अपडेट का उपयोग करने वाले छोटे बुफर्स की तुलना में काफी बेहतर प्रदर्शन किया।
- "एंकरिंग" का प्रमाण: जब उन्होंने "चेनिंग" को हटा दिया (स्थितियों को फिर से अनएंकर्ड बना दिया), तो प्रदर्शन गिर गया। इसने साबित किया कि "अनएंकर्ड बूटस्ट्रैप" की समस्या वास्तविक थी और उनकी चेनिंग समाधान ही मुख्य कुंजी थी।
- गणित मायने रखता है: जब उन्होंने विशेष "एक्सपेक्टाइल" गणित को हटा दिया और मानक औसत का उपयोग किया, तो प्रदर्शन थोड़ा कम हो गया, जिससे पता चला कि "पुरानी खबर" के पूर्वाग्रह को संभालना भी महत्वपूर्ण था।
निष्कर्ष
यह पेपर केवल यह सुझाव नहीं देता कि छोटे बफर्स शायद काम कर सकते हैं; यह दिखाता है कि सही संरचना के साथ, वे वास्तव में काम करते हैं। लेखकों ने गणितीय रूप से सिद्ध किया कि उनका तरीका सुदृढ़ है और सिमुलेशन के माध्यम से प्रदर्शित किया कि यह विशाल मेमोरी बैंकों के प्रदर्शन से मेल खाता है जबकि बहुत कम स्थान का उपयोग करता है।
उन्होंने केवल नोटबुक को छोटा नहीं किया; उन्होंने इस बात को फिर से लिखा कि पन्ने आपस में कैसे जुड़े होते हैं। लंबी शृंखलाओं के "एंडपॉइंट्स" को रखकर और उन्हें आपस में जोड़कर, उन्होंने उस समस्या को हल किया जिससे AI अपनी ही मेमोरी में खो जाता है। इसका अर्थ है कि भविष्य के AI एजेंट बहुत कम मेमोरी वाले उपकरणों पर जटिल कार्य सीख सकते हैं, जिससे शक्तिशाली आर्टिफिशियल इंटेलिजेंस अधिक सुलभ और कुशल बन सकती है।
अपने क्षेत्र के पेपरों की भीड़ में उलझे हुए हैं?
आपके रिसर्च कीवर्ड से मेल खाने वाले सबसे नए और अलग सोच वाले पेपरों का रोज़ाना Digest पाएँ—तकनीकी सारांश के साथ, आपकी भाषा में।