arXiv:2509.04601cs.LGcs.AI2025-09被引 2

用量子特征和自适应加权提升药物毒性和药代动力学预测精度

Quantum-Enhanced Multi-Task Learning with Learnable Weighting for Pharmacokinetic and Toxicity Prediction

  • 融合量子化学描述符与可学习权重,实现多任务联合训练
  • 13个TDC基准中12项超越单任务模型,推理更快且模型更轻
  • 首次在标准评测下系统验证量子增强多任务学习的有效性

ADMET(吸收、分布、代谢、排泄和毒性)预测在药物研发中至关重要,能加速新药筛选与优化。现有方法多依赖单任务学习(STL),难以充分利用任务间的互补性,且训练和推理需更多计算资源。为此,我们提出一种新型统一的量子增强任务加权多任务学习框架(QW-MTL),专为ADMET分类设计。基于Chemprop-RDKit架构,QW-MTL引入量子化学描述符,丰富分子表征中的电子结构与相互作用信息;同时提出一种新的指数型任务加权机制,结合数据集规模先验与可学习参数,实现任务间动态损失平衡。据我们所知,这是首个在全部13个治疗数据共同体(TDC)分类基准上进行联合多任务训练的工作,采用排行榜式数据划分以确保标准化、真实的评估设置。大量实验表明,QW-MTL在13个任务中的12个上显著优于单任务基线,在保持极低模型复杂度的同时实现高效推理,证明了量子信息引导的多任务学习在分子预测中的有效性与效率。

原文摘要 · Abstract (English)

Prediction for ADMET (Absorption, Distribution, Metabolism, Excretion, and Toxicity) plays a crucial role in drug discovery and development, accelerating the screening and optimization of new drugs. Existing methods primarily rely on single-task learning (STL), which often fails to fully exploit the complementarities between tasks. Besides, it requires more computational resources while training and inference of each task independently. To address these issues, we propose a new unified Quantum-enhanced and task-Weighted Multi-Task Learning (QW-MTL) framework, specifically designed for ADMET classification tasks. Built upon the Chemprop-RDKit backbone, QW-MTL adopts quantum chemical descriptors to enrich molecular representations with additional information about the electronic structure and interactions. Meanwhile, it introduces a novel exponential task weighting scheme that combines dataset-scale priors with learnable parameters to achieve dynamic loss balancing across tasks. To the best of our knowledge, this is the first work to systematically conduct joint multi-task training across all 13 Therapeutics Data Commons (TDC) classification benchmarks, using leaderboard-style data splits to ensure a standardized and realistic evaluation setting. Extensive experimental results show that QW-MTL significantly outperforms single-task baselines on 12 out of 13 tasks, achieving high predictive performance with minimal model complexity and fast inference, demonstrating the effectiveness and efficiency of multi-task molecular learning enhanced by quantum-informed features and adaptive task weighting.

药物发现多任务学习量子化学ADMET预测

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