arXiv:2608.07405cs.CVcs.AI2026-08

无需标注,用伪掩码训练出高精度航天器分割模型

GeoDistill-Refine: Silhouette-First Geometry Distillation for Annotation-Free Spacecraft Segmentation

论文配图:GeoDistill-Refine: Silhouette-First Geometry Distillation for Annotation-Free Spacecraft Segmentation
图 1 · 摘自论文原文
  • 先学轮廓再优化几何,利用伪掩码提升分割精度
  • 在航天器数据集上边界准确率提升13.8%,重叠率提高4.56%
  • 适合资源受限场景,模型仅26万参数,推理快至1.1毫秒

基础分割模型可为航天器图像提供无标注的监督信号,但其预测受文本提示影响且存在几何误差,易在知识蒸馏中被放大。本文提出两阶段框架GeoDistill-Refine,将离线SAM 3生成的伪掩码迁移至轻量级分割网络。采用六个固定提示的无权重50%投票机制稳定教师输出。学生模型先学习前景轮廓,再通过符号距离场、骨架和面积目标进行精修。样本级门控机制结合提示一致性、有效提示比例及伪掩码合理性,降低不可靠伪几何的影响。在SpaceSense-Bench HJM锁盒数据集上,相较于普通伪标签学生模型,图像交并比提升0.0456,边界F1提升0.1380。外部评估在SPEED+ Lightbox、Sunlamp域及TANGO数据集上表现优异,区域重叠度竞争性,边界质量或前景精度均有提升。部署的TinyUNet仅含0.263 M参数,RTX 4090上每张图像推理耗时约1.1毫秒,SAM 3伪掩码构建与辅助几何分支仅用于训练。

原文摘要 · Abstract (English)

Foundation segmentation models can provide supervision for spacecraft imagery without manual training masks, but their predictions vary with textual prompts and may contain geometric errors that are amplified during distillation. This paper presents GeoDistill-Refine, a two-stage framework that transfers offline SAM 3 pseudo-masks to a compact segmentation network. Six fixed prompts are fused by an unweighted 50% vote to stabilize the teacher output. The student first learns the foreground silhouette and is then refined with signed-distance-field, skeleton, and area objectives derived from the pseudo-mask. A sample-level gate, computed from prompt agreement, the valid-prompt ratio, and pseudo-mask area plausibility, reduces the influence of unreliable pseudo-geometry. On the SpaceSense-Bench HJM lockbox set, GeoDistill-Refine improves Image IoU and Boundary F1 by 0.0456 and 0.1380, respectively, over a plain pseudo-label student. External evaluations on the SPEED+ Lightbox and Sunlamp domains and on TANGO show competitive regional overlap together with gains in boundary quality or foreground precision. The deployed TinyUNet contains 0.263 M parameters and requires approximately 1.1 ms per image on an RTX 4090; SAM 3 pseudo-mask construction and the auxiliary geometry branches are used only during training.

航天器分割知识蒸馏无标注训练轻量化模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。