HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Scaling Efficient Masked Image Modeling on Large Remote Sensing Dataset

Fengxiang Wang Hongzhen Wang Di Wang Zonghao Guo Zhenyu Zhong Long Lan Jing Zhang Zhiyuan Liu Maosong Sun

Scaling Efficient Masked Image Modeling on Large Remote Sensing Dataset

Abstract

Masked Image Modeling (MIM) has become an essential method for building foundational visual models in remote sensing (RS). However, the limitations in size and diversity of existing RS datasets restrict the ability of MIM methods to learn generalizable representations. Additionally, conventional MIM techniques, which require reconstructing all tokens, introduce unnecessary computational overhead. To address these issues, we present a new pre-training pipeline for RS models, featuring the creation of a large-scale RS dataset and an efficient MIM approach. We curated a high-quality dataset named OpticalRS-13M by collecting publicly available RS datasets and processing them through exclusion, slicing, and deduplication. OpticalRS-13M comprises 13 million optical images covering various RS tasks, such as object detection and pixel segmentation. To enhance efficiency, we propose SelectiveMAE, a pre-training method that dynamically encodes and reconstructs semantically rich patch tokens, thereby reducing the inefficiencies of traditional MIM models caused by redundant background pixels in RS images. Extensive experiments demonstrate that OpticalRS-13M significantly improves classification, detection, and segmentation performance, while SelectiveMAE increases training efficiency over 2 times. This highlights the effectiveness and scalability of our pipeline in developing RS foundational models.

Code Repositories

Fengxiang23/SelectiveMAE
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
object-detection-in-aerial-images-on-diorSelectiveMAE+ViT-B
AP50: 77.80
semantic-segmentation-on-lovedaSelectiveMAE+ViT-L
Category mIoU: 54.31
semantic-segmentation-on-spacenet-1SelectiveMAE+ViT-B
Mean IoU: 79.50

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Scaling Efficient Masked Image Modeling on Large Remote Sensing Dataset | Papers | HyperAI