Home > Research > Publications & Outputs > An Attention-Based U-Net for Detecting Deforest...

Electronic data

  • JAG_accepted

    Accepted author manuscript, 2.98 MB, PDF document

    Available under license: CC BY: Creative Commons Attribution 4.0 International License

Links

Text available via DOI:

View graph of relations

An Attention-Based U-Net for Detecting Deforestation Within Satellite Sensor Imagery

Research output: Contribution to Journal/MagazineJournal articlepeer-review

Published
Article number102685
<mark>Journal publication date</mark>31/03/2022
<mark>Journal</mark>International Journal of Applied Earth Observation and Geoinformation
Volume107
Number of pages9
Publication StatusPublished
Early online date18/01/22
<mark>Original language</mark>English

Abstract

In this paper, we implement and analyse an Attention U-Net deep network for semantic segmentation using Sentinel-2 satellite sensor imagery, for the purpose of detecting deforestation within two forest biomes in South America, the Amazon Rainforest and the Atlantic Forest. The performance of Attention U-Net is compared with U-Net, Residual U-Net, ResNet50-SegNet and FCN32-VGG16 across three different datasets (three-band Amazon, four-band Amazon and Atlantic Forest). Results indicate that Attention U-Net provides the best deforestation masks when tested on each dataset, achieving average pixel-wise F1-scores of 0.9550, 0.9769 and 0.9461 for each dataset, respectively. Mask reproductions from each classifier were also analysed, showing that compared to the ground reference Attention U-Net could detect non-forest polygons more accurately than U-Net and overall it provides the most accurate segmentation of forest/deforest compared with benchmark approaches despite its reduced complexity and training time, thus being the first application of an Attention U-Net to an important deforestation segmentation task. This paper concludes with a brief discussion on the ability of the attention mechanism to offset the reduced complexity of Attention U-Net, as well as ideas for further research into optimising the architecture and applying attention mechanisms into other architectures for deforestation detection. Our code is available at https://github.com/davej23/attention-mechanism-unet.