Equivariant Geometric Scattering Networks via Vector Diffusion Wavelets
2510.01022v1
cs.LG, eess.SP, stat.ML
2025-10-04
Авторы:
David R. Johnson, Rishabh Anand, Smita Krishnaswamy, Michael Perlmutter
Abstract
We introduce a novel version of the geometric scattering transform for
geometric graphs containing scalar and vector node features. This new
scattering transform has desirable symmetries with respect to rigid-body
roto-translations (i.e., $SE(3)$-equivariance) and may be incorporated into a
geometric GNN framework. We empirically show that our equivariant
scattering-based GNN achieves comparable performance to other equivariant
message-passing-based GNNs at a fraction of the parameter count.