基于视觉几何的Transformer与可学习令牌合并技术的顺序航天器姿态估计

《Aerospace Science and Technology》:Sequential Spacecraft Pose Estimation via Visual Geometry Grounded Transformersand Learnable Token Merging

【字体: 大 中 小 】 时间:2026年04月09日 来源:Aerospace Science and Technology 5.8

编辑推荐:

  针对单目航天器姿态估计中计算效率与精度平衡的难题,本文提出融合FastVGGT与混合专家模型(MoE)的框架,创新性地设计可学习令牌合并(L-ToMe)机制,通过动态压缩冗余背景令牌并保留关键几何特征,在瑞士立方体数据集上实现21.6%的精度提升,同时保持线性计算复杂度。

  
Jian Chen|Ziyi He|Fan Wu|Zichen Zhao|Xueqin Chen
哈尔滨工业大学卫星技术研究中心,中国哈尔滨,150006

摘要

单目航天器姿态估计对于自主在轨服务至关重要,但由于太空环境中的强烈光照、高对比度和尺度不确定性,这一任务仍然具有挑战性。尽管基于视觉几何的Transformer(VGGT)在多视图分析方面具有强大的能力,但其高昂的计算成本限制了其性能,因此催生了FastVGGT的发展以解决这一效率问题。在FastVGGT和专家混合(MoE)的基础上,我们提出了一个专为序列航天器姿态估计设计的增强框架。我们的核心创新在于一种新颖的可学习令牌合并(L-ToMe)机制,用于替代标准的静态合并策略。通过明确学习识别和保留显著区域,同时压缩冗余的背景信息并将令牌路由到专家模块,我们的方法显著降低了计算开销,而不会影响几何重建的准确性。在SwissCube数据集上的实验表明,与基线方法相比,我们的方法将姿态估计精度提高了21.6%。这些结果验证了我们增强架构在可靠的单目自主太空导航中的潜力。

引言

精确的六自由度(6-DoF)姿态估计和轨迹跟踪是自主航天器操作的基石,包括在轨服务(OOS)[1]、[2]、主动碎片清除(ADR)[3]以及交会对接(RVD)[4]。虽然像LiDAR [5]和射频(RF)[6]这样的主动传感器能够提供可靠的深度测量,但它们会带来显著的尺寸、重量和功耗(SWaP)问题,使得它们不适用于资源受限的平台。相比之下,服务航天器上的光学传感器提供了轻量级的解决方案,无需依赖协作硬件即可运行。在这种情况下,单目相机因其极低的SWaP和广阔的操作范围而成为主动传感器或立体相机的有吸引力的替代品[7]。单目方法的有效性已在飞行中得到验证。例如,在PRISMA任务中,成功部署了基于单目视觉的传感器来支持自主非合作交会实验[8]。
然而,在轨上的精确单目估计面临着太空环境的极端视觉条件带来的挑战,这些条件包括强烈的光照、高对比度的阴影和固有的尺度不确定性。由于稳健的特征提取是姿态估计的先决条件,传统的姿态估计方法依赖于手工制作的描述符来识别目标体上的关键点、角点或边缘(例如SURF、SIFT和ORB)。然而,这些方法非常脆弱;它们严重依赖于精确的初始化,并且当手工制作的特征被背景噪声干扰时经常失败。最近的研究表明,基于深度学习的方法是必要的[9],显著提高了航天器姿态估计的准确性。在这个领域,方法通常分为两个不同的流派:
  • 1.
    关键点预测:这种方法基于目标的离线3D模型检测预定义的2D地标[10]、[11],然后使用透视n点(PnP)求解器(如高效透视n点(EPnP)[12]或随机样本一致性(RANSAC)将2D地标与相应的3D属性对齐,从而得出6-DoF姿态;
  • 2.
    直接回归:这种方法利用神经网络直接从输入图像中回归6-DoF姿态参数,或者学习一组模板姿态之间的插值权重[13]。
  • 基于卷积神经网络(CNNs)的关键点方法在早期研究中占据主导地位,并在卫星姿态估计挑战(SPEC)[14]、[15]中取得了令人瞩目的性能,这主要是由于PnP过程的几何可解释性。值得注意的是,对PRISMA任务数据的实证评估表明,基于关键点的架构在定向精度上显著优于早期的端到端回归模型,通常高出一个数量级,从而确立了它们在高精度任务中的首选地位。然而,这种两阶段流程的依赖性引入了脆弱性。因为PnP求解器通常作为不可微的后处理步骤应用,这些方法存在一个根本性的限制,即“代理误差”:网络的优化目标(最小化2D地标误差)与最终任务目标(最小化6-DoF姿态误差)并不完全一致。此外,PnP求解器对异常关键点非常敏感,在噪声成像条件下会导致显著的方差[16]。
    这种差异需要结合端到端学习和显式几何推理的优势的架构。受多任务学习[13]、[17]进展的推动,最近的研究进一步探索了利用几何基础的混合方法。通过强制执行类似于PnP的显式几何约束,同时保持直接回归的可微性,这些方法在“黑盒”预测和解耦的几何优化之间架起了一座桥梁,为高精度姿态估计提供了稳健的路径。
    最近,将几何先验集成到预训练的视觉Transformer(ViTs)[18]、[19]中,例如基于视觉几何的Transformer(VGGT)[20],在3D表示学习方面树立了新的基准,其性能超过了最先进的模型。从根本上说,VGGT扩展了自注意力机制,以处理多帧、无序的图像序列,证明了在前置几何监督下训练的Feed-forward Transformer可以获得稳健的几何表示。此外,VGGT表明,通过密集预测Transformer(DPT)[21]预测几何特征(如深度图或点云)可以提高直接回归的准确性。然而,标准的自注意力机制相对于序列长度会产生二次方的计算成本。这种低效率主要限制了VGGT在长图像序列的离线处理上应用,使其不适合实时在轨应用。
    为了缓解这些可扩展性限制,FastVGGT代表了高效VGGT架构的进步[22]。它通过引入基于令牌合并[23]的加速范式来解决密集视觉输入中的冗余问题。其核心机制采用了一种混合分区策略:它明确保留了通过固定间隔采样识别的显著令牌,同时利用随机分组策略来聚合剩余的令牌。通过有效地修剪注意力层中的冗余,FastVGGT将计算复杂度从二次方(O(N2< />)降低到接近线性(O(N< />),从而显著提高了吞吐量和减少了内存使用。这一架构演变从理论上证明了可以使用稀疏令牌表示来实现高保真度的几何重建。
    然而,FastVGGT的固定间隔采样策略对所有非参考令牌一视同仁,无论其语义内容如何都予以丢弃。在航天器场景中,目标只占视野的一小部分,周围是均匀的空白区域,这种静态方法可能会移除卫星边界或对姿态精度至关重要的结构特征的令牌。我们的可学习令牌合并(L-ToMe)通过学习评估令牌的显著性来解决这一限制,明确保留几何上重要的区域,同时积极压缩无信息的背景。
    此外,基于专家混合(MoE)[24]的稀疏激活架构为在不增加计算成本的情况下增强模型容量提供了有希望的方向。通过将每个输入令牌路由到一小部分专家模块,MoE能够将计算资源分配到输入中最具信息量的区域,从而在表示丰富性和推理效率之间实现平衡。这一特性对于航天器姿态估计特别有益,因为在这些场景中,视觉输入通常由广阔的均匀背景区域和结构复杂的卫星表面交织而成。在这种情况下,MoE允许模型选择性地激活专门处理卫星几何的专家模块,同时在无信息的空白区域上节省计算资源,自然补充了上述的令牌合并范式。
    然而,大多数现有的航天器姿态估计方法主要集中在单帧估计上。这一限制主要源于早期公共基准的性质,这些基准主要是为单帧估计设计的,例如Spacecraft PosE Estimation Dataset+(SPEED+)[14]。因此,对于下游导航任务来说很重要的时间连续性在历史上较少被探索。随着高保真序列数据集的引入,这种情况发生了根本性的变化,最著名的是为SPAcecraft Recognition leveraging Knowledge of Space Environment(SPARK)挑战[25]、Satellite Hardware-In-the-loop Rendezvous Trajectories(SHIRT)数据集[26]和SwissCube数据集[27]发布的数据集。这些数据集提供了包含数千帧的扩展图像序列,激发了专注于序列轨迹跟踪的快速发展的研究方向。
    利用SHIRT数据集,Cassinis等人开创了一种基于过滤的范式[28]。他们提出的自适应无迹卡尔曼滤波器(A-UKF)将基于CNN的姿态回归器与轨道力学相结合。关键的是,他们的方法利用网络的随机不确定性来动态加权传入的测量值,证明了时间融合在鲁棒性方面显著优于单帧基线。同时,最近利用SPARK数据集的研究开始将时间约束直接内置于深度网络中。新的方法,如使用运动一致性损失或循环架构[29]、[30],尝试通过端到端学习帧间相关性来平滑预测。
    在这些进展的基础上,我们提出了一个新框架,将FastVGGT的计算效率与序列建模的时间连续性相结合。我们引入了一个可学习令牌合并机制,以解决原始FastVGGT的固定采样策略的问题。该模块优化了将航天器结构的语义稀疏性与空间背景分离,同时积极压缩冗余的空白区域。与使用光流或迭代细化的序列方法不同,我们的框架在端到端网络中集成了直接的几何监督。通过摄入时间有序的图像序列,我们的方法同时预测6-DoF姿态、置信度掩码和密集深度图。这种联合优化确保了轨迹的平滑性和高精度的几何对齐,无需其他后处理。
    我们在具有挑战性的航天器图像数据集SwissCube和SPEED+上验证了我们的方法。实验结果表明,该框架提高了准确性和稳定性。与Wide-range-depth Net [27]在SwissCube上的结果相比,准确度相对提高了21.6%。
    本文的其余部分组织如下。第2节介绍了所提出的序列框架,详细介绍了网络架构、注意力机制和可学习令牌合并。第3节深入阐述了几何学习目标,制定了预测头和复合监督策略。第4节展示了实验评估,包括数据集详情、实现细节和定量结果。最后,第5节总结了本文。

    章节片段

    架构

    在本节中,我们详细介绍了所提出的序列航天器姿态估计框架。我们框架的总体结构如图1所示。

    几何学习目标

    我们在Transformer编码器之后介绍了预测头和用于监督的损失函数。

    实验

    我们首先介绍数据集和实现细节,然后评估和分析结果。

    结论

    在本文中,我们提出了一个用于单目6-DoF姿态估计和非合作航天器几何重建的序列框架,结合了L-ToMe和相对表观旋转监督。L-ToMe通过自适应压缩视觉令牌,将计算开销降低了13%,同时保持了与未压缩基线相比0.5%以内的准确度。

    未引用的引用

    缺少引用 图6

    CRediT作者贡献声明

    Jian Chen:概念化、方法论。Ziyi He:数据整理、方法论、软件、撰写——原始草稿。Fan Wu:项目管理、资源。Zichen Zhao:调查、验证。Xueqin Chen:项目管理、资源、监督、撰写——审阅与编辑。

    利益冲突声明

    作者声明他们没有已知的竞争财务利益或个人关系可能影响本文报告的工作。
    相关新闻
    生物通微信公众号
    微信
    新浪微博

    热点排行

      今日动态 | 人才市场 | 新技术专栏 | 中国科学人 | 云展台 | BioHot | 云讲堂直播 | 会展中心 | 特价专栏 | 技术快讯 | 免费试用

      版权所有 生物通

      Copyright© eBiotrade.com, All Rights Reserved

      联系信箱:

      粤ICP备09063491号