Home > Published Issues > 2024 > Volume 15, No. 12, 2024 >
JAIT 2024 Vol.15(12): 1304-1314
doi: 10.12720/jait.15.12.1304-1314

Automated GI Tract Segmentation with U-Net: A Comparative Study of Loss Functions

Bindu Madhavi Tummala 1,*, Subba Reddy Chavva 2, Yallamandaiah S. 3, A. Radhika 4, Dasari Chinna Veeraiah 1, Rishitha Jaladi 1, and Aruna Kumari Peruri 1
1. Department of Computer Science and Engineering, Velagapudi Ramakrishna Siddhartha Engineering College, Vijayawada, India
2. Department of Information Technology, Velagapudi Ramakrishna Siddhartha Engineering College, India
3. Department of Electronics and Communication Engineering, Vignan’s Nirula Institute of Technology and Science for Women, Guntur, India
4. Department of Computer Science and Engineering, SRK Institute of Technology, Vijayawada, India
Email: bindumadhavi@vrsiddhartha.ac.in (B.M.T.); chavvasubbareddy@gmail.com (S.R.C.); yallamandaiah@gmail.com (Y.S.); radhikaankala@gmail.com (A.R.); veeranani222@gmail.com (D.C.V.); rishitha.jaladi03@gmail.com (R.J.); arunakumarip1003@gmail.com (A.K.P.)
*Corresponding author

Manuscript received January 21, 2024; revised March 20, 2024; accepted April 7, 2024; published December 6, 2024.

Abstract—Medical image processing has significantly transformed the landscape of healthcare, particularly in the diagnosis and treatment of various diseases. Gastrointestinal (GI) cancer has emerged as a rapidly growing concern, with an estimated 5 million new cases reported annually. To achieve this precision, healthcare professionals now leverage the cutting-edge Magnetic Resonance Imaging (MRI) modality known as MR-Linacs, which provides a daily view of tumor positions. However, a bottleneck in this process arises during the manual segmentation of healthy organs at risk, such as the stomach and intestines, from the obtained medical images. This task, performed by radiologists, is time-consuming and can significantly prolong treatment times, consequently amplifying the suffering of patients. Thus, automation of the GI tract segmentation helps oncologists seamlessly. Our research proposes a model that automates the segmentation of the GI tract. This study presents a U-Net model that segments the stomach and intestines from MRI scans. The dataset, sourced from the UW-Madison Carbone Cancer Center, comprises RLE-encoded masks for training annotations alongside 16-bit grayscale PNG images. Each case consists of multiple scan slices, split either by time or the entire case. Our approach used various combinations of loss functions on U-Net to improve the accuracy and efficiency of the automated segmentation of the gastrointestinal tract. Our model achieves high accuracy with the Dice+BCE loss function when compared with other loss functions. On the training dataset, the U-Net model with Dice+BCE loss function received the highest dice score of 0.9082 and IOU score of 0.8594. On the validation dataset, the model obtained a dice score of 0.8974 and an Intersection over Union (IoU) score of 0.8181. This research contributes to addressing the challenges associated with manual GI tract segmentation, offering a viable solution through automated segmentation using deep learning techniques.
 
Keywords—Magnetic Resonance Imaging (MRI), combined loss functions, U-Net, UW-Madison Carbone cancer center, gastro-intestinal tract segmentation

Cite: Bindu Madhavi Tummala, Subba Reddy Chavva, Yallamandaiah S., A. Radhika, Dasari Chinna Veeraiah, Rishitha Jaladi, and Aruna Kumari Peruri, "Automated GI Tract Segmentation with U-Net: A Comparative Study of Loss Functions," Journal of Advances in Information Technology, Vol. 15, No. 12, pp. 1304-1314, 2024.

Copyright © 2024 by the authors. This is an open access article distributed under the Creative Commons Attribution License (CC BY-NC-ND 4.0), which permits use, distribution and reproduction in any medium, provided that the article is properly cited, the use is non-commercial and no modifications or adaptations are made.