arXiv:2503.13322cs.LG2025-03被引 3

基于药物结构的多模态模型,提升药物重定位与冷启动预测能力

SMPR: A structure-enhanced multimodal drug-disease prediction model for drug repositioning and cold start

  • 融合药物分子结构与疾病异构网络,生成双向嵌入表示
  • 重定位AUC达99%,冷启动AUC达80%,正样本召回超70%
  • 支持本地部署,适合需快速预测新药关联的科研人员

药物重定位研究持续活跃,但实际生物学验证案例仍有限,现有模型未能充分挖掘药物结构信息,且多数仅用于构建关系矩阵,难以应对药物冷启动问题。本文提出结构增强型多模态药物-疾病关系预测模型(SMPR)。该模型基于药物的SMILES结构,采用Mol2Vec生成药物嵌入表示,并通过异构网络图神经网络学习疾病嵌入表示,最终构建药物-疾病关系矩阵。为降低使用门槛,SMPR还提供基于结构相似性的冷启动接口,可快速预测新药相关疾病。模型在多个维度验证了重定位与冷启动能力:重定位的AUC和ACUPR分别达到99%和61%;冷启动AUC为80%,召回率超过70%,表明对正样本更敏感。案例分析证实模型实用性,可视化展示结构增强对性能的提升。为便于使用,已提供本地部署版本及可执行程序包。

原文摘要 · Abstract (English)

Repositioning drug-disease relationships has always been a hot field of research. However, actual cases of biologically validated drug relocation remain very limited, and existing models have not yet fully utilized the structural information of the drug. Furthermore, most repositioning models are only used to complete the relationship matrix, and their practicality is poor when dealing with drug cold start problems. This paper proposes a structure-enhanced multimodal relationship prediction model (SMRP). SMPR is based on the SMILE structure of the drug, using the Mol2VEC method to generate drug embedded representations, and learn disease embedded representations through heterogeneous network graph neural networks. Ultimately, a drug-disease relationship matrix is constructed. In addition, to reduce the difficulty of users' use, SMPR also provides a cold start interface based on structural similarity based on reposition results to simply and quickly predict drug-related diseases. The repositioning ability and cold start capability of the model are verified from multiple perspectives. While the AUC and ACUPR scores of repositioning reach 99% and 61% respectively, the AUC of cold start achieve 80%. In particular, the cold start Recall indicator can reach more than 70%, which means that SMPR is more sensitive to positive samples. Finally, case analysis is used to verify the practical value of the model and visual analysis directly demonstrates the improvement of the structure to the model. For quick use, we also provide local deployment of the model and package it into an executable program.

药物重定位冷启动图神经网络多模态建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。