Quantitative regularization in robust vision transformer for remote sensing image classification

Huaxiang Song,Yuxuan Yuan,Zhiwei Ouyang,Yu Yang,Hui Xiang
DOI: https://doi.org/10.1111/phor.12489
2024-04-26
The Photogrammetric Record
Abstract:Qualitative regularization consistently transforms remote sensing images (RSIs) from standard imaging conditions into non‐standard conditions. This substantial change in data distribution statistics within datasets reduces the accuracy of Vision Transformers (ViTs). This study introduces the quantitative regularization (QR) strategy, which effectively enhances the classification capabilities of open‐source ViTs for RSIs. Importantly, QR does not require any structural modifications to the models, thereby significantly lowering the entry barriers for geoscience researchers who aim to utilise ViT for RSI applications. Vision Transformers (ViTs) are exceptional at vision tasks. However, when applied to remote sensing images (RSIs), existing methods often necessitate extensive modifications of ViTs to rival convolutional neural networks (CNNs). This requirement significantly impedes the application of ViTs in geosciences, particularly for researchers who lack the time for comprehensive model redesign. To address this issue, we introduce the concept of quantitative regularization (QR), designed to enhance the performance of ViTs in RSI classification. QR represents an effective algorithm that adeptly manages domain discrepancies in RSIs and can be integrated with any ViTs in transfer learning. We evaluated the effectiveness of QR using three ViT architectures: vanilla ViT, Swin‐ViT and Next‐ViT, on four datasets: AID30, NWPU45, AFGR50 and UCM21. The results reveal that our Next‐ViT model surpasses 39 other advanced methods published in the past 3 years, maintaining robust performance even with a limited number of training samples. We also discovered that our ViT and Swin‐ViT achieve significantly higher accuracy and robustness compared to other methods using the same backbone. Our findings confirm that ViTs can be as effective as CNNs for RSI classification, regardless of the dataset size. Our approach exclusively employs open‐source ViTs and easily accessible training strategies. Consequently, we believe that our method can significantly lower the barriers for geoscience researchers intending to use ViT for RSI applications.
geosciences, multidisciplinary,geography, physical,remote sensing,imaging science & photographic technology
What problem does this paper attempt to address?