通过几何边界监督提升航天器分割精度,兼顾性能与轻量化。
GABI: Geometry-Aware Boundary Integration for Spacecraft Segmentation

- 引入距离场辅助头,强化边界处的几何一致性学习
- 在SPARK上提升基准模型5%的平均精度,跨域泛化超基线50%
- 轻量版仅占变压器模型1/10体积,性能差距小于5%
精确分割对自主航天器至关重要,直接影响三维态势感知等下游任务。然而太空环境光照剧烈变化,导致图像外观差异大,阻碍分割方法在不同航天器和场景间的泛化。本文提出GABI,一种轻量级边界感知多任务分割架构,通过在卷积主干网络上增加距离场预测分支,提供密集的几何边界监督,促使网络学习一致的空间结构表征,同时保持低模型复杂度,适合星载感知系统。在SPARK基准测试中,距离场监督使基线模型平均精度提升最高达5%,性能接近基于变压器的模型。泛化实验显示,GABI较基线平均精度提升超过50%。跨域评估中,轻量版GABI在交并比和F1分数上仅比重型变压器模型低5%,体积约为其十分之一;而重型版GABI性能超越变压器模型,仍轻约三分之二。
原文摘要 · Abstract (English)
Accurate segmentation is crucial for autonomous spacecraft, as it directly affects downstream tasks related to 3D situational awareness. The harsh illumination conditions of space, however, produce images with high variability in appearance, hindering the generalization of segmentation approaches across different spacecraft and environments. In this work, we propose GABI, a lightweight boundary-aware multi-task segmentation architecture that augments a convolutional backbone with an auxiliary distance-field prediction head. The distance field provides dense geometric supervision around object boundaries, encouraging the network to learn spatially consistent representations of spacecraft structures while maintaining low model complexity suitable for onboard perception systems. We evaluated GABI against both an established convolutional baseline and a heavier transformer-based architecture. On the SPARK benchmark, distance-field supervision improves the baseline by up to $5\%$ in Average Precision while achieving performance comparable to the transformer models. In generalization experiments, GABI improves Average Precision by more than $50\%$ over the baseline. In cross-domain evaluation, the lightweight GABI variant performs within $5\%$ in IoU and F1-score of the heavier transformer model while being approximately ten times smaller. At the same time, the heavier GABI variant surpasses the transformer architectures while remaining nearly three times lighter.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。