Tuning a SAM-Based Model with Multi-Cognitive Visual Adapter to Remote Sensing Instance Segmentation

Linghao Zheng,Xinyang Pu,Feng Xu
2024-08-16
Abstract:The Segment Anything Model (SAM), a foundational model designed for promptable segmentation tasks, demonstrates exceptional generalization capabilities, making it highly promising for natural scene image segmentation. However, SAM's lack of pretraining on massive remote sensing images and its interactive structure limit its automatic mask prediction capabilities. In this paper, a Multi-Cognitive SAM-Based Instance Segmentation Model (MC-SAM SEG) is introduced to employ SAM on remote sensing domain. The SAM-Mona encoder utilizing the Multi-cognitive Visual Adapter (Mona) is conducted to facilitate SAM's transfer learning in remote sensing applications. The proposed method named MC-SAM SEG extracts high-quality features by fine-tuning the SAM-Mona encoder along with a feature aggregator. Subsequently, a pixel decoder and transformer decoder are designed for prompt-free mask generation and instance classification. The comprehensive experiments are conducted on the HRSID and WHU datasets for instance segmentation tasks on Synthetic Aperture Radar (SAR) images and optical remote sensing images respectively. The evaluation results indicate the proposed method surpasses other deep learning algorithms and verify its effectiveness and generalization.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
### The Problem Addressed by This Paper This paper aims to address the issue of instance segmentation in remote sensing images based on the Segment Anything Model (SAM). Specifically, the paper proposes a Multi-Cognitive SAM-based instance segmentation model (MC-SAM SEG) to overcome the following problems: 1. **Insufficient Pre-training Dataset**: The pre-training dataset SA-1B for SAM lacks diversity in large-scale remote sensing images, resulting in poor performance when dealing with remote sensing images that have unique scattering characteristics, dense instance layouts, and multi-scale variations. 2. **Interactive Structure Limitation**: SAM relies on high-quality prompt inputs to exclude irrelevant features, making it unsuitable for automatic image interpretation. Therefore, SAM is typically used for interactive segmentation tasks that require user-provided prompt inputs. 3. **Lack of Classification Capability**: The segmentation masks generated by SAM lack category information and cannot classify specific targets. To address these issues, the paper introduces a Multi-Cognitive Visual Adapter (Mona) into SAM's image encoder and achieves high-quality transfer through Parameter-Efficient Fine-Tuning (PEFT) techniques. Additionally, the proposed MC-SAM SEG algorithm can automatically generate masks and predict the category of each instance without using prompt inputs, thereby improving the performance and generalization ability of instance segmentation in remote sensing images.