DiSPA: Differential Substructure-Pathway Attention for Drug Response Prediction

TL;DR

DiSPA is a novel deep learning framework that models bidirectional interactions between chemical substructures and pathway gene expressions, improving drug response prediction accuracy and interpretability.

Contribution

It introduces differential cross-attention to suppress noise and enhance relevant interactions, achieving state-of-the-art results and better interpretability in drug response prediction.

Findings

DiSPA outperforms existing models on GDSC benchmark.

Differential attention improves pathway prioritization.

Model generalizes well to external datasets and spatial transcriptomics.

Abstract

Accurate prediction of drug response in precision medicine requires models that capture how specific chemical substructures interact with cellular pathway states. However, most existing deep learning approaches treat chemical and transcriptomic modalities independently or combine them only at late stages, limiting their ability to model fine-grained, context-dependent mechanisms of drug action. In addition, vanilla attention mechanisms are often sensitive to noise and sparsity in high-dimensional biological networks, hindering both generalization and interpretability. We present DiSPA (Differential Substructure-Pathway Attention), a framework that models bidirectional interactions between chemical substructures and pathway-level gene expression. DiSPA introduces differential cross-attention to suppress spurious associations while enhancing context-relevant interactions. On the GDSC benchmark, DiSPA achieves state-of-the-art performance, with strong improvements in the disjoint setting. These gains are consistent across random and drug-blind splits, suggesting improved robustness. Analyses of attention patterns indicate more selective and concentrated interactions compared to standard cross-attention. Exploratory evaluation shows that differential attention better prioritizes predefined target-related pathways, although this does not constitute mechanistic validation. DiSPA also shows promising generalization on external datasets (CTRP) and cross-dataset settings, although further validation is needed. It further enables zero-shot application to spatial transcriptomics, providing exploratory insights into region-specific drug sensitivity patterns without ground-truth validation.