论文

PMosFM:物理约束单步生成新方法

PMosFM: Preconditioned Manifold Matching for One-Step Physics-Constrained Generation

精选理由

PMosFM解决了物理约束生成模型的高采样成本问题,单步生成就能满足物理约束,训练和推理速度更快。

PMosFM是一种预调节流形匹配框架,通过流形解码器编码约束,在内在坐标中学习传输,无需单独的残差损失或终端残差展开。该方法使用几何预调节器重新缩放坐标,并使用正则化协方差变换近似白化插值状态输入。在多个基准测试中,PMosFM比多步基线方法具有更低的训练和采样时间,同时保持相当的物理和分布保真度。

原文 · arXiv cs.LG

PMosFM: Preconditioned Manifold Matching for One-Step Physics-Constrained Generation

Physics-constrained generative models aim to generate physical fields that match a target distribution and satisfy prescribed constraints. However, enforcing these constraints often increases sampling costs through iterative corrections or training costs through residual optimization and trajectory unrolling. To address this issue, we introduce \textbf{P}reconditioned \textbf{M}anifold \textbf{o}ne-\textbf{s}tep \textbf{F}low \textbf{M}atching (\textbf{PMosFM}), a preconditioned manifold matching framework for one-step physics-constrained generation. By encoding constraints in a manifold decoder, PMosFM learns transport in intrinsic coordinates without separate residual losses or terminal residual unrolling. A geometric preconditioner rescales coordinates using the decoder-induced metric, while a regularized covariance transform approximately whitens the interpolation-state inputs. A finite-interval objective couples velocity supervision with consistency between decoded endpoints in physical space. We show that exact parameterization removes residual-induced Gauss--Newton curvature, that geometric and covariance effects separate in a local conditioning bound, and that physical flow-map error bounds endpoint distributional error. Controlled ablations examine conditioning, and experiments evaluate optimizer-update time and memory footprint. At inference, PMosFM uses one neural transport evaluation followed by physical decoding. Experiments across benchmarks show lower training and sampling time than the multi-step baselines at comparable physical and distributional fidelity. Code and datasets will be released publicly.