Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Advancing Plain Vision Transformer Towards Remote Sensing Foundation Model

About

Large-scale vision foundation models have made significant progress in visual tasks on natural images, with vision transformers being the primary choice due to their good scalability and representation ability. However, large-scale models in remote sensing (RS) have not yet been sufficiently explored. In this paper, we resort to plain vision transformers with about 100 million parameters and make the first attempt to propose large vision models tailored to RS tasks and investigate how such large models perform. To handle the large sizes and objects of arbitrary orientations in RS images, we propose a new rotated varied-size window attention to replace the original full attention in transformers, which can significantly reduce the computational cost and memory footprint while learning better object representation by extracting rich context from the generated diverse windows. Experiments on detection tasks show the superiority of our model over all state-of-the-art models, achieving 81.24% mAP on the DOTA-V1.0 dataset. The results of our models on downstream classification and segmentation tasks also show competitive performance compared to existing advanced methods. Further experiments show the advantages of our models in terms of computational complexity and data efficiency in transferring.

Di Wang, Qiming Zhang, Yufei Xu, Jing Zhang, Bo Du, Dacheng Tao, Liangpei Zhang• 2022

Related benchmarks

TaskDatasetResultRank
Oriented Object DetectionDOTA v1.0 (test)
SV81.55
378
Object DetectionDOTA 1.0 (test)
Plane AP88.97
256
Change DetectionLEVIR-CD
F1 Score90.86
188
Semantic segmentationLoveDA
mIoU53.02
142
Semantic segmentationPotsdam (test)--
104
Scene ClassificationAID TR=50%
Accuracy98.5
94
Scene ClassificationAID TR=20%
Accuracy97.03
93
Oriented Object DetectionDOTA (test)--
92
Semantic segmentationLoveDA (test)
mIoU52.4
81
Semantic segmentationiSAID
mIoU67.2
68
Showing 10 of 33 rows

Other info

Code

Follow for update