(Department of Informatics, Universitas Sebelas Maret, Indonesia)(2) Maulana Daffa Athaullah Yahya
(Department of Informatics, Universitas Sebelas Maret, Indonesia)(3) Umi Salamah
(Department of Informatics, Universitas Sebelas Maret, Indonesia)*corresponding author
AbstractOne of the most common types of cancer is colorectal cancer, which is commonly caused by colon polyps. Early detection via colonoscopy plays a crucial role in preventing the advancement of cancer, even though initial symptoms may not always be noticeable. In this context, computer-assisted methods for image segmentation in colonoscopy significantly aid in polyp detection, reduce the workload on healthcare professionals, and enhance diagnostic precision. A widely used approach in medical image segmentation is the U-Net model, known for its ability to merge information across multiple resolutions through its encoder-decoder structure. Nonetheless, several U-Net-derived architectures struggle to effectively capture deeper features and maintain computational efficiency. This research introduces a modified Double U-Net architecture that integrates MobileNetV2 as its encoder to alleviate computational complexity, while incorporating Depthwise Separable Convolutions (DWSC) and Convolutional Block Attention Modules (CBAM) in the skip connections to enhance both accuracy and efficiency. The proposed model demonstrated notable performance, achieving an F1-score of 95.28% and IoU of 90.84% on the CVC-Clinic DB dataset, and an F1-score of 89.08% with an IoU of 80.69% on the Kvasir-SEG dataset, with a parameter count of 1.96 million. This approach not only increases segmentation precision but also maintains computational efficiency, making it a viable option for detecting colorectal polyps.
KeywordsColorectal Cancer; CBAM; Double U-Net; MobileNetV2; Polyp Segmentation
|
DOIhttps://doi.org/10.26555/ijain.v12i3.1783 |
Article metricsAbstract views : 172 | PDF views : 4 |
Cite |
Full Text Download
|
References
[1] I. Labeda et al., “Colorectal cancer survival rates in Makassar, Eastern Indonesia: A retrospective Cohort Study,” Ann. Med. Surg., vol. 74, Feb. 2022, doi: 10.1016/j.amsu.2021.103211.
[2] F. Bray et al., “Global cancer statistics 2022: GLOBOCAN estimates of incidence and mortality worldwide for 36 cancers in 185 countries,” CA. Cancer J. Clin., vol. 74, no. 3, pp. 229–263, May 2024, doi: 10.3322/caac.21834.
[3] H. P. Irianto and A. S. Putranto, “Correlation between clinical characteristics of colorectal cancer and nutritional status in patients undergoing surgery,” Bali Med. J., vol. 12, no. 1, pp. 1059–1062, Apr. 2023, doi: 10.15562/bmj.v12i1.4057.
[4] E. Rondonotti et al., “Clinical Validation of BASIC Classification for the Resect and Discard Strategy for Diminutive Colorectal Polyps,” Clin. Gastroenterol. Hepatol., vol. 18, no. 10, pp. 2357-2365.e4, Sep. 2020, doi: 10.1016/j.cgh.2019.12.028.
[5] B. C. Bag, H. K. Maity, and C. Koley, “U-Net MobileNetV2: Medical Image Segmentation Using Deep Neural Network (DNN),” J. Mech. Contin. Math. Sci., vol. 18, no. 1, Jan. 2023, doi: 10.26782/jmcms.2023.01.00002.
[6] R. Wang, T. Lei, R. Cui, B. Zhang, H. Meng, and A. K. Nandi, “Medical image segmentation using deep learning: A survey,” IET Image Process., vol. 16, no. 5, pp. 1243–1267, Apr. 2022, doi: 10.1049/ipr2.12419.
[7] A. M. Khan, A. Ashrafee, F. S. Khan, M. B. Hasan, and M. H. Kabir, “AttResDU-Net: Medical Image Segmentation Using Attention-based Residual Double U-Net,” in 2023 International Joint Conference on Neural Networks (IJCNN), Jun. 2023, pp. 1–8, doi: 10.1109/IJCNN54540.2023.10191528.
[8] G. Litjens et al., “A survey on deep learning in medical image analysis,” Med. Image Anal., vol. 42, pp. 60–88, Dec. 2017, doi: 10.1016/j.media.2017.07.005.
[9] D. Jha, M. A. Riegler, D. Johansen, P. Halvorsen, and H. D. Johansen, “DoubleU-Net: A Deep Convolutional Neural Network for Medical Image Segmentation,” in 2020 IEEE 33rd International Symposium on Computer-Based Medical Systems (CBMS), Jul. 2020, vol. 2020-July, pp. 558–564, doi: 10.1109/CBMS49503.2020.00111.
[10] O. Ronneberger, P. Fischer, and T. Brox, “U-net: Convolutional networks for biomedical image segmentation,” in International Conference on Medical image computing and computer-assisted intervention, 2015, pp. 234–241, doi: 10.1007/978-3-319-24574-4_28.
[11] N. Siddique, S. Paheding, C. P. Elkin, and V. Devabhaktuni, “U-Net and Its Variants for Medical Image Segmentation: A Review of Theory and Applications,” IEEE Access, vol. 9, pp. 82031–82057, 2021, doi: 10.1109/ACCESS.2021.3086020.
[12] N. K. Tomar et al., “DDANet: Dual Decoder Attention Network for Automatic Polyp Segmentation,” 2021, pp. 307–314, doi: 10.1007/978-3-030-68793-9_23.
[13] D. Jha et al., “ResUNet++: An Advanced Architecture for Medical Image Segmentation,” in 2019 IEEE International Symposium on Multimedia (ISM), Dec. 2019, pp. 225–2255, doi: 10.1109/ISM46123.2019.00049.
[14] Z. Sultana, M. Foysal, S. Islam, and A. Al Foysal, “Lung Cancer Detection and Classification from Chest CT Images Using an Ensemble Deep Learning Approach,” in 2024 6th International Conference on Electrical Engineering and Information & Communication Technology (ICEEICT), May 2024, pp. 364–369, doi: 10.1109/ICEEICT62016.2024.10534468.
[15] N. Bor, T. T. Sivri, N. P. Akman, A. Berkol, and Y. Ekici, “Breast Cancer Detection Using Various Classification Models Combined with Transfer Learning,” in 2022 International Conference on Artificial Intelligence of Things (ICAIoT), Dec. 2022, pp. 1–6, doi: 10.1109/ICAIoT57170.2022.10121840.
[16] M. Sandler, A. G. Howard, M. Zhu, A. Zhmoginov, and L.-C. Chen, “Inverted Residuals and Linear Bottlenecks: Mobile Networks for Classification, Detection and Segmentation,” CoRR, vol. abs/1801.0, pp. 1–14, 2018, [Online]. Available at: https://arxiv.org/pdf/1801.04381.
[17] A. G. Howard et al., “MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications,” no. April 2017, 2017, [Online]. Available at: http://arxiv.org/abs/1704.04861.
[18] J. Lewis, Y.-J. Cha, and J. Kim, “Dual encoder–decoder-based deep polyp segmentation network for colonoscopy images,” Sci. Rep., vol. 13, no. 1, p. 1183, Jan. 2023, doi: 10.1038/s41598-023-28530-2.
[19] Y. Zhou, X. Kang, F. Ren, H. Lu, S. Nakagawa, and X. Shan, “A multi-attention and depthwise separable convolution network for medical image segmentation,” Neurocomputing, vol. 564, p. 126970, Jan. 2024, doi: 10.1016/j.neucom.2023.126970.
[20] P. Zhao, J. Zhang, W. Fang, and S. Deng, “SCAU-Net: Spatial-Channel Attention U-Net for Gland Segmentation,” Front. Bioeng. Biotechnol., vol. 8, Jul. 2020, doi: 10.3389/fbioe.2020.00670.
[21] T. L. B. Khanh et al., “Enhancing U-Net with Spatial-Channel Attention Gate for Abnormal Tissue Segmentation in Medical Imaging,” Appl. Sci., vol. 10, no. 17, p. 5729, Aug. 2020, doi: 10.3390/app10175729.
[22] J. Bernal, F. J. Sánchez, G. Fernández-Esparrach, D. Gil, C. Rodríguez, and F. Vilariño, “WM-DOVA maps for accurate polyp highlighting in colonoscopy: Validation vs. saliency maps from physicians,” Comput. Med. Imaging Graph., vol. 43, pp. 99–111, Jul. 2015, doi: 10.1016/j.compmedimag.2015.02.007.
[23] D. Jha et al., “Kvasir-SEG: A Segmented Polyp Dataset,” 2020, pp. 451–462. [Online]. Available at: http://link.springer.com/10.1007/978-3-030-37734-2_37.
[24] L. Nanni, S. Brahnam, M. Paci, and S. Ghidoni, “Comparison of Different Convolutional Neural Network Activation Functions and Methods for Building Ensembles for Small to Midsize Medical Data Sets,” Sensors, vol. 22, no. 16, p. 6129, Aug. 2022, doi: 10.3390/s22166129.
[25] S. Woo, J. Park, J.-Y. Lee, and I. S. Kweon, “CBAM: Convolutional Block Attention Module,” in Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), vol. 11211 LNCS, Springer Verlag, 2018, pp. 3–19, doi: 10.1007/978-3-030-01234-2_1.
[26] S. Jahangard, M. Bonyani, and A. Khosravi, “DoubleU-Net++: Architecture with Exploit Multiscale Features for Vertebrae Segmentation,” Electr. Eng. Syst. Sci., pp. 1–23, Jan. 2022, [Online]. Available at: https://arxiv.org/abs/2201.12389.
[27] D. Jha et al., “Real-Time Polyp Detection, Localization and Segmentation in Colonoscopy Using Deep Learning,” IEEE Access, vol. 9, pp. 40496–40510, 2021, doi: 10.1109/ACCESS.2021.3063716.
[28] Z. Zhou, M. M. Rahman Siddiquee, N. Tajbakhsh, and J. Liang, “UNet++: A Nested U-Net Architecture for Medical Image Segmentation,” 2018, pp. 3–11, doi: 10.1007/978-3-030-00889-5_1.
[29] Z. Zhang, Q. Liu, and Y. Wang, “Road Extraction by Deep Residual U-Net,” IEEE Geosci. Remote Sens. Lett., vol. 15, no. 5, pp. 749–753, May 2018, doi: 10.1109/LGRS.2018.2802944.
[30] G. C. Ates, P. Mohan, and E. Celik, “Dual Cross-Attention for medical image segmentation,” Eng. Appl. Artif. Intell., vol. 126, p. 107139, Nov. 2023, doi: 10.1016/j.engappai.2023.107139.
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
___________________________________________________________
International Journal of Advances in Intelligent Informatics
ISSN 2442-6571 (print) | 2548-3161 (online)
Organized by UAD and ASCEE Computer Society
Published by Universitas Ahmad Dahlan
W: http://ijain.org
E: info@ijain.org (paper handling issues)
andri.pranolo.id@ieee.org (publication issues)
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0