Radio-astronomical Image Reconstruction with Conditional Denoising Diffusion Model

Abstract

Reconstructing sky models from dirty radio images for accurate source localization and flux estimation is crucial for studying galaxy evolution at high redshift, especially in deep fields using instruments like the Atacama Large Millimetre Array (ALMA). With new projects like the Square Kilometre Array (SKA), there's a growing need for better source extraction methods. Current techniques, such as CLEAN and PyBDSF, often fail to detect faint sources, highlighting the need for more accurate methods. This study proposes using stochastic neural networks to rebuild sky models directly from dirty images. This method can pinpoint radio sources and measure their fluxes with related uncertainties, marking a potential improvement in radio source characterization. We tested this approach on 10164 images simulated with the CASA tool simalma, based on ALMA's Cycle 5.3 antenna setup. We applied conditional Denoising Diffusion Probabilistic Models (DDPMs) for sky models reconstruction, then used Photutils to determine source coordinates and fluxes, assessing the model's performance across different water vapor levels. Our method showed excellence in source localization, achieving more than 90% completeness at a signal-to-noise ratio (SNR) as low as 2. It also surpassed PyBDSF in flux estimation, accurately identifying fluxes for 96% of sources in the test set, a significant improvement over CLEAN+ PyBDSF's 57%. Conditional DDPMs is a powerful tool for image-to-image translation, yielding accurate and robust characterisation of radio sources, and outperforming existing methodologies. While this study underscores its significant potential for applications in radio astronomy, we also acknowledge certain limitations that accompany its usage, suggesting directions for further refinement and research.

arXiv commentIn production in Astronomy&Astrophyics

References in corpus (15)

• scikit-image: Image processing in Python
• LOFAR: The LOw-Frequency ARray
• The Murchison Widefield Array: the Square Kilometre Array Precursor at low radio frequencies
• CASA, the Common Astronomy Software Applications for Radio Astronomy
• First M87 Event Horizon Telescope Results. II. Array and Instrumentation
• A multi-scale multi-frequency deconvolution algorithm for synthesis imaging in radio interferometry
• Australian Square Kilometre Array Pathfinder: I. System Description
• The ALPINE-ALMA [CII] Survey: data processing, catalogs, and statistical source properties
• ALMA Spectroscopic Survey in the Hubble Ultra Deep Field: Survey Description
• Reionization Era Bright Emission Line Survey: Selection and Characterization of Luminous Interstellar Medium Reservoirs in the z>6.5 Universe
• The ALPINE-ALMA [CII] Survey: Multi-Wavelength Ancillary Data and Basic Physical Measurements
• Automated Mining of the ALMA Archive in the COSMOS Field (A3COSMOS): II. Cold Molecular Gas Evolution out to Redshift 6
• Autoencoders for unsupervised anomaly detection in high energy physics
• Deep Learning-based Imaging in Radio Interferometry
• Challenging interferometric imaging: Machine learning-based source localization from uv-plane observations

Cited by in corpus (5)

• Self-Supervised Learning on MeerKAT Wide-Field Continuum Images
• Tackling the Problem of Distributional Shifts: Correcting Misspecified, High-Dimensional Data-Driven Priors for Inverse Problems
• The spatially-resolved effect of mergers on the stellar mass assembly of MaNGA galaxies
• Simulating images of radio galaxies with diffusion models
• Very-Long Baseline Interferometry Imaging with Closure Invariants using Conditional Image Diffusion