Learning to denoise astronomical images with U-nets

Abstract

ABSTRACT Astronomical images are essential for exploring and understanding the Universe. Optical telescopes capable of deep observations, such as the Hubble Space Telescope (HST), are heavily oversubscribed in the Astronomical Community. Images also often contain additive noise, which makes denoising a mandatory step in post-processing the data before further data analysis. In order to maximize the efficiency and information gain in the post-processing of astronomical imaging, we turn to machine learning. We propose Astro U-net, a convolutional neural network for image denoising and enhancement. For a proof-of-concept, we use HST images from Wide Field Camera 3 instrument UV/visible channel with F555W and F606W filters. Our network is able to produce images with noise characteristics as if they are obtained with twice the exposure time, and with minimum bias or information loss. From these images, we are able to recover $95.9{{\ \rm per\ cent}}$ of stars with an average flux error of $2.26{{\ \rm per\ cent}}$. Furthermore, the images have, on average, 1.63 times higher signal-to-noise ratio than the input noisy images, equivalent to the stacking of at least three input images, which means a significant reduction in the telescope time needed for future astronomical imaging campaigns.