Current browse context:
astro-ph.IM
Change to browse by:
References & Citations
Astrophysics > Instrumentation and Methods for Astrophysics
Title: E(2) Equivariant Self-Attention for Radio Astronomy
(Submitted on 8 Nov 2021 (v1), last revised 30 Nov 2021 (this version, v2))
Abstract: In this work we introduce group-equivariant self-attention models to address the problem of explainable radio galaxy classification in astronomy. We evaluate various orders of both cyclic and dihedral equivariance, and show that including equivariance as a prior both reduces the number of epochs required to fit the data and results in improved performance. We highlight the benefits of equivariance when using self-attention as an explainable model and illustrate how equivariant models statistically attend the same features in their classifications as human astronomers.
Submission history
From: Micah Bowles [view email][v1] Mon, 8 Nov 2021 15:58:30 GMT (1533kb,D)
[v2] Tue, 30 Nov 2021 16:50:04 GMT (1533kb,D)
Link back to: arXiv, form interface, contact.