CBAM UNet+++: Attention mechanism to guide change detection studies of full-scale connected networks

  • role: First author第一作者
  • Affiliation:

    School of Surveying and Mapping Science and Technology, Xi'an University of Science and Technology, Xi'an 710054, China

  • Email:liuying712100@163.com
  • Introduction:E-mailliuying712100@163.com
LIU Ying1,  
  • Affiliation:

    School of Surveying and Mapping Science and Technology, Xi'an University of Science and Technology, Xi'an 710054, China

HE Xue1,  
  • Affiliation:

    National Geographic Information System Engineering Technology Research Center, China University of Geosciences, Wuhan 430000, China

LI Danyang2,  
  • Affiliation:

    School of Surveying and Mapping Science and Technology, Xi'an University of Science and Technology, Xi'an 710054, China

YUE Hui1,  
  • Affiliation:

    School of Surveying and Mapping Science and Technology, Xi'an University of Science and Technology, Xi'an 710054, China

WEI Jiali1

ملخص

Existing change detection networks rely heavily on layer-by-layer convolution for feature extraction. However, the use of this method leads to a loss of information, and it lacks the ability to mine important change features. Therefore, knowing how to effectively suppress the influence of the background and identifying ways to increase the ability of the network to learn salient features and generate recognizable feature information are highly important for change detection tasks. Traditional skip connections lack the ability to obtain change information from a full-scale perspective and perform encoder feature extraction. Thus, a UNet+++ high-resolution remote sensing image change detection network called CBAM UNet+++ combined with a coupled attention mechanism (i.e., a convolutional block attention module [CBAM]) was designed in this research.CBAM UNet+++ is based on the semantic segmentation structure UNet+++. The unique full-scale concatenation operation of UNet+++ effectively fuses the semantic and spatial information from the full-scale perspective to avoid information loss. The basic convolutional unit can be replaced by a residual attention module (Residual Block_CBAM and ResBlock_CBAM) to suppress background effects and enhance the learning ability of the encoder to handle significant features. The residual attention module was validated on two remote sensing image change detection datasets—LEBEDV and LEVIR-CD—involving different high-resolution change regions.The proposed method has the highest accuracy on the LEBEDEV multifeature change dataset, with F1 and OA values of 88.9% and 97.3%, respectively, and the second highest accuracy on the LEVIR-CD building change dataset, with F1 and OA values of 86.7% and 96.8%, respectively. The proposed method can obtain deep semantics in a targeted manner, and its qualitative results are better than those of other benchmark networks.The CBAM UNet+++ method can accurately locate and detect change regions with better detection and accuracy than can the benchmark method. The accuracy results of the two selected datasets were slightly different, but they were not inconsistent. The accuracy of the CBAM UNet+++ model was disrupted by pseudochange information in the building dataset. Future work may focus on the usability of this network for change detection in heterogeneous dual-temporal images to further address the impact of early fusion on change detection accuracy.

مفهوم

remote sensing;change detection;UNet+++;attention mechanism;encoding and decoding

References

  1. 1.
    Chen H and Shi Z W. 2020. A spatial-temporal attention-based method and a new dataset for remote sensing image change detection. Remote Sensing, 12(10): 1662
  2. 2.
    Daudt R C, Saux B L and Boulch A. 2018. Fully convolutional siamese networks for change detection//2018 25th IEEE International Conference on Image Processing (ICIP). Athens: IEEE: 4063-4067
  3. 3.
    Han X, Han L, Li L Z and Li H H. 2022. Building change detection in high-resolution remote-sensing images based on deep learning. Laser and Optoelectronics Progress, 59(10): 1001003
  4. 4.
    Huang H M, Lin L F, Tong R F, Hu H J, Zhang Q W, Iwamoto Y, Han X H, Chen Y W and Wu J. 2020. Unet 3+: a full-scale connected unet for medical image segmentation//ICASSP 2020-2020 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). Barcelona: IEEE: 1055-1059
  5. 5.
    Krizhevsky A, Sutskever I and Hinton G E. 2012. ImageNet classification with deep convolutional neural networks//Proceedings of the 26th International Conference on Neural Information Processing Systems. Lake Tahoe: NIPS: 1106-1114
  6. 6.
    Li C, Tan Y S, Chen W, Luo X, Gao Y M, Jia X G and Wang Z Y. 2020. Attention Unet++: a nested attention-aware U-Net for liver CT image segmentation//2020 IEEE International Conference on Image Processing (ICIP). Abu Dhabi: IEEE: 345-349
  7. 7.
    Liu T F. 2020. Research on Land Cover Change Detection Method Based on Spatial-Spectral Feature Combination Using High Resolution Remote Sensing Image. Xi'an: Xi'an University of Technology
  8. 8.
    Long J, Shelhamer E and Darrell T. 2015. Fully convolutional networks for semantic segmentation//Proceedings of 2015 IEEE Conference on Computer Vision and Pattern Recognition. Boston: IEEE: 3431-3440
  9. 9.
    Oktay O, Schlemper J, Folgoc L L, Lee M, Heinrich M, Misawa K, Mori K, McDonagh S, Hammerla N Y, Kainz B, Glocker B and Rueckert D. 2018. Attention U-Net: learning where to look for the pancreas. arXiv: 1804.03999
  10. 10.
    Peng D F, Zhang Y J and Guan H Y. 2019. End-to-end change detection for high resolution satellite images using improved UNet++. Remote Sensing, 11(11): 1382
  11. 11.
    Ronneberger O, Fischer P and Brox T. 2015. U-Net: convolutional networks for biomedical image segmentation//18th International Conference on Medical Image Computing and Computer-Assisted Intervention. Munich: Springer: 234-241
  12. 12.
    Ruan C, Guo H and An J B. 2021. SAR inshore ship detection algorithm in complex background. Journal of Image and Graphics, 26(5): 1058-1066
  13. 13.
    Selvaraju R R, Cogswell M, Das A, Vedantam R, Parikh D and Batra D. 2017. Grad-CAM: visual explanations from deep networks via gradient-based localization//Proceedings of 2017 IEEE International Conference on Computer Vision. Venice: IEEE: 618-626
  14. 14.
    Simonyan K and Zisserman A. 2015. Very deep convolutional networks for large-scale image recognition//3rd International Conference on Learning Representations. San Diego: ICLR
  15. 15.
    Stramondo S, Bignami C, Chini M, Pierdicca N and Tertulliani A. 2006. Satellite radar and optical remote sensing for earthquake damage detection: results from different case studies. International Journal of Remote Sensing, 27(20): 4433-4447
  16. 16.
    Tian J, Reinartz P, d’Angelo P and Ehlers M. 2013. Region-based automatic building and forest change detection on Cartosat-1 stereo imagery. ISPRS Journal of Photogrammetry and Remote Sensing, 79: 226-239
  17. 17.
    Wang H Y, Zhu Y K, Green B, Adam H, Yuille A and Chen L C. 2020. Axial-DeepLab: stand-alone axial-attention for panoptic segmentation//16th European Conference on Computer Vision. Glasgow: Springer: 108-126
  18. 18.
    Woo S, Park J, Lee J Y and Kweon I S. 2018. CBAM: convolutional block attention module//Proceedings of the 15th European Conference on Computer Vision. Munich: Springer: 3-19
  19. 19.
    Xiang Y. 2020. Change Detection Based on Encoder Decoder for High-Resolution Remote Sensing Image. Xuzhou: China University of Mining and Technology
  20. 20.
    Xu W H, Pei Y J, Gao D L, Zhu J D and Liu Y K. 2021. Mass classification of breast mammogram based on attention mechanism and transfer learning. Laser and Optoelectronics Progress, 58(4): 0410007
  21. 21.
    Yuan Z, Guo H T, Lu J, Lu W and Lin Y Z. 2021. High-resolution remote sensing image change detection technology based on UNet++ and attention mechanism. Journal of Geomatics Science and Technology, 38(2): 155-159
  22. 22.
    Zhang C X, Yue P, Tapete D, Jiang L C, Shangguan B, Huang L and Liu G C. 2020. A deeply supervised image fusion network for change detection in high resolution bi-temporal remote sensing images. ISPRS Journal of Photogrammetry and Remote Sensing, 166: 183-200

قراءة النص الكامل

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website