A Multi-Modal Unified Representation Learning Framework with Masked Image Modeling for Remote Sensing Images

Dakuan Du,Tianzhu Liu,Yanfeng Gu
DOI: https://doi.org/10.1109/tgrs.2024.3494244
IF: 8.2
2024-01-01
IEEE Transactions on Geoscience and Remote Sensing
Abstract:The coordinated utilization of diverse types of satellite sensors provides a more comprehensive view of the Earth’s surface. However, due to the significant heterogeneity across modalities and the scarcity of high-quality labels, most existing methods face bottlenecks in the underutilization of massive unlabeled multimodal satellite data, making it challenging to understand the scene comprehensively. To this end, we propose a multi-modal unified representation learning framework (MURLF) based on masked image modeling for remote sensing images, aiming to make better use of massive unlabeled multi-modal remote sensing data. MURLF leverages the consistency and complementarity relationships among modalities to extract both common and distinctive features, mitigating the challenges faced by encoders due to significant heterogeneity across various data types. Additionally, MURLF employs multi-level masking independently across different modalities, utilizing visual tokens both within the same modality and across modalities to jointly recover masked pixels as the pretext task, facilitating comprehensive cross-modal information interaction. Furthermore, we design a pre-selected sensor-specific feature extractor (PSFE) to exploit the heterogeneous characteristics of various data sources, thereby extracting discriminative features. By integrating the multi-stage PSFE with the ViT backbone, MURLF can naturally extract multimodal hierarchical representations for downstream tasks, fully preserving valuable information from each modality. The proposed MURLF is not restricted to multi-modal inputs but also supports single-modal inputs during the fine-tuning stage, significantly broadening the framework’s application. Extensive experiments across multiple tasks demonstrate the superiority of the proposed MURLF compared to several advanced multi-modal models. The code will be released soon.
What problem does this paper attempt to address?