journal article·2024-02-04Open access

Dual Hybrid Attention Mechanism-Based U-Net for Building Segmentation in Remote Sensing Images

Jingxiong Lei, Xuzhi Liu, Haolang Yang, Zeyu Zeng, Jun Feng

Applied Sciences· Volume 14· Issue 3

Abstract

High-resolution remote sensing images (HRRSI) have important theoretical and practical value in urban planning. However, current segmentation methods often struggle with issues like blurred edges and loss of detailed information due to the intricate backgrounds and rich semantics in high-resolution remote sensing images. To tackle these challenges, this paper proposes an end-to-end attention-based Convolutional Neural Network (CNN) called Double Hybrid Attention U-Net (DHAU-Net). We designed a new Double Hybrid Attention structure consisting of dual-parallel hybrid attention modules to replace the skip connections in U-Net, which can eliminate redundant information interference and enhances the collection and utilization of important shallow features. Comprehensive experiments on the Massachusetts remote sensing building dataset and the Inria aerial image labeling dataset demonstrate that our proposed method achieves effective pixel-level building segmentation in urban remote sensing images by eliminating redundant information interference and making full use of shallow features, and improves the segmentation performance without significant time costs (approximately 15%). The evaluation metrics reveal significant results, with an accuracy rate of 0.9808, precision reaching 0.9300, an F1 score of 0.9112, a mean intersection over union (mIoU) of 0.9088, and a recall rate of 0.8932.
Source: crossref · Quality: baseline

Research context

OpenAlex topics: Advanced Neural Network Applications · Automated Road and Building Extraction · Remote Sensing and LiDAR Applications

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

Affiliations

Geomathematics Key Laboratory of Sichuan Province, Chengdu University of Technology, Chengdu 610059, China

Department of Basic Courses, Officers College of People’s Armed Police Force, Chengdu 610213, China

College of Information and Intelligent Engineering, Yunnan College of Business Management, Kunming 650304, China

Chengdu Gas Group Corporation Ltd., Chengdu 610041, China

Chengdu Municipal Bureau of Statistics, Chengdu 610095, China

College of Mathematics and Physics, Chengdu University of Technology, Chengdu 610059, China

Funding

Sichuan Science and Technology Program · Natural Science Foundation of Sichuan Province · Opening Fund of Geomathematics Key Laboratory of Sichuan Province

References (34)

Show source-provided references
1. Mostafa, E., Li, X., Sadek, M., and Dossou, J.F. (2021). Monitoring and Forecasting of Urban Expansion Using Machine Learning-Based Techniques and Remotely Sensed Data: A Case Study of Gharbia Governorate, Egypt. Remote Sens., 13.10.3390/rs13224498 ↗
2. Detection of Damaged Buildings after an Earthquake with Convolutional Neural Networks in Conjunction with Image Segmentation10.1007/s00371-020-02043-9 ↗
3. Estimating Traffic in Urban Areas from Very-High Resolution Aerial Images10.1080/01431161.2020.1815891 ↗
4. Updating land cover map based on change detection of high-resolution remote sensing images10.1117/1.JRS.15.044507 ↗
5. An Efficient Approach for Automatic Rectangular Building Extraction From Very High Resolution Optical Satellite Imagery10.1109/LGRS.2014.2347332 ↗
6. Segmentation of bright targets using wavelets and adaptive thresholding10.1109/83.931096 ↗
7. Efficient graph-based image segmentation10.1023/B:VISI.0000022288.19776.77 ↗
8. ImageNet Classification with Deep Convolutional Neural Networks
9. Rotation-Insensitive and Context-Augmented Object Detection in Remote Sensing Images10.1109/TGRS.2017.2778300 ↗
10. Xia, G.S., Bai, X., Ding, J., Zhu, Z., Belongie, S., Luo, J., Datcu, M., Pelillo, M., and Zhang, L. (2018, January 18–23). DOTA: A Large-Scale Dataset for Object Detection in Aerial Images. Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA.10.1109/CVPR.2018.00418 ↗
11. Wei, Z., Huang, Y., Chen, Y., Zheng, C., and Gao, J. (2023, January 15–19). A-ESRGAN: Training Real-World Blind Super-Resolution with Attention U-Net Discriminators. Proceedings of the PRICAI 2023: Trends in Artificial Intelligence: 20th Pacific Rim International Conference on Artificial Intelligence, PRICAI 2023, Jakarta, Indonesia. Proceedings, Part III.10.1007/978-981-99-7025-4_2 ↗
12. Rehman, M.U., Ryu, J., Nizami, I.F., and Chong, K.T. (2023). RAAGR2-Net: A brain tumor segmentation network using parallel processing of multiple spatial frames. Comput. Biol. Med., 152.10.1016/j.compbiomed.2022.106426 ↗
13. Ryu, J., Rehman, M.U., Nizami, I.F., and Chong, K.T. (2023). SegR-Net: A deep learning framework with multi-scale feature fusion for robust retinal vessel segmentation. Comput. Biol. Med., 163.10.1016/j.compbiomed.2023.107132 ↗
14. Algorithms for semantic segmentation of multispectral remote sensing imagery using deep learning10.1016/j.isprsjprs.2018.04.014 ↗
15. Depth-Wise Asymmetric Bottleneck with Point-Wise Aggregation Decoder for Real-Time Semantic Segmentation in Urban Scenes10.1109/ACCESS.2020.2971760 ↗
16. Joint Deep Learning for land cover and land use classification10.1016/j.rse.2018.11.014 ↗
17. TransAttUnet: Multi-Level Attention-Guided U-Net with Transformer for Medical Image Segmentation10.1109/TETCI.2023.3309626 ↗
18. RoadNet: Learning to Comprehensively Analyze Road Networks in Complex Urban Scenes from High-Resolution Remotely Sensed Images10.1109/TGRS.2018.2870871 ↗
19. Examining the impacts of future land use/land cover changes on climate in Punjab province, Pakistan: Implications for environmental sustainability and economic growth10.1007/s11356-020-08984-x ↗
20. Long, J., Shelhamer, E., and Darrell, T. (2014). Fully Convolutional Networks for Semantic Segmentation. arXiv.10.1109/CVPR.2015.7298965 ↗
21. Ronneberger, O., Fischer, P., and Brox, T. (2015, January 5–9). U-Net: Convolutional Networks for Biomedical Image Segmentation. Proceedings of the Medical Image Computing and Computer-Assisted Intervention—MICCAI 2015, Munich, Germany.10.1007/978-3-319-24574-4_28 ↗
22. Sun, K., Zhao, Y., Jiang, B., Cheng, T., Xiao, B., Liu, D., Mu, Y., Wang, X., Liu, W., and Wang, J. (2019). High-Resolution Representations for Labeling Pixels and Regions. arXiv.
23. SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation10.1109/TPAMI.2016.2644615 ↗
24. Multiscale feature U-Net for remote sensing image segmentation10.1117/1.JRS.16.016507 ↗
25. DenseU-Net-Based Semantic Segmentation of Small Objects in Urban Remote Sensing Images10.1109/ACCESS.2019.2917952 ↗
26. Oktay, O., Schlemper, J., Folgoc, L.L., Lee, M.C.H., Heinrich, M.P., Misawa, K., Mori, K., McDonagh, S.G., Hammerla, N.Y., and Kainz, B. (2018). Attention U-Net: Learning Where to Look for the Pancreas. arXiv.
27. MACU-Net for Semantic Segmentation of Fine-Resolution Remotely Sensed Images
28. Lu, M., Zhang, Y.X., Du, X., Chen, T., Liu, S., and Lei, T. (2020, January 1–3). Attention-Based DSM Fusion Network for Semantic Segmentation of High-Resolution Remote-Sensing Images. Proceedings of the International Conference on Natural Computation, Fuzzy Systems and Knowledge Discovery, Xi’an, China.10.1007/978-3-030-70665-4_67 ↗
29. Khanh, T.L.B., Dao, D.P., Ho, N.H., Yang, H.J., Baek, E.T., Lee, G., Kim, S.H., and Yoo, S.B. (2020). Enhancing U-Net with Spatial-Channel Attention Gate for Abnormal Tissue Segmentation in Medical Imaging. Appl. Sci., 10.10.3390/app10175729 ↗
30. Multistage Attention ResU-Net for Semantic Segmentation of Fine-Resolution Remote Sensing Images
31. Yu, M., Chen, X., Zhang, W., and Liu, Y. (2022). AGs-Unet: Building Extraction Model for High Resolution Remote Sensing Images based on Attention Gates U Network. Sensors, 22.10.3390/s22082932 ↗
32. HA U-Net: Improved Model for Building Extraction From High Resolution Remote Sensing Imagery10.1109/ACCESS.2021.3097630 ↗
33. Mnih, V. (2013). Machine Learning for Aerial Image Labeling. [Ph.D. Thesis, University of Toronto].
34. Cordts, M., Omran, M., Ramos, S., Rehfeld, T., Enzweiler, M., Benenson, R., Franke, U., Roth, S., and Schiele, B. (2016, January 27–30). The Cityscapes Dataset for semantic urban scene understanding. Proceedings of the Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA.10.1109/CVPR.2016.350 ↗