Gradient-Skipping Relevance Propagation for Efficient Explainability of Vision Transformers
यह शोध पत्र GradSkip को प्रस्तुत करता है, जो विजन ट्रांसफॉर्मर्स के लिए एक नवीन प्रासंगिकता प्रसार (relevance propagation) विधि है जो अटेंशन हेड्स को अनुकूल रूप से भारित करके और रेसिडुअल कनेक्शनों के माध्यम से प्रासंगिकता को गतिशील रूप से वितरित करके निष्ठा (faithfulness) और स्थानीयकरण (localization) में सुधार करती है, जिससे मौजूदा दृष्टिकोणों की तुलना में 14 गुना से अधिक कम GFLOPs के साथ अत्याधुनिक प्रदर्शन प्राप्त होता है।
मूल पेपर CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/) के तहत लाइसेंस किया गया है। नीचे दिए गए पेपर की यह व्याख्या AI से तैयार की गई है। इसे लेखकों ने न तो लिखा है, न इसका समर्थन किया है। तकनीकी सटीकता के लिए मूल पेपर देखें। पूरा डिस्क्लेमर पढ़ें
कल्पना कीजिए कि आपके पास एक सुपर-स्मार्ट रोबोट दिमाग है जिसे विज़न ट्रांसफ़ॉर्मर (ViT) कहा जाता है। यह तस्वीरों को देखने और उनमें क्या है—जैसे कि एक लेडीबग या एक ट्रक को पहचानने में अद्भुत है। लेकिन समस्या यह है कि रोबोट थोड़ा "ब्लैक बॉक्स" जैसा है। आप उससे पूछ सकते हैं, "आपको क्यों लगा कि वह एक ट्रक था?" और वह बस इतना कहता है, "क्योंकि मैंने ऐसा कहा।" वह आपको यह नहीं दिखाता कि किस हिस्से ने उसे निर्णय लेने के लिए प्रेरित किया।
वैज्ञानिकों ने रोबोट के दिमाग के अंदर झाँकने की कोशिश की है कि वह कैसे सोचता है, लेकिन उनके पुराने उपकरण ऐसे थे जैसे अखरोट तोड़ने के लिए हथौड़े का उपयोग करना। उन्होंने रोबोट के दिमाग के हर हिस्से के साथ ऐसा व्यवहार किया जैसे कि वे सभी समान रूप से महत्वपूर्ण हों, और उन्होंने उन "शॉर्टकट" तारों को अनदेखा कर दिया जो सूचना को बिना बदले सीधे प्रवाहित होने देते हैं। इसने उनकी व्याख्याओं को अव्यवस्थित और अक्सर गलत बना दिया।
पेश है GradSkip, एक नया, शानदार टूल जिसे क्रिस्टोफर बुराटी और उनकी टीम ने इस गड़बड़ को ठीक करने के लिए बनाया है। GradSkip को एक स्मार्ट डिटेक्टिव (जासूस) के रूप में समझें जो केवल पूरे दिमाग को एक साथ नहीं देखता, बल्कि जानता है कि कौन से न्यूरॉन्स भारी काम कर रहे हैं और कौन से सिर्फ झपकी ले रहे हैं।
दो बड़ी गलतियाँ जो पुराने उपकरणों ने कीं
पेपर बताता है कि पिछले जासूसों के काम करने के तरीके में दो बड़ी त्रुटियाँ थीं:
- "सब बराबर हैं" वाली गलती: एक गायक मंडली (choir) की कल्पना करें जहाँ कुछ गायक सोलोइस्ट (soloists) एकदम सटीक ऊंचे सुर लगा रहे हैं, जबकि अन्य केवल पृष्ठभूमि में गुनगुना रहे हैं। पुराने उपकरणों ने हर गायक के साथ ऐसा व्यवहार किया जैसे कि वे सभी एक ही वॉल्यूम में गा रहे हों। GradSkip समझता है कि कुछ "अटेंशन हेड्स" (गायक मंडली के हिस्से) निर्णय के लिए बहुत महत्वपूर्ण हैं, जबकि अन्य केवल शोर (noise) हैं। यह सोलोइस्ट को सुनने और गुनगुनाहट को अनदेखा करने के बारे में सीखता है।
- "शॉर्टकट" वाली गलती: ट्रांसफॉर्मर्स के पास ये शानदार "स्किप कनेक्शन" होते हैं—जैसे एक गुप्त सुरंग जो एक संदेश को पूरे कमरे को छोड़कर सीधे अगले कमरे में जाने देती है। पुराने उपकरणों ने इन सुरंगों के साथ ऐसा व्यवहार किया जैसे कि वे केवल खाली गलियारे हैं जिनमें कुछ भी नहीं बदलता। लेकिन कभी-कभी, संदेश मुख्य कमरे में बदल जाता है, और कभी-कभी नहीं। GradSkip पहला है जो यह जाँचता है: "क्या इस संदेश ने सुरंग का रास्ता लिया, या इसे मुख्य कमरे में बदला गया?" यह पता लगाता है कि प्रत्येक पथ को कितना श्रेय दिया जाए।
GradSkip रहस्य को कैसे सुलझाता है
GradSkip रोबोट की सोच को समझने के लिए एक चतुर दो-चरणीय रणनीति का उपयोग करता है:
- चरण 1: "कौन कौन है" फ़िल्टर: यह रोबोट के दिमाग को देखता है और पूछता है, "कौन से अटेंशन हेड्स वास्तव में काम कर रहे हैं?" यह एक विशेष गणितीय ट्रिक (ग्रेडिएंट फ्लो और एक "स्पैरसिटी" माप को मिलाना, जो कि एक स्पॉटलाइट के केंद्रित होने की जांच करने जैसा है) का उपयोग करके सबसे महत्वपूर्ण हेड्स को चुनता है। यह एक क्लब के बाउंसर की तरह है जो केवल कूल और प्रासंगिक न्यूरॉन्स को अंदर आने देता है और शोर मचाने वालों को बाहर निकाल देता है।
- चरण 2: "पाथ स्प्लिटर" (पथ विभाजक): जब जासूस विचार प्रक्रिया को पीछे की ओर ट्रैक करता है, तो वह "प्रासंगिकता" (निर्णय के लिए श्रेय) को मुख्य पथ और स्किप पथ के बीच विभाजित करता है। यदि रोबोट ने जानकारी को बिना बदले बस आगे पहुँचा दिया, तो GradSkip स्किप पथ को अधिक श्रेय देता है। यदि रोबोट ने कुछ गहन सोच-विचार किया, तो वह मुख्य पथ को श्रेय देता है। यह श्रेय साझा करने का एक गतिशील और निष्पक्ष तरीका है।
प्रमाण: यह काम करता है, और यह तेज़ है!
टीम ने कई तस्वीरों पर GradSkip का परीक्षण किया, जिसमें एक मानक 1,000 छवियों (ImageNet1K) से लेकर मेडिकल ब्लड सेल फोटो (BloodMNIST) तक शामिल हैं। उन्होंने इसे एक कार्य पर भी आज़माया जहाँ रोबोट को छवि के हर हिस्से को कलर-कोड करना होता है (सेगमेंटेशन)।
यहाँ उन्हें क्या पता चला:
- यह सबसे ईमानदार व्याख्या है: जब उन्होंने परीक्षण किया कि कितनी अच्छी तरह से व्याख्याएं रोबोट के वास्तविक निर्णयों से मेल खाती हैं, तो GradSkip सबसे अच्छा था। ImageNet1K टेस्ट पर, यह एक प्रमुख मीट्रिक में दूसरे सर्वश्रेष्ठ तरीके से 12.50% बेहतर था। मेडिकल ब्लड सेल टेस्ट पर, यह उतना ही अच्छा था।
- यह अविश्वसनीय रूप से सटीक है: जब उन्होंने रोबiously रोबोट को वस्तु की ओर इशारा करने के लिए कहा, तो GradSkip सबसे सटीक था।
- यह एक स्पीड डेमन (गति का सौदागर) है: यह सबसे कूल हिस्सा है। पुराने, सटीक तरीके एक टैंक चलाने की तरह थे—वे बहुत अच्छा काम करते थे लेकिन बहुत अधिक ईंधन खर्च करते थे। GradSkip एक चिकनी स्पोर्ट्स कार की तरह है। यह दूसरे सर्वश्रेष्ठ सटीक तरीके की तुलना में 14.45 गुना कम GFLOPs (कंप्यूटर कार्य का एक माप) का उपयोग करता है। क्लासिफिकेशन के लिए, इसे केवल 69.39 GFLOPs की आवश्यकता थी, और सेगमेंटेशन के लिए, केवल 28.52 GFLOPs।
GradSkip क्या नहीं है
पेपर बहुत स्पष्ट है कि GradSkip क्या नहीं करता है:
- यह तब काम नहीं करता जब आप रोबोट के दिमाग के अंदर नहीं देख सकते। आपको एक "बैकवर्ड पास" (ग्रेडिएंट्स की जाँच करना) चलाने में सक्षम होना चाहिए, इसलिए इसका उपयोग उन मॉडलों पर नहीं किया जा सकता जहाँ आप उनके आंतरिक गियर नहीं देख सकते।
- यह सब कुछ जादू से ठीक नहीं कर देता। कुछ पेचीदा मामलों में, जैसे कि जब एक तस्वीर में एक ही प्रकार की कई वस्तुएं होती हैं, तो यह अभी भी बहुत अच्छा है, लेकिन पेपर स्वीकार करता है कि इसमें परफेक्ट स्कोर पाना कठिन है।
- यह "CLS टोकन" (एक विशेष टोकन जिसका उपयोग कुछ रोबोट पूरी छवि का सारांश देने के लिए करते हैं) को नहीं देखता है। लेखक नोट करते हैं कि मानक रोबोटों के लिए जो इस टोकन का उपयोग करते हैं, उन्हें अनदेखा करने से कुछ सुराग छूट सकते हैं, लेकिन अन्य प्रकार के (जैसे SegFormer) के लिए, यह पूरी तरह से काम करता है।
निचोड़
GradSkip केवल एक छोटा सा बदलाव नहीं है; यह रोबोट को सुनने का एक स्मार्ट तरीका है। यह महसूस करके कि मस्तिष्क के सभी भाग समान नहीं हैं और "शॉर्टकट" तारों का सम्मान करके, यह हमें एक स्पष्ट, ईमानदार और सुपर-फास्ट नज़र देता है कि विज़न ट्रांसफॉर्मर निर्णय कैसे लेते हैं। लेखक सुझाव देते हैं कि यह मेडिकल इमेजिंग जैसे क्षेत्रों के लिए गेम-चेंजर हो सकता है, जहाँ यह समझना कि रोबोट ने निदान (diagnosis) क्यों किया, निदान करने जितना ही महत्वपूर्ण है। और सबसे अच्छी बात? यह यह सब बिना किसी सुपरकंप्यूटर की आवश्यकता के करता है।
अपने क्षेत्र के पेपरों की भीड़ में उलझे हुए हैं?
आपके रिसर्च कीवर्ड से मेल खाने वाले सबसे नए और अलग सोच वाले पेपरों का रोज़ाना Digest पाएँ—तकनीकी सारांश के साथ, आपकी भाषा में।