Optimizing machine learning methods to discover strong gravitational lenses in the Deep Lens Survey
Authors:
Keerthi Vasan G. C.,
Stephen Sheng,
Tucker Jones,
Chi Po Choi,
James Sharpnack
Abstract:
Machine learning models can greatly improve the search for strong gravitational lenses in imaging surveys by reducing the amount of human inspection required. In this work, we test the performance of supervised, semi-supervised, and unsupervised learning algorithms trained with the ResNetV2 neural network architecture on their ability to efficiently find strong gravitational lenses in the Deep Len…
▽ More
Machine learning models can greatly improve the search for strong gravitational lenses in imaging surveys by reducing the amount of human inspection required. In this work, we test the performance of supervised, semi-supervised, and unsupervised learning algorithms trained with the ResNetV2 neural network architecture on their ability to efficiently find strong gravitational lenses in the Deep Lens Survey (DLS). We use galaxy images from the survey, combined with simulated lensed sources, as labeled data in our training datasets. We find that models using semi-supervised learning along with data augmentations (transformations applied to an image during training, e.g., rotation) and Generative Adversarial Network (GAN) generated images yield the best performance. They offer 5--10 times better precision across all recall values compared to supervised algorithms. Applying the best performing models to the full 20 deg$^2$ DLS survey, we find 3 Grade-A lens candidates within the top 17 image predictions from the model. This increases to 9 Grade-A and 13 Grade-B candidates when $1$% ($\sim2500$ images) of the model predictions are visually inspected. This is $\gtrsim10\times$ the sky density of lens candidates compared to current shallower wide-area surveys (such as the Dark Energy Survey), indicating a trove of lenses awaiting discovery in upcoming deeper all-sky surveys. These results suggest that pipelines tasked with finding strong lens systems can be highly efficient, minimizing human effort. We additionally report spectroscopic confirmation of the lensing nature of two Grade-A candidates identified by our model, further validating our methods.
△ Less
Submitted 15 August, 2023; v1 submitted 31 October, 2022;
originally announced November 2022.
An Unsupervised Hunt for Gravitational Lenses
Authors:
Stephen Sheng,
Keerthi Vasan G. C,
Chi Po Choi,
James Sharpnack,
Tucker Jones
Abstract:
Strong gravitational lenses allow us to peer into the farthest reaches of space by bending the light from a background object around a massive object in the foreground. Unfortunately, these lenses are extremely rare, and manually finding them in astronomy surveys is difficult and time-consuming. We are thus tasked with finding them in an automated fashion with few if any, known lenses to form posi…
▽ More
Strong gravitational lenses allow us to peer into the farthest reaches of space by bending the light from a background object around a massive object in the foreground. Unfortunately, these lenses are extremely rare, and manually finding them in astronomy surveys is difficult and time-consuming. We are thus tasked with finding them in an automated fashion with few if any, known lenses to form positive samples. To assist us with training, we can simulate realistic lenses within our survey images to form positive samples. Naively training a ResNet model with these simulated lenses results in a poor precision for the desired high recall, because the simulations contain artifacts that are learned by the model. In this work, we develop a lens detection method that combines simulation, data augmentation, semi-supervised learning, and GANs to improve this performance by an order of magnitude. We perform ablation studies and examine how performance scales with the number of non-lenses and simulated lenses. These findings allow researchers to go into a survey mostly ``blind" and still classify strong gravitational lenses with high precision and recall.
△ Less
Submitted 20 October, 2022;
originally announced October 2022.
