利用3D分子信息与对比学习,提升小样本下的药物分子搜索效果。
S-MolSearch: 3D Semi-supervised Contrastive Learning for Bioactive Molecule Search
- 基于3D分子结构和半监督对比学习,结合逆最优传输思想生成软标签。
- 在LIT-PCBA和DUD-E数据集上,AUROC、BEDROC、EF均超越现有方法。
- 适合药物发现早期筛选,尤其适用于标注数据少、噪声多的场景。
虚拟筛选是药物研发早期的关键技术,旨在从庞大的分子库中识别有潜力的候选药物。近年来,基于配体的虚拟筛选因无需依赖特定蛋白结合位点信息,在大规模数据库筛查中备受关注。然而,复合物结合亲和力数据获取成本高,导致可用数据量有限,覆盖化学空间较小,且存在大量不一致噪声。如何在数据增强中保持分子活性的完整性,仍缺乏稳定归纳偏置。为此,我们提出S-MolSearch——首个融合分子3D信息与亲和力信息的半监督对比学习框架,用于基于配体的虚拟筛选。基于逆最优传输原理,S-MolSearch高效处理有标签与无标签数据,训练分子结构编码器的同时为无标签数据生成软标签,实现学习过程中对无标签数据的自适应利用。实验证明,S-MolSearch在主流基准LIT-PCBA和DUD-E上表现优异,其AUROC、BEDROC和EF均优于现有结构基与配体基方法。
原文摘要 · Abstract (English)
Virtual Screening is an essential technique in the early phases of drug discovery, aimed at identifying promising drug candidates from vast molecular libraries. Recently, ligand-based virtual screening has garnered significant attention due to its efficacy in conducting extensive database screenings without relying on specific protein-binding site information. Obtaining binding affinity data for complexes is highly expensive, resulting in a limited amount of available data that covers a relatively small chemical space. Moreover, these datasets contain a significant amount of inconsistent noise. It is challenging to identify an inductive bias that consistently maintains the integrity of molecular activity during data augmentation. To tackle these challenges, we propose S-MolSearch, the first framework to our knowledge, that leverages molecular 3D information and affinity information in semi-supervised contrastive learning for ligand-based virtual screening. Drawing on the principles of inverse optimal transport, S-MolSearch efficiently processes both labeled and unlabeled data, training molecular structural encoders while generating soft labels for the unlabeled data. This design allows S-MolSearch to adaptively utilize unlabeled data within the learning process. Empirically, S-MolSearch demonstrates superior performance on widely-used benchmarks LIT-PCBA and DUD-E. It surpasses both structure-based and ligand-based virtual screening methods for AUROC, BEDROC and EF.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。