用AI模型预测航天器近距离操作,提升自主对接成功率与安全性
GPU-Accelerated Astrodynamics World Models for Spacecraft Rendezvous and Proximity Operations

- 基于Transformer的AI世界模型,融合运动状态与摄像头图像预测未来轨迹
- 在国际空间站对接任务中,成功率比传统强化学习高53%(对比29%)
- 能识别异常物体,准确率达98%,适合航天器自主导航研究者使用
世界模型是表示学习中的新兴范式,通过离线轨迹数据联合学习状态-动作动态与观测模型,实现多步规划和带不确定性估计的轨迹预测。该方法在机器人和游戏领域表现优异,但尚未应用于太空领域。本文提出基于世界模型的航天器交会与近距离操作方法:首先构建开源、基于JAX的国际空间站对接仿真环境,支持并行GPU模拟轨道与姿态动力学,生成训练所需数千条状态-动作转移数据;其次提出Out-of-this-World-Model,一种基于Transformer的世界模型,将相对运动状态与本体相机图像编码为潜在状态,利用一步流匹配预测在指令推力与扭矩下的演化过程,输出未来观测分布,捕捉随机动态与每一步不确定性;其性能优于DreamerV3类后验修正基线,在参数与超参数更少的情况下表现更佳;最后将其应用于无人舱段自主对接国际空间站,在保持禁区约束条件下,相比强化学习基线提升样本效率与任务成功率(跨端口对接成功率达53%对29%),在未见端口上泛化能力显著增强(成功率达40%对17%),并以98%准确率检测接近过程中的异常物体。本文开源仿真环境与模型架构,推动该范式进一步研究。
原文摘要 · Abstract (English)
World models are an emerging paradigm in representation learning in which an agent jointly learns state-action dynamics and observation models from offline trajectory data, enabling multi-step planning and trajectory prediction with uncertainty estimates. They have shown strong results in robotics and game environments, but, to the best of our knowledge, have not previously been applied to the space domain. This paper introduces a world model-based approach to cooperative and non-cooperative spacecraft rendezvous and proximity operations. First, we introduce an open-source, JAX-based International Space Station (ISS) docking environment supporting parallel GPU simulation of spacecraft orbit and attitude dynamics, generating the thousands of state-action transitions that world model training requires. Second, we introduce Out-of-this-World-Model, a transformer-based world model that encodes relative kinematic states and body-fixed camera imagery into a latent state and predicts its evolution under commanded thrusts and torques using one-step flow matching. It produces a distribution over future observations, capturing stochastic dynamics and per-timestep uncertainty, and outperforms DreamerV3-style posterior-correction baselines with fewer trainable parameters and hyperparameters. Third, we apply the approach to a capsule autonomously docking with the ISS under keep-out-zone constraints, demonstrating improved sample efficiency and task performance over reinforcement learning baselines (53% versus 29% docking success across ports), better out-of-distribution generalization (on held-out ports the world model more than doubles baseline success, 40% versus 17%), and detection of anomalous objects encountered during approach with 98% classification accuracy. We open-source the simulation environment and model architecture to enable further study of this paradigm.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。