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


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 known symmetry 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 more general baselines in prediction tasks over the symmetric group and Artin braid group. We additionally show that MatrixNet has better generalization to unseen group elements of greater word length than the training set.

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