Minuet: A Diffusion Autoencoder for Compact Semantic Compression of Multi-Band Galaxy Images

TL;DR

Minuet is a low-dimensional diffusion autoencoder that creates compact, meaningful representations of galaxy images, enabling efficient analysis of galaxy properties and morphology with high-fidelity reconstructions.

Contribution

It introduces Minuet, a diffusion autoencoder with only five latent dimensions for galaxy images, improving interpretability and utility in astrophysical analyses.

Findings

Achieves high-quality reconstructions with only five latent dimensions.

Latent features strongly correlate with galaxy morphology and physical properties.

Enables efficient nearest neighbor searches in galaxy image space.

Abstract

The Vera C. Rubin Observatory is slated to observe nearly 20 billion galaxies during its decade-long Legacy Survey of Space and Time. The rich imaging data it collects will be an invaluable resource for probing galaxy evolution across cosmic time, characterizing the host galaxies of transient phenomena, and identifying novel populations of anomalous systems. While machine learning models have shown promise for extracting galaxy features from multi-band astronomical imaging, the large dimensionality of the learned latent space presents a challenge for mechanistic interpretability studies. In this work, we present Minuet, a low-dimensional diffusion autoencoder for multi-band galaxy imaging. Minuet is trained to reconstruct 72x72-pixel $grz$ image cutouts of 6M galaxies within $z<1$ from the Dark Energy Camera Legacy Survey using only five latent dimensions. By using a diffusion model conditioned on the transformer-based autoencoder's output for image reconstruction, we achieve semantically-meaningful latent representations of galaxy images while still allowing for high-fidelity, probabilistic reconstructions. We train a series of binary classifiers on Minuet's latent features to quantify their connection to morphological labels from Galaxy Zoo, and a conditional flow to produce posterior distributions of SED-derived redshifts, stellar masses, and star-formation rates. We further show the value of Minuet for nearest neighbor searches in the learned latent space. Minuet provides strong evidence for the low intrinsic dimensionality of galaxy imaging, and introduces a class of astrophysical models that produce highly compact representations for diverse science goals.