English

Using transfer learning to detect galaxy mergers

Instrumentation and Methods for Astrophysics 2018-06-13 v2 Machine Learning

Abstract

We investigate the use of deep convolutional neural networks (deep CNNs) for automatic visual detection of galaxy mergers. Moreover, we investigate the use of transfer learning in conjunction with CNNs, by retraining networks first trained on pictures of everyday objects. We test the hypothesis that transfer learning is useful for improving classification performance for small training sets. This would make transfer learning useful for finding rare objects in astronomical imaging datasets. We find that these deep learning methods perform significantly better than current state-of-the-art merger detection methods based on nonparametric systems like CAS and GM20_{20}. Our method is end-to-end and robust to image noise and distortions; it can be applied directly without image preprocessing. We also find that transfer learning can act as a regulariser in some cases, leading to better overall classification accuracy (p=0.02p = 0.02). Transfer learning on our full training set leads to a lowered error rate from 0.038 ±\pm 1 down to 0.032 ±\pm 1, a relative improvement of 15%. Finally, we perform a basic sanity-check by creating a merger sample with our method, and comparing with an already existing, manually created merger catalogue in terms of colour-mass distribution and stellar mass function.

Keywords

Cite

@article{arxiv.1805.10289,
  title  = {Using transfer learning to detect galaxy mergers},
  author = {Sandro Ackermann and Kevin Schawinski and Ce Zhang and Anna K. Weigel and M. Dennis Turp},
  journal= {arXiv preprint arXiv:1805.10289},
  year   = {2018}
}

Comments

Accepted for publication in MNRAS. Code and data on https://space.ml/proj/transfer_learning

R2 v1 2026-06-23T02:08:44.439Z