Skip to yearly menu bar Skip to main content


Poster

MatrixNet: Learning over symmetry groups using learned group representations

Lucas Laird · Circe Hsu · Asilata Bapat · Robin Walters

East Exhibit Hall A-C #2405
[ ]
Fri 13 Dec 11 a.m. PST — 2 p.m. PST

Abstract:

Group theory has been used in machine learning to provide a theoretically grounded approach for incorporating known symmetry transformations in tasks from robotics to protein modeling. In these applications, equivariant neural networks use knownsymmetry groups with predefined representations to learn over geometric input data. We propose MatrixNet, a neural network architecture that learns matrix representations of group element inputs instead of using predefined representations. MatrixNet achieves higher sample efficiency and generalization over several standard baselines in prediction tasks over the several finite groups and the Artin braid group. We also show that MatrixNet respects group relations allowing generalization to group elements of greater word length than in the training set. Our code is available at https://github.com/lucas-laird/MatrixNet.

Live content is unavailable. Log in and register to view live content