用电子密度图提升分子性质预测,低数据时更高效,量子性质更准。
Beyond Atoms: Evaluating Electron Density Representation for 3D Molecular Learning
- 用体素化的电子密度、梯度和原子类型作输入,对比三种表示方法。
- 低数据下电子密度比原子类型表现更好,量子任务中精度显著提升。
- 适合需要高精度电子结构信息的分子建模与小样本学习场景。
3D分子性质预测的机器学习模型通常依赖原子级表示,可能忽略细微物理信息。电子密度图(来自X射线晶体学和冷冻电镜)提供了连续且物理基础的替代方案。本文在两个任务上比较了三种体素化输入:原子类型、原始电子密度和密度梯度幅值,分别用于蛋白质-配体结合亲和力预测(PDBbind)和量子性质预测(QM9)。由于电子密度本质为体数据,体素网格是最自然的表示方式。在完整数据下,所有方法表现相似;但在低数据条件下,基于密度的输入优于原子类型,而基于形状的基线表现相当,表明空间占据在此任务中占主导。在QM9任务中,尽管标签来自密度泛函理论(DFT),输入密度来自较低层级方法(XTB),密度输入仍优于原子输入,反映其蕴含丰富的结构与电子信息。总体表明,密度输入的效果取决于任务与数据规模,在亲和力预测中提升数据效率,在量子建模中提高精度。
原文摘要 · Abstract (English)
Machine learning models for 3D molecular property prediction typically rely on atom-based representations, which may overlook subtle physical information. Electron density maps -- the direct output of X-ray crystallography and cryo-electron microscopy -- offer a continuous, physically grounded alternative. We compare three voxel-based input types for 3D convolutional neural networks (CNNs): atom types, raw electron density, and density gradient magnitude, across two molecular tasks -- protein-ligand binding affinity prediction (PDBbind) and quantum property prediction (QM9). We focus on voxel-based CNNs because electron density is inherently volumetric, and voxel grids provide the most natural representation for both experimental and computed densities. On PDBbind, all representations perform similarly with full data, but in low-data regimes, density-based inputs outperform atom types, while a shape-based baseline performs comparably -- suggesting that spatial occupancy dominates this task. On QM9, where labels are derived from Density Functional Theory (DFT) but input densities from a lower-level method (XTB), density-based inputs still outperform atom-based ones at scale, reflecting the rich structural and electronic information encoded in density. Overall, these results highlight the task- and regime-dependent strengths of density-derived inputs, improving data efficiency in affinity prediction and accuracy in quantum property modeling.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。