Using Multiple Vector Channels Improves E(n)-Equivariant Graph Neural Networks

Daniel Levy, Sékou-Oumar Kaba, Carmelo Gonzales +2

We present a natural extension to E(n)-equivariant graph neural networks that uses multiple equivariant vectors per node. We formulate the extension and show that it improves perfo…