तकनीकी सारांश: VisCo – विजुअल टोकन संपीड़न के लिए इंट्रिंसिक एनकोडर के रूप में लार्ज लैंग्वेज मॉडल्स का लाभ उठाना
1. समस्या विवरण
विज़न-लैंग्वेज मॉडल्स (VLMs) आमतौर पर उच्च-रिज़ॉल्यूशन वाली छवियों को एक बड़ी संख्या में विजुअल टोकन में एनकोड करके प्रोसेस करते हैं। इसके परिणामस्वरूप सेल्फ-अटेंशन मैकेनिज्म की क्वाड्रेटिक कॉम्प्लेक्सिटी और की-वैल्यू (KV) कैश के स्टोरेज आवश्यकताओं के कारण पर्याप्त इंफरेंस लेटेंसी और मेमोरी ओवरहेड होता है। यह बाधा VLMs को संसाधन-सीमित और रियल-टाइम परिदृश्यों में तैनाती के लिए गंभीर रूप से सीमित करती है।
विजुअल टोकन संपीड़न के मौजूदा समाधान दो श्रेणियों में आते हैं, दोनों की महत्वपूर्ण सीमाएँ हैं:
- ट्रेनिंग-फ्री विधियाँ: ये हीयुरिस्टिक मेट्रिक्स (जैसे अटेंशन स्कोर या फीचर सिमिलैरिटी) पर निर्भर करती हैं ताकि टोकन को हटाना (prune) या मर्ज करना जा सके। हालांकि ये कुशल हैं, लेकिन उच्च संपीड़न अनुपात (high compression ratios) के तहत इनका प्रदर्शन काफी गिर जाता है। ये "अटेंशन शिफ्ट" के प्रति संवेदनशील हैं, जिससे मतिभ्रम (hallucinations) होता है, और अक्सर टेक्स्ट-गाइडेड ग्लोबल सेमantics या सूक्ष्म विवरणों को संरक्षित करने में विफल रहती हैं।
- ट्रेनिंग-आधारित विधियाँ: ये बाहरी संपीड़न मॉड्यूल पेश करती हैं या व्यापक रिट्रेनिंग (री-अलाइनमेंट और इंस्ट्रक्शन ट्यूनिंग सहित) की आवश्यकता होती है। हालांकि ये अधिक मजबूत हैं, लेकिन इनमें भारी रिट्रेनिंग लागत आती है, ये प्रीट्रेनड VLM के प्रायर्स (priors) से समझौता करती हैं, और अक्सर इनमें प्लग-एंड-प्ले लचीलेपन की कमी होती है।
मुख्य चुनौती यह है कि बिना किसी बड़े रिट्रेनिंग या बाहरी मॉड्यूल के, आक्रामक संपीड़न (जैसे, टोकन को एक इकाई तक कम करना) के तहत भी सिमेंटिक फिडेलिटी को बनाए रखते हुए प्रभावी विजुअल टोकन संपीड़न कैसे प्राप्त किया जाए।
2. कार्यप्रणाली: VisCo फ्रेमवर्क
लेखक VisCo का प्रस्ताव करते हैं, जो एक ट्रेनिंग-एफिशिएंट, सेल्फ-कंप्रेशन फ्रेमवर्क है जो प्रीट्रेनड VLM का उपयोग स्वयं एक इंट्रिंसिक कंप्रेसर के रूप में करता है। VisCo को एक साझा पैरामीटर्स वाले एसिमेट्रिक VLM ऑटोएनकोडर के रूप में डिज़ाइन किया गया है, जो दो चरणों में कार्य करता है:
A. एसिमेट्रिक आर्किटेक्चर
- एनकोडिंग चरण: फ्रेमवर्क प्रीट्रेनड VLM बैकबोन को एक लाइटवेट ट्रेन करने योग्य मॉड्यूल के साथ संवर्धित करता है। विशेष रूप से, यह लर्न करने योग्य मेमोरी टोकन्स (Xm) का एक छोटा सेट पेश करता है और Q/V प्रोजेक्शन पर LoRA एडेप्टर्स का उपयोग करता है। एनकोडर विजुअल टोकन (Xv) और मेमोरी टोकन्स (Xm) के संयुक्त अनुक्रम (joint sequence) को प्रोसेस करता है।
- डिकोडिंग चरण: LoRA एडेप्टर्स को हटा दिया जाता है, और मूल फ्रोजन प्रीट्रेनड VLM बैकबोन का उपयोग सीधे डिकोडर के रूप में किया जाता है। यह डिज़ाइन सुनिश्चित करता है कि VisCo को मल्टीमॉडल जनरेशन प्रक्रिया को शून्य से सीखने के बजाय मौजूदा क्षमताओं के केवल हल्के अनुकूलन (lightweight adaptation) की आवश्यकता है।
B. सीक्वेंस कंस्ट्रक्शन और कॉज़ल मास्किंग
VisCo डिकोडर-ओनली VLMs के मानक कॉज़ल मास्किंग मैकेनिज्म का कड़ाई से पालन करता है। इनपुट सीक्वेंस में विजुअल टोकन के बाद मेमोरी टोकन जोड़े जाते हैं। यह मॉडल के इंट्रिंसिक अटेंशन प्रायर का उपयोग करके मेमोरी टोकन को सभी पूर्ववर्ती विजुअल टोकन पर स्वाभाविक रूप से ध्यान केंद्रित करने की अनुमति देता है, जिससे अनुकूलित इंटरैक्शन नियमों को पेश किए बिना समृद्ध विजुअल जानकारी एकत्रित की जा सकती है।
C. पदानुक्रमित सूचना एकत्रीकरण और पासिंग (Hierarchical Information Aggregation and Passing)
यह पहचानते हुए कि ट्रांसफार्मर लेयर्स कार्यात्मक विशेषज्ञता (निचली लेयर्स स्थानीय पैटर्न को एनकोड करती हैं, उच्च लेयर्स अमूर्त सेमantics को एनकोड करती हैं) प्रदर्शित करती हैं, VisCo एक पदानुक्रमित सूचना-पासिंग मैकेनिज्म का उपयोग करता है:
- मेमोरी बैंक निर्माण: एनकोडिंग के दौरान, डिकोडर को फाइनल हिडन स्टेट्स पास करने के बजाय, सिस्टम हर लेयर l पर केवल मेमोरी टोकन के अनुरूप की (Key - K) और वैल्यू (V) मैट्रिसेस को सुरक्षित रखता है। इन्हें एक लेयर-वाइज मेमोरी बैंक (Kmem,Vmem) में संग्रहीत किया जाता है।
- पदानुक्रमित डिकोडिंग: डिकोडिंग के दौरान, इन लेयर-वाइज मेमोरी KV कैश को सीधे डिकोडर के KV कैश में भरा जाता है। यह डिकोडर को प्रोजेक्शन लेयर्स की आवश्यकता के बिना मल्टी-ग्रैनुलैरिटी सिमेंटिक जानकारी (लो-लेवल टेक्सचर से लेकर हाई-लेवल डिस्कोर्स तक) तक पहुँचने की अनुमति देता है, जो पैरामीटर शेयरिंग का लाभ उठाता है।
D. ट्रेनिंग ऑब्जेक्टिव
VisCo को ग्राउंड-ट्रुथ आंसर की कंडीशनल लाइकलीहुड को मैक्सिमाइज करने के लिए टीचर फोर्सिंग का उपयोग करके एंड-टू-एंड फाइन-ट्यून किया जाता है। कोई अतिरिक्त प्री-ट्रेनिंग ऑब्जेक्टिव पेश नहीं किया गया है।
3. प्रमुख योगदान
- एसिमेट्रिक इंट्रिंसिक सेल्फ-कंप्रेशन फ्रेमवर्क: VisCo प्रीट्रेनड VLM का उपयोग स्वयं कंप्रेसर के रूप में करता है, बैकबोन को बरकरार रखता है और केवल लाइटवेट ट्रेनिंग (LoRA + लर्न करने योग्य मेमोरी टोकन्स) की आवश्यकता होती है।
- पदानुक्रमित सूचना-पासिंग मैकेनिज्म: यह विधि लेयर-वाइज KV कैश के माध्यम से एनकोडर लेयर्स से डिकोडर को मल्टी-ग्रैनुलैरिटी सेमantics ट्रांसफर करने के लिए एक तंत्र पेश करती है, जो संपीड़न के तहत विजुअल जानकारी के स्ट्रक्चरल पदानुक्रम को बनाए रखती है।
- व्यापक मूल्यांकन: 3 VLM बैकबोन (LLaVA-1.5-7B, Qwen2-VL-2B, Qwen2-VL-7B) और 6 बेंचमार्क पर विस्तृत प्रयोग निरंतर श्रेष्ठता प्रदर्शित करते हैं, विशेष रूप से आक्रामक संपीड़न के तहत।
4. प्रयोगात्मक परिणाम
पेपर का मूल्यांकन GQA, MMB, MMB-CN, MME, POPE, और MMVet जैसे बेंचमार्क पर किया गया है।
आक्रामक संपीड़न के तहत प्रदर्शन:
- LLaVA-1.5-7B (576 टोकन → 32 टोकन) पर, VisCo मूल प्रदर्शन का 91.8% बनाए रखता है, जो औसत स्कोर में सर्वश्रेष्ठ बेसलाइन (VisPruner) से 4.0 अंक आगे है।
- एक्सट्रीम सिंगल-टोकन सेटिंग (1 टोकन) में, VisCo मूल प्रदर्शन का 85.3% बनाए रखता है, जो ट्रेनिंग-फ्री मेथड्स (जैसे SparseVLM ~72.6%) से काफी बेहतर है और विशिष्ट मेट्रिक्स पर MatryoshkaQuery और VoCo-LLaMA जैसे मजबूत ट्रेनिंग-बेस्ड रेफरेंस को भी पीछे छोड़ देता है।
- Qwen2-VL-2B (जिसमें पहले से ही बिल्ट-इन कंप्रेशन है) पर, VisCo 25% टोकन रिटेंशन पर 95.2% प्रदर्शन और 12.5% रिटेंशन पर 91.4% प्रदर्शन बनाए रखता है, जो VisionZip जैसे फाइन-ट्यून्ड बेसलाइन्स से बेहतर है।
गुणात्मक विश्लेषण (Qualitative Analysis):
- VisCo बहुत कम टोकन के साथ भी समग्र दृश्य संदर्भ (overall scene context) और प्रमुख क्षेत्रों को सफलतापूर्वक कैप्चर करता है, जबकि अटेंशन-आधारित प्रूनिंग मेथड्स ग्लोबल कॉन्टेक्स्ट और डिटेल्स खो देते हैं।
- सिंगल-टोकन सेटिंग में, VisCo ऑब्जेक्ट-लेवल विवरणों और सुसंगत ग्लोबल सेमantics वाली विस्तृत कैप्शन जेनरेट कर सकता है।
दक्षता (Efficiency):
- VisCo KV कैश स्टोरेज को काफी कम कर देता है।
- हालांकि प्रारंभिक एनकोडिंग चरण कुछ ओवरहेड पेश करता है, VisCo तेज़ डिकोडिंग गति प्राप्त करता है। मल्टी-टर्न डायलॉग परिदृश्यों में, कैश किए गए रिप्रेजेंटेशन्स का पुन: उपयोग करने से VisCo केवल तीन डायलॉग टर्न के बाद ट्रेनिंग-फ्री मेथड्स (जैसे FastV) की लेटेंसी को पार कर जाता है।
पूरक प्रतिनिधित्व (Complementary Representations):
- एक एब्लेशन स्टडी (VisCo+) दिखाती है कि सीखे गए मेमोरी टोकन्स को मूल विजुअल टोकन्स के साथ जोड़ने से वास्तव में कई बेंचमार्क पर बेस मॉडल का प्रदर्शन सुधार सकता है। यह सुझाव देता है कि मेमोरी टोकन्स केवल एक संपीड़ित उपसमुच्चय (compressed subset) के बजाय पूरक जानकारी कैप्चर करते हैं।
5. महत्व और दावे
पेपर दावा करता है कि प्रीट्रेनड VLMs के इंट्रिंसिक प्रायर्स का लाभ उठाना केवल लाइटवेट ट्रेनिंग के साथ मजबूत विजुअल टोकन संपीड़न को सक्षम बनाता है।
- ट्रेड-ऑफ को पार करना: VisCo ट्रेनिंग-फ्री मेथड्स के गंभीर प्रदर्शन ह्रास और बाहरी-मॉड्यूल-आधारित दृष्टिकोणों की भारी रिट्रेनिंग लागत से बचता है।
- नेटिव प्रोसेसिंग पैटर्न: VLM के नेटिव प्रोसेसिंग पैटर्न (कॉज़ल मास्किंग और पैरामीटर शेयरिंग का उपयोग करके) का पालन करके, VisCo हेवीवेट मेथड्स के प्रतिस्पर्धी प्रदर्शन हासिल करता है और लाइटवेट एडैप्टेशन का लचीलापन बनाए रखता है।
- संपीड़न से परे: लेखक सुझाव देते हैं कि सीखे गए मेमोरी टोकन्स पूरक प्रतिनिधित्व के रूप में कार्य करते हैं जो विजुअल फीचर्स को समृद्ध करते हैं, जो संभावित रूप से केवल संपीड़न दक्षता से परे मॉडल के प्रदर्शन को बेहतर बनाते हैं।
कार्य इस निष्कर्ष पर निकलता है कि प्रभावी विजुअल टोकन संपीड़न, प्रीट्रेनड VLMs में पहले से मौजूद मजबूत सूचना एनकोडिंग क्षमताओं पर निर्भर करता है, जिसे VisCo जैसे कुशल सेल्फ-कंप्रेशन फ्रेमवर्क के माध्यम से अनलॉक किया जा सकता है। भविष्य के कार्य के रूप में एडेप्टिव टोकन एलोकेशन और वीडियो कंप्रेशन तक विस्तार का प्रस्ताव दिया गया है।