← नवीनतम पेपर
🤖 machine learning

Agent Memory Distillation: Empowering Small LLM Agents with Hierarchical Teacher Memory

यह शोध पत्र एजेंट मेमोरी डिस्टिलेशन (AMD) को प्रस्तुत करता है, जो एक प्रशिक्षण-मुक्त ढांचा है जो एक बड़े शिक्षक एजेंट से पदानुक्रमित ज्ञान—जिसमें वर्कफ़्लो रणनीतियाँ, उप-कार्य उदाहरण और फ़ंक्शन कन्वेंशन शामिल हैं—को स्थानांतरित करके छोटे भाषा मॉडल एजेंटों को उन्नत करता है, जिसके परिणामस्वरूप कई टूल-उपयोग बेंचमार्क में महत्वपूर्ण प्रदर्शन सुधार होता है।

मूल लेखक: Taeil Kim, Kangsan Kim, Sung Ju Hwang

प्रकाशित 2026-08-10
📖 9 मिनट में पढ़ें🧠 गहराई से पढ़ें

मूल लेखक: Taeil Kim, Kangsan Kim, Sung Ju Hwang

मूल पेपर CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/) के तहत लाइसेंस किया गया है। नीचे दिए गए पेपर की यह व्याख्या AI से तैयार की गई है। इसे लेखकों ने न तो लिखा है, न इसका समर्थन किया है। तकनीकी सटीकता के लिए मूल पेपर देखें। पूरा डिस्क्लेमर पढ़ें

कल्पना कीजिए कि आप एक नए, उत्साही प्रशिक्षु (apprentice) को एक जटिल मशीन को ठीक करना सिखाने की कोशिश कर रहे हैं। आपके पास एक मास्टर मैकेनिक है जो सेकंडों में किसी भी समस्या को हल कर सकता है, लेकिन आपका प्रशिक्षु छोटा, अनुभवहीन और गलतियाँ करने के प्रति प्रवृत्त है। आर्टिफिशियल इंटेलिजेंस की दुनिया में, ये "प्रशिक्षु" छोटे भाषा मॉडल (small language models) हैं—स्मार्ट, लेकिन सीमित। वे कार्यों को हल करने के लिए डिजिटल उपकरणों, जैसे API या कोड का उपयोग करने की कोशिश करते हैं, लेकिन वे अक्सर अटक जाते हैं, चरणों को भूल जाते हैं, या गलत उपकरणों को कॉल कर देते हैं। उनकी मदद करने के लिए, शोधकर्ताओं ने उन्हें एक "मेमोरी" देने की कोशिश की है, यानी अतीत की सफलताओं और विफलताओं की एक नोटबुक, जिसे वे देख सकें। हालाँकि, इसमें एक पेच है: क्योंकि प्रशिक्षु बहुत नया है, उनकी नोटबुक ज्यादातर "मैं यहाँ विफल रहा" और "मैं वहाँ खो गया" वाले पन्नों से भरी है। उनके पास सीखने के लिए पर्याप्त सफल कहानियाँ नहीं हैं। यह वह पहेली है जिसे वैज्ञानिक हल करने की कोशिश कर रहे हैं: आप एक छोटे, संघर्षरत छात्र को एक मास्टर के अनुभव का लाभ कैसे दे सकते हैं, बिना उनकी मेज पर एक विशाल, भ्रमित करने वाली पाठ्यपुस्तक डाल दिए?

यहीं पर पेपर "एजेंट मेमोरी डिस्टिलेशन" (Agent Memory Distillation) आता है। शोधकर्ता, जो KAIST के ताईल किम और सहयोगियों के नेतृत्व में हैं, एक शक्तिशाली "शिक्षक" AI से एक छोटे "छात्र" AI को ज्ञान स्थानांतरित करने का एक चतुर तरीका प्रस्तावित करते हैं। वे अपने इस तरीके को एजेंट मेमोरी डिस्टिलेशन (AMD) कहते हैं। शिक्षक की विशाल, अव्यवस्थित यादों का ढेर छात्र को देने के बजाय, AMD एक कुशल संपादक की तरह कार्य करता है। यह शिक्षक की सफल यात्राओं को लेता है और उन्हें तीन विशिष्ट, आसानी से समझ में आने वाले प्रकार की यादों में तोड़ देता है: एक उच्च-स्तरीय वर्कफ्लो (Workflow) (बड़ी योजना), एक सबटास्क (Subtask) गाइड (चीजों को करने के विशिष्ट चरण), और एक फंक्शन (Function) संदर्भ शीट (उपकरणों का सही उपयोग कैसे करें)। पेपर सुझाव देता है कि अपने ज्ञान को इस तरह व्यवस्थित करके, छोटा छात्र कच्चे डेटा को पढ़ने की तुलना में इसे बहुत बेहतर तरीके से समझ और उपयोग कर सकता है।

समस्या: एक बड़ा अंतर रखने वाला एक छोटा छात्र

एक छोटे AI एजेंट को एक टेक कंपनी में एक नए इंटर्न की तरह समझें। जब वे कोई काम करने की कोशिश करते हैं, जैसे "इस ऐप में लॉग इन करें और एक संदेश भेजें," तो वे अक्सर विफल हो जाते हैं। यदि आप उनसे उनके द्वारा किए गए कार्यों का एक डायरी रखने के लिए कहते हैं, तो उनकी डायरी ज्यादातर "मैंने लॉग इन करने की कोशिश की, लेकिन मुझे एक एरर मिला" या "मैंने गलत बटन दबा दिया" से भरी होती है। क्योंकि वे अक्सर विफल होते हैं, उनके पास "कैसे सफल हों" की कहानियों का एक अच्छा पुस्तकालय नहीं है जिससे वे सीख सकें।

शोधकर्ताओं ने एक सरल समाधान आजमाया: "चलो इंटर्न को मास्टर मैकेनिक की डायरी दे देते हैं!" उन्होंने एक विशाल, सुपर-स्मार्ट AI (शिक्षक) के सफल नोट्स लिए और उन्हें छोटे AI (छात्र) को दे दिया। लेकिन यह अच्छी तरह से काम नहीं किया। यह एक बच्चे को उन्नत भौतिकी (advanced physics) में लिखी गई पाठ्यपुस्तक देने जैसा था; बच्चा जटिल भाषा को समझ ही नहीं पाया या यह समझ नहीं पाया कि इसे कैसे लागू किया जाए। "क्षमता का अंतर" (capability gap) बहुत अधिक था। छात्र बहुत छोटा था कि वह शिक्षक के बड़े, अमूर्त विचारों को समझ सके।

समाधान: तीन-भागों वाला मेमोरी लंचबॉक्स

लेखकों ने महसूस किया कि इस अंतर को पाटने के लिए, वे केवल शिक्षक की स्मृति को छात्र पर थोप नहीं सकते। उन्हें इसे "डिस्टिल" (distill) करने की आवश्यकता थी—जैसे एक जटिल सूप को एक सांद्रित, आसानी से निगले जाने वाले शोरबे (broth) में बदलना। उन्होंने एक प्रणाली बनाई जो शिक्षक की सफलता की कहानियों को तीन अलग-अलग परतों में व्यवस्थित करती है, जिनमें से प्रत्येक का एक अलग उद्देश्य है:

  1. वर्कफ्लो मेमोरी (नक्शा - The Map): यह बड़ी तस्वीर है। छात्र को कार्य शुरू करने से पहले ही, सिस्टम उन्हें एक उच्च-स्तरीय नक्शा देता है। यह कहता है, "पहले, आपको लॉग इन करने की आवश्यकता है। फिर, डेटा खोजें। अंत में, संदेश भेजें।" यह कोड नहीं देता; यह रणनीति देता है। यह प्रशिक्षु को यह बताने जैसा है कि, "इंजन को ठीक करने के लिए, पहले आप तेल की जाँच करते हैं, फिर स्पार्क प्लग की, फिर बैटरी की।" यह छात्र को यह जानने में मदद करता है कि इससे पहले कि वे खो जाएँ, संचालन का क्रम क्या होना चाहिए।
  2. सबटास्क मेमोरी (रेसिपी - The Recipe): एक बार जब छात्र को योजना पता चल जाती है, तो उन्हें यह जानने की आवश्यकता होती है कि प्रत्येक चरण को कैसे किया जाए। यह मेमोरी विशिष्ट कार्यों को शिक्षक ने कैसे संभाला, इसके ठोस उदाहरण प्रदान करती है। यदि कार्य "वस्तुओं की सूची प्राप्त करना" है, तो सबटास्क मेमोरी छात्र को ठीक से दिखाती है कि शिक्षक ने उस सूची को प्राप्त करने के लिए कोड कैसे लिखा था, जिसमें पेजिनेशन (पेज 1, फिर पेज 2 आदि प्राप्त करना) को कैसे संभालना है, यह भी शामिल है। यह प्रशिक्षु को एक रेसिपी कार्ड देने जैसा है जो कहता है, "2 कप मैदा और 1 कप चीनी मिलाएं," बजाय इसके कि केवल "केक बेक करें" कहा जाए।
  3. फंक्शन मेमोरी (टूल मैनुअल - The Tool Manual): यह सुरक्षा जाल है। जब छात्र किसी उपकरण का उपयोग करने का प्रयास करता है (जैसे किसी विशिष्ट फंक्शन को कॉल करना) और उसे एरर मिलता है, तो सिस्टम तुरंत एक "फंक्शन मेमोरी" प्रविष्टि निकाल लेता है। यह प्रविष्टि उस विशिष्ट टूल को उपयोग करने का शिक्षक का सही तरीका दिखाती है, जिसमें आवश्यक सटीक तर्क (arguments) भी शामिल हैं। यह एक मैकेनिक के लिए मैनुअल देखने जैसा है जैसे ही उसे एहसास होता है कि वह रेंच (wrench) को उल्टा पकड़ रहा है।

व्यवहार में यह कैसे काम करता है

यह प्रक्रिया एक स्मार्ट ट्यूटरिंग सत्र की तरह है।

  • प्रोएक्टिव इंजेक्शन (Proactive Injection): कार्य की शुरुआत में ही, छात्र को वर्कफ्ल और सबटास्क मेमोरी उनके "मस्तिष्क" (उनके सिस्टम प्रॉम्प्ट) में इंजेक्ट कर दी जाती है। वे एक भी कदम उठाने से पहले नक्शा और रेसिपी देखते हैं।
  • रिएक्टिव इंजेक्शन (Reactive Injection): यदि छात्र कोई गलती करता है और कोई एरर आता है, तो सिस्टम तुरंत उस विशिष्ट एरर से संबंधित फंक्शन मेमोरी को पकड़ लेता है और उसे हिंट के रूप में छात्र को दिखाता है। यह केवल तभी होता है जब आवश्यक हो, ताकि जब चीजें ठीक चल रही हों तो छात्र सूचना से अभिभूत न हो।

परिणाम: बड़े उछाल प्राप्त करते छोटे मॉडल

शोधकर्ताओं ने चार अलग-अलग छोटे छात्र मॉडलों (4 बिलियन से 8 बिलियन पैरामीटर्स तक) और एक शक्तिशाली GPT-5-mini (शिक्षक) का उपयोग करके तीन अलग-अलग बेंचमार्क (AppWorld, BFCL V3, और ToolSandbox) पर इस पद्धति का परीक्षण किया।

परिणाम प्रभावशाली थे। छोटे छात्र, जब इस पदानुक्रमित (hierarchical) मेमोरी से लैस थे, तो उनमें भारी सुधार देखा गया:

  • AppWorld पर, उनकी सटीकता औसतन 27.2 प्रतिशत अंक बढ़ गई।
  • BFCL V3 पर, उनमें 11.2 प्रतिशत अंक का सुधार हुआ।
  • ToolSandbox पर, उन्हें 3.4 प्रतिशत अंक का लाभ हुआ।

कुछ मामलों में, छोटे छात्र केवल बेहतर ही नहीं हुए; वे वास्तव में विशाल शिक्षक मॉडल के बराबर या उससे भी बेहतर प्रदर्शन करने लगे! उदाहरण के लिए, AppWorld बेंचमार्क पर, 4-बिलियन-पैरामीटर वाला छात्र जो AMD का उपयोग कर रहा था, उसने 49.40% स्कोर किया, जो शिक्षक के 50.00% के लगभग समान था।

डेटा हमें क्या बताता है

पेपर यह समझने के लिए गहराई से जांच करता है कि यह इतना अच्छा क्यों काम कर रहा था।

  • "सबटास्क" राजा है: उन्होंने पाया कि सबटास्क मेमोरी (ठोस उदाहरण) सबसे महत्वपूर्ण हिस्सा था। इसने सबसे बड़ा उछाल प्रदान किया। यह सुझाव देता है कि छोटे मॉडलों को वास्तव में यह देखने की आवश्यकता होती है कि चीजों को कैसे किया जाए, न कि केवल यह कि क्या किया जाए।
  • आकार मायने रखता है (लेकिन बहुत अधिक नहीं): 4-बिलियन-पैरामीटर वाले मॉडल को सबसे अधिक लाभ हुआ। वे शिक्षक की मेमोरी को समझने के लिए पर्याप्त स्मार्ट थे लेकिन अभी भी सुधार के लिए पर्याप्त जगह थी। बड़े 8-बिलियन मॉडल पहले से ही अपने आप में काफी अच्छा कर रहे थे, इसलिए उछाल कम था, हालांकि सकारात्मक था।
  • शिक्षक संगत (Compatible) होना चाहिए: दिलचस्प बात यह है कि सबसे शक्तिशाली संभव शिक्षक होने का मतलब हमेशा छात्र के लिए सबसे अच्छे परिणाम नहीं होता। कभी-कभी, एक शिक्षक जो थोड़ा कम शक्तिशाली लेकिन छात्र की शैली के साथ अधिक "संगत" था, बेहतर काम करता था। यह केवल एक जीनियस शिक्षक होने के बारे में नहीं है; यह एक ऐसे शिक्षक के बारे में है जिसकी शैली छात्र वास्तव में सीख सके।
  • कम ही अधिक है: जब उन्होंने छात्र को अधिक मेमोरी देने की कोशिश की (1 के बजाय 5 रिक्रिव करना), तो प्रदर्शन गिर गया। छोटे मॉडल बहुत अधिक जानकारी से भ्रमित हो गए। उन्हें बस सही, उच्च-गुणवत्ता वाले हिंट की आवश्यकता थी, न कि डेटा के सैलाब की।

इसका क्या अर्थ है (और क्या नहीं है)

पेपर सुझाव देता है कि हमें इन छोटे मॉडलों को शून्य से प्रशिक्षित करने या उन्हें स्मार्ट बनाने के लिए महंगी कंप्यूटिंग शक्ति का उपयोग करने की आवश्यकता नहीं है। इसके बजाय, हम बड़े मॉडलों के अनुभवों को एक संरचित मेमोरी सिस्टम में "डिस्टिल" कर सकते हैं जिसे छोटे मॉडल वास्तव में उपयोग कर सकें। यह एक "ट्रेनिंग-फ्री" दृष्टिकोण है, जिसका अर्थ है कि छात्र मॉडल के आंतरिक वेट्स (weights) नहीं बदलते; यह केवल बाहरी मेमोरी का बेहतर उपयोग करना सीखता है।

हालाँकि, लेखक इसकी सीमाओं पर सावधानी बरतते हैं। इस पद्धति का परीक्षण कोड और संरचित टूल उपयोग (जैसे API कॉल करना) से जुड़े कार्यों पर किया गया था। वे अभी तक निश्चित नहीं हैं कि क्या यह उन कार्यों के लिए काम करेगा जिनमें चित्रों को देखने या मौलिक रूप से रचनात्मक कहानियाँ लिखने की आवश्यकता होती है। साथ ही, मेमोरी "फ्रोजन" (frozen) है—यह शिक्षक की पिछली सफलताओं से बनी है और छात्र जैसे नई चीजें सीखता है वैसे वास्तविक समय में अपडेट नहीं होती है।

अंत में, एजेंट मेमोरी डिस्टिलेशन एक समान स्तर लाने का एक नया और आशाजनक तरीका प्रदान करता है। यह सुझाव देता है कि छोटे, सीमित AI एजेंट भी शक्तिशाली समस्या-समाधानकर्ता बन सकते हैं यदि हम उनके शिक्षक के ज्ञान को एक ऐसे प्रारूप में व्यवस्थित करने के लिए समय लें जिसे वे वास्तव में समझ सकें। यह एक याद दिलाता है कि कभी-कभी सीखने का सबसे अच्छा तरीका पूरी विश्वकोश (encyclopedia) को पढ़ना नहीं होता है, बल्कि एक स्मार्ट गाइड का आपको सही समय पर सही पन्ना थमाना होता है।

अपने क्षेत्र के पेपरों की भीड़ में उलझे हुए हैं?

आपके रिसर्च कीवर्ड से मेल खाने वाले सबसे नए और अलग सोच वाले पेपरों का रोज़ाना Digest पाएँ—तकनीकी सारांश के साथ, आपकी भाषा में।

Digest आज़माएँ →