Data-Adaptive Relaxed Equivariant Networks for Symmetry Breaking
Abstract
Equivariant neural networks improve generalization by incorporating symmetry, but real-world data often breaks symmetry in complex ways. In particular, different orbits in the data may follow different symmetry patterns or degrees of breaking. Existing relaxed equivariant models use shared weights across all orbits, which is less effective in such heterogeneous settings. We propose Data-Adaptive Relaxed Equivariant Networks (DAREN), a method that learns to adjust symmetry behavior separately for each orbit. By generating orbit-specific relaxed weights and using a gating mechanism to control symmetry strength, our model adapts to varying symmetry level across the data. Experiments on synthetic datasets show that DAREN outperforms equivariant models, relaxed models and unconstrained models, especially when symmetry varies across regions.