PromptSeg: Prompt for Universal Remote Sensing Semantic Segmentation

Semantic segmentation in Remote Sensing (RS) involves classifying each pixel of an image into predefined categories. Despite existing deep learning-based methods having significantly advanced semantic segmentation performance, two key challenges remain: (1) these methods are task-specific and exhibit limited generalization across different RS scenario tasks, and (2) they require prior knowledge of the task types to select the appropriate model. To address the above challenges, we propose PromptSeg, a prompt learning method for universal RS semantic segmentation based on the Segment Anything Model (SAM). Specifically, (1) to achieve efficient adaptation to diverse RS tasks with minimal fine-tuning, PromptSeg leverages the unparalleled generalization capabilities of SAM. (2) we propose task-specific prompts that encode essential image features and key distinctions to reduce the reliance on prior knowledge. These task-specific prompts consist of two implicit prompts and one explicit prompt: the Task-related Implicit Prompt (TI-Prompt), which learns task scenario types; the Feature-aware Implicit Prompt (FI-Prompt), which captures essential image features; and the Explicit Prompt (E-Prompt), which facilitates accurate boundary identification and delineation. To effectively integrate these prompts, we introduce the Prompt Learning Module (PLM), which coordinates their fusion to enhance segmentation performance. Extensive experiments demonstrate the superior performance of PromptSeg across various scenarios, including road extraction, field extraction, cloud segmentation, and building segmentation.