用神经网络加速系外行星轨道参数估计,速度快近400倍。
Estimating Orbital Parameters of Direct Imaging Exoplanet Using Neural Network
- 先用生成模型压缩参数范围,再用马尔可夫链蒙特卡洛精确推断
- 比传统方法快77.8到365.4倍,且后验似然更高
- 适合未来大规模系外行星数据处理,可推广至其他科学领域
本文提出一种流匹配马尔可夫链蒙特卡洛(FM-MCMC)算法,用于估算仅含一颗系外行星系统的轨道参数。相较于依赖贝叶斯框架内随机采样的传统方法,本方法首先利用流匹配后验估计(FMPE)高效约束物理参数的先验范围,再通过MCMC精确推断后验分布。以β Pictoris b的轨道参数推断为例,该模型运行速度比并行退火蒙特卡洛(PTMCMC)快77.8倍,比嵌套采样快365.4倍,同时保持相当的精度。此外,本方法在所有对比方法中取得了最高的平均对数似然值,证明其具有更优的采样效率与准确性。该方法展现出良好的可扩展性与高效性,适用于未来大规模系外行星巡天的数据处理。除天体物理外,该范式为深度生成模型与传统采样方法的协同提供了通用框架,可用于宇宙学、生物医学成像及粒子物理等复杂推断问题。
原文摘要 · Abstract (English)
In this work, we propose a flow-matching Markov chain Monte Carlo (FM-MCMC) algorithm for estimating the orbital parameters of exoplanetary systems, especially for those only one exoplanet is involved. Compared to traditional methods that rely on random sampling within the Bayesian framework, our approach first leverages flow matching posterior estimation (FMPE) to efficiently constrain the prior range of physical parameters, and then employs MCMC to accurately infer the posterior distribution. For example, in the orbital parameter inference of beta Pictoris b, our model achieved a substantial speed-up while maintaining comparable accuracy-running 77.8 times faster than Parallel Tempered MCMC (PTMCMC) and 365.4 times faster than nested sampling. Moreover, our FM-MCMC method also attained the highest average log-likelihood among all approaches, demonstrating its superior sampling efficiency and accuracy. This highlights the scalability and efficiency of our approach, making it well-suited for processing the massive datasets expected from future exoplanet surveys. Beyond astrophysics, our methodology establishes a versatile paradigm for synergizing deep generative models with traditional sampling, which can be adopted to tackle complex inference problems in other fields, such as cosmology, biomedical imaging, and particle physics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。