TRAPS: Treatment-Assignment Prediction via Pathway-informed Stratification
Sujoy Banik, Sayantan Chakraborty, Boishakhi Das Toma, Zainab Ghafoor, Ushashi Bhattacharjee, Koushik Howlader, Tirtho Roy
cs.LG, cs.MA, q-bio.QM
Submitted: 2026-08-10
Updated: 2026-08-21
License: http://creativecommons.org/licenses/by/4.0/
The gist: Cancer treatment involves decisions across multiple clinical outcomes, yet pathway-informed deep learning models are typically evaluated in isolation, making their relative benefits unclear.
Terminology
Abstract
Cancer treatment involves decisions across multiple clinical outcomes, yet pathway-informed deep learning models are typically evaluated in isolation, making their relative benefits unclear. We present a harmonized benchmark of three biologically informed architectures, BINN, GraphPath, and PATH, for predicting treatment exposure and short-term survival across five TCGA cancer cohorts comprising 2,622 patients represented by Reactome pathway activity scores. Treatment labels indicate recorded exposure in TCGA rather than therapeutic response. All models jointly predict targeted molecular therapy (TMT), radiation therapy (RT), and six-month overall survival (OS) from a shared pathway representation and are evaluated on identical stratified folds using five repeated splits and paired-bootstrap testing. Under this controlled evaluation, most differences between architectures fall within 95 percent confidence intervals, indicating that rankings suggested by isolated evaluations are largely not statistically resolved. The main exception is survival prediction: the sparse-hierarchy BINN significantly outperforms both graph models on breast-cancer OS, with an AUROC improvement of up to 0.14 and p less than or equal to 0.01, and leads on lung and prostate OS. For treatment exposure, TMT is best discriminated in prostate cancer, with AUROC approximately 0.80 for all models, but no architecture significantly outperforms another on any TMT cohort. RT prediction remains weak across models, suggesting that its determinants may be more clinical than transcriptomic. Overall, architecture choice has limited impact under a unified evaluation, while short-term survival provides the clearest differentiation among pathway-informed models.
Sources
- Graph Attention Networks
- A Generalization of Transformer Networks to Graphs
- Graph Transformer-Based Pathway Embedding for Cancer Prognosis
- Adam: A Method for Stochastic Optimization
- PathMoG: A Pathway-Centric Modular Graph Neural Network for Multi-Omics Survival Prediction
- Graph Neural Networks in Multi-Omics Cancer Research: A Structured Survey
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: Local-Consistency Optimisation for Bayesian Neural Networks
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks