Skip to yearly menu bar Skip to main content


Poster

Learning symmetries via weight-sharing with doubly stochastic tensors

Putri van der Linden · Alejandro GarcĂ­a-Castellanos · Thijs Kuipers · Sharvaree Vadgama · Erik Bekkers

East Exhibit Hall A-C #2104
[ ]
Wed 11 Dec 11 a.m. PST — 2 p.m. PST

Abstract: Group equivariance has emerged as a valuable inductive bias in deep learning, enhancing generalization, data efficiency, and robustness. Classically, group equivariant methods require the groups of interest to be known beforehand, which may not be realistic for real-world data. Additionally, baking in fixed group equivariance may impose overly restrictive constraints on model architecture. This highlights the need for methods that can dynamically discover and apply symmetries as soft constraints. For neural network architectures, equivariance is commonly achieved through group transformations of a canonical weight tensor, resulting in weight sharing over a given group $G$. In this work, we propose to \textit{learn} such a weight-sharing scheme by defining a collection of learnable doubly stochastic matrices that act as soft permutation matrices on canonical weight tensors, which can take regular group representations as a special case. This yields learnable kernel transformations that are jointly optimized with downstream tasks. We show that when the dataset exhibits strong symmetries, the permutation matrices will converge to regular group representations and our weight-sharing networks effectively become regular group convolutions. We also show that our proposed method learns partial symmetries.

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