跳转到正文

MolGPT

MolGPT 使用 MOSES 和 Guacamol 数据集,通过下一词元预测训练一个小型自定义 GPT,用于无条件和有条件分子生成,并链接了训练后的权重和基于 Ecco 的显著性分析。

条目更新 ·

概览

MolGPT 是一个分子生成模型项目,核心是通过下一词元预测训练的小型自定义 GPT。README 描述了使用 MOSES 和 Guacamol 数据集进行训练,并使用训练得到的模型进行无条件和有条件分子生成。其引用的预印本将该方法称为 transformer 解码器模型。在研究工作流中,该仓库可作为探究这种生成方法的入口,而不是一个有完整端到端文档的分子设计服务。

文档中说明的训练输入是经过处理的 CSV 格式数据集文件,必须放在与代码文件相同的目录中。README 链接了这些处理后的数据集、其上游 MOSES 和 Guacamol 仓库,以及单独的训练权重集合。README 列出了针对不同数据集的训练脚本和两个生成脚本。分子生成是明确说明的输出任务,但来源摘录没有说明生成结果的表示形式、输出文件格式或如何提供条件输入。

所述工作还包括可解释性:使用 Ecco 库生成显著性图。README 表示 MolGPT 与此前的方法在两个数据集上进行了比较,但未提供数值结果或评估方案。因此,现有证据支持训练、生成和显著性分析工作流,但无法据此确认模型性能、化学有效性或生成候选分子的实际效用。来源摘录未说明安装要求、硬件需求或详细参数配置。

主要功能

  • 使用 MOSES 和 Guacamol 数据集,以下一词元预测目标训练的小型自定义 GPT。
  • 按 README 所述,支持无条件和有条件分子生成。
  • 提供单独的训练入口 `train_moses.sh` 和 `train_guacamol.sh`。
  • 列出生成入口 `generate_guacamol_prop.sh` 和 `generate_moses_prop_scaf.sh`。
  • 链接处理后的 CSV 数据集和单独的训练模型权重集合。
  • 介绍基于 Ecco 的显著性图,用于模型可解释性分析。

使用场景

  • 拟评估用途:使用所链接的数据集和训练权重,研究无条件和有条件分子生成。
  • 拟评估用途:研究针对 MOSES 和 Guacamol 的数据集专属训练工作流,并依据独立制定的评估方案评估生成候选分子。
  • 拟评估用途:探索基于 Ecco 的显著性分析,以考察模型在分子生成过程中的行为。

使用方式

  1. 阅读官方 README,区分训练和生成入口。所提供的说明未指定依赖项、硬件要求或配置详情。
  2. 从所链接的下载文件夹获取处理后的 CSV 数据集。有关数据集背景,请参阅上游 Guacamol 和 MOSES 仓库。
  3. 按照说明,将所选数据集 CSV 放在与代码文件相同的目录中以进行训练。在仓库中查看 train_moses.sh 或 train_guacamol.sh,然后再选择相应的工作流。
  4. 如需基于权重进行评估,请参阅训练权重集合。查看生成脚本,以确定其预期输入和权重加载配置;摘录中未提供这些详情。
  5. 选择相关的已记录生成入口,并规划对其输出进行独立检查。如需研究可解释性,请查看仓库中与 Ecco 相关的实现;README 提到了显著性图,但未说明具体流程。

相关资源

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 的实用规划示例。