arXiv:2512.01242cs.CVcs.AI2025-12

用强化学习和蒙特卡洛树搜索解决扩散模型在约束生成中的失效问题

When Diffusion Breaks Constraints: Sequential Autoregressive Generation with RL and MCTS

  • 将约束生成转化为离散自回归序列生成,提升可行性
  • 强化学习使任务成功率显著提升,树搜索增强前瞻决策能力
  • 适合需要严格几何或物理约束的工程设计、分子生成等场景

数据驱动生成模型在语言与视觉领域表现优异,但扩散模型在受约束的规划与设计任务中常出现严重约束违反,如工程逆向设计、分子生成、多机器人路径规划及户型/场景合成等,即便使用投影或引导也难以避免。这类任务兼具难以明确定义的语义目标与严格的几何或物理约束(如不重叠、连通性),可行解位于输出空间中低维、稀疏且可能不连通的区域。本文通过从语言生成拼图(七块固定形状需构成文字描述轮廓,保持连通且不重叠)和简化矩形组合任务(含学习得到的边界框约束)研究该失败模式,发现扩散模型在满足约束方面表现不佳,与在低维子流形附近生成样本困难的现象一致。受局部可行重参数化启发,本文将约束生成重构为离散自回归序列生成。强化学习提升了可行性和任务成功率,蒙特卡洛树搜索量化了当可行区域缩小后前瞻搜索的价值。实证、理论及已有工作证据共同指向连续密度匹配在这一类约束生成问题上的结构性局限,并建议采用序列化约束感知生成作为替代方案。

原文摘要 · Abstract (English)

Data-driven generative models excel in language and vision, but diffusion models often fail in constrained planning and design tasks, exhibiting severe constraint violations in engineering inverse design, molecular generation, multi-robot planning, and floorplan/scene synthesis even with projection or guidance. Such tasks combine hard-to-specify semantic goals with strict geometric or physical constraints (e.g., non-overlap, connectivity), yielding feasible solutions that lie on low-dimensional, small, and sometimes disconnected regions of the output space. This paper studies the failure mode through tangram generation from language, where seven fixed shapes must form a text-described silhouette while remaining connected and non-overlapping, and a simplified rectangle composition task with a learned bounding-box constraint. We find diffusion models struggle to satisfy constraints, consistent with difficulty generating samples near low-dimensional submanifolds. Motivated by locally feasible reparameterizations, we reformulate constrained generation as discrete autoregressive sequential generation. Reinforcement learning improves feasibility and task success, and Monte Carlo tree search quantifies the value of look-ahead when feasible regions shrink. Overall, the empirical, theoretical, and prior-work evidence points to a structural limitation of continuous density matching on this class of constrained-generation problems, and suggests sequential constraint-aware generation as a promising alternative.

扩散模型约束生成强化学习树搜索

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