Using convolutional neural networks to identify gravitational lenses in astronomical images

Abstract

Abstract The Euclid telescope, due for launch in 2021, will perform an imaging and slitless spectroscopy survey over half the sky, to map baryon wiggles and weak lensing. During the survey, Euclid is expected to resolve 100 000 strong gravitational lens systems. This is ideal to find rare lens configurations, provided they can be identified reliably and on a reasonable time-scale. For this reason, we have developed a convolutional neural network (CNN) that can be used to identify images containing lensing systems. CNNs have already been used for image and digit classification as well as being used in astronomy for star-galaxy classification. Here, our CNN is trained and tested on Euclid-like and KiDS (Kilo-Degree Survey)-like simulations from the Euclid Strong Lensing Group, successfully classifying 77 per cent of lenses, with an area under the ROC curve of up to 0.96. Our CNN also attempts to classify the lenses in COSMOS Hubble Space Telescope F814W-band images. After convolution to the Euclid resolution, we find we can recover most systems that are identifiable by eye. The python code is available on Github.