MoM: Mixtures of Scenario-Aware Document Memories for Retrieval-Augmented Generation Systems
Zhao, Ji, Niu et al.
The traditional RAG paradigm, which typically engages in the comprehension of relevant text chunks in response to received queries, inherently restricts both the depth of knowledge internalization and reasoning capabilities. To address this limitation, our research transforms the text processing in RAG from passive chunking to proactive understanding, defining this process as document memory extraction with the objective of simulating human cognitive processes during reading. Building upon this, we propose the Mixtures of scenario-aware document Memories (MoM) framework, engineered to efficiently handle documents from multiple domains and train small language models (SLMs) to acquire the ability to proactively explore and construct document memories. The MoM initially instructs large language models (LLMs) to simulate domain experts in generating document logical outlines, thereby directing structured chunking and core content extraction. It employs a multi-path sampling and multi-perspective evaluation mechanism, specifically designing comprehensive metrics that represent chunk clarity and extraction completeness to select the optimal document memories. Additionally, to infuse deeper human-like reading abilities during the training of SLMs, we incorporate a reverse reasoning strategy, which deduces refined expert thinking paths from high-quality outcomes. Finally, leveraging diverse forms of content generated by MoM, we develop a three-layer document memory retrieval mechanism, which is grounded in our theoretical proof from the perspective of probabilistic modeling. Extensive experimental results across three distinct domains demonstrate that the MoM framework not only resolves text chunking challenges in existing RAG systems, providing LLMs with semantically complete document memories, but also paves the way for SLMs to achieve human-centric intelligent text processing.
academic
MoM: पुनर्प्राप्ति-संवर्धित पीढ़ी प्रणालियों के लिए परिदृश्य-जागरूक दस्तावेज़ स्मृति का मिश्रण
परंपरागत पुनर्प्राप्ति-संवर्धित पीढ़ी (RAG) प्रतिमान आमतौर पर प्रासंगिक पाठ खंडों को समझकर प्रश्नों का उत्तर देते हैं, जो ज्ञान आंतरीकरण की गहराई और तर्क क्षमता को सीमित करता है। इस सीमा को दूर करने के लिए, यह अनुसंधान RAG में पाठ प्रसंस्करण को निष्क्रिय खंडन से सक्रिय समझ में परिवर्तित करता है, जिसे दस्तावेज़ स्मृति निष्कर्षण प्रक्रिया के रूप में परिभाषित किया गया है, जो मानव पढ़ने के दौरान संज्ञानात्मक प्रक्रिया का अनुकरण करता है। इसके आधार पर, लेखकों ने परिदृश्य-जागरूक दस्तावेज़ स्मृति मिश्रण (MoM) ढांचा प्रस्तावित किया है, जिसका उद्देश्य बहु-डोमेन दस्तावेज़ों को कुशलतापूर्वक संभालना और छोटे भाषा मॉडल (SLM) को सक्रिय रूप से दस्तावेज़ स्मृति की खोज और निर्माण करने की क्षमता प्रदान करना है।
परंपरागत RAG प्रणालियों में एक मौलिक संज्ञानात्मक अंतराल है: दस्तावेज़ प्रसंस्करण को यांत्रिक पूर्व-प्रसंस्करण चरण में सरल बनाना, "पहले विभाजित करें फिर समझें" की निष्क्रिय विधि अपनाना, जो मानव विशेषज्ञों की संज्ञानात्मक प्रक्रिया के विपरीत है।
शब्दार्थ पूर्णता की कमी: परंपरागत खंडन विधियां (निश्चित लंबाई, पुनरावर्ती खंडन आदि) दस्तावेज़ की गहन शब्दार्थ सुसंगतता और तार्किक संरचना को नजरअंदाज करती हैं
ज्ञान विखंडन: मौजूदा विधियां नीचे से ऊपर की ओर निर्माण तर्क का पालन करती हैं, दस्तावेज़ की समग्र वास्तुकला की व्यापक समझ की कमी है
तर्क क्षमता सीमित: निष्क्रिय खंडन मॉडल की ज्ञान आंतरीकरण गहराई और तर्क क्षमता को सीमित करता है
जटिल दस्तावेज़ों को पढ़ने की मानव विशेषज्ञों की संज्ञानात्मक प्रक्रिया का अनुकरण करना: पहले व्यापक तार्किक संरचना को समझना, मुख्य तर्कों की पहचान करना, अंततः संरचित, स्तरीय स्मृति बनाना।
सक्रिय स्मृति निष्कर्षण प्रतिमान: निष्क्रिय पाठ खंडन के स्थान पर सक्रिय स्मृति निष्कर्षण प्रस्तावित करना, वैश्विक समझ के माध्यम से संरचित दस्तावेज़ स्मृति का निर्माण करना
त्रि-स्तरीय दस्तावेज़ स्मृति पुनर्प्राप्ति तंत्र: संभाव्य मॉडलिंग पर आधारित सैद्धांतिक प्रमाण के साथ पुनर्प्राप्ति एल्गोरिदम विकसित करना, जो परंपरागत संलयन रणनीतियों की तुलना में अधिक प्रभावी रूप से सूचना हानि को कम करता है
विपरीत तर्क रणनीति: CoM (Chain of Memory extraction) निर्माण विधि डिजाइन करना, जो SLM को स्वतंत्र रूप से जटिल स्मृति निष्कर्षण कार्य निष्पादित करने में सक्षम बनाता है
बहु-डोमेन सत्यापन: तीन विभिन्न डोमेन डेटासेट पर MoM ढांचे की प्रभावशीलता को सत्यापित करना, 40K प्रशिक्षण नमूने बनाना और कई MemReader मॉडल प्रशिक्षित करना
विशेषज्ञ अनुकरण: विशिष्ट डोमेन विशेषज्ञों का अनुकरण करने के लिए बड़े भाषा मॉडल MG का उपयोग करना, परिदृश्य-जागरूक संकेत के माध्यम से दस्तावेज़ तार्किक रूपरेखा O उत्पन्न करना।
बहु-पथ नमूनाकरण: MG के डिकोडिंग पैरामीटर को समायोजित करना, एक ही दस्तावेज़ D के लिए N उम्मीदवार दस्तावेज़ स्मृति सेट उत्पन्न करना।
बहु-आयामी मूल्यांकन: दो मुख्य मात्रात्मक मूल्यांकन संकेतक डिजाइन करना:
निर्देशक मॉडल MG का उपयोग करना, मूल दस्तावेज़ D और इष्टतम दस्तावेज़ स्मृति Mdoc को इनपुट करना, तर्क पथ P उत्पन्न करना, उच्च-गुणवत्ता वाले CoM डेटा का गठन करना।
O, C, A के अनुरूप त्रि-स्तरीय पुनर्प्राप्ति तंत्र का निर्माण करना, स्वतंत्र रूप से पुनर्प्राप्ति के बाद परिणाम संलयन करना, सैद्धांतिक रूप से सूचना हानि को अधिक प्रभावी ढंग से रोकने में सक्षम है।
परमाणु खंड स्पष्टता और ROUGE-L के बीच सहसंबंध गुणांक तीन मूल्यांकन मॉडल के तहत क्रमशः 0.7044, 0.7585 और 0.7248 तक पहुंचते हैं, मजबूत सकारात्मक सहसंबंध दिखाते हैं।
सूचना समर्थन स्कोर डिजाइन करना जो पुनर्प्राप्त सामग्री के उत्तर के लिए समर्थन का मूल्यांकन करता है:
Ssupport(A|C) = -1/m * Σ log P(ai|a1,...,ai-1,C)
MemReader-3B सभी मूल्यांकन मॉडल पर सर्वोत्तम प्रदर्शन प्राप्त करता है, यह साबित करता है कि निष्कर्षित स्मृति डाउनस्ट्रीम कार्यों के लिए अधिक सूचना प्रदान कर सकती है।
पेपर ने 32 संबंधित संदर्भों का हवाला दिया है, जिसमें RAG मूल सिद्धांत, पाठ खंडन विधियां, स्मृति प्रणाली डिजाइन आदि मुख्य क्षेत्र शामिल हैं, जो अनुसंधान के लिए एक ठोस सैद्धांतिक आधार प्रदान करते हैं।
समग्र मूल्यांकन: यह RAG क्षेत्र में महत्वपूर्ण नवीन महत्व वाला एक पेपर है, जो संज्ञानात्मक विज्ञान के दृष्टिकोण को पेश करके दस्तावेज़ प्रसंस्करण प्रतिमान को पुनः परिभाषित करता है, न केवल सैद्धांतिक रूप से सफलता प्राप्त करता है, बल्कि व्यावहारिक रूप से भी उल्लेखनीय परिणाम प्राप्त करता है। कुछ सीमाओं के बावजूद, इसके अग्रणी विचार और ठोस प्रयोगात्मक सत्यापन इसे इस क्षेत्र का एक महत्वपूर्ण योगदान बनाते हैं।