Binary Segmentation of Medical Images using U-Net

U-Net 을 이용한 의료 영상 이진 분할

  • Subin Lim (Dept. of Computer Software Engineering, Soonchunhyang University) ;
  • Junseok Kim (Dept. of Computer Software Engineering, Soonchunhyang University) ;
  • Misun Kang (Dept. of Computer Software Engineering, Soonchunhyang University)
  • 임수빈 (순천향대학교 컴퓨터소프트웨어공학과) ;
  • 김준석 (순천향대학교 컴퓨터소프트웨어공학과) ;
  • 강미선 (순천향대학교 컴퓨터소프트웨어공학과)
  • Published : 2024.10.31

Abstract

컴퓨터 비전 및 영상 처리 분야에서 정확하고 신뢰할 수 있는 분할을 달성하는 것은 중요하다. 특히 의료 분야에서는 분할의 부정확성이 부적절한 특징의 추출과 분류를 야기한다. 이 논문에서는 이중 어텐션 메커니즘을 U-Net 아키텍처에 통합하여 이진 영상 분할을 개선하는 방법을 제안한다. 인코더와 디코더 사이의 중요 접합점에 위치 및 채널 어텐션 모듈을 배치하여 기능 표현을 향상시켜, 벤치마크 실험에서 기존 방법보다 우수한 성능을 확인하였다. 접근 방식은 의료 영상에서의 이진 분할 작업에 대한 잠재력을 보여주며, 자동 분할을 통해 의료 전문가의 효율성을 높일 수 있다.

Keywords

Acknowledgement

본 연구는 2024 년 과학기술정보통신부 및 정보통신기획평가원의 SW 중심대학사업의 연구 결과로 수행되었음(2021-0-01399)

References

  1. J. Long, E. Shelhamer, and T. Darrell, "Fully Convolutional Networks for Semantic Segmentation," 2015, arXiv:1411.4038 [cs.CV].
  2. L.-C. Chen, G. Papandreou, I. Kokkinos, K. Murphy, and A. L. Yuille, "DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs," IEEE Transactions on Pattern Analysis and Machine Intelligence, vol. 40, no. 4, pp. 834-848, April 2018. DOI: 10.1109/TPAMI.2017.2699184.
  3. 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.
  4. V. Badrinarayanan, A. Kendall, and R. Cipolla, "SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation," 2016. Available: https://doi.org/10.48550/arXiv.1511.00561
  5. O. Oktay, J. Schlemper, L. L. Folgoc, M. Lee, M. Heinrich, K. Misawa, K. Mori, S. McDonagh, N. Y. Hammerla, B. Kainz, B. Glocker, and D. Rueckert, "Attention U-Net: Learning Where to Look for the Pancreas," arXiv e-prints, Apr 2018, arXiv:1804.03999. DOI: 10.48550/arXiv.1804.03999.
  6. S. Woo, J. Park, J. Lee, and I. S. Kweon, "CBAM: Convolutional Block Attention Module," in Computer Vision - ECCV 2018, edited by V. Ferrari, M. Hebert, C. Sminchisescu, and Y. Weiss, Springer International Publishing, Cham, 2018, pp. 3-19. DOI: 10.1007/978-3-030-01234-2_1.
  7. J. Fu, J. Liu, H. Tian, Y. Li, Y. Bao, Z. Fang, and H. Lu, "Dual attention network for scene segmentation," in Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2019, pp. 3146-3154.