E(3)-Equivariant Mesh Neural Networks

AISTATS 2024

Abstract:

Triangular meshes are widely used to represent three-dimensional objects. As a result, many recent works have addressed the need for geometric deep learning on 3D mesh. However, we observe that the complexities in many of these architectures do not translate to practical performance, and simple deep models for geometric graphs are competitive in practice. Motivated by this observation, we minimally extend the update equations of E(n)-Equivariant Graph Neural Networks (EGNNs) (Sartogas et. al, 2022) to incorporate mesh face information, and further improve it to account for long-range interactions through the hierarchy. The resulting architecture, Equivariant Mesh Neural Network (EMNN), outperforms other, more complicated equivariant methods on mesh tasks, with a fast run-time and no expensive preprocessing.

Paper: https://arxiv.org/pdf/2402.04821.pdf

Code: https://github.com/Fsoft-AIC/EquiMesh/tree/main

Authors

Thuan Trang (Batch-1 AI Resident)*, Nhat Khang Ngo (Batch-3 AI Resident)*, Daniel Levy*, Thieu Vo, Siamak Ravanbakhsh, Truong Son Hy (*: equal contribution)

Leave A Comment