Interventionally-guided representation learning for robust and interpretable AI models in cancer medicine
Kirkham, D.; Masina, R.; Sammut, S.-J.; Mukherjee, S.; Rueda, O. M.
Show abstract
Machine learning models hold promise in cancer medicine but often lack robustness and interpretability. We introduce a new class of model for high-dimensional molecular data that incorporate interventional auxiliary information to learn latent representations that are informative and interpretable by design. By using causal signals from genetic loss-of-function screens, our approach generates representations that generalize well across data distributions and biological contexts. In cancer cell line datasets, we show that causal guidance enables "zero-shot" transfer to cancer types unseen during training. Moreover, models trained solely on cell line data translate effectively to clinical cohorts, demonstrating strong "bench-to-bedside" generalization without fine-tuning. This strategy highlights a scalable way to leverage tractable laboratory assays for clinical modeling. More broadly, our results establish how integrating causal biological information within generative frame-works enhances data efficiency, interpretability, and robustness, opening avenues for a new generation of scientifically informed AI models in molecular medicine.
Matching journals
The top 6 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
- TUGDA: Task uncertainty guided domain adaptation for robust generalization of cancer drug response prediction from in vitro to in vivo settings 97%
- ELISL: Early-Late Integrated Synthetic Lethality Prediction in Cancer 97%
- ARTEMIS integrates autoencoders and schrodinger bridges to predict continuous dynamics of gene expression, cell population and perturbation from time-series single-cell data 96%
Similar papers in this journal
Similar papers in this journal
- Sagittarius: Extrapolating Heterogeneous Time-Series Gene Expression Data 96%
- Learning the rules of cell competition without prior scientific knowledge 95%
- Inferring spatial single-cell-level interactions through interpreting cell state and niche correlations learned by self-supervised graph transformer 95%
Similar papers in this journal
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.