Source-linked AI summary

BDANet: Multiscale Convolutional Neural Network with Cross-directional Attention for Building Damage Assessment from Satellite Images

Yu Shen, Sijie Zhu, Taojiannan Yang, Chen Chen, Delu Pan, Jianyu Chen, Liang Xiao, Qian Du

arXiv:2105.07364v1cs.CVcs.LGeess.IV

TL;DR

Satellite-based building damage assessment matters for time-critical disaster response, but paired-image methods must model pre- and post-disaster correlations. BDANet combines building segmentation and damage assessment in a two-stage network, achieving state-of-the-art xBD performance and improving F1d over its vanilla network.

  • Problem

    Existing methods often concatenate paired pre- and post-disaster images without modeling their correlations, while damage assessment must predict building damage levels from those images.

    Source: §I · p. 1
  • Method

    BDANet first segments buildings from pre-disaster imagery, then assesses damage with separate pre- and post-disaster branches, multi-scale fusion, cross-directional attention, and targeted CutMix.

    Source: §I · p. 2
  • Results

    BDANet achieves state-of-the-art performance on xBD, with F1d of 0.782 versus 0.757 for the vanilla network.

    Source: §I · p. 2, C. Results Analysis · p. 8
  • Takeaways & Limitations

    Automated satellite-image information extraction addresses time-critical disaster-response needs where expert analysis over large areas may be time-consuming.

    Source: §I · p. 1

Abstract

from arXiv · show

Fast and effective responses are required when a natural disaster (e.g., earthquake, hurricane, etc.) strikes. Building damage assessment from satellite imagery is critical before relief effort is deployed. With a pair of pre- and post-disaster satellite images, building damage assessment aims at predicting the extent of damage to buildings. With the powerful ability of feature representation, deep neural networks have been successfully applied to building damage assessment. Most existing works simply concatenate pre- and post-disaster images as input of a deep neural network without considering their correlations. In this paper, we propose a novel two-stage convolutional neural network for Building Damage Assessment, called BDANet. In the first stage, a U-Net is used to extract the locations of buildings. Then the network weights from the first stage are shared in the second stage for building damage assessment. In the second stage, a two-branch multi-scale U-Net is employed as backbone, where pre- and post-disaster images are fed into the network separately. A cross-directional attention module is proposed to explore the correlations between pre- and post-disaster images. Moreover, CutMix data augmentation is exploited to tackle the challenge of difficult classes. The proposed method achieves state-of-the-art performance on a large-scale dataset -- xBD. The code is available at https://github.com/ShaneShen/BDANet-Building-Damage-Assessment.

I. INTRODUCTION

The paper addresses the need for timely satellite-image damage assessment and the difficulty of using paired pre- and post-disaster imagery effectively. It proposes a two-stage framework with multi-scale fusion, cross-directional attention, and targeted CutMix, reporting state-of-the-art performance on xBD.

  • Automated satellite-image information extraction is motivated by time-critical disaster response, since expert analysis can be time-consuming over large areas.Source: §I · p. 1
  • Post-disaster images alone may lack precise contours of intact buildings, while damage assessment must classify damage levels and include undamaged buildings.Unlike change detection, building damage assessment also requires identifying unchanged buildings.Source: §I · p. 1
  • About 24.2% of minor-damage samples were misclassified as no damage by the ResNet-50 baseline on xBD, illustrating the visual similarity of difficult classes.Source: §I · p. 2
  • The two-stage framework first segments buildings, then assesses damage using a two-branch multi-scale CNN that processes pre- and post-disaster images separately.Building segmentation guides the locations assessed in Stage 2, which reuses network weights from Stage 1.Source: §I · p. 2
  • The proposed MFF module aggregates features across scales, while CDA uses channel-wise and spatial-wise correlations to fuse pre- and post-disaster features.CutMix is applied selectively to difficult classes; experiments report better classification performance than applying CutMix to all classes.Source: §I · p. 2
  • BDANet achieves state-of-the-art performance on the large-scale xBD building-damage-assessment benchmark.Source: §I · p. 2

II. RELATED WORK

Prior work spans post-disaster-only and paired-image damage assessment, including both binary and multiple damage-level classification. U-Net's multiscale feature aggregation and skip connections have made it useful in segmentation, but the paper notes it cannot be directly applied to building damage assessment.

A. Building Damage Assessment

  • Post-disaster-only assessment methods include SVM and object-based machine-learning approaches,, but may lack precise contours of intact buildings; paired imagery has also been used to extract textural and structural features.Source: A. Building Damage Assessment · p. 3
  • Some prior work simplified assessment to binary damage/no-damage labels, while later approaches considered multiple levels, including minor and major damage,.Source: A. Building Damage Assessment · p. 3

B. U-Net

  • U-Net combines downsampling and upsampling to aggregate features at multiple scales, with skip connections integrating low- and high-level features.Source: B. U-Net · p. 3
  • Although U-Net variants have been used for remote-sensing segmentation and related tasks,,, the paper says the architecture cannot be directly used for building damage assessment.Source: B. U-Net · p. 3

C. Attention-based Models

The related work describes attention methods for remote-sensing tasks and notes their limited application to building damage assessment. The paper then presents BDANet's two-stage design, alongside CutMix as an augmentation strategy for difficult classes.

C. Attention-based Models

  • Attention methods have been used in remote-sensing classification and segmentation,, but the cited non-local attention approach to paired-image damage assessment has high computational cost on high-resolution features [13],.Source: C. Attention-based Models · p. 3

D. Data Augmentation Strategies

III. PROPOSED METHOD

  • BDANet uses pre-disaster imagery in Stage 1 to produce building masks, then feeds pre- and post-disaster images into separate Stage 2 branches with MFF and CDA modules.Source: §III · p. 4

A. Two-stage Framework

Building damage assessment must classify damage levels and detect undamaged buildings, beyond identifying changes. BDANet first locates buildings from pre-disaster imagery, then assesses damage using paired-image branches initialized with Stage 1 weights.

A. Two-stage Framework

  • A two-stage framework separates building localization from damage assessment, addressing the need to classify damage levels and detect undamaged buildings.The paper notes that change-detection networks focus on image differences and may ignore unchanged objects such as undamaged buildings.Source: A. Two-stage Framework · p. 4
  • Stage 1 uses pre-disaster imagery in a U-Net to produce a binary building-distribution map for guiding the assessment stage.The paper motivates this stage by noting that damaged post-disaster buildings make simultaneous segmentation and damage assessment difficult.Source: A. Two-stage Framework · pp. 4–5
  • Stage 2 feeds pre- and post-disaster images into two branches and initializes their shared convolutional kernels with weights from Stage 1.The resulting segmentation predictions identify building locations, while the damage output is guided by those locations.Source: A. Two-stage Framework · p. 5

B. Multi-scale Feature Fusion Module

BDANet fuses encoder features from three image scales to represent buildings of varying sizes. Its cross-directional attention module aggregates channel and spatial information across the pre- and post-disaster branches.

B. Multi-scale Feature Fusion Module

  • Three image resolutions—1×, 0.5×, and 0.25×—feed separate streams whose features are embedded at encoder layers for multi-scale representation.The 1× and 0.5× stream features are concatenated and dimension-reduced; the 0.25× stream is concatenated with a later ResNet block.Source: B. Multi-scale Feature Fusion Module · p. 5

C. Cross-directional Attention Module

  • CDA cross-aggregates channel and spatial information from pre- and post-disaster features to recalibrate both branches.Channel feature fusion feeds into spatial feature fusion; the module uses convolution and matrix operations.Source: C. Cross-directional Attention Module · pp. 5–6

D. Data Augmentation with CutMix

CutMix addresses visually similar damage classes by adding patches from difficult-class samples to randomly selected training images. The authors state that the increased sample size encourages more robust representations.

D. Data Augmentation with CutMix

  • The authors use CutMix to increase difficult-class sample sizes, aiming to improve feature representations and encourage attention to those classes.The motivating challenge is visual similarity among damage classes.Source: D. Data Augmentation with CutMix · p. 6
  • CutMix takes patches from difficult-class samples—minor and major damage in xBD—and pastes them into randomly selected sample images.The augmented image pairs and labels are generated using a binary mask.Source: D. Data Augmentation with CutMix · p. 6

A. Dataset and Evaluation

The experiments use xBD, a large-scale dataset of paired pre- and post-disaster satellite images with building and damage annotations. Performance is assessed separately for building segmentation and damage classification, with an overall score combining both.

A. Dataset and Evaluation

  • xBD covers 19 disasters and more than 800,000 building annotations, with paired 1024 × 1024 RGB images at 0.8 meter per pixel.Damage labels comprise no damage, minor damage, major damage, and destroyed; the distribution is highly skewed toward no damage.Source: A. Dataset and Evaluation · pp. 6–7
  • Building segmentation is evaluated with F1b, while damage assessment uses F1d, the harmonic mean of class-wise damage-classification F1 scores.F1b uses true-positive, false-positive, and false-negative pixels; F1s is a weighted average of F1b and F1d.Source: A. Dataset and Evaluation · p. 7

B. Implementation Details

The implementation uses a shared ResNet-50 backbone setup and evaluates BDANet against comparison networks on xBD. Results report advantages for the full framework and show that its Stage 1 segmentation outputs can guide damage localization.

B. Implementation Details

  • Training uses 512 × 512 crops, ResNet-50 for all compared networks, cross-entropy loss, and Adam; the segmentation and damage stages run for 120 and 25 epochs.Testing predictions use the original 1024 × 1024 image size.Source: B. Implementation Details · p. 7

C. Results Analysis

  • The vanilla framework exceeds RescueNet and Weber’s method on F1s, F1b, and F1d, with over 2% improvement in F1b and minor-damage F1 rising from 0.493 to 0.616.The authors attribute the segmentation advantage in part to the decoder and skip connections.Source: C. Results Analysis · p. 7
  • 0.782 vs 0.757 in F1d: the full framework improves on the vanilla network, with gains across damage levels and higher F1s than RescueNet and Weber’s method.The comparison considers building segmentation and damage assessment together; Table V reports xBD results.Source: Table V · p. 8, C. Results Analysis · p. 8
  • Stage 1 segmentation results are reported to match ground-truth maps well and guide building locations for Stage 2 damage assessment.Figure 5 orders pre-disaster images, segmentation ground truth, and Stage 1 results by row.Source: C. Results Analysis · p. 8, Figure 5 · p. 8
  • In visual comparisons, BDANet has fewer damage-assessment errors overall; minor and major damage remain difficult classes for the compared methods.Figure 6 places ground truth beside FCN, SegNet, DeepLab, and BDANet predictions.Source: C. Results Analysis · p. 8

D. Ablation Study

Ablations examine the effects of multi-scale feature fusion, cross-directional attention, and CutMix. The reported gains vary with feature scale, attention placement, and which damage classes receive augmentation.

D. Ablation Study

  • 1.3% in F1s and 2.7% in minor-damage accuracy (0.578 vs 0.605): CutMix improves overall performance, especially when focused on minor and major damage.Applying CutMix to all classes yields barely any improvement, whereas class-specific augmentation produces a significant gain.Source: D. Ablation Study · pp. 8, 11, Table VII · p. 9
  • 0.5% in F1s: using all three MFF streams improves overall damage-assessment performance; the 0.5× and 0.25× streams individually add about 0.4% and 0.2%.The 1× stream captures local detail, while lower-resolution streams capture more global information.Source: D. Ablation Study · p. 8, Table VIII · p. 9, Figure 7 · p. 10
  • About 0.6% improvement over the vanilla network: using all three CDA modules performs better than placing attention only at dconv1 or dconv3.The authors report a better gain at larger-scale dconv3 than at small-scale dconv1; Table IX concerns these positions.Source: Table IX · p. 9, D. Ablation Study · p. 10
  • BDANet’s CDA outperforms the evaluated SE attention variants; SE channel attention alone adds only 0.1% in F1s.The comparison evaluates channel attention, spatial attention, and their combination.Source: D. Ablation Study · p. 10, Table X · p. 10

E. Computational Cost and Training Efficiency

The paper reports the added computational cost of MFF and CDA alongside the effect of Stage 1 initialization on Stage 2 training. The added inference time is described as negligible, while initialization saves training epochs.

E. Computational Cost and Training Efficiency

  • 107.6 vs 92.9 GFLOPs: adding MFF and CDA raises computational cost, while the full framework adds less than 2M parameters and about 60G FLOPs over vanilla.CutMix is applied during preprocessing and adds no network computational resources; added runtime is 0.016 seconds per image.Source: E. Computational Cost and Training… · p. 11
  • At least 5 epochs are saved when Stage 1 weights initialize Stage 2; validation accuracy starts higher and loss lower, with faster convergence.The comparison is reported for Stage 2 fine-tuning against training without Stage 1 initialization.Source: E. Computational Cost and Training… · p. 11

F. Evaluation on Building Change Detection Dataset

On LEVIR-CD, BDANet achieved the best precision, recall, and F1 scores among the compared methods. The conclusion also reports state-of-the-art building damage assessment performance on xBD.

F. Evaluation on Building Change Detection Dataset

  • LEVIR-CD evaluation used 509 image pairs for training and 128 for testing, with Stage 2 alone trained for change detection.The dataset contains 637 high-resolution images from Texas cities, captured from 2002 to 2018.Source: F. Evaluation on Building Change… · p. 11
  • BDANet achieved the best precision, recall, and F1 scores among the compared change-detection methods on LEVIR-CD.The authors also report better visual performance than other methods.Source: F. Evaluation on Building Change… · pp. 11–12

V. CONCLUSION

  • The authors report state-of-the-art building damage assessment performance on xBD and improvements from CutMix on difficult damage levels.They describe MFF as learning features at multiple scales and CDA as exploring correlations between pre- and post-disaster features.Source: §V · p. 12
Loading 2105.07364v1…