Source-linked AI summary
Remote Sensing Change Detection With Transformers Trained from Scratch
Mubashir Noman, Mustansar Fiaz, Hisham Cholakkal, Sanath Narayan, Rao Muhammad Anwer, Salman Khan, Fahad Shahbaz Khan
TL;DR
Transformer-based change detection commonly depends on pre-training, while direct training on small target datasets is challenging. ScratchFormer uses sparse attention and change-enhanced feature fusion, and reports state-of-the-art results on four public datasets, including a 14.27% IoU gain on DSIFN-CD.
Problem
Transformer-based change detection methods generally rely on ImageNet or another CD dataset for pre-training, and direct training on the target CD dataset performs poorly.
Source: §1 · p. 1Method
ScratchFormer is a Siamese transformer CD approach that uses shuffled sparse attention and per-channel change-enhanced feature fusion.
Source: §1 · pp. 1–2Results
Takeaways & Limitations
The reported results show that ScratchFormer achieves state-of-the-art CD performance on four challenging datasets when trained from scratch.
Source: §6 · p. 8Standard self-attention over dense patches requires large training data for optimal convergence, while pre-training on another CD dataset increases training time.
Source: §2.1 · p. 3
Abstract
from arXiv · showhide
Current transformer-based change detection (CD) approaches either employ a pre-trained model trained on large-scale image classification ImageNet dataset or rely on first pre-training on another CD dataset and then fine-tuning on the target benchmark. This current strategy is driven by the fact that transformers typically require a large amount of training data to learn inductive biases, which is insufficient in standard CD datasets due to their small size. We develop an end-to-end CD approach with transformers that is trained from scratch and yet achieves state-of-the-art performance on four public benchmarks. Instead of using conventional self-attention that struggles to capture inductive biases when trained from scratch, our architecture utilizes a shuffled sparse-attention operation that focuses on selected sparse informative regions to capture the inherent characteristics of the CD data. Moreover, we introduce a change-enhanced feature fusion (CEFF) module to fuse the features from input image pairs by performing a per-channel re-weighting. Our CEFF module aids in enhancing the relevant semantic changes while suppressing the noisy ones. Extensive experiments on four CD datasets reveal the merits of the proposed contributions, achieving gains as high as 14.27\% in intersection-over-union (IoU) score, compared to the best-published results in the literature. Code is available at \url{https://github.com/mustansarfiaz/ScratchFormer}.
1. Introduction
Remote-sensing change detection seeks relevant semantic changes between co-registered satellite images, while excluding noisy variations. ScratchFormer addresses limits of training transformer-based change detection from scratch and reports state-of-the-art results on four public datasets.
1. Introduction
- Change detection identifies relevant changes between co-registered satellite images; CNN-based approaches can struggle to capture long-range dependencies between image regions.The target includes changes in man-made facilities while ignoring noisy changes such as shadows and seasonal or environmental variations.Source: §1 · p. 1
- Existing transformer-based CD methods generally rely on ImageNet or another CD dataset for pre-training, while direct training on the target dataset performs poorly.The introduction attributes this challenge in part to dense self-attention’s quadratic token complexity, slower convergence, and overfitting tendency.Source: §1 · p. 1
- Common two-stream methods fuse features by difference, summation, or concatenation without explicit re-weighting, which may inadequately aggregate semantic changes.ScratchFormer’s CEFF addresses this by performing per-channel re-calibration.Source: §1 · pp. 1–2
- In DSIFN-CD examples, ScratchFormer reduces the false positives and false negatives shown for ChangeFormer [1].The figure marks false positives in purple and false negatives in red.Source: Figure 1 · p. 2
- ScratchFormer combines shuffled sparse attention (SSA), which attends to sparse informative regions, with CEFF, which re-weights channels to enhance semantic changes and suppress noisy ones.The architecture is designed for training directly on the target CD dataset.Source: §1 · p. 2
- On DSIFN-CD, ScratchFormer achieves 90.75% IoU, outperforming the best existing method [1] in the literature by 14.27%, and sets new state-of-the-art performance across all four evaluated datasets.Source: §1 · p. 2
2. Preliminaries
The paper formulates change detection as predicting a binary mask of relevant semantic changes between paired, co-registered satellite images. Its baseline uses a Siamese transformer encoder, feature fusion, and a decoder; standard dense self-attention creates a training-data and convergence challenge.
2. Preliminaries
- Change detection predicts a binary mask of relevant semantic changes between paired images, excluding seasonal, illumination, shadow, and atmospheric variations.The input images are co-registered satellite images acquired at distinct times.Source: §2 · p. 2
2.1. Baseline Change Detection Framework
- The baseline shares transformer-encoder weights across pre- and post-change image streams, fuses multi-scale difference features, and decodes them into a semantic change map.The encoder uses standard self-attention; the decoder upsamples fused features before mask prediction.Source: §2.1 · pp. 2–3
- Standard self-attention over uniformly sampled dense patches requires large training data for optimal convergence; pre-training on another CD dataset adds training time.Figure 2 compares IoU across training epochs for scratch-trained, pre-trained, and proposed approaches; ScratchFormer reaches similar performance to the scratch baseline’s final result with 10% of its training time.Source: §2.1 · p. 3, Figure 2 · p. 3
3. Method
ScratchFormer pairs shuffled sparse attention (SSA), which samples data-dependent feature subsets, with change-enhanced feature fusion (CEFF), which reweights paired features per channel. Its shared-weight, four-stage Siamese encoder and decoder produce a binary change map, addressing the challenges of learning from small CD datasets and distinguishing semantic from noisy changes.
3.1. Motivation
- The method targets two CD challenges: learning useful representations from small datasets and distinguishing relevant semantic changes from noisy ones.The motivation identifies sparse informative regions and explicit per-channel modeling between pre- and post-change features as design priorities.Source: §3.1 · p. 3
3.2. Overall Architecture
- ScratchFormer uses a shared-weight, four-stage Siamese encoder, CEFF to fuse paired features at each stage, and a decoder to predict a binary change map.The architecture computes features at different spatial resolutions; CEFF reweights paired features per channel to retain relevant changes while suppressing noisy ones.Source: §3.2 · p. 3
3.3. Shuffled Sparse Attention Layer
- SSA samples data-dependent feature subsets, performs token mixing on them, then returns attended features to the full-resolution feature map.Learned position offsets guide the sampling; sparse attention maintains a global receptive field while reducing computation by a factor reported as O(γ^2).Source: §3.3 · pp. 4–5
3.4. Change Enhanced Features Fusion Module
- CEFF combines paired features and reweights their channels to emphasize semantic changes and suppress noisy changes.It predicts channel weights for both streams from a pooled feature vector, normalizes them across the streams, and fuses the weighted features by addition.Source: §3.4 · p. 5
4. Experiments
ScratchFormer is evaluated on four change-detection datasets and reports state-of-the-art performance while training from scratch. Ablations assess shuffled sparse attention (SSA), change-enhanced feature fusion (CEFF), and the SSA sparsity setting.
4.1. Experimental Setup
- The experiments use four public datasets, 256 × 256 input patches, and random initialization; training runs for 300 epochs on four NVIDIA A100 GPUs.The setup also describes separate dataset splits, data augmentation, and evaluation using F1, IoU, and overall accuracy.Source: §4.1 · pp. 5–6
4.2. State-of-the-art Comparison
- ScratchFormer sets state-of-the-art on four datasets without pre-training; on DSIFN-CD, it gains 8.48% F1, 2.8% OA, and 14.27% IoU over ChangeFormer [1].The comparison covers DSIFN-CD, LEVIR-CD, WHU-CD, and CDD-CD.Source: Table 1 · p. 6, §4.2 · pp. 6–7
- Qualitative comparisons show clearer semantic-change boundaries on DSIFN-CD and LEVIR-CD, including better localization of changes at multiple scales.The examples compare ScratchFormer with BIT [2] and ChangeFormer [1].Source: §4.2 · p. 7, Figure 4 · p. 7, Figure 5 · p. 7
4.3. Ablation Study
- On DSIFN-CD, IoU rises from 78.11% for the scratch baseline to 83.94% after LEVIR-CD pre-training, 86.62% with SSA, and 90.75% with SSA plus CEFF trained from scratch.The ablation also reports consistent gains from adding SSA and CEFF to the baseline on LEVIR-CD.Source: §4.3 · pp. 7–8, Table 2 · p. 8
- γ = 4 provides the best reported sparsity performance on both DSIFN-CD and LEVIR-CD, so the experiments use that setting throughout.Source: §4.3 · p. 8, Table 3 · p. 8
- CEFF's per-channel feature re-weighting outperforms addition, subtraction, and concatenation on DSFIN-CD.The comparison evaluates the fusion techniques on all reported metrics.Source: §4.3 · p. 8
5. Relation to Prior Art
Prior change-detection work uses CNNs and transformers to capture discriminative features or long-range context. ScratchFormer is distinguished by its SSA and CEFF contributions.
- CNN-based change detection uses approaches such as dual attention, channel attention, and feature pyramids to capture discriminative features or long-range dependencies.Source: §5 · p. 8
- Transformer-based prior work includes BIT, TransUNetCD, multi-scale Swin transformers, hybrid transformers, and hierarchical Siamese transformers [1].Source: §5 · p. 8
- Unlike the described prior approaches, ScratchFormer uses SSA for scratch training and CEFF for channel re-weighting; CEFF outperforms addition, subtraction, and concatenation on DSFIN-CD.Source: §5 · p. 8, Table 4 · p. 8
6. Conclusion
The paper presents ScratchFormer, a Siamese transformer for remote-sensing change detection, with SSA and CEFF. It reports state-of-the-art results on four datasets and points to natural-image and medical-imaging change detection as future directions.
- ScratchFormer is a Siamese transformer architecture for remote-sensing change detection that uses SSA and CEFF.Source: §6 · p. 8
- The authors report state-of-the-art change-detection performance on all four evaluated datasets.Source: §6 · p. 8
- Future work could explore change detection in natural images and medical imaging.Source: §6 · p. 9