Enhancing Lung Image Segmentation Using Hybrid U-Net and Transfer Learning: A Comprehensive Approach on Tuberculosis Diagnosis

Enhancing Lung Image Segmentation Using Hybrid U-Net and Transfer Learning: A Comprehensive Approach on Tuberculosis Diagnosis | Research Square window.SnipcartSettings = { analytics: { enabled: false } }; (function() { var accessVector = localStorage.getItem('access_vector') || ''; window.dataLayer = window.dataLayer || []; if (accessVector) { window.dataLayer.push({ user: { profile: { profileInfo: { snid: accessVector } } } }); } })(); (function(w,d,s,l,i){w[l]=w[l]||[];w[l].push({'gtm.start':new Date().getTime(),event:'gtm.js'});var f=d.getElementsByTagName(s)[0],j=d.createElement(s),dl=l!='dataLayer'?'&l='+l:'';j.async=true;j.src='https://www.googletagmanager.com/gtm.js?id='+i+dl;f.parentNode.insertBefore(j,f);})(window,document,'script','dataLayer','GTM-K279D39R'); Browse Preprints In Review Journals COVID-19 Preprints AJE Video Bytes Research Tools Research Promotion AJE Professional Editing AJE Rubriq About Preprint Platform In Review Editorial Policies Our Team Advisory Board Help Center Sign In Submit a Preprint Cite Share Download PDF Research Article Enhancing Lung Image Segmentation Using Hybrid U-Net and Transfer Learning: A Comprehensive Approach on Tuberculosis Diagnosis Deepika Gupta, Dr. Suma Dawn This is a preprint; it has not been peer reviewed by a journal. https://doi.org/ 10.21203/rs.3.rs-6759181/v1 This work is licensed under a CC BY 4.0 License Status: Posted Version 1 posted You are reading this latest preprint version Abstract Tuberculosis (TB) stands as the foremost global cause of mortality, a highly contagious lung ailment trailed closely by malaria and HIV/AIDS. To facilitate precise lung X-ray image analysis, which is crucial for diagnoses like lung tuberculosis, lung X-ray image segmentation takes precedence. The formidable U-net architecture, renowned in deep learning for image segmentation, is prominent in this endeavor. This architectural marvel comprises a contracting pathway, adept at extracting high-level information, and a symmetrically expanding pathway, adept at restoring vital details. Setting itself apart, this network outshines many counterparts and exhibits the capacity for comprehensive training even with a limited dataset. In this context, the primary goal of our work is to provide an automated lung segmentation method aimed at addressing the challenge of reconstructing damaged lung sections, which makes a significant contribution to our field of medical science and in the application of artificial intelligence that automatically segments an image of the lung to aid TB detection and classification. The proposed approach can be distilled into three fundamental steps: (a)Image Acquisition: This initial step involves describing the materials and techniques employed for image collection. (b)Initial Segmentation: This critical phase utilizes the power of the U-net deep convolutional network (CNN) model and employs three distinct approaches. These approaches play a pivotal role in the initial segmentation of lung regions. (c) Ensemble Modeling: Subsequently, all three models are amalgamated through ensemble modeling. This consolidation process combines the best outputs from all three right at every pixel in each of the three approaches to yield a final result. In summary, our focus is achieving accurate lung segmentation, particularly for damaged sections, using a comprehensive method that leverages semantic segmentation, transfer learning, and deep learning techniques, notably the Hybrid U-Net model. This approach also enhances lung nodule detection, making it a valuable contribution to the field. Three transfer learning methods trained on large-size image datasets are ResNet34, Inception V3, and VGG 16. The results demonstrated impressive performance metrics, including Mean IoU, Dice-Score, and F-Score. The result section shows that the hybrid U-net model with inception V3 gives better results than the rest, with 0.975 Mean IoU, 0.987 Dice Score, and 0.9963 F-Score. After the ensemble method improved, the result of (0.976 )IoU and (0.988) Dice-Score was achieved. Health Economics & Outcomes Research Artificial Intelligence and Machine Learning Computer Architecture and Engineering Semantic Segmentation Transfer Learning Deep Learning Hybrid U-Net Tuberculosis accuracy Full Text Additional Declarations The authors declare no competing interests. Cite Share Download PDF Status: Posted Version 1 posted You are reading this latest preprint version Research Square lets you share your work early, gain feedback from the community, and start making changes to your manuscript prior to peer review in a journal. As a division of Research Square Company, we’re committed to making research communication faster, fairer, and more useful. We do this by developing innovative software and high quality services for the global research community. Our growing team is made up of researchers and industry professionals working together to solve the most critical problems facing scientific publishing. Also discoverable on Platform About Our Team In Review Editorial Policies Advisory Board Help Center Resources Author Services Accessibility API Access RSS feed Manage Cookie Preferences © Research Square 2026 | ISSN 2693-5015 (online) Privacy Policy Terms of Service Do Not Sell My Personal Information {"props":{"pageProps":{"initialData":{"identity":"rs-6759181","acceptedTermsAndConditions":true,"allowDirectSubmit":true,"archivedVersions":[],"articleType":"Research Article","associatedPublications":[],"authors":[{"id":462495526,"identity":"8f581220-7b81-41ad-88f7-1767df6c586a","order_by":0,"name":"Deepika Gupta","email":"data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAZAAAAAyAQMAAABI0h/eAAAABlBMVEX///8AAABVwtN+AAAACXBIWXMAAA7EAAAOxAGVKw4bAAAA5klEQVRIie3PsQrCMBCA4UIhXY66RhB9hYJQBaG+SqRQl/gEigqFOumzOHUWiukS92wqgoOLgiBO4mVxrBkF80OOC+SD1nFstl9sg4fhIF6qV9owJz4IpgmYET2alAd6/U78MhWHg4yiDOT9oiZdcLxiu64idSmGAVNxnHnLvMcFfhgkiaoigeIhZTc3JrDL25wgoRBWk/31gWQWE8rPbf4yIQoIZaqIkLinUWZA6jLpBEyWjIAI3dGKAvn2L35ZnI9PMe63Funpzh/TZs0rRCXBiB6DOS70czUhfTzuzeC1zWaz/WFvRGVN1vcoxCwAAAAASUVORK5CYII=","orcid":"https://orcid.org/0009-0006-1093-495X","institution":"Jaypee Institute of Information and Technology","correspondingAuthor":true,"prefix":"","firstName":"Deepika","middleName":"","lastName":"Gupta","suffix":""},{"id":462495527,"identity":"539eb657-ea29-49a2-800f-c168ee913aff","order_by":1,"name":"Dr. Suma Dawn","email":"","orcid":"","institution":"Jaypee Institute of Information and Technology","correspondingAuthor":false,"prefix":"Dr.","firstName":"Suma","middleName":"","lastName":"Dawn","suffix":""}],"badges":[],"createdAt":"2025-05-27 12:00:52","currentVersionCode":1,"declarations":{"humanSubjects":false,"vertebrateSubjects":false,"conflictsOfInterestStatement":false,"humanSubjectEthicalGuidelines":false,"humanSubjectConsent":false,"humanSubjectClinicalTrial":false,"humanSubjectCaseReport":false,"vertebrateSubjectEthicalGuidelines":false},"doi":"10.21203/rs.3.rs-6759181/v1","doiUrl":"https://doi.org/10.21203/rs.3.rs-6759181/v1","draftVersion":[],"editorialEvents":[],"editorialNote":"","failedWorkflow":false,"files":[{"id":83644408,"identity":"3958d8d3-295c-4bf3-8653-9389d8a76798","added_by":"auto","created_at":"2025-05-30 04:49:03","extension":"pdf","order_by":1,"title":"","display":"","copyAsset":false,"role":"manuscript-pdf","size":477711,"visible":true,"origin":"","legend":"","description":"","filename":"Paper1Lungsegmentation.edited.docxpreprint.pdf","url":"https://assets-eu.researchsquare.com/files/rs-6759181/v1_covered_56f560c0-40ff-43a5-8639-b9f77e928a14.pdf"}],"financialInterests":"The authors declare no competing interests.","formattedTitle":"\u003cp\u003eEnhancing Lung Image Segmentation Using Hybrid U-Net and Transfer Learning: A Comprehensive Approach on Tuberculosis Diagnosis\u003c/p\u003e","fulltext":[],"fulltextSource":"","fullText":"","funders":[],"hasAdminPriorityOnWorkflow":false,"hasManuscriptDocX":false,"hasOptedInToPreprint":true,"hasPassedJournalQc":"","hasAnyPriority":true,"hideJournal":true,"highlight":"","institution":"Jaypee Institute of Information Technology","isAcceptedByJournal":false,"isAuthorSuppliedPdf":true,"isDeskRejected":"","isHiddenFromSearch":false,"isInQc":false,"isInWorkflow":false,"isPdf":true,"isPdfUpToDate":true,"isWithdrawnOrRetracted":false,"journal":{"display":true,"email":"[email protected]","identity":"researchsquare","isNatureJournal":false,"hasQc":true,"allowDirectSubmit":true,"externalIdentity":"","sideBox":"","snPcode":"","submissionUrl":"/submission","title":"Research Square","twitterHandle":"researchsquare","acdcEnabled":true,"dfaEnabled":false,"editorialSystem":"","reportingPortfolio":"","inReviewEnabled":false,"inReviewRevisionsEnabled":true},"keywords":"Semantic Segmentation, Transfer Learning, Deep Learning, Hybrid U-Net, Tuberculosis, accuracy","lastPublishedDoi":"10.21203/rs.3.rs-6759181/v1","lastPublishedDoiUrl":"https://doi.org/10.21203/rs.3.rs-6759181/v1","license":{"name":"CC BY 4.0","url":"https://creativecommons.org/licenses/by/4.0/"},"manuscriptAbstract":"\u003cp\u003eTuberculosis (TB) stands as the foremost global cause of mortality, a highly contagious lung ailment trailed closely by malaria and HIV/AIDS. To facilitate precise lung X-ray image analysis, which is crucial for diagnoses like lung tuberculosis, lung X-ray image segmentation takes precedence. The formidable U-net architecture, renowned in deep learning for image segmentation, is prominent in this endeavor. This architectural marvel comprises a contracting pathway, adept at extracting high-level information, and a symmetrically expanding pathway, adept at restoring vital details. Setting itself apart, this network outshines many counterparts and exhibits the capacity for comprehensive training even with a limited dataset.\u003c/p\u003e\n\u003cp\u003eIn this context, the primary goal of our work is to provide an automated lung segmentation method aimed at addressing the challenge of reconstructing damaged lung sections, which makes a significant contribution to our field of medical science and in the application of artificial intelligence that automatically segments an image of the lung to aid TB detection and classification. The proposed approach can be distilled into three fundamental steps: (a)Image Acquisition: This initial step involves describing the materials and techniques employed for image collection. (b)Initial Segmentation: This critical phase utilizes the power of the U-net deep convolutional network (CNN) model and employs three distinct approaches. These approaches play a pivotal role in the initial segmentation of lung regions. (c) Ensemble Modeling: Subsequently, all three models are amalgamated through ensemble modeling. This consolidation process combines the best outputs from all three right at every pixel in each of the three approaches to yield a final result.\u003c/p\u003e\n\u003cp\u003eIn summary, our focus is achieving accurate lung segmentation, particularly for damaged sections, using a comprehensive method that leverages semantic segmentation, transfer learning, and deep learning techniques, notably the Hybrid U-Net model. This approach also enhances lung nodule detection, making it a valuable contribution to the field. Three transfer learning methods trained on large-size image datasets are ResNet34, Inception V3, and VGG 16. The results demonstrated impressive performance metrics, including Mean IoU, Dice-Score, and F-Score. The result section shows that the hybrid U-net model with inception V3 gives better results than the rest, with 0.975 Mean IoU, 0.987 Dice Score, and 0.9963 F-Score. After the ensemble method improved, the result of (0.976 )IoU and (0.988) Dice-Score was achieved.\u003c/p\u003e","manuscriptTitle":"Enhancing Lung Image Segmentation Using Hybrid U-Net and Transfer Learning: A Comprehensive Approach on Tuberculosis Diagnosis","msid":"","msnumber":"","nonDraftVersions":[{"code":1,"date":"2025-05-30 04:40:56","doi":"10.21203/rs.3.rs-6759181/v1","editorialEvents":[{"type":"communityComments","content":0}],"status":"published","journal":{"display":true,"email":"[email protected]","identity":"researchsquare","isNatureJournal":false,"hasQc":true,"allowDirectSubmit":true,"externalIdentity":"","sideBox":"","snPcode":"","submissionUrl":"/submission","title":"Research Square","twitterHandle":"researchsquare","acdcEnabled":true,"dfaEnabled":false,"editorialSystem":"","reportingPortfolio":"","inReviewEnabled":false,"inReviewRevisionsEnabled":true}}],"origin":"","ownerIdentity":"b4008797-a9de-4835-ba3b-a14178505b36","owner":[],"postedDate":"May 30th, 2025","published":true,"recentEditorialEvents":[],"rejectedJournal":[],"revision":"","amendment":"","status":"posted","subjectAreas":[{"id":49108623,"name":"Health Economics \u0026 Outcomes Research"},{"id":49108624,"name":"Artificial Intelligence and Machine Learning"},{"id":49108625,"name":"Computer Architecture and Engineering"}],"tags":[],"updatedAt":"2025-05-30T04:40:56+00:00","versionOfRecord":[],"versionCreatedAt":"2025-05-30 04:40:56","video":"","vorDoi":"","vorDoiUrl":"","workflowStages":[]},"version":"v1","identity":"rs-6759181","journalConfig":"researchsquare"},"__N_SSP":true},"page":"/article/[identity]/[[...version]]","query":{"redirect":"/article/rs-6759181","identity":"rs-6759181","version":["v1"]},"buildId":"8U1c8b4HqxoKbykW_rLl7","isFallback":false,"isExperimentalCompile":false,"dynamicIds":[84888],"gssp":true,"scriptLoader":[]}