用多尺度专家路由提升分子三维属性预测精度
Topology-Aware Multiscale Mixture of Experts for Efficient Molecular Property Prediction
- 根据距离分段设计专家集群,灵活捕捉长短程相互作用
- 通过拓扑特征动态分配专家,适应不同几何结构
- 可嵌入主流模型,在多种分子数据集上稳定增效
许多分子性质依赖于三维几何结构,其中非共价相互作用、立体化学效应及中长程力由空间距离和角度决定,无法仅通过二维键图唯一表征。然而,现有3D分子图神经网络仍依赖全局固定的邻域启发式规则(如距离截断和最大邻居数)定义消息传递邻域,导致交互预算僵化且与数据无关。本文提出多尺度交互混合专家(MI-MoE),实现几何环境下交互建模的自适应。贡献有三:(1) 设计基于距离截断的专家集合,无需固定截断值即可显式捕捉短、中、长程相互作用;(2) 构建基于滤波的拓扑门控编码器,利用持久同调特征等描述连通性随半径演变的方式,动态路由输入至专家;(3) 验证MI-MoE作为即插即用模块,可在多个强基准3D分子骨干模型上持续提升性能,覆盖分子与聚合物性质预测的回归与分类任务。结果表明,拓扑感知的多尺度路由是3D分子图学习的有效范式。
原文摘要 · Abstract (English)
Many molecular properties depend on 3D geometry, where non-covalent interactions, stereochemical effects, and medium- to long-range forces are determined by spatial distances and angles that cannot be uniquely captured by a 2D bond graph. Yet most 3D molecular graph neural networks still rely on globally fixed neighborhood heuristics, typically defined by distance cutoffs and maximum neighbor limits, to define local message-passing neighborhoods, leading to rigid, data-agnostic interaction budgets. We propose Multiscale Interaction Mixture of Experts (MI-MoE) to adapt interaction modeling across geometric regimes. Our contributions are threefold: (1) we introduce a distance-cutoff expert ensemble that explicitly captures short-, mid-, and long-range interactions without committing to a single cutoff; (2) we design a topological gating encoder that routes inputs to experts using filtration-based descriptors, including persistent homology features, summarizing how connectivity evolves across radii; and (3) we show that MI-MoE is a plug-in module that consistently improves multiple strong 3D molecular backbones across diverse molecular and polymer property prediction benchmark datasets, covering both regression and classification tasks. These results highlight topology-aware multiscale routing as an effective principle for 3D molecular graph learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。