arXiv:2510.10402cs.LGcs.AI2025-10中稿 · WWW 2026被引 5

用树搜索引导扩散模型,实现精准可控的图生成。

Controllable Graph Generation with Diffusion Models via Inference-Time Tree Search Guidance

  • 引入树搜索与双空间去噪,扩展生成路径并保持结构准确
  • 在分子生成任务中达到当前最优,且计算越多效果越好
  • 适合需要精细控制图结构的应用,如药物设计

图生成是图学习中的基础问题,广泛应用于大规模网络、知识图谱及药物与材料发现等领域。现有扩散模型虽能逐步生成图,但无条件生成难以控制目标属性,常导致质量不稳定且难融入新目标。推理阶段引导方法虽可调整采样过程而无需重训练,但仍局限于局部、启发式且控制能力有限。为此,我们提出TreeDiff,一种基于蒙特卡洛树搜索(MCTS)的双空间扩散框架,用于可控图生成。TreeDiff为即插即用的推理阶段方法,可扩大搜索空间同时保持计算可行性。具体包含三项设计:(1) 宏步扩展策略,将多个去噪步骤合并为单一转移,降低树深度,支持长时程探索;(2) 双空间去噪机制,结合高效的隐空间去噪与轻量的图空间离散修正,兼顾可扩展性与结构保真度;(3) 双空间验证器,从部分去噪图中预测长期奖励,实现早期价值评估,避免完整回滚。在二维和三维分子生成基准上,无论无条件或有条件设置,实验均表明TreeDiff达到当前最佳性能。尤其值得注意的是,其推理时间具有良好扩展性:随着计算资源增加持续提升,而现有方法在资源受限下迅速饱和。

原文摘要 · Abstract (English)

Graph generation is a fundamental problem in graph learning with broad applications across Web-scale systems, knowledge graphs, and scientific domains such as drug and material discovery. Recent approaches leverage diffusion models for step-by-step generation, yet unconditional diffusion offers little control over desired properties, often leading to unstable quality and difficulty in incorporating new objectives. Inference-time guidance methods mitigate these issues by adjusting the sampling process without retraining, but they remain inherently local, heuristic, and limited in controllability. To overcome these limitations, we propose TreeDiff, a Monte Carlo Tree Search (MCTS) guided dual-space diffusion framework for controllable graph generation. TreeDiff is a plug-and-play inference-time method that expands the search space while keeping computation tractable. Specifically, TreeDiff introduces three key designs to make it practical and scalable: (1) a macro-step expansion strategy that groups multiple denoising updates into a single transition, reducing tree depth and enabling long-horizon exploration; (2) a dual-space denoising mechanism that couples efficient latent-space denoising with lightweight discrete correction in graph space, ensuring both scalability and structural fidelity; and (3) a dual-space verifier that predicts long-term rewards from partially denoised graphs, enabling early value estimation and removing the need for full rollouts. Extensive experiments on 2D and 3D molecular generation benchmarks, under both unconditional and conditional settings, demonstrate that TreeDiff achieves state-of-the-art performance. Notably, TreeDiff exhibits favorable inference-time scaling: it continues to improve with additional computation, while existing inference-time methods plateau early under limited resources.

图生成扩散模型可控生成分子设计

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