Medical Image Segmentation via Multi-scale Attention Guided Network
2
조회
0
다운로드
3
인용
0
참고문헌
키워드
초록
Even though deep learning (DL) based methods have been achieving superior performance in medical image segmentation, such methods still have some downsides. First, the use of skip-connections in encoder-decoder architecture like U-Net allows transferring redundant and superfluous low-level features information at multiple scales. Second, prior methods cannot capture long-range dependencies and hence fail to reconstruct the feature maps adeptly. To subdue these problems, we propose an architecture that adaptively captures global correlations from different scales and utilizes the attention mechanism. This approach integrates the local-features at different scales and underlines the essential features by suppressing noises and unwanted information. We evaluate the proposed architecture in the context of medical image segmentation on two different datasets: Kvasir-SEG and nuclei segmentation. Experimental results show that the proposed model yields better accuracy and outperforms previous methods.
Abstract
Even though deep learning (DL) based methods have been achieving superior performance in medical image segmentation, such methods still have some downsides. First, the use of skip-connections in encoder-decoder architecture like U-Net allows transferring redundant and superfluous low-level features information at multiple scales. Second, prior methods cannot capture long-range dependencies and hence fail to reconstruct the feature maps adeptly. To subdue these problems, we propose an architecture that adaptively captures global correlations from different scales and utilizes the attention mechanism. This approach integrates the local-features at different scales and underlines the essential features by suppressing noises and unwanted information. We evaluate the proposed architecture in the context of medical image segmentation on two different datasets: Kvasir-SEG and nuclei segmentation. Experimental results show that the proposed model yields better accuracy and outperforms previous methods.
참고문헌 (26)
- S. Poudel, "Colorectal disease classification using efficiently scaled dilation in convolutional neural network", IEEE Access, 8, pp.99227~99238 (2020)
- Sahadev Poudel, "A Novel Integrated Convolutional Neural Network via Deep Transfer Learning in Colorectal Images", 정보기술융합공학논문지, 9(2), pp.9~21 (2019), IT연구소[DOI]
- 이민경, "컨볼루션 신경망의 신뢰도 규제를 이용한 CT 영상에서 악성 난소 종양 판별 방법", 한국차세대컴퓨팅학회 논문지, 16(2), pp.80~87 (2020), 한국차세대컴퓨팅학회[DOI]
- 마누엘 에우제니오 모로초 카얌셀라, "Application of Image Classification using Machine Learning Technique on Smart Device", 한국차세대컴퓨팅학회 논문지, 14(1), pp.16~26 (2018), 한국차세대컴퓨팅학회
- N. -Q. Nguyen, "Robust boundary segmentation in medical images using a consecutive deep encoder-decoder network", IEEEAccess, 7, pp.33795~33808 (2019)
- C. Cortes, "Support-vector networks", Machine learning, 20(3), pp.273~297 (1995)
- L. Breiman, "Random forests", Machine learning, 45(1), pp.5~32 (2001)
- 이현규, "명암도 조절을 통한 베이스 분류기 기반 영상분할기법", 한국차세대컴퓨팅학회 논문지, 8(5), pp.73~86 (2012), 한국차세대컴퓨팅학회
- C. Sutton, "An introduction to conditional random fields", Foundations and Trends® in Machine Learning, 4(4), pp.267~373 (2012)
- O. Ronneberger, "U-net : Convolutional networks for biomedical image segmentation", International Conference on Medical image computing and computerassisted intervention, pp.234~241 (2015), Springer
- L. -C. Chen, "Deeplab : Semantic image segmentation with deep convolutional nets, atrous convolution, and fully connected crfs", IEEE transactions on pattern analysis and machine intelligence, 40(4), pp.834~848 (2017)
- H. Zhao, "Pyramid scene parsing network", Proceedings of the IEEE conference on computer vision and pattern recognition, pp.2881~2890 (2017)
- L. -C. Chen, "Encoder-decoder with atrous separable convolution for semantic image segmentation", Proceedings of the European conference on computer vision (ECCV), pp.801~818 (2018)
- F. Yu, "Multi-scale context aggregation by dilated convolutions", arXiv preprint arXiv : 1511. 07122, ~ (2015)
- K. Li, "Tell me where to look : Guided attention inference network", Proceedings of the IEEEConference on Computer Vision and Pattern Recognition, pp.9215~9223 (2018)
- L. -C. Chen, "Attention to scale : Scale-aware semantic image segmentation", Proceedings of the IEEE conference on computer vision and pattern recognition, pp.3640~3649 (2016)
- J. Hu, "Squeezeand-excitation networks", Proceedings of the IEEE conference on computer vision and pattern recognition, pp.7132~7141 (2018)
- D. Jha, "Kvasir-seg : A segmented polyp dataset", inInternational Conference on Multimedia Modeling, pp.451~462 (2020), Springer
- J. Schlemper, "Attention gated networks : Learning to leverage salient regions in medical images", Medical image analysis, 53, pp.197~207 (2019)
- O. Oktay, "Attention u-net : Learning where to look for the pancreas", arXiv preprintarXiv : 1804. 03999, ~ (2018)
- Y. Wang, "Deep attentional features for prostate segmentation in ultrasound", inInternational Conference on Medical Image Computing and Computer-Assisted Intervention, pp.523~530 (2018), Springer
- A. Sinha, "Multi-scale self-guided attention for medical image segmentation", IEEE Journal of Biomedical and Health Informatics, June (2020)
- M. Tan, "Efficientnet : Rethinking model scaling for convolutional neural networks", arXiv preprint arXiv:1905.11946, ~ (2019)
- Z. Zhou, "DeepLearning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support", pp.3~11 (2018), Springer
- K. Simonyan, "Very deep convolutional networks for large-scale image recognition", arXiv preprint arXiv:1409.1556, ~ (2014)
- K. He, "Deep residual learning for image recognition", Proceedings of the IEEE conference on computer vision and pattern recognition, pp.770~778 (2016)