arXiv:2607.07611cs.LGphysics.comp-ph2026-07

用新型损失函数提升药物相互作用预测准确率

Asymmetric Focal Loss Improves Graph Neural Network Prediction of Drug-Drug Interactions

  • 设计不对称焦点损失,聚焦难分类的正样本
  • 准确率和F1提升近20个百分点,误报率降为9.1%
  • 无需改模型结构,适合医疗用药安全研究者

图神经网络可提升多药副作用的计算预测,但标准二元交叉熵训练对易分与难分样本投入相同精力,可能遗漏临床重要相互作用。本文采用一种不对称焦点损失(ClinicalFocal loss),集成于关系感知图卷积网络,利用分子指纹、理化描述符和学习嵌入进行药物相互作用(DDI)预测。在TWOSIDES数据集上,五折交叉验证中,模型使用相同架构、特征、数据划分、超参数与随机种子,对比基线。结果表明,该方法将准确率从0.699提升至0.892(+19.3百分点),F1从0.700升至0.894(+19.4百分点)。AUROC从0.766增至0.914,AUCPR从0.714升至0.860。假阴性率由29.8%降至9.1%,特异性从69.6%升至87.5%。总体分类错误率从30.1%降至10.8%,相对减少64.1%。各折结果一致。结论:通过优化损失函数即可显著提升图神经网络在DDI预测中的表现,且达到90.9%召回率,无需修改模型结构。

原文摘要 · Abstract (English)

Background: Graph neural networks improve computational prediction of polypharmacy side effects, but standard binary cross-entropy training allocates equal capacity to well-classified and difficult examples, potentially missing clinically significant interactions. We evaluated whether an asymmetric focal objective could improve multi-relational drug-drug interaction (DDI) prediction by emphasizing difficult positive interactions. Methods: ClinicalFocal loss was integrated into a relation-aware graph convolutional network using molecular fingerprints, physicochemical descriptors, and learned embeddings. The model was evaluated on TWOSIDES using five-fold cross-validation with identical experimental conditions (architecture, features, data partitions, hyperparameters, and random seeds) for ClinicalFocal loss and binary cross-entropy baseline. Results: ClinicalFocal loss increased accuracy from 0.699 to 0.892 (+19.3 percentage points) and F1 score from 0.700 to 0.894 (+19.4 percentage points). AUROC increased from 0.766 to 0.914, and AUCPR increased from 0.714 to 0.860. The false-negative rate decreased from 29.8% to 9.1%, while specificity increased from 69.6% to 87.5%. Overall classification error decreased from 30.1% to 10.8%, corresponding to a 64.1% relative reduction. Improvements were consistent across all five folds. Conclusions: Asymmetric focal optimization improved classification and ranking performance while achieving 90.9% recall for observed interaction triples, without modifying the underlying architecture. Loss-function design is a direct, tunable lever for improving graph-based DDI prediction.

药物相互作用图神经网络损失函数医疗AI

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