Ax550Vokon@OpenReview

Total: 1

#1 Steerable Transformers for Volumetric Data [PDF] [Copy] [Kimi] [REL]

Authors: Soumyabrata Kundu, Risi Kondor

We introduce Steerable Transformers, an extension of the Vision Transformer mechanism that maintains equivariance to the special Euclidean group $\mathrm{SE}(d)$. We propose an equivariant attention mechanism that operates on features extracted by steerable convolutions. Operating in Fourier space, our network utilizes Fourier space non-linearities. Our experiments in both two and three dimensions show that adding steerable transformer layers to steerable convolutional networks enhances performance.

Subject: ICML.2025 - Poster