Search papers, labs, and topics across Lattice.
This paper systematically investigates the impact of target context on molecular property prediction across diverse protein families and data regimes. They find that FiLM-based fusion architectures (NestDrug) significantly outperform concatenation and additive conditioning methods for incorporating target context. Crucially, target context enables accurate predictions in data-scarce scenarios but can degrade performance due to distribution mismatch, while also exposing flaws in standard benchmarking practices.
Context-aware models for molecular property prediction can unlock accurate predictions in data-scarce regimes, but beware – distribution shifts can cause them to backfire, and your benchmarks might be meaningless.
We present the first systematic study of when target context helps molecular property prediction, evaluating context conditioning across 10 diverse protein families, 4 fusion architectures, data regimes spanning 67-9,409 training compounds, and both temporal and random evaluation splits. Using NestDrug, a FiLM-based architecture that conditions molecular representations on target identity, we characterize both success and failure modes with three principal findings. First, fusion architecture dominates: FiLM outperforms concatenation by 24.2 percentage points and additive conditioning by 8.6 pp; how you incorporate context matters more than whether you include it. Second, context enables otherwise impossible predictions: on data-scarce CYP3A4 (67 training compounds), multi-task transfer achieves 0.686 AUC where per-target Random Forest collapses to 0.238. Third, context can systematically hurt: distribution mismatch causes 10.2 pp degradation on BACE1; few-shot adaptation consistently underperforms zero-shot. Beyond methodology, we expose fundamental flaws in standard benchmarking: 1-nearest-neighbor Tanimoto achieves 0.991 AUC on DUD-E without any learning, and 50% of actives leak from training data, rendering absolute performance metrics meaningless. Our temporal split evaluation (train up to 2020, test 2021-2024) achieves stable 0.843 AUC with no degradation, providing the first rigorous evidence that context-conditional molecular representations generalize to future chemical space.
