【ICML2023】几何潜在扩散模型的三维分子生成

生成模型，尤其是扩散模型（DMs），在生成具有丰富特征的几何形状和推进分子设计等基础科学问题方面取得了令人瞩目的成果。受到近期稳定（潜在）扩散模型巨大成功的启发，我们提出了一种名为几何潜在扩散模型（GEOLDM）的新颖且有原则的3D分子生成方法。GEOLDM是第一个用于分子几何领域的潜在DM模型，由将结构编码为连续潜在代码的自动编码器和在潜在空间中操作的DMs组成。我们的关键创新是，在对3D分子几何进行建模时，我们通过构建具有不变标量和等变张量的点结构潜在空间来捕捉其关键的旋转-平移等变性约束。大量实验证明，GEOLDM在多个分子生成基准测试上的性能均有所提高，对于大型生物分子的有效百分比提高了高达7%。结果还表明，GEOLDM在可控生成方面具有更高的能力，这得益于潜在建模。代码可在以下地址获取：https://github.com/MinkaiXu/GeoLDM。