通过分层交互机制提升分子属性预测准确率
Learning Hierarchical Interaction for Accurate Molecular Property Prediction
- 设计分层注意力消息传递机制,融合原子、基团和分子多层级特征
- 在11个数据集上表现优异,尤其在血脑屏障通透性等关键任务中领先
- 具备化学直觉契合的可解释性,适合药物研发早期决策
发现具有理想分子属性(如ADMET特性)的分子在新药研发中至关重要。现有方法通常使用图神经网络(GNNs)和Transformer等深度学习模型,从多样化的化学信息中学习预测这些属性,但往往无法有效捕捉分子结构的层次特性,且缺乏多层级特征间的有效交互机制。为此,我们提出分层交互消息传递机制,作为新型模型Hierarchical Interaction Message Net(HimNet)的基础。该方法通过分层注意力引导的消息传递,在原子、基团和分子三个层次实现交互感知的表征学习,有效平衡全局与局部信息,确保下游任务如血脑屏障通透性(BBBP)预测中提取丰富且任务相关的特征。我们在11个数据集上系统评估了HimNet,包括8个常用的MoleculeNet基准数据集及3个高价值挑战性数据集(代谢稳定性、疟疾活性、肝微粒体清除率),覆盖广泛的药理相关属性。大量实验表明,HimNet在多数分子属性预测任务中达到最佳或接近最佳性能。此外,该方法展现出良好的分层可解释性,与代表性分子的化学直觉高度一致。我们认为,HimNet为分子活性与ADMET属性预测提供了一种准确高效的解决方案,有助于推动新药研发早期阶段的智能决策。
原文摘要 · Abstract (English)
Discovering molecules with desirable molecular properties, including ADMET profiles, is of great importance in drug discovery. Existing approaches typically employ deep learning models, such as Graph Neural Networks (GNNs) and Transformers, to predict these molecular properties by learning from diverse chemical information. However, these models often fail to efficiently capture and utilize the hierarchical nature of molecular structures, and often lack mechanisms for effective interaction among multi-level features. To address these limitations, we propose a Hierarchical Interaction Message Passing Mechanism, which serves as the foundation of our novel model, the Hierarchical Interaction Message Net (HimNet). Our method enables interaction-aware representation learning across atomic, motif, and molecular levels via hierarchical attention-guided message passing. This design allows HimNet to effectively balance global and local information, ensuring rich and task-relevant feature extraction for downstream property prediction tasks, such as Blood-Brain Barrier Permeability (BBBP). We systematically evaluate HimNet on eleven datasets, including eight widely-used MoleculeNet benchmarks and three challenging, high-value datasets for metabolic stability, malaria activity, and liver microsomal clearance, covering a broad range of pharmacologically relevant properties. Extensive experiments demonstrate that HimNet achieves the best or near-best performance in most molecular property prediction tasks. Furthermore, our method exhibits promising hierarchical interpretability, aligning well with chemical intuition on representative molecules. We believe that HimNet offers an accurate and efficient solution for molecular activity and ADMET property prediction, contributing to advanced decision-making in the early stages of drug discovery.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。