arXiv:2602.05549cs.LG2026-02

让扩散模型精准执行复杂逻辑条件生成,支持组合式约束。

Logical Guidance for the Exact Composition of Diffusion Models

  • 基于布尔代数构建精确逻辑引导机制,支持复杂条件组合。
  • 在特定条件下可精确计算引导信号,提升生成准确性。
  • 适用于图像与蛋白质结构生成,适合需要精确控制的场景。

我们提出 LOGDIFF(逻辑引导用于扩散模型的精确组合),一种可在推理阶段实现复杂逻辑表达式约束生成的引导框架。研究发现,当复杂逻辑公式满足特定电路表示结构时,可获得精确的基于分数的引导:若合取结合条件独立子公式,析取结合条件独立或互斥子公式,则可精确计算引导信号。此时,引导信号可通过原子得分与后验概率利用高效递归算法计算。此外,我们证明对于常见分布类,任何期望的布尔公式均可编译为该电路形式。通过结合原子引导得分与后验概率估计,我们提出一种混合引导方法,融合分类器引导与无分类器引导,既适用于组合逻辑引导,也适用于标准条件生成。我们在多个图像与蛋白质结构生成任务中验证了该框架的有效性。

原文摘要 · Abstract (English)

We propose LOGDIFF (Logical Guidance for the Exact Composition of Diffusion Models), a guidance framework for diffusion models that enables principled constrained generation with complex logical expressions at inference time. We study when exact score-based guidance for complex logical formulas can be obtained from guidance signals associated with atomic properties. First, we derive an exact Boolean calculus that provides a sufficient condition for exact logical guidance. Specifically, if a formula admits a circuit representation in which conjunctions combine conditionally independent subformulas and disjunctions combine subformulas that are either conditionally independent or mutually exclusive, exact logical guidance is achievable. In this case, the guidance signal can be computed exactly from atomic scores and posterior probabilities using an efficient recursive algorithm. Moreover, we show that, for commonly encountered classes of distributions, any desired Boolean formula is compilable into such a circuit representation. Second, by combining atomic guidance scores with posterior probability estimates, we introduce a hybrid guidance approach that bridges classifier guidance and classifier-free guidance, applicable to both compositional logical guidance and standard conditional generation. We demonstrate the effectiveness of our framework on multiple image and protein structure generation tasks.

扩散模型逻辑引导生成控制蛋白质结构

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