用AI自动调优射电望远镜源搜参数,100步就超人工基准。
AI Agent for Source Finding by SoFiA-2 for SKA-SDC2
- 用强化学习智能调节SoFiA参数,自动试错优化。
- 100次评估内性能超越团队基准,耗时更少。
- 适合需要复杂参数调优的天文数据处理任务。
射电波段下一代大型巡天(如平方公里阵列SKA)的数据分析中,源提取至关重要。现有工具如SoFiA和Aegean虽能处理,但其参数配置对结果影响大,难以手动调优。本文提出基于软演员-评论家(SAC)强化学习算法的AI代理框架,利用SKA科学数据挑战2(SDC2)数据集进行评估。该代理通过调整参数并接收SDC2评分反馈,逐步学习最优配置。经充分训练后,仅需100次评估即可找到优于团队基准的参数组合,且耗时更低。该方法可推广至其他需复杂调参的任务,但依赖高质量含真实标注的训练数据。
原文摘要 · Abstract (English)
Source extraction is crucial in analyzing data from next-generation, large-scale sky surveys in radio bands, such as the Square Kilometre Array (SKA). Several source extraction programs, including SoFiA and Aegean, have been developed to address this challenge. However, finding optimal parameter configurations when applying these programs to real observations is non-trivial. For example, the outcomes of SoFiA intensely depend on several key parameters across its preconditioning, source-finding, and reliability-filtering modules. To address this issue, we propose a framework to automatically optimize these parameters using an AI agent based on a state-of-the-art reinforcement learning (RL) algorithm, i.e., Soft Actor-Critic (SAC). The SKA Science Data Challenge 2 (SDC2) dataset is utilized to assess the feasibility and reliability of this framework. The AI agent interacts with the environment by adjusting parameters based on the feedback from the SDC2 score defined by the SDC2 Team, progressively learning to select parameter sets that yield improved performance. After sufficient training, the AI agent can automatically identify an optimal parameter configuration that outperform the benchmark set by Team SoFiA within only 100 evaluation steps and with reduced time consumption. Our approach could address similar problems requiring complex parameter tuning, beyond radio band surveys and source extraction. Yet, high-quality training sets containing representative observations and catalogs of ground truth are essential.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。