cvignac/DiGress

解决的问题

DiGress 旨在生成合成图和分子结构。它通过将去噪扩散模型应用于离散空间，解决了生成真实离散图数据的挑战，从而能够生成复杂的网络和化学分子。

工作原理

该项目实现了离散去噪扩散模型。它可以在离散和连续两种模式下生成图。系统使用 Graph Transformer 架构来处理和生成这些结构，支持多 GPU 训练和自动数据集下载。

适用人群

适用于从事图生成、分子发现和离散结构生成式 AI 的研究人员和开发者。

主要亮点

• 支持用于图生成的离散和连续扩散模型。
• 兼容分子数据集（如 Guacamol 和 MOSES）和抽象图数据集（如 QM9 和 SBM）。
• 包含 Graph Transformer 实现。
• 支持多 GPU 加速训练和推理。