Affiliation:
1. Electrical & Electronic Engineering, Stellenbosch University , Stellenbosch, 7600 , South Africa
2. Computer Science, Stellenbosch University , Stellenbosch, 7600 , South Africa
Abstract
ABSTRACT
In the realm of radio astronomy, the detection of radio frequency interference (RFI) stands as a pivotal pursuit. This study performs a novel comparative analysis of deep learning methodologies and introduces a novel transfer learning method, called fine-tuning. We compare various aspects and factors relating to this problem, including supervised fully convolutional network (FCN) architectures used within the literature, loss functions, regularization techniques, and training methodologies, to establish the most effective strategies for RFI detection. Moreover, the relationship between parameters, FLOPS, and inference times are examined. Fine-tuning involves pre-training models with low-quality aoflagger reference outputs, a very popular and accessible RFI flagging software package, and thereafter re-training the models with high-quality reference outputs. We utilize two data sets: real observations from LOFAR and simulated data from HERA. The mean squared error (MSE) loss function emerges as a robust performer if a high recall is desired. In contrast, the binary cross-entropy (BCE) loss function excels in generalization but falls short in classification performance. The Dice loss function emerges as the top performer, maximizing the F1 score and thereby serving as the choice for our further investigations. Notably, we highlight the important role of data quality and model capacity. In particular, we find that low-capacity models exhibit resilience when trained with low-quality flags from aoflagger, showcasing their ability to mitigate overfitting and overflagging tendencies. In contrast, high-capacity models excel when trained with high-quality flags. Fine-tuning proved to be an effective method to unlearn the overflagging tendencies of aoflagger, whilst requiring very little data.
Funder
National Research Foundation
Publisher
Oxford University Press (OUP)
Cited by
3 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献