TY - JOUR
T1 - Transfer learning for radio galaxy classification
AU - Tang, H.
AU - Scaife, A. M.M.
AU - Leahy, J. P.
N1 - Publisher Copyright:
© 2019 The Author(s) Published by Oxford University Press on behalf of the Royal Astronomical Society
PY - 2019
Y1 - 2019
N2 - In the context of radio galaxy classification, most state-of-the-art neural network algorithms have been focused on single survey data. The question of whether these trained algorithms have cross-survey identification ability or can be adapted to develop classification networks for future surveys is still unclear. One possible solution to address this issue is transfer learning, which reuses elements of existing machine learning models for different applications. Here we present radio galaxy classification based on a 13-layer Deep Convolutional Neural Network (DCNN) using transfer learning methods between different radio surveys. We find that our machine learning models trained from a random initialization achieve accuracies comparable to those found elsewhere in the literature. When using transfer learning methods, we find that inheriting model weights pre-trained on FIRST images can boost model performance when retraining on lower resolution NVSS data, but that inheriting pre-trained model weights from NVSS and retraining on FIRST data impairs the performance of the classifier. We consider the implication of these results in the context of future radio surveys planned for next-generation radio telescopes such as ASKAP, MeerKAT, and SKA1-MID.
AB - In the context of radio galaxy classification, most state-of-the-art neural network algorithms have been focused on single survey data. The question of whether these trained algorithms have cross-survey identification ability or can be adapted to develop classification networks for future surveys is still unclear. One possible solution to address this issue is transfer learning, which reuses elements of existing machine learning models for different applications. Here we present radio galaxy classification based on a 13-layer Deep Convolutional Neural Network (DCNN) using transfer learning methods between different radio surveys. We find that our machine learning models trained from a random initialization achieve accuracies comparable to those found elsewhere in the literature. When using transfer learning methods, we find that inheriting model weights pre-trained on FIRST images can boost model performance when retraining on lower resolution NVSS data, but that inheriting pre-trained model weights from NVSS and retraining on FIRST data impairs the performance of the classifier. We consider the implication of these results in the context of future radio surveys planned for next-generation radio telescopes such as ASKAP, MeerKAT, and SKA1-MID.
KW - Methods: statistical
KW - Radio continuum: galaxies
KW - Surveys
UR - http://www.scopus.com/inward/record.url?scp=85082621698&partnerID=8YFLogxK
U2 - 10.1093/mnras/stz1883
DO - 10.1093/mnras/stz1883
M3 - Article
AN - SCOPUS:85082621698
SN - 0035-8711
VL - 488
SP - 3358
EP - 3375
JO - Monthly Notices of the Royal Astronomical Society
JF - Monthly Notices of the Royal Astronomical Society
IS - 3
ER -