Search papers, labs, and topics across Lattice.
This paper introduces a short-term graph memory module that enhances molecular optimization by efficiently utilizing a limited oracle budget to prioritize candidate evaluations. By integrating an online graph neural surrogate, the method ensures that the oracle queries focus on molecules with higher predicted utility, leading to improved performance without increasing costs. The approach consistently outperforms baseline methods across multiple generators, demonstrating significant gains in optimization efficiency while maintaining robust performance under budget constraints.
Spending a fixed oracle budget more selectively can lead to substantial improvements in molecular optimization outcomes without incurring additional costs.
Molecular optimization is commonly performed under a limited oracle budget, which makes deciding what to evaluate as important as deciding what to generate. We introduce short-term graph memory, a plug-in module that preserves the generator architecture and native update rule while learning from previously evaluated molecules to prioritize subsequent oracle queries. The module maintains an online graph neural surrogate that pre-screens each round's candidate pool, so the fixed oracle budget is spent on molecules with higher predicted utility. Applied to a fragment-based generator on a standard molecular optimization benchmark, it improves the mean top-10 score at no extra oracle cost and never falls behind the base on any oracle; the gain extends to all four generators we tested at a tight budget of one thousand calls. We then analyze how surrogate-guided selection interacts with the exploration and exploitation behavior of different generators. Its benefit at larger budgets is consistent with two properties of the backbone: how broadly it searches, and how effectively its native search already exploits oracle feedback. We provide a simple way to spend a fixed oracle budget more selectively, and evidence on which generators benefit from it.
