Variational views for self-supervised learning in radio astronomy

TL;DR

This paper introduces a novel self-supervised learning approach using a pre-trained Variational Autoencoder to generate augmented views for radio galaxy morphology classification, improving performance in large, unlabelled datasets.

Contribution

It demonstrates that combining generative views from a β-VAE with standard augmentations enhances downstream classification, highlighting the potential of disentanglement-aware SSL in radio astronomy.

Findings

Generative views improve classification accuracy.

Moderate β regularization balances reconstruction and disentanglement.

Fanaroff-Riley class varies continuously in latent space.

Abstract

Modern astronomical surveys are producing progressively larger and more complex datasets, making traditional supervised approaches that rely on extensive labelled catalogues increasingly difficult. Consequently, pre-training using self-supervised learning (SSL), which offers a scalable route by extracting structure directly from unlabelled images, is becoming attractive for many downstream applications. In this work we consider the use of coupled self-supervised representation learning approaches for radio galaxy morphology pre-training. In order to account for the more nuanced variations in radio galaxy morphology than are typically included in the augmented views of view-based SSL algorithms, we use a pre-trained Variational Autoencoder (VAE) to generate views for training a larger view-based self-supervised model. To do this, a $\beta$-VAE was trained on the Radio Galaxy Zoo (RGZ) dataset, where moderate regularization ($\beta = 2.3$) was found to provide a good balance between reconstruction quality and disentanglement of generative factors such as source multiplicity and lobe asymmetry. An analysis of the $\beta$-VAE reveals that Fanaroff-Riley class identity manifests as a continuous transition across the latent space, rather than being associated to a single discrete dimension. $\beta$-VAE reconstructions were then incorporated as generative augmentations within a view-based SSL pipeline. Our experiments show that combining these generative views with standard image augmentations improves downstream classification performance, and we present ablation studies clarifying the relative contribution of each augmentation type. These results indicate that generative and contrastive approaches are complementary, and point toward disentanglement-aware self-supervised learning as a promising direction for future radio astronomy surveys.