ATOM：用多智能体树状搜索破解多目标分子优化难题
多目标分子优化是药物发现和材料设计中的核心挑战：不仅需要搜索庞大的化学空间，还要在活性、可合成性、ADMET（吸收、分布、代谢、排泄、毒性）等多个相互冲突的目标之间取得平衡。传统方法通常依赖单一策略或固定标量化，难以同时探索多种有前景的设计路径。来自中国多所高校的研究团队在 arXiv 上发表的论文中提出了 ATOM（Agents on a Tree） 框架，将分子优化形式化为树状搜索过程，通过多智能体路径协调来应对这一难题。
核心思想：树上的智能体协作
ATOM 的灵感来源于决策树：每个节点代表一次原子操作（如添加或替换化学基团），并部署一个专门负责某个目标或决策上下文的智能体。与要求全局共识的传统多智能体系统不同，ATOM 的智能体沿着树的不同路径独立协调。这意味着系统可以同时维护并比较多条分子演化轨迹，从而保持多样化的权衡方案。
全局记忆与长程依赖
ATOM 还引入了一个全局记忆模块，记录过去优化行为中的成功与失败经验。这有助于平衡不同目标之间的探索与利用，避免陷入局部最优。由于分子设计中的早期决策会强烈影响后续结果，树状结构的交互方式使得模型能够推理长程依赖关系——这正是传统方法难以处理的。
实验表现：全面超越基线
研究团队在多个具有挑战性的多目标基准上测试了 ATOM，这些基准涉及分子活性、可合成性以及 ADMET 相关性质。结果显示，ATOM 在 帕累托覆盖率（Pareto coverage） 和 超体积（hypervolume） 两个关键指标上持续优于强基线方法。这意味着 ATOM 能够找到更多样化且更高质量的候选分子，覆盖更广泛的权衡空间。
对 AI 药物研发的启示
ATOM 的工作展示了多智能体强化学习与结构化搜索结合的巨大潜力。与近年来大热的扩散模型或生成式 AI 不同，ATOM 更侧重于决策过程的分解与协调。这种方法特别适合那些需要显式权衡多个约束条件的现实场景，例如先导化合物优化。随着代码已开源（GitHub），该框架有望被集成到更广泛的分子设计工作流中。
总体而言，ATOM 不仅提出了一种新颖的算法框架，也为多目标优化问题提供了一个可解释、可扩展的解决方案。未来，将树状搜索与更强大的分子表示（如预训练图神经网络）结合，可能会进一步突破现有性能天花板。