Activity-cliff awareness enables robust graph learning for molecular property prediction

Read full text
Authors: Chao Cui, Xiaorui Su, Zaixi Zhang, Alejandro Velez-Arce, Jianmin Wang, Xiang Cheng Shi, Yan Bing Zhang, J R Wu, Yu Zong Chen, Marinka Žitnik, Wan Xiang Shen
Cited by: 1

Abstract

Accurately predicting molecular activity is hindered by activity cliffs (ACs), which are sharp potency changes between highly similar compounds that distort the smoothness assumed by modern quantitative structure–activity relationship (SAR) models and graph neural networks (GNNs). Here we introduce AC awareness (ACA), an inductive bias that reshapes GNN latent spaces to account for these discontinuities. Implemented through an ACA loss combining regression with softmargin triplet contrastive learning, the method dynamically mines high-value activity-cliff triplets during training and corrects inconsistent neighbourhoods in latent space. This process yields progressively fewer cliff violations, more coherent activity gradients, and reduced label incoherence across diverse chemical spaces. Evaluated on 52 datasets spanning low-sample narrow-scaffold series, large mixed-scaffold benchmarks, matched-pair cliff classification, and absorption, distribution, metabolism, excretion and toxicity (ADMET) property prediction, ACA consistently improves predictive accuracy over strong extended-connectivity fingerprint (ECFP) and GNN baselines. The approach generalizes across multiple GNN backbones and remains robust under fixed hyperparameters. These results suggest that ACA provides a principled strategy for enhancing molecular property prediction by aligning latent representations with the nonadditive behaviour underlying ACs.

Fields of Study & Subjects

Computational Drug Discovery Methods Advanced Graph Neural Networks Machine Learning in Materials Science

Publication details

Journal / source
ChemRxiv
Publisher
American Chemical Society (ACS)
Publication date
15 June 2026
Document type
Preprint