Smiles2Class 直接从 SMILES 预测化合物所属的分子类别;一个结构可以同时属于多个类别。

输入与输出

  • 输入:一条 SMILES。
  • 输出:426 个类别各自的存在概率。每类是独立的二分类判断,不是互斥的单一归属。

类别体系

426 个类别取自 HMDB 的 Alternative Parents 注释,均为 ClassyFire 术语。类别词表及其顺序与模型固定绑定,因此输出第 k 列始终对应同一类别。

模型

  • 主模型:15 层、512 维的 Uni-Mol 风格编码器,接 3 层 Transformer 解码器(426 个类别查询);每个查询投影为“存在/不存在”的二分类 softmax。
  • 所用编码器为 UniMol-Lite,即 Uni-Mol 架构的轻量实现(MIT 许可)。
  • 另一种实现:在冻结的 512 维嵌入上,为每个类别训练一个 AutoGluon 二分类器。

训练与评估

144,665
HMDB 来源结构
426
类别(多标签)
28,935
留出测试分子

训练 / 验证 / 测试划分为 86,685 / 28,935 / 28,935 个分子,按类别划分。每个类别作为独立的 one-vs-rest 问题评估:AUROC 与平均精度用连续分数计算;精确率、召回率、F1 与平衡准确率按 0.5 阈值计算。

与 NPClassifier 的比较

两套工具使用不同的类别体系(这里为 ClassyFire 术语,NPClassifier 为天然产物分类),共有 22 个同名类别;因此比较只针对这 22 类,并在 28,935 个分子的共同队列上进行。NPClassifier 只有 V1 可公开分发,且其分数与概率尺度不同,故按其遗留阈值 1×10⁻⁷ 报告——这是 V1 可用的最有利设定。

使用位置

  • 网页平台:AI Tools 模块。
  • 代码:smiles2class(MIT 许可)。