arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2507.11994cs.CV

SAMST:一种基于SAM伪标签过滤的Transformer框架用于遥感半监督语义分割

SAMST: A Transformer framework based on SAM pseudo label filtering for remote sensing semi-supervised semantic segmentation

Zhejiang University(浙江大学);Zhejiang Dahua Technology Co., Ltd(浙江大华技术有限公司)

Jun Yin, Fei Wu, Yupeng Ren, Jisheng Huang, Qiankun Li, Heng jin, Jianhai Fu, Chanjie Cui

AI总结:

本文提出SAMST,一种基于SAM伪标签过滤的遥感半监督语义分割Transformer框架,通过结合大模型泛化能力与小模型训练效率迭代优化伪标签,提升分割性能。

摘要:

公共遥感数据集常因分辨率变化和土地覆盖类别定义不一致而在通用性上受限。为利用海量未标记遥感数据,我们提出了SAMST,一种半监督语义分割方法。SAMST利用了Segment Anything Model(SAM)在零样本泛化和边界检测方面的优势。SAMST通过两个主要组件迭代优化伪标签:使用标记和伪标记数据进行监督模型自训练,以及基于SAM的伪标签精炼器。该精炼器包含三个模块:用于预处理的阈值过滤模块,提取连通区域并为SAM生成提示的提示生成模块,以及用于最终标签拼接的标签精炼模块。通过结合大模型的泛化能力与小模型的训练效率,SAMST提高了伪标签准确性,从而增强整体模型性能。在Potsdam数据集上的实验验证了SAMST的有效性和可行性,展示了其应对遥感语义分割中标记数据有限挑战的潜力。

英文摘要:

Public remote sensing datasets often face limitations in universality due to resolution variability and inconsistent land cover category definitions. To harness the vast pool of unlabeled remote sensing data, we propose SAMST, a semi-supervised semantic segmentation method. SAMST leverages the strengths of the Segment Anything Model (SAM) in zero-shot generalization and boundary detection. SAMST iteratively refines pseudo-labels through two main components: supervised model self-training using both labeled and pseudo-labeled data, and a SAM-based Pseudo-label Refiner. The Pseudo-label Refiner comprises three modules: a Threshold Filter Module for preprocessing, a Prompt Generation Module for extracting connected regions and generating prompts for SAM, and a Label Refinement Module for final label stitching. By integrating the generalization power of large models with the training efficiency of small models, SAMST improves pseudo-label accuracy, thereby enhancing overall model performance. Experiments on the Potsdam dataset validate the effectiveness and feasibility of SAMST, demonstrating its potential to address the challenges posed by limited labeled data in remote sensing semantic segmentation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑