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.
 
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/