TY - JOUR
T1 - DDU-Net
T2 - Dual-Decoder-U-Net for Road Extraction Using High-Resolution Remote Sensing Images
AU - Wang, Ying
AU - Peng, Yuexing
AU - Li, Wei
AU - Alexandropoulos, George C.
AU - Yu, Junchuan
AU - Ge, Daqing
AU - Xiang, Wei
N1 - Publisher Copyright:
© 1980-2012 IEEE.
PY - 2022
Y1 - 2022
N2 - Extracting roads from high-resolution remote sensing images (HRSIs) is vital in a wide variety of applications, such as autonomous driving, path planning, and road navigation. Due to the long and thin shape as well as the shades induced by vegetation and buildings, small-sized roads are more difficult to discern. In order to improve the reliability and accuracy of small-sized road extraction when roads of multiple sizes coexist in an HRSI, an enhanced deep neural network model termed dual-decoder-U-net (DDU-Net) is proposed in this article. Motivated by the U-Net model, a small decoder is added to form a dual-decoder structure for more detailed features. In addition, we introduce the dilated convolution attention module (DCAM) between the encoder and decoders to increase the receptive field as well as to distill multiscale features through cascading dilated convolution and global average pooling. The convolutional block attention module (CBAM) is also embedded in the parallel dilated convolution and pooling branches to capture more attention-aware features. Extensive experiments are conducted on the Massachusetts Roads dataset with experimental results showing that the proposed model outperforms the state-of-the-art DenseUNet, DeepLabv3+, and D-LinkNet by 6.5%, 3.3%, and 2.1% in the mean intersection over union (mIoU), and by 4%, 4.8%, and 3.1% in the F1 score, respectively. Both ablation and heatmap analysis are presented to validate the effectiveness of the proposed model. Moreover, the designed small decoder and introduced DCAM can be used as a portable module to be embedded in other U-Net-like models with encoder-decoder structure to enhance the road detection performance, especially for small-sized roads. The high portability of the designed module is validated by embedding it in the LinkNet, which greatly improves the road segmentation performance.
AB - Extracting roads from high-resolution remote sensing images (HRSIs) is vital in a wide variety of applications, such as autonomous driving, path planning, and road navigation. Due to the long and thin shape as well as the shades induced by vegetation and buildings, small-sized roads are more difficult to discern. In order to improve the reliability and accuracy of small-sized road extraction when roads of multiple sizes coexist in an HRSI, an enhanced deep neural network model termed dual-decoder-U-net (DDU-Net) is proposed in this article. Motivated by the U-Net model, a small decoder is added to form a dual-decoder structure for more detailed features. In addition, we introduce the dilated convolution attention module (DCAM) between the encoder and decoders to increase the receptive field as well as to distill multiscale features through cascading dilated convolution and global average pooling. The convolutional block attention module (CBAM) is also embedded in the parallel dilated convolution and pooling branches to capture more attention-aware features. Extensive experiments are conducted on the Massachusetts Roads dataset with experimental results showing that the proposed model outperforms the state-of-the-art DenseUNet, DeepLabv3+, and D-LinkNet by 6.5%, 3.3%, and 2.1% in the mean intersection over union (mIoU), and by 4%, 4.8%, and 3.1% in the F1 score, respectively. Both ablation and heatmap analysis are presented to validate the effectiveness of the proposed model. Moreover, the designed small decoder and introduced DCAM can be used as a portable module to be embedded in other U-Net-like models with encoder-decoder structure to enhance the road detection performance, especially for small-sized roads. The high portability of the designed module is validated by embedding it in the LinkNet, which greatly improves the road segmentation performance.
KW - High-resolution remote sensing
KW - U-Net
KW - road extraction
KW - semantic segmentation
UR - http://www.scopus.com/inward/record.url?scp=85136061586&partnerID=8YFLogxK
U2 - 10.1109/TGRS.2022.3197546
DO - 10.1109/TGRS.2022.3197546
M3 - Article
AN - SCOPUS:85136061586
SN - 0196-2892
VL - 60
JO - IEEE Transactions on Geoscience and Remote Sensing
JF - IEEE Transactions on Geoscience and Remote Sensing
M1 - 4412612
ER -