Group Symmetries in Diffusion Models: Formulation, Generalization, and Enforcement
Abstract
Group symmetries are fundamental structures in many real-world datasets, and lever- aging them is crucial for building robust and data-efficient machine learning models. Diffusion models have achieved state-of-the-art performance in generative tasks but typically rely on standard neural network architectures for score estimation. This thesis investigates whether such standard configurations enable diffusion models to implicitly learn and generalize underlying data symmetries purely from examples, particularly when data is partially observed. Drawing motivation from Neural Tangent Kernel (NTK) theory, which suggests limitations in the ability of standard supervised networks to generalize symmetries beyond local data structure, we hypothesize and empirically demonstrate that score networks in diffusion models exhibit similar constraints. Using a 2D toy dataset with inherent SO(2) rotational symmetry, we show a consistent failure of standard models trained on incomplete data (interpolation and extrapolation settings) to generalize symmetry, exhibiting significant score field distortions in unobserved regions. To address this limitation, we propose and evaluate a novel per-timestep symmetry loss that regularizes the denoising process to encourage approximate equivariance. Empirical results on both the toy dataset and higher-dimensional MNIST data confirm that this loss significantly enhances symmetry generalization even in standard architectures, yielding geometrically…
Read the paper
Related studies
- Learning from First-Generation Qualitative Approaches in the IS Discipline: An Evolutionary View and Some Implications for Authors and Evaluators (PART 1/2)
- Commentary on the clinical and preclinical dosage limits of interstitially administered magnetic fluids for therapeutic hyperthermia based on current practice and efficacy models
- Prosthesis: New Technological Opportunities and Innovative Biomedical Devices
- The history and future of neural modeling for cochlear implants
- Ryanodine receptors: waking up from refractoriness
- Novel Transgenic Rabbit Model Sheds Light on the Puzzling Role of Matrix Metalloproteinase-12 in Atherosclerosis
Go deeper in Evidence Hub
Evidence Hub adds a study data sheet (design, sample size, evidence level, key numbers), an AI summary and a chat grounded in the paper's text.