提出相似性感知评估框架,解决药物靶点亲和力预测泛化性误导问题
Rethinking the generalization of drug target affinity prediction algorithms via similarity aware evaluation
- 设计新型分组方法,使测试集分布可定制且避免高相似样本主导
- 在4个数据集上验证,模型在低相似性样本上性能下降超50%
- 适合关注真实场景泛化能力的药物发现研究者
药物-靶点结合亲和力预测是药物发现的基础任务,已有大量研究并报告了良好结果。然而本文指出,现有结果可能具有误导性,难以真实推广。核心发现是:传统随机划分测试集时,测试集仍被高度相似于训练集的样本主导。模型在低相似性样本上的性能严重下降,但该缺陷在当前评估中被严重忽视。因此,当模型遇到低相似性样本时,性能不可信。为此,我们提出相似性感知评估框架,通过优化问题建模实现任意目标分布的分组,利用梯度下降近似高效求解。我们在四个数据集上对五种代表性方法进行两阶段典型靶点评估,结果表明所提分组方法能显著匹配目标分布,有效引导模型发展。代码已公开于https://github.com/Amshoreline/SAE/tree/main。
原文摘要 · Abstract (English)
Drug-target binding affinity prediction is a fundamental task for drug discovery. It has been extensively explored in literature and promising results are reported. However, in this paper, we demonstrate that the results may be misleading and cannot be well generalized to real practice. The core observation is that the canonical randomized split of a test set in conventional evaluation leaves the test set dominated by samples with high similarity to the training set. The performance of models is severely degraded on samples with lower similarity to the training set but the drawback is highly overlooked in current evaluation. As a result, the performance can hardly be trusted when the model meets low-similarity samples in real practice. To address this problem, we propose a framework of similarity aware evaluation in which a novel split methodology is proposed to adapt to any desired distribution. This is achieved by a formulation of optimization problems which are approximately and efficiently solved by gradient descent. We perform extensive experiments across five representative methods in four datasets for two typical target evaluations and compare them with various counterpart methods. Results demonstrate that the proposed split methodology can significantly better fit desired distributions and guide the development of models. Code is released at https://github.com/Amshoreline/SAE/tree/main.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。