Poster
in
Workshop: NeurIPS 2024 Workshop: Machine Learning and the Physical Sciences
DiffLense: A Conditional Diffusion Model for Super-Resolution of Gravitational Lensing Data
Pranath Reddy Kumbam · Michael Toomey · Hanna Parul · Sergei Gleyzer
Gravitational lensing data is frequently collected at low resolution due to instrumental limitations and observing conditions. Machine learning-based super-resolution techniques offer a method to enhance the resolution of these images, enabling more precise measurements of lensing effects and a better understanding of the matter distribution in the lensing system. In this work, we introduce DiffLense, a novel super-resolution pipeline based on a conditional diffusion model specifically designed to enhance the resolution of gravitational lensing images obtained from the Hyper Suprime-Cam Subaru Strategic Program (HSC-SSP). The diffusion model, trained to generate Hubble Space Telescope (HST) data, is conditioned on HSC data pre-processed with denoising techniques and thresholding to significantly reduce noise and background interference. This process leads to a more distinct and less overlapping conditional distribution during the model's training phase. We demonstrate that DiffLense outperforms existing state-of-the-art single-image super-resolution techniques, particularly in retaining the fine details necessary for astrophysical analyses.