Timezone: »

 
Poster
3D Steerable CNNs: Learning Rotationally Equivariant Features in Volumetric Data
Maurice Weiler · Wouter Boomsma · Mario Geiger · Max Welling · Taco Cohen

Tue Dec 04 07:45 AM -- 09:45 AM (PST) @ Room 210 #45
in Tue Poster Session A »

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 R^3. 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.

Keywords: Classification · CNN Architectures

Author Information

Maurice Weiler (University of Amsterdam)
Wouter Boomsma (University of Copenhagen)
Mario Geiger (EPFL)
Max Welling (University of Amsterdam / Qualcomm AI Research)
Taco Cohen (University of Amsterdam)

Taco Cohen is a machine learning research scientist at Qualcomm AI Research in Amsterdam and a PhD student at the University of Amsterdam, supervised by prof. Max Welling. He was a co-founder of Scyfer, a company focussed on active deep learning, acquired by Qualcomm in 2017. He holds a BSc in theoretical computer science from Utrecht University and a MSc in artificial intelligence from the University of Amsterdam (both cum laude). His research is focussed on understanding and improving deep representation learning, in particular learning of equivariant and disentangled representations, data-efficient deep learning, learning on non-Euclidean domains, and applications of group representation theory and non-commutative harmonic analysis, as well as deep learning based source compression. He has done internships at Google Deepmind (working with Geoff Hinton) and OpenAI. He received the 2014 University of Amsterdam thesis prize, a Google PhD Fellowship, ICLR 2018 best paper award for “Spherical CNNs”, and was named one of 35 innovators under 35 in Europe by MIT in 2018.

More from the Same Authors