Saved in:
Bibliographic Details
Main Authors: Zhou, Xingchen, Li, Nan, Jia, Peng, Liu, Yingfeng, Deng, Furen, Shu, Shuanghao, Li, Ying, Cao, Liang, Shan, Huanyuan, Ibitoye, Ayodeji
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2512.00769
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909934725103616
author Zhou, Xingchen
Li, Nan
Jia, Peng
Liu, Yingfeng
Deng, Furen
Shu, Shuanghao
Li, Ying
Cao, Liang
Shan, Huanyuan
Ibitoye, Ayodeji
author_facet Zhou, Xingchen
Li, Nan
Jia, Peng
Liu, Yingfeng
Deng, Furen
Shu, Shuanghao
Li, Ying
Cao, Liang
Shan, Huanyuan
Ibitoye, Ayodeji
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2512_00769
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle AI Agent for Source Finding by SoFiA-2 for SKA-SDC2
Zhou, Xingchen
Li, Nan
Jia, Peng
Liu, Yingfeng
Deng, Furen
Shu, Shuanghao
Li, Ying
Cao, Liang
Shan, Huanyuan
Ibitoye, Ayodeji
Machine Learning
Astrophysics of Galaxies
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.
title AI Agent for Source Finding by SoFiA-2 for SKA-SDC2
topic Machine Learning
Astrophysics of Galaxies
url https://arxiv.org/abs/2512.00769