arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2305.09481q-bio.BMcs.LG
收藏

上下文丰富的分子表示改进少样本药物发现

Context-enriched molecule representations improve few-shot drug discovery

Johannes Kepler University(约翰内斯·开普勒大学);Merck Healthcare(默克医疗)

Johannes Schimunek, Philipp Seidl, Lukas Friedrich, Daniel Kuhn, Friedrich Rippmann, Sepp Hochreiter, Günter Klambauer

AI总结:

提出一种基于Modern Hopfield Network的少样本药物发现新方法,通过关联参考分子丰富分子表示,在FS-Mol基准上达到最先进水平。

摘要:

计算药物发现中的一个核心任务是从已知活性分子构建模型,以寻找更多有前景的分子用于后续筛选。然而,通常已知的活性分子非常少。因此,少样本学习方法有望提升药物发现过程这一关键阶段的有效性。我们引入了一种用于少样本药物发现的新方法。其主要思想是通过关于已知上下文或参考分子的知识来丰富分子表示。我们关于分子表示丰富的新颖概念是通过Modern Hopfield Network将来自支持集和查询集的分子与一大组参考(上下文)分子关联起来。直观地说,这一丰富步骤类似于人类专家将给定分子与其属性已知的熟悉分子联系起来。该丰富步骤强化并放大了数据的协方差结构,同时消除了由分子装饰引起的虚假相关性。我们的方法在FS-Mol基准数据集上与其他用于药物发现的少样本方法进行了比较。在FS-Mol上,我们的方法优于所有比较方法,因此为药物发现中的少样本学习设定了新的最先进水平。消融研究表明,我们方法的丰富步骤是提高预测质量的关键。在领域偏移实验中,我们进一步证明了该方法的鲁棒性。代码可在https://github.com/ml-jku/MHNfs获取。

英文摘要:

A central task in computational drug discovery is to construct models from known active molecules to find further promising molecules for subsequent screening. However, typically only very few active molecules are known. Therefore, few-shot learning methods have the potential to improve the effectiveness of this critical phase of the drug discovery process. We introduce a new method for few-shot drug discovery. Its main idea is to enrich a molecule representation by knowledge about known context or reference molecules. Our novel concept for molecule representation enrichment is to associate molecules from both the support set and the query set with a large set of reference (context) molecules through a Modern Hopfield Network. Intuitively, this enrichment step is analogous to a human expert who would associate a given molecule with familiar molecules whose properties are known. The enrichment step reinforces and amplifies the covariance structure of the data, while simultaneously removing spurious correlations arising from the decoration of molecules. Our approach is compared with other few-shot methods for drug discovery on the FS-Mol benchmark dataset. On FS-Mol, our approach outperforms all compared methods and therefore sets a new state-of-the art for few-shot learning in drug discovery. An ablation study shows that the enrichment step of our method is the key to improve the predictive quality. In a domain shift experiment, we further demonstrate the robustness of our method. Code is available at https://github.com/ml-jku/MHNfs.

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

↑