Should the Boundary Term Be Learned in Reflected Diffusion? Conormal Trace and Reflection Masking
Not on the map. Why?
Abstract (arXiv)
We study score learning for reflected diffusion on bounded domains. Reflection keeps trajectories feasible but does not ensure that the learned score satisfies the boundary behavior implied by the forward process. With implicit score matching, integration by parts leaves a boundary term, and we show that it depends on one scalar at each boundary point: the diffusion- weighted normal component of the score, or conormal trace. The no-flux condition fixes this value while leaving the re- maining boundary components unrestricted; under anisotropic diffusion it generally differs from the ordinary normal score component. On hyperrectangles, our parametrization enforces the required trace without additional trainable parameters or a stochastic boundary estimator and, under regularity assump- tions, can represent the true score, whereas fixing an incorrect value creates an error that more data cannot remove. We ex- tend the construction to simplices and polygonal domains and identify reflection masking: hard reflection can keep samples feasible even when the learned trace is wrong, so post-reflection metrics may hide the error. Experiments show the clearest separation with less frequent reflection, anisotropic diffusion, and mass near intersections of constraints; under full reflection, final sample placement improves inconsistently, illustrating how hard repair can mask boundary-score errors and decouple score accuracy from downstream generation quality.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 8 of its 12 references that are in the map. First posted on arXiv 4 Aug 2026.
Builds on
8 of 12 references.
References · 8
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2023Mirror Diffusion Models for Constrained and Watermarked Generationcited by 38
- 2019Estimating Density Models with Truncation Boundaries using Score Matchingcited by 20
- 2024Constrained Diffusion Models via Dual Trainingcited by 19
- 2020Generalized Score Matching for General Domainscited by 13
- 2020Score matching for compositional distributionscited by 5
- 2025Score-based constrained generative modeling via Langevin diffusions with boundary conditionscited by 2
Built on by
Not cited yet.