arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2503.03360cs.LGcs.AIcs.CL
收藏

用于分子性质预测的Transformer:领域自适应有效提升性能

Transformers for molecular property prediction: Domain adaptation efficiently improves performance

Spoken Language Systems, Saarland Informatics Campus, Saarland University(spoken language systems,萨尔兰信息学校园,萨尔兰大学);Medical Faculty, Saarland University(医学学院,萨尔兰大学);Center for Bioinformatics, Saarland University(生物信息学中心,萨尔兰大学);Data Driven Drug Design, Center for Bioinformatics, Saarland University(数据驱动药物设计,生物信息学中心,萨尔兰大学)

Afnan Sultan, Max Rausch-Dupont, Shahrukh Khan, Olga Kalinina, Dietrich Klakow, Andrea Volkamer

AI总结:

本研究评估Transformer在分子性质预测中的表现,发现大规模预训练数据量超过40万分子后性能不再提升,而基于小规模领域数据(≤4K分子)的多任务回归自适应能显著提升性能,并优于MolFormer、媲美MolBERT。

摘要:

在过去六年中,分子Transformer模型已成为药物发现中的关键工具。大多数现有模型在大型无标注数据集(如ZINC或ChEMBL)上进行预训练。然而,大规模预训练在多大程度上改善了分子性质预测仍不清楚。本研究评估了Transformer模型在此任务上的表现,同时解决其局限性。我们探讨了预训练数据集规模和化学信息目标对性能的影响。我们的结果表明,将来自大规模无标注数据库的数据集从约40万增加到80万分子,并未在覆盖五个ADME终点(即脂溶性、渗透性、溶解度(两个数据集)、微粒体稳定性(两个数据集)和血浆蛋白结合)的七个数据集上提升性能。相比之下,在小型领域特定数据集(小于等于4000分子)上使用物理化学性质的多任务回归进行领域自适应,显著提升了性能(P值小于0.001)。一个在40万分子上预训练并经过领域特定数据自适应的模型,优于MolFormer等更大模型,并与MolBERT表现相当。与使用描述符和Morgan指纹的随机森林(RF)基线进行基准比较表明,化学和物理信息特征在各种模型类型中始终产生更好的性能。虽然RF仍是强基线,但我们确定了提升Transformer性能的具体实践。将预训练和自适应与化学意义任务及领域相关数据对齐,为分子性质预测提供了一个有前景的方向。我们的模型可在HuggingFace上获取,便于使用和自适应。

英文摘要:

Over the past six years, molecular transformer models have become key tools in drug discovery. Most existing models are pre-trained on large, unlabeled datasets such as ZINC or ChEMBL. However, the extent to which large-scale pre-training improves molecular property prediction remains unclear. This study evaluates transformer models for this task while addressing their limitations. We explore how pre-training dataset size and chemically informed objectives impact performance. Our results show that increasing the dataset beyond approximately 400K to 800K molecules from large-scale unlabeled databases does not enhance performance across seven datasets covering five ADME endpoints: lipophilicity, permeability, solubility (two datasets), microsomal stability (two datasets), and plasma protein binding. In contrast, domain adaptation on a small, domain-specific dataset (less than or equal 4K molecules) using multi-task regression of physicochemical properties significantly boosts performance (P-value less than 0.001). A model pre-trained on 400K molecules and adapted with domain-specific data outperforms larger models such as MolFormer and performs comparably to MolBERT. Benchmarks against Random Forest (RF) baselines using descriptors and Morgan fingerprints show that chemically and physically informed features consistently yield better performance across model types. While RF remains a strong baseline, we identify concrete practices to enhance transformer performance. Aligning pre-training and adaptation with chemically meaningful tasks and domain-relevant data presents a promising direction for molecular property prediction. Our models are available on HuggingFace for easy use and adaptation.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑