arrow
Return

A segment anything model-based geological remote sensing interpretation method with a distributed data-parallel deep learning framework

delete2025-08-14
delete0
delete
OA
AI
X
Xiaohui Huang
W
Wei Han
陈
陈云亮 (Yunliang Chen)
G
Geyong Min
D
Dongmei Yan *
DOI:10.1080/17538947.2025.2542913delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
The interpretation of remote sensing images is pivotal in extracting geological elements of interest. Recent studies using deep learning models often fail to provide accurate boundaries between geological elements due to high interclass similarity and imbalanced data distribution. Furthermore, these models are computationally intensive. Inspired by expert interpretation practices, which involve first delineating boundaries and then identifying semantics, we leverage the vision foundation model and propose a distributed interpretation framework including distributed training and inference phases based on data parallelism in distributed architectures. First, a conventional semantic segmentation model, DeepLabV3, is trained. Subsequently, we integrate the Segment Anything Model (SAM) with the trained model, completing the final mapping during the inference phase. Specifically, the inference model features two branches: one for extracting masks by SAM and another for categorizing geological elements. Then, a semantic voting module combines the information from both branches to determine the category for each mask. Experiments demonstrate that the method enhanced the performance metric by more than 2%. Moreover, the method achieves a 3.85× speedup in training and a 3.95× speedup in inference on a 4-GPU machine. We further exploit simulation experiments of 16 GPUs and achieve up to 15.1× training speedup.
Keywords:
Geological remote sensing
semantic segmentation
SAM
distributed deep learning

Journal

International Journal of Digital Earth cover
International Journal of Digital Earth
IF:
4.9
Papers:
2.0K
Citations:
4.7K

Organization

C
China University of Geosciences
Scholars:
3.7W
Papers: 2.8W
Citations: 4.3W
U
University of Exeter
Scholars:
2.0W
Papers: 2.1W
Citations: 3.6W
researcher View more organizations
Cited Papers

Cited Papers

researcher View more