RADiff: Controllable Diffusion Models for Radio Astronomical Maps Generation

Journal
Date Issued
2024
Author(s)
•
•
DeMarco, Andrea
•
Fiameni, Giuseppe
•
Pilzer, Andrea
•
•
Hopkins, Andrew M.
•
•
•
•
•
•
Spampinato, Concetto
DOI
Abstract
Along with the nearing completion of the Square Kilometre Array (SKA), comes an increasing demand for accurate and reliable automated solutions to extract valuable information from the vast amount of data it will allow acquiring. Automated source finding is a particularly important task in this context, as it enables the detection and classification of astronomical objects. Deep-learning-based object detection and semantic segmentation models have proven to be suitable for this purpose. However, training such deep networks requires a high volume of labeled data, which is not trivial to obtain in the context of radio astronomy. Since data needs to be manually labeled by experts, this process is not scalable to large dataset sizes, limiting the possibilities of leveraging deep networks to address several tasks. In this work, we propose RADiff, a generative approach based on conditional diffusion models trained over an annotated radio dataset to generate synthetic images, containing radio sources of different morphologies, to augment existing datasets and reduce the problems caused by class imbalances. We also show that it is possible to generate fully synthetic image-annotation pairs to automatically augment any annotated dataset. We evaluate the effectiveness of this approach by training a semantic segmentation model on a real dataset augmented in two ways: 1) using synthetic images obtained from real masks, and 2) generating images from synthetic semantic masks. Finally, we also show how the model can be applied to populate background noise maps for simulating radio maps for Data Challenges. Code is available at: https://github.com/SKA-INAF/radiff
Start page
1
Uri
Url
Issn Identifier
2691-4581
Rights
open.access
File(s)
Loading...
Thumbnail Image
Name

RADiff_IEEE TRANS..pdf

Description
[Administrators only]
Size

1.69 MB

Format

Adobe PDF

Checksum (MD5)

65dd1cf96d6a1a30b5c52e9533f05d83

Loading...
Thumbnail Image
Name

2307.02392v1_compressed.pdf

Description
Preprint
Size

1.04 MB

Format

Adobe PDF

Checksum (MD5)

1a189c9e77b5a3b9cefda387ea05e842