跳转到正文

JT-VAE

JT-VAE 是 Junction Tree Variational Autoencoder 的官方实现,用于分子图生成,包含 VAE 训练代码,以及用于贝叶斯优化和联合训练性质预测器的脚本。

条目更新 ·

概览

JT-VAE 提供了用于分子图生成的 Junction Tree Variational Autoencoder 官方实现,相关论文链接为 https://arxiv.org/abs/1802.04364。其工作流程面向分子图生成及相关模型训练实验。README 将分子图列为生成目标,但来源摘录未说明数据集准备、可接受的输入格式、输出序列化方式或检查点是否可用。

该仓库将模型实现与训练及实验脚本分开。README 指引用户查看加速实现 fast_jtnn/ 和 VAE 训练代码 fast_molvae/。文档还列出了支持原始 ICML 论文相关实验的目录:bo/ 用于贝叶斯优化,molvae/ 用于仅训练 VAE,molopt/ 用于联合训练 VAE 和性质预测器,jtnn/ 用于模型形式。这些是文档所述的工作流程组件,并不能证明相关结果已复现。

列出的要求包括 Linux、Python 2.7、RDKit 2017.09 或更高版本,以及 PyTorch 0.2 或更高版本;作者称只在 Ubuntu 上测试过。规划评估环境时应考虑这些历史要求。README 还建议使用独立的 hgraph2graph 仓库来改进架构。其所述的 ChEMBL 预训练模型和性质引导生成脚本不应归入此 JT-VAE 仓库。虽然 README 称较新的实现经过加速,但所提供的证据没有给出数值速度对比。

主要功能

  • 用于分子图生成的 Junction Tree Variational Autoencoder 官方实现,并附有研究论文链接。
  • `fast_jtnn/` 中的加速模型实现,以及配套的 `fast_molvae/` VAE 训练代码。
  • `molvae/` 中用于原始 ICML 论文相关实验的仅训练 VAE 脚本。
  • `molopt/` 中用于联合训练 VAE 和性质预测器的脚本。
  • `bo/` 中的贝叶斯优化实验脚本;主文档提到了该目录的专属 README。

使用场景

  • 预期评估用途:使用文档所述的实现和训练流程,研究 JT-VAE 作为分子图生成模型的表现。
  • 预期评估用途:通过 `molopt/` 训练脚本考察联合表示学习与分子性质预测。
  • 预期评估用途:研究仓库中的贝叶斯优化实验,将其作为分子优化研究流程的起点。

使用方式

  1. 阅读官方 README和关联论文,了解模型用途,并区分实现说明与研究结论。
  2. 评估所列环境要求:Linux、Python 2.7、RDKit >= 2017.09 和 PyTorch >= 0.2。查阅所链接的 RDKit 安装文档;README 建议使用 conda,并说明仅在 Ubuntu 上测试过。
  3. 在仓库中检查 fast_jtnn/,并在尝试加速 VAE 训练前阅读 fast_molvae/README.md。应在那里确认输入准备和输出内容,不要自行假定格式。
  4. 根据评估目标选择实验目录:bo/、molvae/ 或 molopt/。阅读相应 README 了解实际流程;来源摘录没有列出命令。
  5. 查看仓库代码许可证。如果需要了解架构改进,请单独检查 hgraph2graph,不要将其功能视为 JT-VAE 的特性。

相关资源

AIDDISON Explorer 是一个托管式药物发现平台,可根据目标特性要求、设计约束、预测属性和合成可行性生成并排序分子候选物。

分子生成 · 药物发现

Chemistry42 是一个小分子发现平台,将生成式设计、逆合成、ADMET 与选择性预测以及基于物理学的优先级排序结合起来,用于命中识别和先导化合物优化。

分子生成 · 药物发现

用于生成 3D 分子的 E(3) 等变扩散模型官方研究代码,包含 QM9 和 GEOM-Drugs 训练工作流、样本分析及性质条件生成。

开源Python

分子生成

FEgrow

开源工具

FEgrow 支持交互式构建配体系列以用于自由能计算准备,并提供有文档说明的主动学习示例,以及用于分子设计工作流的基于 Dask 的加速。

开源

分子生成 · 药物发现

GeoDiff

模型

GeoDiff 是一种用于分子构象生成的几何扩散模型,提供官方代码,支持基于 GEOM 的训练、检查点采样,以及构象和性质评估。

开源Python

分子生成 · 计算化学

GEOM

数据集

GEOM 提供 3,700 万个分子构象,涵盖超过 450,000 个分子,并标注能量和统计权重;数据包含 MessagePack、RDKit 对象以及加载和分析教程。

Python

分子生成 · 计算化学

相关指南

工作流

AI 药物发现工具指南:Agent、MCP 与科研平台

按工作流程阶段选择工具:靶点证据、蛋白质结构、已知结合位点、分子生成、性质分析和筛选。比较研究智能体、MCP 集成、托管 API 与平台,并参考一个拟议的端到端示例和实用评估关卡。

工作流

化学 AI 工作流的可复现性

构建精简的评估记录,将科学问题与数据来源及处理记录、转换步骤、模型配置、输出和失败情况关联起来,并提供化学 AI 的实用规划示例。