Scalable Precision Imaging in Radio Astronomy: from Learned denoisers on GPU to Science (SPIRALS)
Lead Research Organisation:
Heriot-Watt University
Department Name: Sch of Engineering and Physical Science
Abstract
The ambitious science goals in radio astronomy for the next decades have triggered the development of a new generation of telescopes targeting imaging the sky with much higher precision (i.e. resolution and sensitivity) than current instruments. Endowing these telescopes with their expected acute vision requires image formation algorithms capable of transforming radio interferometry data into images at target precision, while being robust (i.e. including calibration and uncertainty quantification functionalities), and ultimately scalable to exascale data volumes. The "Scalable Precision Imaging in Radio Astronomy: from Learned denoisers on GPU to Science" (SPIRALS) work programme aims to design transformative deep learning methodology to address this challenge, and apply it on cutting-edge science cases, from the detection of halos and relics in galaxy clusters and detailed morphology mapping of radio galaxies from surveys of the MeerKAT telescope, to black hole imaging with Event Horizon Telescope (EHT) data.
In detail, firstly, artificial neural networks will be trained as simple "denoisers" encapsulating advanced learned physical models of the both radio sky and interfering observation effects. These denoisers will be integrated into a parallel algorithmic structure to define a new image formation algorithm with simultaneous capability for precision, robustness, and scalability. A parallel Python software implementation will be designed for, and mapped onto the latest and most efficient high performance computing hardware technologies, primarily large scale GPU systems. Secondly, as a by-product of the learning of denoisers, a low-cost "post-processor" will be developed to enhance legacy images. Before addressing the science cases, algorithms and software will be validated up to Terabyte image size, using both simulations from the future Square Kilometre Array (SKA) telescope and from the Deep Synoptic Array (DSA-2000) telescope concept, and real data from the Jansky Very Large Array (JVLA) and MeerKAT telescopes, with particular focus on wideband imaging of diffuse emission with complex and faint structure across the field of view.
In detail, firstly, artificial neural networks will be trained as simple "denoisers" encapsulating advanced learned physical models of the both radio sky and interfering observation effects. These denoisers will be integrated into a parallel algorithmic structure to define a new image formation algorithm with simultaneous capability for precision, robustness, and scalability. A parallel Python software implementation will be designed for, and mapped onto the latest and most efficient high performance computing hardware technologies, primarily large scale GPU systems. Secondly, as a by-product of the learning of denoisers, a low-cost "post-processor" will be developed to enhance legacy images. Before addressing the science cases, algorithms and software will be validated up to Terabyte image size, using both simulations from the future Square Kilometre Array (SKA) telescope and from the Deep Synoptic Array (DSA-2000) telescope concept, and real data from the Jansky Very Large Array (JVLA) and MeerKAT telescopes, with particular focus on wideband imaging of diffuse emission with complex and faint structure across the field of view.
Publications


Aghabiglou A
(2023)
Deep Network Series for Large-Scale High-Dynamic Range Imaging

Aghabiglou A
(2022)
Deep network series for large-scale high-dynamic range imaging

Aghabiglou A
(2024)
The R2D2 Deep Neural Network Series Paradigm for Fast Precision Imaging in Radio Astronomy
in The Astrophysical Journal Supplement Series

Dabbech A
(2024)
CLEANing Cygnus A Deep and Fast with R2D2
in The Astrophysical Journal Letters

Dabbech A
(2022)
First AI for Deep Super-resolution Wide-field Imaging in Radio Astronomy: Unveiling Structure in ESO 137-006
in The Astrophysical Journal Letters


Dabbech A
(2023)
CLEANing Cygnus A deep and fast with R2D2

Repetti A
(2022)
Dual Forward-Backward Unfolded Network for Flexible Plug-and-Play

Terris M
(2023)
Image reconstruction algorithms in radio interferometry: From handcrafted to learned regularization denoisers
in Monthly Notices of the Royal Astronomical Society
Title | BASPLib |
Description | BASPLib is an open-source library on GitHub, gathering Python and Matlab codes to solve challenging inverse imaging problems in astronomy and medicine. The primary imaging modality of focus is synthesis imaging by interferometry in radio astronomy, with functionality currently being developed for magnetic resonance imaging and ultrasound imaging in medicine. The BASPLib software suite gathers implementations of the most advanced computational imaging algorithms at the interface of optimisation and deep learning theories. The proposed algorithms can be seen as intermediate steps in the quest for an ultimate "intelligent" imaging algorithm (yet to be devised) providing the joint precision, robustness, efficiency, and scalability required by modern applications. A key feature on this past is algorithm modularity, with regularisation modules (enforcing image and calibration models) alternating with data-fidelity modules (enforcing consistency with the observed data). BASPLib algorithms and software are developed at Edinburgh's Biomedical and Astronomical Signal Processing Laboratory (https://basp.site.hw.ac.uk/) headed by Prof. Wiaux. |
Type Of Technology | Software |
Year Produced | 2024 |
Open Source License? | Yes |
Impact | Makes advance computational imaging algorithms available to the community and recently triggered new algorithmic developments in astronomical imaging. |
URL | https://basp-group.github.io/BASPLib/ |