Rethinking Molecular OOD Generalization via Target-Aware Source Selection

TL;DR

This paper introduces SCOPE-BENCH, a new benchmark for molecular OOD evaluation, and POMA, a policy-based framework for source selection and domain adaptation, significantly improving prediction robustness in drug discovery.

Contribution

It proposes a novel benchmark and a reinforcement learning-based source selection framework to enhance molecular property prediction under extreme OOD conditions.

Findings

Prediction errors increase up to 8.0x on SCOPE-BENCH for state-of-the-art models.

POMA reduces mean absolute error by up to 11.2%.

Average relative improvement of 6.2% across backbone architectures.

Abstract

Robust prediction of molecular properties under extreme out-of-distribution (OOD) scenarios is a pivotal bottleneck in AI-driven drug discovery. Current scaffold-splitting protocols fail to obstruct microscopic semantic overlap, predisposing models to shortcut learning and overestimating their true extrapolation capability; meanwhile, conventional domain adaptation paradigms suffer under extreme structural shifts, as blindly aligning heterogeneous source libraries injects topological noise and triggers negative transfer. To address these two challenges, scaffold-cluster out-of-distribution performance evaluation benchmark (SCOPE-BENCH), a benchmark built on cluster-level partitioning in an explicit physicochemical descriptor space, is proposed alongside policy optimization for multi-source adaptation (POMA), a framework that formulates knowledge transfer as a retrieve-compose-adapt pipeline: labeled source scaffolds structurally close to the unlabeled target are first identified as proxy targets; a reinforcement learning policy then adaptively selects the optimal source subset from an exponentially large candidate pool; and dual-scale domain adaptation is finally performed at macroscopic topological and microscopic pharmacophore scales. Evaluations show that prediction errors of state-of-the-art 3D molecular models surge by up to 8.0x on SCOPE-BENCH with a mean of 5.9x, while POMA achieves up to an 11.2% reduction in mean absolute error with an average relative improvement of 6.2% across diverse backbone architectures. Code is available at https://anonymous.4open.science/r/Molecular-OOD-Code-73F6.