用扩散模型加速低推力航天器轨迹全局搜索,提升效率百倍以上。
Global Search for Optimal Low Thrust Spacecraft Trajectories using Diffusion Models and the Indirect Method
- 结合扩散模型与间接法,学习代价变量随推力变化的结构规律。
- 对不同复杂度任务,新推力下求解速度提升10~100倍。
- 适合需频繁调整参数的深空探测任务快速设计,尤其适合高精度轨迹优化。
长时间低推力非线性最优航天器轨迹全局搜索计算量大、耗时长,且局部最优解呈现聚集模式。在任务初期设计阶段,参数频繁变动,要求轨迹设计者能高效生成高质量控制解。生成式机器学习模型可学习解结构随条件参数的变化规律,从而加速参数更新后的全局搜索。本文将最先进的扩散模型与间接法相结合,构建全局搜索框架。该框架在圆形限制三体问题中两种不同复杂度的低推力转移任务上进行了测试。通过生成并分析训练数据集,我们建立了数学关系与技术,揭示了局部最优解在代价域中的复杂结构。基于此数据训练的扩散模型成功加速了两类问题的全局搜索。模型能预测代价解结构随最大飞船推力的变化。以扩散模型生成的初始时刻代价变量作为数值求解器的热启动,相较于均匀分布或伴随控制变换样本,在未见过的推力值下,每分钟生成解的数量提升一到两个数量级。
原文摘要 · Abstract (English)
Long time-duration low-thrust nonlinear optimal spacecraft trajectory global search is a computationally and time expensive problem characterized by clustering patterns in locally optimal solutions. During preliminary mission design, mission parameters are subject to frequent changes, necessitating that trajectory designers efficiently generate high-quality control solutions for these new scenarios. Generative machine learning models can be trained to learn how the solution structure varies with respect to a conditional parameter, thereby accelerating the global search for missions with updated parameters. In this work, state-of-the-art diffusion models are integrated with the indirect approach for trajectory optimization within a global search framework. This framework is tested on two low-thrust transfers of different complexity in the circular restricted three-body problem. By generating and analyzing a training data set, we develop mathematical relations and techniques to understand the complex structures in the costate domain of locally optimal solutions for these problems. A diffusion model is trained on this data and successfully accelerates the global search for both problems. The model predicts how the costate solution structure changes, based on the maximum spacecraft thrust magnitude. Warm-starting a numerical solver with diffusion model samples for the costates at the initial time increases the number of solutions generated per minute for problems with unseen thrust magnitudes by one to two orders of magnitude in comparison to samples from a uniform distribution and from an adjoint control transformation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。