Photometric redshifts from SDSS images with an interpretable deep capsule network

Journal Article · 29 July 2022 · Monthly Notices of the Royal Astronomical Society

DOI: https://doi.org/10.1093/mnras/stac2105 · OSTI ID:1882234

Dey, Biprateep ; Andrews, Brett H.; Newman, Jeffrey A.; Mao, Yao-Yuan ; Rau, Markus Michael; Zhou, Rongpu

ABSTRACT Studies of cosmology, galaxy evolution, and astronomical transients with current and next-generation wide-field imaging surveys like the Rubin Observatory Legacy Survey of Space and Time are all critically dependent on estimates of photometric redshifts. Capsule networks are a new type of neural network architecture that is better suited for identifying morphological features of the input images than traditional convolutional neural networks. We use a deep capsule network trained on ugriz images, spectroscopic redshifts, and Galaxy Zoo spiral/elliptical classifications of ∼400 000 Sloan Digital Sky Survey galaxies to do photometric redshift estimation. We achieve a photometric redshift prediction accuracy and a fraction of catastrophic outliers that are comparable to or better than current methods for SDSS main galaxy sample-like data sets (r ≤ 17.8 and zspec ≤ 0.4) while requiring less data and fewer trainable parameters. Furthermore, the decision-making of our capsule network is much more easily interpretable as capsules act as a low-dimensional encoding of the image. When the capsules are projected on a two-dimensional manifold, they form a single redshift sequence with the fraction of spirals in a region exhibiting a gradient roughly perpendicular to the redshift sequence. We perturb encodings of real galaxy images in this low-dimensional space to create synthetic galaxy images that demonstrate the image properties (e.g. size, orientation, and surface brightness) encoded by each dimension. We also measure correlations between galaxy properties (e.g. magnitudes, colours, and stellar mass) and each capsule dimension. We publicly release our code, estimated redshifts, and additional catalogues at https://biprateep.github.io/encapZulate-1.

• Research Organization: Argonne National Laboratory (ANL), Lemont, IL (United States); Lawrence Berkeley National Laboratory (LBNL), Berkeley, CA (United States)
• Sponsoring Organization: National Aeronautics and Space Administration (NASA); National Science Foundation (NSF); USDOE Office of Science (SC), High Energy Physics (HEP)
• Grant/Contract Number: AC02-05CH11231; AC02-06CH11357
• OSTI ID: 1882234
• Journal Information: Monthly Notices of the Royal Astronomical Society, Journal Name: Monthly Notices of the Royal Astronomical Society Journal Issue: 4 Vol. 515; ISSN 0035-8711
• Publisher: Oxford University PressCopyright Statement
• Country of Publication: United Kingdom
• Language: English