Softly Constrained Denoisers for Diffusion Models Applied to Partial Differential Equations

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yeom-Song, Victor M., Rissanen, Severi, Solin, Arno, Kaski, Samuel, Sun, Mingfei
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913170887540736
author Yeom-Song, Victor M.
Rissanen, Severi
Solin, Arno
Kaski, Samuel
Sun, Mingfei
author_facet Yeom-Song, Victor M.
Rissanen, Severi
Solin, Arno
Kaski, Samuel
Sun, Mingfei
contents Diffusion models have become a powerful generative prior for solutions of partial differential equations (PDEs). Existing approaches enforce physical constraints either by adding the PDE residuals as loss regularizers or through inference-time adjustments. These methods bias the model away from the true data distribution, which is especially problematic when the governing PDE is misspecified. To circumvent these issues while making the most out of the PDE constraint, we introduce soft inductive biases into the denoiser architecture derived from the PDEs. We show that these softly constrained denoisers exploit constraint knowledge to improve compliance over standard denoisers, while maintaining enough flexibility to deviate from it in case of misspecification with respect to observed data.
format Preprint
id arxiv_https___arxiv_org_abs_2512_14980
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Softly Constrained Denoisers for Diffusion Models Applied to Partial Differential Equations
Yeom-Song, Victor M.
Rissanen, Severi
Solin, Arno
Kaski, Samuel
Sun, Mingfei
Machine Learning
68T07 (Primary) 68T01, 37M05 (Secondary)
I.2.0; I.2.1; J.2
Diffusion models have become a powerful generative prior for solutions of partial differential equations (PDEs). Existing approaches enforce physical constraints either by adding the PDE residuals as loss regularizers or through inference-time adjustments. These methods bias the model away from the true data distribution, which is especially problematic when the governing PDE is misspecified. To circumvent these issues while making the most out of the PDE constraint, we introduce soft inductive biases into the denoiser architecture derived from the PDEs. We show that these softly constrained denoisers exploit constraint knowledge to improve compliance over standard denoisers, while maintaining enough flexibility to deviate from it in case of misspecification with respect to observed data.
title Softly Constrained Denoisers for Diffusion Models Applied to Partial Differential Equations
topic Machine Learning
68T07 (Primary) 68T01, 37M05 (Secondary)
I.2.0; I.2.1; J.2
url https://arxiv.org/abs/2512.14980