MeDi: ट्यूमर वर्गीकरण में पूर्वाग्रह (Biases) को कम करने के लिए मेटाडेटा-निर्देशित डिफ्यूजन मॉडल

पृष्ठभूमि और अकादमिक वंशावली

मेडिकल AI में "शॉर्टकट लर्निंग" (Shortcut Learning) की समस्या इस तथ्य से उत्पन्न होती है कि डीप लर्निंग मॉडल पैटर्न खोजने में अक्सर अत्यधिक कुशल होते हैं—इतने कुशल कि वे वास्तविक जैविक संकेत (biological signal) के बजाय अप्रासंगिक पर्यावरणीय शोर (environmental noise) पर ध्यान केंद्रित करने लगते हैं। हिस्टोपैथोलॉजी में, इसका अर्थ यह है कि एक मॉडल कैंसर की कोशिकीय आकृति विज्ञान (cellular morphology) के बजाय किसी स्कैनर के विशिष्ट रंग टोन या किसी विशेष अस्पताल के स्टेनिंग प्रोटोकॉल के आधार पर ट्यूमर की पहचान करना सीख सकता है। ऐतिहासिक रूप से, यह समस्या तब उभरी जब AI नियंत्रित प्रयोगशाला डेटासेट से वास्तविक नैदानिक परिनियोजन (clinical deployment) की ओर बढ़ा, जहाँ डेटा विषमता (data heterogeneity) एक सामान्य स्थिति है। इसका मूल "पेन पॉइंट" यह है कि जब प्रशिक्षण डेटा पक्षपाती (skewed) होता है (उदाहरण के लिए, एक अस्पताल किसी विशिष्ट कैंसर प्रकार के सभी नमूने प्रदान करता है), तो मॉडल मेटाडेटा (अस्पताल) और लक्ष्य (कैंसर) के बीच एक छद्म सहसंबंध (spurious correlation) बना लेता है। परिणामस्वरूप, जब मॉडल किसी नए अस्पताल का सामना करता है, तो वह विफल हो जाता है क्योंकि उसका "शॉर्टकट" अब मौजूद नहीं होता।

सहज डोमेन शब्दावली

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

नोटेशन तालिका

गणितीय व्याख्या

लेखक मेटाडेटा को स्पष्ट रूप से जनरेटिव प्रक्रिया में इंजेक्ट करके पूर्वाग्रह की समस्या का समाधान करते हैं। एक मानक डिफ्यूजन मॉडल के बजाय जो केवल $p(\text{image} \mid \text{class})$ मैपिंग सीखता है, वे उद्देश्य को $p(\text{image} \mid \text{class}, \text{metadata})$ सीखने के लिए पुनर्परिभाषित करते हैं।

वे इसे एक कंडीशनिंग वेक्टर $\mathbf{z}_{\text{cond}}$ बनाकर प्राप्त करते हैं जो क्लास जानकारी को सभी प्रासंगिक मेटाडेटा एट्रिब्यूट्स के साथ जोड़ता है:
$$\mathbf{z}_{\text{cond}} = \text{concat}(\mathbf{z}_{\text{class}}, \mathbf{z}_{\text{meta},1}, \dots, \mathbf{z}_{\text{meta},k}) \in \mathbb{R}^{d_t}$$
इसके बाद इस वेक्टर को UNet की आंतरिक डीनोइजिंग प्रक्रिया में टाइमस्टेप एम्बेडिंग $\mathbf{z}_t$ में जोड़कर एकीकृत किया जाता है:
$$\mathbf{z}_{\text{final}} = \mathbf{z}_t + \mathbf{z}_{\text{cond}}$$
ऐसा करने से, मॉडल यह सीखने के लिए बाध्य होता है कि विशिष्ट मेटाडेटा (जैसे अस्पताल की अनूठी स्टेनिंग शैली) ऊतक की जैविक विशेषताओं के साथ कैसे इंटरैक्ट करता है। इन्फरेंस के दौरान, उपयोगकर्ता कम प्रतिनिधित्व वाले या अनदेखे संयोजनों के लिए सिंथेटिक डेटा उत्पन्न करने हेतु इन स्थितियों को "मिक्स एंड मैच" कर सकता है, जिससे डेटासेट प्रभावी रूप से संतुलित हो जाता है और डाउनस्ट्रीम क्लासिफायर मेटाडेटा शॉर्टकट्स को अनदेखा करने के लिए मजबूर हो जाता है।

समस्या परिभाषा और बाधाएं

मुख्य समस्या निरूपण और दुविधा

प्रारंभिक बिंदु (इनपुट/वर्तमान स्थिति):
नैदानिक हिस्टोपैथोलॉजी में, डीप लर्निंग मॉडल को नैदानिक कार्यों (जैसे ट्यूमर सबटाइपिंग) को करने के लिए बड़े डेटासेट (जैसे TCGA) पर प्रशिक्षित किया जाता है। ये डेटासेट स्वाभाविक रूप से पक्षपाती होते हैं, क्योंकि वे विभिन्न चिकित्सा केंद्रों से डेटा एकत्र करते हैं, जिनमें से प्रत्येक के पास अद्वितीय स्टेनिंग प्रोटोकॉल, स्कैनर हार्डवेयर और रोगी जनसांख्यिकी होती है।

वांछित अंतिम बिंदु (आउटपुट/लक्ष्य स्थिति):
लक्ष्य एक ऐसा मजबूत नैदानिक मॉडल बनाना है जो विविध नैदानिक वातावरणों में सामान्यीकरण (generalize) कर सके। विशेष रूप से, लेखक उच्च-विश्वसनीयता वाले सिंथेटिक हिस्टोपैथोलॉजी चित्र उत्पन्न करना चाहते हैं जो कम प्रतिनिधित्व वाले या पूरी तरह से अनदेखे सबपॉपुलेशन का प्रतिनिधित्व करते हैं (उदाहरण के लिए, प्रशिक्षण सेट में मौजूद न होने वाले अस्पताल से एक विशिष्ट कैंसर प्रकार)। इन सिंथेटिक नमूनों के साथ प्रशिक्षण डेटा को संवर्धित (augment) करके, मॉडल को एक संतुलित वितरण प्राप्त करना चाहिए, जो डेटा में मौजूद अंतराल को प्रभावी ढंग से "भर" सके।

लुप्त कड़ी (The Missing Link):
यह अंतर मानक जनरेटिव मॉडल की जैविक विशेषताओं (रोग) को मेटाडेटा-संचालित विविधताओं ("डोमेन" या "साइट" प्रभाव) से अलग करने में असमर्थता में निहित है। जब एक मॉडल को पक्षपाती डेटासेट पर प्रशिक्षित किया जाता है, तो वह वास्तविक ट्यूमर आकृति विज्ञान और विशिष्ट साइट के इमेजिंग आर्टिफैक्ट्स द्वारा शुरू किए गए छद्म सहसंबंधों के बीच अंतर करने में विफल रहता है।

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

कठोर, यथार्थवादी दीवारें:
1. कॉम्बिनेटरियल विस्फोट: मेटाडेटा स्पेस विशाल है। 626 ऊतक स्रोत साइटों और 32 कैंसर प्रकारों के साथ, संभावित संयोजन ($626 \times 32 = 20,032$) वास्तविक दुनिया के डेटा में केवल आंशिक रूप से दर्शाए गए हैं। यह सभी परिदृश्यों को कवर करने के लिए सरल डेटा संग्रह पर निर्भर रहना असंभव बनाता है।
2. छद्म सहसंबंध: डेटा अत्यधिक असंतुलित है; कुछ कैंसर प्रकार प्रशिक्षण सेट में विशेष रूप से विशिष्ट अस्पतालों से जुड़े होते हैं। यह एक "क्लेवर हंस" (Clever Hans) प्रभाव पैदा करता है जहाँ मॉडल कैंसर के जैविक गुणों के बजाय अस्पताल के अनूठे "लुक" को कैंसर के प्रकार के साथ जोड़ना सीख जाता है।
3. गैर-विभेदनीय/विविक्त मेटाडेटा: श्रेणीबद्ध मेटाडेटा (जैसे अस्पताल आईडी) को एक निरंतर डिफ्यूजन प्रक्रिया में एकीकृत करने के लिए एक सावधानीपूर्वक डिज़ाइन की गई एम्बेडिंग रणनीति की आवश्यकता होती है ताकि यह सुनिश्चित हो सके कि मॉडल डेटा वितरण के एक एकल "मोड" में ढहे बिना जनरेशन प्रक्रिया को प्रभावी ढंग से कंडीशन कर सके।

यह दृष्टिकोण क्यों?

इस पेपर द्वारा संबोधित मुख्य चुनौती कंप्यूटेशनल पैथोलॉजी में "क्लेवर हंस" प्रभाव है, जहाँ डीप लर्निंग मॉडल अनजाने में गैर-जैविक मेटाडेटा—जैसे अस्पताल-विशिष्ट स्टेनिंग प्रोटोकॉल, स्कैनर आर्टिफैक्ट्स, या जनसांख्यिकीय झुकाव—पर भरोसा करना सीख जाते हैं, न कि वास्तविक ट्यूमर आकृति विज्ञान पर। जब एक मॉडल को ऐसे डेटासेट पर प्रशिक्षित किया जाता है जहाँ विशिष्ट कैंसर प्रकार विशिष्ट अस्पतालों से जुड़े होते हैं, तो वह इन मेटाडेटा को शॉर्टकट के रूप में मानता है, जिससे एक अलग डेटा वितरण वाले नए नैदानिक वातावरण में तैनात किए जाने पर विनाशकारी विफलता होती है।

विकल्प की अनिवार्यता

लेखकों ने पहचाना कि स्व-पर्यवेक्षित शिक्षण (self-supervised learning) के माध्यम से प्रशिक्षित बड़े पैमाने के फाउंडेशन मॉडल सहित मानक "SOTA" दृष्टिकोण अपर्याप्त हैं क्योंकि वे इन मेटाडेटा पूर्वाग्रहों को अपने लेटेंट रिप्रेजेंटेशन में अंतर्निहित रूप से एन्कोड करते हैं। यदि प्रशिक्षण वितरण पक्षपाती है, तो ये मॉडल केवल पूर्वाग्रह को विरासत में प्राप्त करते हैं। लेखकों ने महसूस किया कि इसे वास्तव में कम करने के लिए, कोई निष्क्रिय शिक्षण पर भरोसा नहीं कर सकता; मेटाडेटा को एक कंडीशनिंग वेरिएबल के रूप में स्पष्ट रूप से मॉडल करना होगा।

• तुलनात्मक श्रेष्ठता: पारंपरिक स्टेन नॉर्मलाइजेशन या स्टाइल-ट्रांसफर तकनीकों (जैसे CycleGAN) के विपरीत, जो छवि को एक विहित शैली में मजबूर करके उसे "ठीक" करने का प्रयास करती हैं, MeDi मेटाडेटा को एक नियंत्रणीय पैरामीटर के रूप में मानता है। क्लास लेबल और मेटाडेटा (जैसे, ऊतक स्रोत साइट) दोनों पर कंडीशन किए गए डिफ्यूजन मॉडल का उपयोग करके, फ्रेमवर्क लक्षित डेटा संवर्धन करने की क्षमता प्राप्त करता है। यह मौजूदा संयोजनों को संतुलित करने के लिए मेटाडेटा स्पेस के भीतर इंटरपोलेट कर सकता है या कम प्रतिनिधित्व वाले या पूरी तरह से अनदेखे सबपॉपुलेशन के लिए सिंथेटिक नमूने उत्पन्न करने के लिए एक्स्ट्रापोलेट कर सकता है। यह संरचनात्मक लाभ मॉडल को प्रशिक्षण वितरण में अंतराल को "भरने" की अनुमति देता है, जिससे रोग लेबल को अस्पताल-विशिष्ट आर्टिफैक्ट्स से प्रभावी ढंग से अलग किया जा सकता है।
• आवश्यकताओं और समाधान का "मिलन": समस्या के लिए एक ऐसे जनरेटिव मॉडल की आवश्यकता है जो उच्च-विश्वसनीयता वाला और अत्यधिक नियंत्रणीय दोनों हो। डिफ्यूजन मॉडल यहाँ एकमात्र व्यवहार्य समाधान हैं क्योंकि वे एक स्थिर, पुनरावृत्ति डीनोइजिंग प्रक्रिया प्रदान करते हैं जिसे हर चरण पर आसानी से कंडीशन किया जा सकता है। $\mathbf{z}_{\text{cond}} = \text{concat}(\mathbf{z}_{\text{class}}, \mathbf{z}_{\text{meta},1}, \dots, \mathbf{z}_{\text{meta},k})$ को परिभाषित करके और इसे $\mathbf{z}_{\text{final}} = \mathbf{z}_t + \mathbf{z}_{\text{cond}}$ के माध्यम से UNet के रेजिडुअल ब्लॉक्स में इंजेक्ट करके, लेखक यह सुनिश्चित करते हैं कि जनरेशन प्रक्रिया वांछित मेटाडेटा द्वारा सख्ती से निर्देशित हो। यह उन छवियों को संश्लेषित करने की आवश्यकता के साथ पूरी तरह से मेल खाता है जो जैविक अखंडता को बनाए रखती हैं और साथ ही एक कम प्रतिनिधित्व वाले अस्पताल साइट की विशिष्ट "शैली" प्रदर्शित करती हैं।

गणितीय और तार्किक तंत्र

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

मास्टर समीकरण

MeDi फ्रेमवर्क का मूल एक कंडीशनिंग वेक्टर का निर्माण है जो डिफ्यूजन मॉडल की डीनोइजिंग प्रक्रिया का मार्गदर्शन करता है। UNet को प्रदान किया गया अंतिम कंडीशनिंग सिग्नल इस प्रकार परिभाषित है:

$$ \mathbf{z}_{\text{final}} = \mathbf{z}_t + \mathbf{z}_{\text{cond}} $$

जहाँ $\mathbf{z}_{\text{cond}}$ को इस प्रकार परिभाषित किया गया है:

$$ \mathbf{z}_{\text{cond}} = \text{concat}(\mathbf{z}_{\text{class}}, \mathbf{z}_{\text{meta},1}, \dots, \mathbf{z}_{\text{meta},k}) \in \mathbb{R}^{d_t} $$

समीकरण का विश्लेषण

1. $\mathbf{z}_t$: यह टाइमस्टेप एम्बेडिंग है। यह डिफ्यूजन प्रक्रिया में वर्तमान "शोर स्तर" का प्रतिनिधित्व करता है। इसकी भूमिका मॉडल को यह सूचित करना है कि वर्तमान चरण में कितनी डीनोइजिंग की आवश्यकता है।
2. $\mathbf{z}_{\text{class}}$: यह कैंसर उपप्रकार (जैसे, फेफड़े का एडेनोकार्सिनोमा) की सीखने योग्य एम्बेडिंग है। यह प्राथमिक अर्थपूर्ण मार्गदर्शन प्रदान करती है कि किस जैविक संरचना को उत्पन्न करना है।
3. $\mathbf{z}_{\text{meta},i}$: ये $k$ मेटाडेटा एट्रिब्यूट्स (जैसे, ऊतक स्रोत साइट) के लिए सीखने योग्य एम्बेडिंग हैं। उनकी भूमिका एक "शैली" या "डोमेन" नियंत्रक के रूप में कार्य करने की है, जो मॉडल को किसी विशेष अस्पताल से जुड़े विशिष्ट दृश्य आर्टिफैक्ट्स को सीखने के लिए मजबूर करती है।
4. $\text{concat}(\dots)$: लेखक इन अलग-अलग सूचना स्रोतों को एक एकल वेक्टर में फ्यूज करने के लिए संयोजन (concatenation) का उपयोग करते हैं। यहाँ इसे जोड़ने (addition) के बजाय प्राथमिकता दी जाती है क्योंकि क्लास और मेटाडेटा स्वतंत्र, श्रेणीबद्ध आयामों का प्रतिनिधित्व करते हैं जिन्हें तब तक मिश्रित नहीं किया जाना चाहिए जब तक कि मॉडल उन्हें UNet परतों के भीतर स्पष्ट रूप से संसाधित न कर ले।
5. $\mathbf{z}_{\text{final}}$: यह संयुक्त कंडीशनिंग वेक्टर है। इसे $\mathbf{z}_t$ में जोड़कर, लेखक यह सुनिश्चित करते हैं कि डीनोइजिंग ऑपरेशन एक साथ "समय" (शोर स्तर) और "संदर्भ" (क्लास + मेटाडेटा) के प्रति जागरूक हो।

परिणाम, सीमाएं और निष्कर्ष

MeDi का विश्लेषण: मेटाडेटा-निर्देशित डिफ्यूजन मॉडल

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

प्रयोगात्मक सत्यापन

लेखकों ने एक चुनौतीपूर्ण आउट-ऑफ-डिस्ट्रीब्यूशन परिदृश्य बनाकर अपनी परिकल्पना का "कठोरता से" परीक्षण किया। उन्होंने विशिष्ट चिकित्सा केंद्र और रोगी जाति संयोजनों के 30% को अलग रखा, यह सुनिश्चित करते हुए कि मॉडल ने प्रशिक्षण के दौरान इन विशिष्ट सबपॉपुलेशन को कभी नहीं देखा था।

• साक्ष्य:

  1. विश्वसनीयता (Fidelity): MeDi ने CLS बेसलाइन के 50.65 की तुलना में 37.73 का कम औसत Fréchet Inception Distance (FID) प्राप्त किया, जो यह साबित करता है कि मेटाडेटा कंडीशनिंग अधिक विश्वसनीय छवि संश्लेषण की ओर ले जाती है।

  2. डाउनस्ट्रीम उपयोगिता: लेखकों ने फाउंडेशन मॉडल (UNI) से एम्बेडिंग के ऊपर रैखिक क्लासिफायर प्रशिक्षित किए। जब अनदेखे सबपॉपुलेशन पर परीक्षण किया गया, तो MeDi-संवर्धित प्रशिक्षण सेट ने NSCLC और गर्भाशय कैंसर कार्यों के लिए संतुलित सटीकता में लगातार CLS-संवर्धित सेट से बेहतर प्रदर्शन किया। यह निश्चित प्रमाण प्रदान करता है कि MeDi उन छद्म सहसंबंधों को सफलतापूर्वक तोड़ता है जो आमतौर पर इन मॉडलों को प्रभावित करते हैं।

लेखकों ने प्रभावी ढंग से प्रदर्शित किया कि "शोर" (मेटाडेटा) को स्पष्ट रूप से मॉडल करके, वे मॉडल को "सिग्नल" (पैथोलॉजी) पर ध्यान केंद्रित करने के लिए मजबूर कर सकते हैं, जिसके परिणामस्वरूप एक अधिक मजबूत और निष्पक्ष प्रणाली प्राप्त होती है।

अन्य क्षेत्रों के साथ समरूपता (Isomorphisms)

MeDi का विश्लेषण: मेटाडेटा-निर्देशित डिफ्यूजन मॉडल

पृष्ठभूमि ज्ञान

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

प्रेरणा और बाधाएं

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

गणितीय व्याख्या

लेखक मेटाडेटा एम्बेडिंग के साथ मानक डिफ्यूजन प्रक्रिया को संवर्धित करके सशर्त डेटा जनरेशन की समस्या को हल करते हैं।

एक मानक डिफ्यूजन मॉडल में, मॉडल छवि $x_t$ और टाइमस्टेप $t$ को देखते हुए शोर $\epsilon$ की भविष्यवाणी करना सीखता है। लेखक एक कंडीशनिंग वेक्टर $\mathbf{z}_{\text{cond}}$ पेश करते हैं जो क्लास लेबल $\mathbf{z}_{\text{class}}$ और मेटाडेटा एट्रिब्यूट्स $\mathbf{z}_{\text{meta},i}$ दोनों को शामिल करता है।

कंडीशनिंग वेक्टर को इस प्रकार परिभाषित किया गया है:
$$\mathbf{z}_{\text{cond}} = \text{concat}(\mathbf{z}_{\text{class}}, \mathbf{z}_{\text{meta},1}, \dots, \mathbf{z}_{\text{meta},k}) \in \mathbb{R}^{d_t}$$

इस वेक्टर को फिर टाइमस्टेप एम्बेडिंग में जोड़कर UNet आर्किटेक्चर में एकीकृत किया जाता है:
$$\mathbf{z}_{\text{final}} = \mathbf{z}_t + \mathbf{z}_{\text{cond}}$$

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

संरचनात्मक कंकाल

एक जनरेटिव तंत्र जो सहायक मेटाडेटा पर लेटेंट स्पेस को कंडीशन करके डोमेन-विशिष्ट शोर को लक्ष्य विशेषताओं से अलग करता है, जिससे कम प्रतिनिधित्व वाले डेटा बिंदुओं के इंटरपोलेशन की अनुमति मिलती है।