Combinatorial Prediction of Therapeutic Perturbations Using Causally-Inspired Neural Networks

Phenotype-driven approaches identify disease-counteracting compounds by analysing the phenotypic signatures that distinguish diseased from healthy states. Here we introduce PDGrapher, a causally inspired graph neural network model that predicts combinatorial perturbagens (sets of therapeutic targets) capable of reversing disease phenotypes. Unlike methods that learn how perturbations alter phenotypes, PDGrapher solves the inverse problem and predicts the perturbagens needed to achieve a desired response by embedding disease cell states into networks, learning a latent representation of these states, and identifying optimal combinatorial perturbations.

In experiments in nine cell lines with chemical perturbations, PDGrapher identifies effective perturbagens in more testing samples than competing methods. It also shows competitive performance on ten genetic perturbation datasets. An advantage of PDGrapher is its direct prediction, in contrast to the indirect and computationally intensive approach common in phenotype-driven models. It trains up to 25× faster than existing methods, providing a fast approach for identifying therapeutic perturbations and advancing phenotype-driven drug discovery.

Target-driven drug discovery, predominant since the 1990s, focuses on the design of highly specific compounds against disease-associated targets such as proteins or enzymes.

  • A prime example of target-driven drug discovery is the development of small molecule kinase inhibitors like Imatinib. Imatinib halts the progression of chronic myeloid leukemia (CML) by inhibiting the BCR-ABL tyrosine kinase, a mutated protein that drives uncontrolled proliferation of leukemic cells in CML patients.
  • A second notable example are monoclonal antibodies like Trastuzumab which specifically targets the HER2 receptor, a protein overexpressed in certain types of breast cancer. Trastuzumab inhibits cell proliferation while engaging the body’s immune system to initiate an attack against the cancer.

These examples illustrate the success of target-driven approaches, yet the past decade has seen a revival of phenotype-driven approaches. This shift has been fueled by the observation that the majority of the first-in-class drugs approved by the US FDA between 1999 and 2008 were discovered empirically without a drug target hypothesis.

Instead of the one drug, one gene, one disease model of target-driven approaches, phenotype-driven approaches aim to identify compounds, or perturbagens, that reverse phenotypic disease effects as measured by high-throughput screening (HTS) assays. Notable recent successes include Ivacaftor, the first available therapy treating the underlying cause of cystic fibrosis, and Risdiplam, the first oral medicine approved to treat spinal muscular atrophy.

Overview of PDGrapher

Current phenotype-driven lead discovery methods are limited as they identify potential leads by predicting and matching gene expression response signatures with known disease signatures. We propose an alternative approach where the objective is to determine which genes a new perturbagen should target to reverse disease effects as represented by a disease signature. We term this approach combinatorial prediction of therapeutic targets due to its focus on predicting gene combinations.

The problem of finding which elements of a system should be perturbed to achieve a desired outcome is typically referred to as optimal intervention design within the causal discovery literature. Here, we build on this concept to introduce PDGrapher, an approach for the combinatorial prediction of therapeutic targets to shift cell gene expression from an initial to a treated state in genetic and chemical intervention datasets. Our predictive task is formulated using a causal model, where genes are the nodes in a causal graph and structural equations encode their causal relationships. The goal is to identify a set of genes a perturbagen should target to transition nodes states from diseased to treated.

We consider protein-protein interaction networks (PPI) or gene regulatory networks (GRN) as approximations of the causal graph, assuming no unobserved confounders. We tackle our objective through a representation learning approach, using graph neural network (GNN)-based models to implicitly represent the structural equations. For a given pair of diseased and treated samples, PDGrapher outputs a ranking of genes, with the top-predicted genes identified as the primary therapeutic targets that shift gene expression phenotypes from a diseased to a treated state.

We evaluate PDGrapher across four datasets of genetic and chemical interventions in held out folds containing novel samples, and held out folds containing novel samples and cell lines:

  • PDGrapher ranks ground-truth therapeutic targets up to 35% higher in chemical intervention datasets and 16% higher in genetic intervention datasets in held out folds containing novel samples, compared with existing approaches.
  • In held out folds containing both novel samples and cell lines, PDGrapher maintained its robust performance.
  • A key innovation of PDGrapher is its ability to directly predict perturbagens that can shift gene expression from diseased to treated states in contrast with prevailing approaches that indirectly predict perturbagens through extensive computational modeling of cell responses. This enables up to 30 times faster training than existing methods.
  • Therapeutic targets predicted by PDGrapher in chemical intervention datasets agree well with ground-truth therapeutic targets, indicating that PDGrapher can elucidate the mode of action for chemical perturbagens and potentially enhance phenotype-driven lead discovery.

Publication

Combinatorial Prediction of Therapeutic Perturbations Using Causally-Inspired Neural Networks
Guadalupe Gonzalez*, Xiang Lin*, Isuru Herath, Kirill Veselkov, Michael Bronstein and Marinka Zitnik
Nature Biomedical Engineering 2025 [bioRxiv] [Harvard Medicine News]

@article{gonzalez2025combinatorial,
  title={Combinatorial Prediction of Therapeutic Perturbations Using Causally-Inspired Neural Networks},
  author={Gonzalez, Guadalupe and Lin, Xiang and Herath, Isuru and Veselkov, Kirill and Bronstein, Michael and Zitnik, Marinka},
  journal={Nature Biomedical Engineering},
  url={https://www.nature.com/articles/s41551-025-01481-x},
  year={2025}
}

Code Availability

Pytorch implementation of PDGrapher is available in the GitHub repository.

Data Availability

Processed data, including the cell line gene expression dataset, protein-protein interaction network, drug targets, and disease-associated genes are shared with the research community:

Processed dataset is available through Figshare.

Authors

Latest News

Oct 2025:   Our research featured by Kempner and Crimson

A news story about PDGrapher in Harvard Crimson. ToolUniverse featured on the Kempner Institute blog.

Oct 2025:   A Scientist's Guide to AI Agents in Nature

A piece on AI agents in Nature highlights ongoing projects in our group, including methods for evaluating scientific hypotheses, challenges in benchmarking AI agents, and the open ToolUniverse ecosystem.

Sep 2025:   ToolUniverse: AI Agents for Science and Medicine

New paper: ToolUniverse introduces an open ecosystem for building AI scientists with 600+ scientific and biomedical tools. Build your AI co-scientists at https://aiscientist.tools.

Sep 2025:   Democratizing "AI Scientists" with ToolUniverse

Our new initiative: Use Tool Universe to build an AI scientist for yourself from any language or reasoning model, whether open or closed. https://aiscientist.tools

Sep 2025:   InfEHR in Nature Communications

Collaboration with Ben and Girish on clinical phenotype resolution through deep geometric learning on electronic health records published in Nature Communications.

Sep 2025:   PDGrapher in Nature Biomedical Engineering

New paper in Nature Biomedical Engineering introducing PDGrapher, a model for phenotype-based target discovery. [Harvard Medicine News]

Sep 2025:   AI and Net Medicine: Path to Precision Medicine

Aug 2025:   CUREBench - Reasoning for Therapeutics

Update from CUREBench: 650+ entrants, 100+ teams and 500+ submissions. Thank you to the CUREBench community. Working on AI for drug discovery and reasoning in medicine? New teams welcome. Tasks, rules, and leaderboard: https://curebench.ai.

Aug 2025:   Drug Discovery Workshop at NeurIPS 2025

Excited to organize a NeurIPS workshop on Virtual Cells and Digital Instruments. Submit your papers.

Aug 2025:   AI for Science Workshop at NeurIPS

Excited to organize a NeurIPS workshop on AI for Science. This is our 6th workshop in the AI for Science series. Submit your papers.

Jul 2025:   Launching CUREBench

Launched CUREBench, the first competition in AI reasoning for therapeutics. Colocated with NeurIPS 2025. Start at https://curebench.ai.

Jul 2025:   Launching TxAgent Evaluation Portal

Launched TxAgent evaluation portal, our global evaluation of AI for drug decision-making and therapeutic reasoning. Participate in TxAgent evaluations! [TxAgent project]

Jul 2025:   SPATIA Model of Spatial Cell Phenotypes

Jul 2025:   AI-Enabled Drug Discovery Reaches Clinical Milestone

Jun 2025:   Knowledge Tracing for Biomedical AI Education

New preprint on biologically inspired architecture for knowledge tracing. The study on the use of generative AI in education with prospective evaluation of knowledge tracing in the classroom.

Jun 2025:   Few shot learning for rare disease diagnosis

Jun 2025:   One Patient, Many Contexts: Scaling Medical AI

Jun 2025:   ToolUniverse - 211+ Tools for "AI Scientist" Agents

ToolUniverse now offers access to over 211 cutting-edge biological and medical tools, all integrated with Model Context Protocol (MCP). Any “AI Scientist” agent can tap into these tools for biomedical research. [Tutorial] [ToolUniverse] [TxAgent]

May 2025:   What Perturbation Can Reverse Disease Effects?

In press at Nature Biomedical Engineering: PDGrapher AI predicts chemicals to reverse disease phenotypic effects — with applications to drug target identification.

May 2025:   Decision Transformers for Cell Reprogramming

New preprint: Decision transformers for generating reach-avoid policies in sequential decision making — with applications from robotics to cell reprogramming.

Zitnik Lab  ·  Artificial Intelligence in Medicine and Science  ·  Harvard  ·  Department of Biomedical Informatics