2025-11-18T01:13:20.274920

Advanced Lung Nodule Segmentation and Classification for Early Detection of Lung Cancer using SAM and Transfer Learning

V, K

Lung cancer is an extremely lethal disease primarily due to its late-stage diagnosis and significant mortality rate, making it the major cause of cancer-related demises globally. Machine Learning (ML) and Convolution Neural network (CNN) based Deep Learning (DL) techniques are primarily used for precise segmentation and classification of cancerous nodules in the CT (Computed Tomography) or MRI images. This study introduces an innovative approach to lung nodule segmentation by utilizing the Segment Anything Model (SAM) combined with transfer learning techniques. Precise segmentation of lung nodules is crucial for the early detection of lung cancer. The proposed method leverages Bounding Box prompts and a vision transformer model to enhance segmentation performance, achieving high accuracy, Dice Similarity Coefficient (DSC) and Intersection over Union (IoU) metrics. The integration of SAM and Transfer Learning significantly improves Computer-Aided Detection (CAD) systems in medical imaging, particularly for lung cancer diagnosis. The findings demonstrate the proposed model effectiveness in precisely segmenting lung nodules from CT scans, underscoring its potential to advance early detection and improve patient care outcomes in lung cancer diagnosis. The results show SAM Model with transfer learning achieving a DSC of 97.08% and an IoU of 95.6%, for segmentation and accuracy of 96.71% for classification indicates that ,its performance is noteworthy compared to existing techniques.

academic

SAM और Transfer Learning का उपयोग करके फेफड़ों के कैंसर की प्रारंभिक पहचान के लिए उन्नत फेफड़ों की नोड्यूल विभाजन और वर्गीकरण

बुनियादी जानकारी

पेपर ID: 2501.00586
शीर्षक: Advanced Lung Nodule Segmentation and Classification for Early Detection of Lung Cancer using SAM and Transfer Learning
लेखक: Asha V, Bhavanishankar K (RNS Institute of Technology, Bengaluru & Visvesvaraya Technological University, Belagavi)
वर्गीकरण: eess.IV cs.CV cs.LG
प्रकाशन समय: Preprint submitted to Image and Vision Computing दिसंबर 31, 2024
पेपर लिंक: https://arxiv.org/abs/2501.00586

सारांश

यह अनुसंधान फेफड़ों के कैंसर की प्रारंभिक पहचान की महत्वपूर्ण चिकित्सा समस्या के समाधान के लिए फेफड़ों की नोड्यूल विभाजन और वर्गीकरण की एक नवीन विधि प्रस्तुत करता है। यह विधि Segment Anything Model (SAM) और Transfer Learning तकनीकों को जोड़ती है, जो सीमा बॉक्स संकेतों और Vision Transformer मॉडल का उपयोग करके विभाजन प्रदर्शन को बढ़ाती है। प्रायोगिक परिणाम दर्शाते हैं कि यह विधि विभाजन कार्य पर 97.08% Dice Similarity Coefficient (DSC) और 95.6% Intersection over Union (IoU) प्राप्त करती है, वर्गीकरण कार्य पर 96.71% सटीकता प्राप्त करती है, जो मौजूदा तकनीकों की तुलना में महत्वपूर्ण लाभ दर्शाती है।

अनुसंधान पृष्ठभूमि और प्रेरणा

समस्या की महत्ता

फेफड़ों के कैंसर की गंभीरता: फेफड़ों का कैंसर विश्व स्तर पर कैंसर से संबंधित मृत्यु का प्रमुख कारण है, 2024 के आंकड़ों से पता चलता है कि यह सभी कैंसर मृत्यु का लगभग 18% है, जिससे प्रतिवर्ष लगभग 1.8 मिलियन लोगों की मृत्यु होती है
प्रारंभिक पहचान की महत्ता: फेफड़ों के कैंसर की उच्च मृत्यु दर मुख्य रूप से देर से निदान के कारण है, प्रारंभिक सटीक पहचान रोगी के पूर्वानुमान में सुधार के लिए महत्वपूर्ण है
चिकित्सा इमेजिंग विश्लेषण की चुनौतियाँ: CT स्कैन छवियों की जटिलता और डेटा की मात्रा के लिए कंप्यूटर-सहायक पहचान (CAD) प्रणाली की आवश्यकता होती है जो रेडियोलॉजिस्ट को सटीक निदान में सहायता करे

मौजूदा विधियों की सीमाएं

पारंपरिक इमेज प्रोसेसिंग विधियाँ: किनारे की पहचान, थ्रेसहोल्डिंग विभाजन और क्षेत्र-आधारित विधियाँ चिकित्सा छवियों की जटिलता और परिवर्तनशीलता को संभालने में सीमाएं रखती हैं
गहन शिक्षा विधियों की चुनौतियाँ: हालांकि CNN जैसी गहन शिक्षा विधियाँ चिकित्सा छवि विश्लेषण में उत्कृष्ट प्रदर्शन करती हैं, फिर भी विभाजन और वर्गीकरण सटीकता में चुनौतियों का सामना करती हैं
डेटा गुणवत्ता और लेबलिंग समस्याएं: उच्च गुणवत्ता वाले डेटा प्राप्त करना, सटीक विभाजन प्राप्त करना और सटीक लेबलिंग प्राप्त करना मुख्य बाधाएं हैं

अनुसंधान प्रेरणा

यह अनुसंधान SAM की शक्तिशाली विभाजन क्षमता और Transfer Learning के लाभों को जोड़कर एक अधिक सटीक और कुशल फेफड़ों की नोड्यूल पहचान प्रणाली विकसित करने का लक्ष्य रखता है, जो प्रारंभिक फेफड़ों के कैंसर निदान की सटीकता में सुधार करे।

मुख्य योगदान

SAM और Transfer Learning का नवीन एकीकरण: पहली बार Segment Anything Model को Transfer Learning तकनीकों के साथ जोड़ा गया है, जो फेफड़ों की नोड्यूल विभाजन के लिए उपयोग किया गया है, जिससे पहचान प्रदर्शन और सटीकता में महत्वपूर्ण सुधार हुआ है
सीमा बॉक्स संकेत और Vision Transformer का अनुकूलित अनुप्रयोग: सीमा बॉक्स संकेत और Vision Transformer मॉडल को एकीकृत करके, उत्कृष्ट विभाजन प्रदर्शन प्राप्त किया गया है, जो सटीकता, DSC और IoU मेट्रिक्स पर उत्कृष्ट प्रदर्शन करता है
MobileNetV2 घातक ट्यूमर वर्गीकरण: SAM विभाजन परिणामों को MobileNetV2 के साथ जोड़कर, फेफड़ों की नोड्यूल की घातकता की डिग्री का कुशल वर्गीकरण प्राप्त किया गया है
व्यापक प्रदर्शन सत्यापन: LUNA16 डेटासेट पर व्यापक प्रायोगिक सत्यापन किया गया है, कई बेंचमार्क विधियों के साथ विस्तृत तुलना की गई है

विधि विवरण

कार्य परिभाषा

इस अनुसंधान में दो मुख्य कार्य शामिल हैं:

फेफड़ों की नोड्यूल विभाजन: CT स्कैन छवियों से फेफड़ों की नोड्यूल क्षेत्र को सटीक रूप से विभाजित करना
घातकता डिग्री वर्गीकरण: विभाजित नोड्यूल को सौम्य/घातक में वर्गीकृत करना

इनपुट CT स्कैन छवियाँ हैं, आउटपुट विभाजन मास्क और घातकता डिग्री वर्गीकरण परिणाम हैं।

मॉडल आर्किटेक्चर

1. SAM बुनियादी आर्किटेक्चर

SAM मॉडल में तीन मुख्य घटक शामिल हैं:

इमेज एनकोडर (Image Encoder):

पूर्व-प्रशिक्षित Vision Transformer (ViT-H/16) का उपयोग करता है
14×14 विंडो ध्यान तंत्र और 4 समान दूरी वाले वैश्विक ध्यान मॉड्यूल का उपयोग करता है
16×16 डाउनसैंपलिंग एम्बेडिंग उत्पन्न करता है, इनपुट छवि रिज़ॉल्यूशन 256×256 है
256 चैनल प्राप्त करने के लिए 1×1 और 3×3 कनवोल्यूशन का उपयोग करता है, प्रत्येक कनवोल्यूशन के बाद लेयर नॉर्मलाइजेशन होता है

प्रॉम्प्ट एनकोडर (Prompt Encoder):

सीमा बॉक्स संकेत को 256-आयामी वेक्टर एम्बेडिंग में परिवर्तित करता है
प्रत्येक सीमा बॉक्स को बाएं ऊपरी कोने और दाएं निचले कोने के बिंदु की एम्बेडिंग जोड़ी द्वारा दर्शाया जाता है
विरल संकेत (बिंदु, बॉक्स, पाठ) और घने संकेत (मास्क) का समर्थन करता है

मास्क डिकोडर (Mask Decoder):

कस्टम Transformer डिकोडर का उपयोग करता है, जिसमें स्व-ध्यान और क्रॉस-ध्यान ब्लॉक शामिल हैं
अपसैंपलिंग और गतिशील रैखिक वर्गीकारक के माध्यम से अंतिम विभाजन मास्क उत्पन्न करता है
परिणाम को इनपुट आकार में समायोजित करने के लिए द्विरेखीय प्रक्षेप का उपयोग करता है

2. LUNA16 डेटासेट के लिए फाइन-ट्यूनिंग

सीमा बॉक्स संकेत के साथ पर्यवेक्षित शिक्षा का उपयोग करता है
पूर्व-प्रशिक्षित वजन के साथ Transfer Learning को जोड़ता है
फेफड़ों की नोड्यूल विशेषताओं के लिए मॉडल पैरामीटर अनुकूलन करता है

3. MobileNetV2 वर्गीकरण नेटवर्क

गणना जटिलता को कम करने के लिए गहराई से अलग करने योग्य कनवोल्यूशन का उपयोग करता है
उलटे अवशिष्ट संरचना और रैखिक बोतल गर्दन डिजाइन का उपयोग करता है
फेफड़ों की नोड्यूल घातकता डिग्री वर्गीकरण कार्य के लिए Transfer Learning के माध्यम से अनुकूलन करता है

तकनीकी नवाचार बिंदु

शून्य-शॉट से पर्यवेक्षित शिक्षा में रूपांतरण: SAM को शून्य-शॉट विभाजन मॉडल से फेफड़ों की नोड्यूल के लिए पर्यवेक्षित शिक्षा मॉडल में परिवर्तित करता है
सीमा बॉक्स संकेत अनुकूलन: चिकित्सा छवि विशेषताओं के लिए सीमा बॉक्स संकेत तंत्र को अनुकूलित करता है
बहु-चरण आर्किटेक्चर डिजाइन: विभाजन और वर्गीकरण का कैस्केड डिजाइन, विभाजन परिणामों का पूर्ण उपयोग करके वर्गीकरण को निर्देशित करता है

प्रायोगिक सेटअप

डेटासेट

LUNA16 डेटासेट:

LIDC-IDRI डेटासेट के चयनित संस्करण से स्रोत
888 CT स्कैन (.mhd प्रारूप) शामिल हैं
10 उप-समूहों (subset 0-9) में विभाजित, प्रत्येक उप-समूह में 88-89 छवियाँ हैं
3D नोड्यूल केंद्र निर्देशांक और व्यास लेबलिंग प्रदान करता है
डेटा प्रीप्रोसेसिंग: .mhd → .npy → .jpg प्रारूप रूपांतरण

डेटा विभाजन:

प्रशिक्षण सेट: 70%
परीक्षण सेट: 30%

मूल्यांकन मेट्रिक्स

विभाजन मेट्रिक्स:

DSC (Dice Similarity Coefficient): 2×TP/(2×TP+FP+FN)
IoU (Intersection over Union): TP/(TP+FP+FN)
संवेदनशीलता (Sensitivity): TP/(TP+FN)
सकारात्मक पूर्वानुमानित मान (PPV): TP/(TP+FP)

वर्गीकरण मेट्रिक्स:

सटीकता, परिशुद्धता, संवेदनशीलता, विशिष्टता, F1 स्कोर

तुलनात्मक विधियाँ

विभाजन विधियाँ: UNet, VNet, FCNUNet, Mask RCNN, EFCM वर्गीकरण विधियाँ: Inception V3, ResNet, VGG16, DenseNet, AlexNet, DenseAlexNet

कार्यान्वयन विवरण

प्रोग्रामिंग भाषा: Python 3.8
प्लेटफॉर्म: Google Colab, Kaggle
हार्डवेयर: 64GB RAM, 6GB NVIDIA GPU
अनुकूलक: Adam (सीखने की दर 0.001)
बैच आकार: 4 (विभाजन), 5 (वर्गीकरण)
प्रशिक्षण युग: 100 epochs

प्रायोगिक परिणाम

मुख्य परिणाम

विभाजन प्रदर्शन:

DSC: 97.08%
IoU: 95.6%
संवेदनशीलता: 97.85%
PPV: 98.1%

वर्गीकरण प्रदर्शन:

सटीकता: 96.71%
परिशुद्धता: 95.25%
संवेदनशीलता: 98.30%
विशिष्टता: 95.45%
F1 स्कोर: 96.50%

तुलनात्मक विश्लेषण

विभाजन कार्य तुलना:

विधि	DSC	IoU
UNet	94.97%	-
RFRVNet	95.01%	83.00%
EFCM	97.10%	91.96%
यह विधि	97.08%	95.60%

वर्गीकरण कार्य तुलना:

विधि	सटीकता	F1 स्कोर
DenseAlexNet	95.65%	95.58%
Inception V3	91.40%	92.31%
यह विधि	96.71%	96.50%

प्रायोगिक निष्कर्ष

SAM चिकित्सा छवि विभाजन में मजबूत सामान्यीकरण क्षमता प्रदर्शित करता है
Transfer Learning विशिष्ट चिकित्सा कार्यों पर मॉडल प्रदर्शन में महत्वपूर्ण सुधार करता है
सीमा बॉक्स संकेत तंत्र विभाजन सटीकता को प्रभावी रूप से बढ़ाता है
MobileNetV2 उच्च सटीकता बनाए रखते हुए गणना दक्षता प्राप्त करता है

निष्कर्ष और चर्चा

मुख्य निष्कर्ष

SAM और Transfer Learning का संयोजन फेफड़ों की नोड्यूल विभाजन कार्य पर उत्कृष्ट प्रदर्शन प्राप्त करता है
सीमा बॉक्स संकेत तंत्र चिकित्सा छवि विभाजन की सटीकता को प्रभावी रूप से बढ़ाता है
प्रस्तावित विधि कई मूल्यांकन मेट्रिक्स पर इष्टतम स्तर पर पहुंचती है या उसके करीब है
यह विधि CAD प्रणाली के प्रदर्शन को बढ़ाने और रोगी के पूर्वानुमान में सुधार करने की क्षमता रखती है

सीमाएं

डेटासेट सीमा: केवल LUNA16 डेटासेट पर सत्यापन किया गया है, सामान्यीकरण क्षमता को आगे सत्यापन की आवश्यकता है
विभाजन सटीकता: SAM सभी फेफड़ों की CT छवियों के लिए सटीक मास्क उत्पन्न नहीं कर सकता है
गणना जटिलता: हालांकि MobileNetV2 का उपयोग किया गया है, समग्र प्रणाली की गणना लागत को अभी भी अनुकूलन की आवश्यकता है
नैदानिक सत्यापन: बड़े पैमाने पर नैदानिक परीक्षण सत्यापन की कमी है

भविष्य की दिशाएं

सभी फेफड़ों की CT छवियों के लिए मास्क उत्पन्न करने की सटीकता में सुधार करना
अधिक चिकित्सा छवि डेटासेट पर सत्यापन के लिए विस्तार करना
गणना दक्षता में सुधार के लिए मॉडल संरचना को अनुकूलित करना
वास्तविक अनुप्रयोग प्रभाव को सत्यापित करने के लिए नैदानिक परीक्षण आयोजित करना

गहन मूल्यांकन

शक्तियाँ

तकनीकी नवाचार: पहली बार SAM को Transfer Learning के साथ व्यवस्थित रूप से फेफड़ों की नोड्यूल पहचान के लिए जोड़ा गया है, जिसमें मजबूत नवाचार है
प्रायोगिक पूर्णता: कई मेट्रिक्स पर कई बेंचमार्क विधियों के साथ व्यापक तुलना की गई है, प्रायोगिक डिजाइन उचित है
उत्कृष्ट प्रदर्शन: विभाजन और वर्गीकरण कार्य दोनों पर उत्कृष्ट प्रदर्शन प्राप्त किया गया है
व्यावहारिक मूल्य: विधि में स्पष्ट नैदानिक अनुप्रयोग मूल्य है, जो फेफड़ों के कैंसर की प्रारंभिक निदान स्तर को बढ़ाने में सहायता करता है

कमियाँ

विधि सीमाएं: चिकित्सा छवियों पर SAM के अनुकूलन की गहराई से विश्लेषण अपर्याप्त है
प्रायोगिक दायरा: केवल एकल डेटासेट पर सत्यापन किया गया है, क्रॉस-डेटासेट सत्यापन की कमी है
सैद्धांतिक विश्लेषण: विधि प्रभावशीलता के सैद्धांतिक विश्लेषण और व्याख्या की कमी है
गणना दक्षता: मॉडल की गणना जटिलता और वास्तविक समय प्रदर्शन विश्लेषण अपर्याप्त है

प्रभाव

शैक्षणिक योगदान: चिकित्सा छवि विश्लेषण क्षेत्र में SAM के अनुप्रयोग के लिए मूल्यवान अन्वेषण प्रदान करता है
व्यावहारिक मूल्य: विधि में मजबूत नैदानिक अनुप्रयोग क्षमता है
पुनरुत्पादनीयता: विस्तृत कार्यान्वयन विवरण प्रदान करता है, अच्छी पुनरुत्पादनीयता है
सामान्यीकरण: विधि ढांचा अन्य चिकित्सा छवि विश्लेषण कार्यों तक विस्तारित किया जा सकता है

लागू परिदृश्य

अस्पताल CAD प्रणाली: मौजूदा कंप्यूटर-सहायक निदान प्रणालियों में एकीकृत किया जा सकता है
फेफड़ों के कैंसर की जांच: बड़े पैमाने पर फेफड़ों के कैंसर की जांच परियोजनाओं के लिए उपयुक्त है
चिकित्सा शिक्षा: चिकित्सा इमेजिंग शिक्षण और प्रशिक्षण के लिए उपयोग किया जा सकता है
वैज्ञानिक अनुप्रयोग: संबंधित चिकित्सा छवि विश्लेषण अनुसंधान के लिए बुनियादी उपकरण प्रदान करता है

संदर्भ

पेपर ने 35 संबंधित संदर्भों का हवाला दिया है, जिसमें SAM सिद्धांत, चिकित्सा छवि विभाजन, फेफड़ों की नोड्यूल पहचान, गहन शिक्षा आदि कई संबंधित क्षेत्रों के महत्वपूर्ण कार्य शामिल हैं, जो अनुसंधान के लिए एक ठोस सैद्धांतिक आधार प्रदान करते हैं।