Mozi: Governed Autonomy for Drug Discovery LLM Agents
Tool-augmented large language model (LLM) agents promise to unify scientific reasoning with computation, yet their deployment in high-stakes domains like drug discovery is bottlenecked by two critical barriers: unconstrained tool-use governance and poor long-horizon reliability. In dependency-heavy pharmaceutical pipelines, autonomous agents often drift into irreproducible trajectories, where early-stage hallucinations multiplicatively compound into downstream failures. To overcome this, we present Mozi, a dual-layer architecture that bridges the flexibility of generative AI with the deterministic rigor of computational biology. Layer A (Control Plane) establishes a governed supervisor--worker hierarchy that enforces role-based tool isolation, limits execution to constrained action spaces, and drives reflection-based replanning. Layer B (Workflow Plane) operationalizes canonical drug discovery stages -- from Target Identification to Lead Optimization -- as stateful, composable skill graphs. This layer integrates strict data contracts and strategic human-in-the-loop (HITL) checkpoints to safeguard scientific validity at high-uncertainty decision boundaries.
  Operating on the design principle of ``free-form reasoning for safe tasks, structured execution for long-horizon pipelines,'' Mozi provides built-in robustness mechanisms and trace-level audibility to completely mitigate error accumulation. We evaluate Mozi on PharmaBench, a curated benchmark for biomedical agents, demonstrating superior orchestration accuracy over existing baselines. Furthermore, through end-to-end therapeutic case studies, we demonstrate Mozi's ability to navigate massive chemical spaces, enforce stringent toxicity filters, and generate highly competitive in silico candidates, effectively transforming the LLM from a fragile conversationalist into a reliable, governed co-scientist.
69a8e4da8b5b27c7ea9fa58e
He Cao
claimed_verified
2026-03-05T10:01:16.556Z
63458f173cc8a5caf9b84e48
https://cdn-avatars.huggingface.co/v1/production/uploads/63458f173cc8a5caf9b84e48/fCk_ijUSRx1KwjiiqzBtY.jpeg
CAO He
hcaoaf
user
hcaoaf
69a8e4da8b5b27c7ea9fa58f
Siyu Liu
claimed_verified
2026-03-06T13:56:54.575Z
6508469df3060ea840fd888f
https://cdn-avatars.huggingface.co/v1/production/uploads/6508469df3060ea840fd888f/3vwlKo0Cd9UWOPlwOvlJw.jpeg
Siyu Liu
SiyuLiu
user
SiyuLiu
69a8e4da8b5b27c7ea9fa590
Fan Zhang
69a8e4da8b5b27c7ea9fa591
Zijing Liu
69a8e4da8b5b27c7ea9fa592
Hao Li
69a8e4da8b5b27c7ea9fa593
Bin Feng
69a8e4da8b5b27c7ea9fa594
Shengyuan Bai
69a8e4da8b5b27c7ea9fa595
Leqing Chen
69a8e4da8b5b27c7ea9fa596
Kai Xie
69a8e4da8b5b27c7ea9fa597
Yu Li
Mozi: Governed Autonomy for Drug Discovery LLM Agents
Tool-augmented large language model (LLM) agents promise to unify scientific reasoning with computation, yet their deployment in high-stakes domains like drug discovery is bottlenecked by two critical barriers: unconstrained tool-use governance and poor long-horizon reliability. In dependency-heavy pharmaceutical pipelines, autonomous agents often drift into irreproducible trajectories, where early-stage hallucinations multiplicatively compound into downstream failures. To overcome this, we present Mozi, a dual-layer architecture that bridges the flexibility of generative AI with the deterministic rigor of computational biology. Layer A (Control Plane) establishes a governed supervisor--worker hierarchy that enforces role-based tool isolation, limits execution to constrained action spaces, and drives reflection-based replanning. Layer B (Workflow Plane) operationalizes canonical drug discovery stages -- from Target Identification to Lead Optimization -- as stateful, composable skill graphs. This layer integrates strict data contracts and strategic human-in-the-loop (HITL) checkpoints to safeguard scientific validity at high-uncertainty decision boundaries.
  Operating on the design principle of ``free-form reasoning for safe tasks, structured execution for long-horizon pipelines,'' Mozi provides built-in robustness mechanisms and trace-level audibility to completely mitigate error accumulation. We evaluate Mozi on PharmaBench, a curated benchmark for biomedical agents, demonstrating superior orchestration accuracy over existing baselines. Furthermore, through end-to-end therapeutic case studies, we demonstrate Mozi's ability to navigate massive chemical spaces, enforce stringent toxicity filters, and generate highly competitive in silico candidates, effectively transforming the LLM from a fragile conversationalist into a reliable, governed co-scientist.
Mozi is a dual-layer architectural framework that combines generative AI flexibility with computational biology rigor to create reliable, governed tool-augmented agents for drug discovery applications.
tool-augmented large language model agents drug discovery autonomous agents computational biology control plane workflow plane skill graphs data contracts human-in-the-loop therapeutic case studies orchestration accuracy