arXiv:2605.12754cs.LG2026-05被引 3

让生成模型严格满足物理约束,同时保持高质量样本。

Constraint-Aware Flow Matching: Decision Aligned End-to-End Training for Constrained Sampling

论文配图:Constraint-Aware Flow Matching: Decision Aligned End-to-End Training for Constrained Sampling
图 1 · 摘自论文原文
  • 将约束投影融入训练目标,实现端到端对齐。
  • 在三个真实场景中验证,生成质量与约束满足度双提升。
  • 适合需要高可靠性约束生成的科研与工程应用。

深度生成模型在众多应用中表现卓越,尤其在科学与工程领域日益普及。尽管已有研究尝试将物理约束融入生成过程,但现有方法难以在保证样本质量的同时严格满足约束。特别是无需训练的约束采样方法虽能确保每样本可行性,却导致训练目标与采样过程不一致,常引发性能下降。本文识别出这一训练-采样错位为当前约束生成建模的核心瓶颈,提出约束感知流匹配(Constraint-Aware Flow Matching)框架,通过在训练目标中显式引入约束投影,使模型学习的动力学与约束采样过程对齐。该方法缓解了基于投影修正带来的分布偏移,实现高质量约束生成。在三个具有挑战性的真实世界基准上评估,验证了方法的通用性与有效性。

原文摘要 · Abstract (English)

Deep generative models provide state-of-the-art performance across a wide array of applications, with recent studies showing increasing applicability for science and engineering. Despite a growing corpus of literature focused on the integration of physics-based constraints into the generation process, existing approaches fail to enforce strict constraint satisfaction while maintaining sample quality. In particular, training-free constrained sampling methods, while providing per-sample feasibility guarantees, introduce a fundamental mismatch between the training objective and the constrained sampling procedure, often leading to performance degradation. Identifying this training-sampling misalignment as a central limitation of current constrained generative modeling approaches, this paper proposes Constraint-Aware Flow Matching, a novel end-to-end framework that explicitly incorporates constraint projections into the training objective. By aligning the model's learned dynamics with the constrained sampling process, the proposed method mitigates distributional shift induced by projection-based corrections, enabling high-quality constrained generation. The proposed approach is evaluated on three challenging real-world benchmarks, illustrating the generality and efficacy of the method.

生成模型约束生成端到端

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。