Исследователи представили GeoDistill-Refine — двухэтапный фреймворк для обучения компактных моделей сегментации космических аппаратов без участия человека. Метод решает проблему нестабильности базовых моделей сегментации, таких как SAM, при работе с узкоспециализированными изображениями. Система объединяет предсказания по нескольким текстовым промптам и корректирует геометрические ошибки, обеспечивая высокую точность выделения объектов в сложных условиях космоса.
Основная сложность при работе с космическими снимками заключается в отсутствии качественных размеченных датасетов. Использование foundation-моделей позволяет генерировать псевдомаски, однако их качество сильно зависит от формулировки промпта. Новый подход GeoDistill-Refine сначала фокусируется на силуэтах объектов, минимизируя влияние шумов и артефактов, а затем переносит эти знания в компактную нейросеть, пригодную для развертывания на бортовых вычислительных системах.
Технология включает этап геометрической дистилляции, который позволяет сгладить противоречивые предсказания базовой модели. За счет слияния шести фиксированных промптов система формирует более устойчивое представление об объекте, что критически важно для задач навигации и стыковки в космосе, где точность сегментации напрямую влияет на безопасность маневров.
Ключевые факты
- Метод GeoDistill-Refine исключает необходимость ручной разметки данных для обучения моделей сегментации.
- Алгоритм использует двухэтапный процесс: сначала формируются силуэты, затем происходит дистилляция знаний в компактную сеть.
- Система объединяет результаты сегментации по шести различным текстовым промптам для повышения стабильности.
- Решение оптимизировано для задач, где требуется высокая точность выделения контуров космических аппаратов в условиях отсутствия обучающих масок.