cvignac/DiGress

code for the paper "DiGress: Discrete Denoising diffusion for graph generation"

解決的問題

DiGress 是為了生成合成圖和分子結構而設計的。它透過將去噪擴散模型應用於離散空間,解決了生成真實離散圖資料的挑戰,從而能夠生成複雜的網路與化學分子。

工作原理

此專案實作了離散去噪擴散模型。它可在離散與連續兩種模式下生成圖。系統使用 Graph Transformer 架構來處理與生成這些結構,支援多 GPU 訓練與自動資料集下載。

適用對象

適用於從事圖生成、分子發現與離散結構生成式 AI 的研究人員與開發者。

主要亮點

  • 支援用於圖生成的離散與連續擴散模型。
  • 兼容分子資料集(如 Guacamol 和 MOSES)與抽象圖資料集(如 QM9 和 SBM)。
  • 包含 Graph Transformer 實作。
  • 支援多 GPU 加速訓練與推論。

相關

  • 專案
  • 專案
  • 專案
  • 專案