Skip to content

Atomic Uncertainty Pinpoints Critical Failure Structures for Trustworthy Molecular Property Prediction

Jul 2026 · Journal of Chemical Information and Modeling · Vol 66, pp. 8076-8086 · 0 citations · 26 references
Medicine Computer Science

TL;DR

AUINet is introduced, an atomic uncertainty-aware iterative network that quantifies and refines uncertainty at the atomic scale and establishes a new paradigm for trustworthy molecular property prediction.

Abstract

Accurate prediction of molecular properties is essential for accelerating drug discovery, yet current deep learning methods generally lack reliable uncertainty estimates, particularly at the atomic level. A limitation of current approaches is the reliance on global molecular uncertainty, which frequently masks localized ambiguities by averaging the uncertainty across the entire molecule. To address this, we introduce AUINet, an atomic uncertainty-aware iterative network that quantifies and refines uncertainty at the atomic scale. Built on a D-MPNN architecture, AUINet uses Monte Carlo dropout to estimate atom-level uncertainty and iteratively refines atomic features through uncertainty-guided updates. Comprehensive evaluations show that AUINet outperforms state-of-the-art models on molecular property benchmarks and protein-protein interaction inhibitor tasks under low-data conditions, all without requiring extensive pretraining. Crucially, rejection sampling experiments reveal that atom uncertainty provides a more robust error signal than classic molecular uncertainty. More importantly, AUINet provides chemically interpretable insights by pinpointing specific functional groups and structural motifs that contribute most to prediction uncertainty, as validated in solubility prediction and activity-cliff analysis. Overall, AUINet's precise localization of atomic-level uncertainty establishes a new paradigm for trustworthy molecular property prediction.

View source

Similar papers

#artificial intelligence Preprint Aug 2026

CoMPASS: Collaborative Molecular Property Prediction via Adaptive Small-Large Model Synergy

CoMPASS is presented, a retrieval-calibrated framework for small-large model collaboration that retains a graph attention network as the predictive anchor, retrieves locally relevant training molecules, provides attention-grounded evidence to an LLM, and converts its proposal into a bounded correction through an agreement-aware gate.

Wen-Tao Li, Jiang-Jie Qiu, Yi-Jun Li et al. · 0 citations
Jul 2026

Chem World: A Large-Scale Benchmark and Physics-Informed Framework for Trustworthy Chemical Property Prediction

Chem World is introduced, a comprehensive benchmark for chemical property prediction that integrates 17 diverse chemical datasets with over 800,000 molecular samples, covering various properties including density, electrical conductivity, solubility, and other molecular characteristics and Mixture-PINN is proposed, a physics-informed neural network based prediction framework that incorporates chemical prior knowledge into data-driven learning.

Tianyou Bai, Huanfei Wang, Ming Gao et al. · 0 citations
Book Jul 2026

Trustworthy Protein-Ligand Binding Affinity Prediction via Reliability-Aware Multi-Engine Fusion

To the knowledge, RELIABLE-BA is the first multi-engine binding affinity prediction framework to combine evidential fusion with context-dependent reliability, offering a principled path toward trustworthy AI-guided drug discovery.

Yongchan Hong, Defu Cao, Wenjin Liu et al. · 0 citations
Open access Aug 2026

A unified framework for molecular property prediction based on hierarchical multi-granularity molecular representation learning

An atom–bond bipartite graph modeling approach that treats atoms and bonds as explicit learnable node types and jointly models atom–atom, atom–bond, and bond–bond local interactions within a unified propagation framework is introduced.

Xing Zhao, Xianlai Chen, Yunbo Wang et al. · 0 citations
#machine learning Preprint Aug 2026

AdaptNTK: Adaptive Uncertainty Quantification and Active Learning for Neural Network Potentials

Machine learning interatomic potentials bridge the gap between quantum chemical precision and classical computational speed, enabling molecular dynamics simulations with first-principles accuracy. Their reliability is often improved through active learning, which iteratively expands the training set by identifying uncertain, out-of-distribution configurations. Existing uncertainty-quantification methods often involve a trade-off between computational cost and reliability, and generally cannot account for redundancy as an acquisition batch is assembled. Here, we introduce AdaptNTK, a single-model framework that measures uncertainty as a regularized Mahalanobis distance in empirical neural tangent kernel (NTK) feature space. With the NTK features fixed during acquisition, the uncertainty depends on the acquired configurations but not their reference labels. This allows the uncertainty to be updated recursively after each selection without retraining, reducing redundancy within an acquisition batch. On held-out rMD17 data, AdaptNTK achieves the highest mean correlations with force errors (Spearman 0.68, Pearson 0.71) and matches a three-member ensemble in error retention. In active learning experiments, AdaptNTK achieves the lowest force errors across rMD17 and Transition-1X, with particularly strong performance on transition-state configurations in Transition-1X. AdaptNTK provides a 2.6-fold speedup per Transition-1X cycle relative to the ensemble, providing efficient single-model uncertainty estimation with sequential updates for data-efficient active learning.

Prajwal Ananth, Shuwen Yue · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.