3D steerable CNNs: Learning rotationally equivariant features in volumetric data

Maurice Weiler, Mario Geiger, Max Welling, Wouter Boomsma, Taco Cohen

31 Citationer (Scopus)

Abstract

We present a convolutional network that is equivariant to rigid body motions. The model uses scalar-, vector-, and tensor fields over 3D Euclidean space to represent data, and equivariant convolutions to map between such representations. These SE(3)-equivariant convolutions utilize kernels which are parameterized as a linear combination of a complete steerable kernel basis, which is derived analytically in this paper. We prove that equivariant convolutions are the most general equivariant linear maps between fields over R3. Our experimental results confirm the effectiveness of 3D Steerable CNNs for the problem of amino acid propensity prediction and protein structure classification, both of which have inherent SE(3) symmetry.

OriginalsprogEngelsk
Publikationsdato2018
Antal sider12
StatusUdgivet - 2018
Begivenhed32nd Annual Conference on Neural Information Processing Systems - Montreal, Montreal, Canada
Varighed: 2 dec. 20188 dec. 2018
Konferencens nummer: 32
https://nips.cc/Conferences/2018

Konference

Konference32nd Annual Conference on Neural Information Processing Systems
Nummer32
LokationMontreal
Land/OmrådeCanada
ByMontreal
Periode02/12/201808/12/2018
Internetadresse

Fingeraftryk

Dyk ned i forskningsemnerne om '3D steerable CNNs: Learning rotationally equivariant features in volumetric data'. Sammen danner de et unikt fingeraftryk.

Citationsformater