不用训练,用检索增强实现更准的开放词汇语义分割
SegRAG: Training-Free Retrieval-Augmented Semantic Segmentation

- 从特征库中提取类别专属点提示,不依赖模型微调
- 在LVIS上最高提升3.92 mIoU,农业数据集平均精度翻倍
- 适合零样本迁移场景,尤其对罕见类表现显著提升
开放词汇分割模型如SAM3虽能通过文本提示覆盖广泛类别,但在目标类别视觉代表性不足或偏离典型图像时性能下降——文本提示无法解决空间问题。我们提出SegRAG,一种无需训练的检索增强分割框架,通过定制化点提示将SAM3与一个经筛选的DINOv3特征库对齐。离线阶段,从标注参考图像中提取密集的块级描述符,并通过类内一致性蒸馏(ICCD)过滤,仅保留可稳定检索到类内前景的原型。推理时,拓扑相似性定位(TSG)计算与检索原型的余弦相似度图,通过连通域分析识别高置信度区域,再用非极大值抑制提取峰值点。这些点提示与类别名称文本共同输入SAM3进行一次前向传播。在四个标准基准上,SegRAG始终优于纯文本基线,在LVIS上最高提升3.92 mIoU;在AgML农业基准上,零样本域迁移下平均IoU从25.27升至59.24(+33.97),部分类别达95以上mIoU。消融实验证明ICCD、TSG和联合提示各自独立贡献,组合后效果叠加。代码已开源。
原文摘要 · Abstract (English)
Open-vocabulary segmentation models such as SAM3 perform well across broad categories via text prompting, yet degrade when target classes are visually underrepresented in pretraining or depart from canonical depictions-limitations text prompts cannot resolve spatially. We present SegRAG, a training-free retrieval-augmented segmentation framework that grounds SAM3 with class-specific point prompts derived from a curated DINOv3 feature bank. Offline, dense patch-level descriptors are extracted from annotated references and filtered by Intra-Class Cohesion Distillation (ICCD), retaining only prototypes that reliably retrieve within-class foreground. At inference, Topographic Similarity Grounding (TSG) computes a cosine-similarity landscape against retrieved prototypes, identifies coherent high-confidence regions via connected-component analysis, and extracts peak locations through non-maximum suppression. The resulting point prompts are delivered jointly with class-name text in a single SAM3 forward pass. On four standard benchmarks, SegRAG consistently outperforms the text-only baseline, gaining up to +3.92 mIoU on LVIS. On AgML agricultural benchmarks under zero-shot domain transfer, it raises mean IoU from 25.27 to 59.24 (+33.97) and recovers individual classes from zero to over 95 mIoU. Ablations confirm that ICCD, TSG, and joint prompting each contribute independently and compound when combined. Code is available at (https://github.com/boudiafA/SegRAG).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。