Attention-Enhanced Deep Generative Modeling Enables Interpretable Prediction of Cancer Drug Sensitivity

Shuangxia Ren, AODONG QIU, Mengyao Lu, Xinghua Lu
Proceedings of the 11th Machine Learning for Healthcare Conference, PMLR 340:1667-1688, 2026.

Abstract

Predicting how tumors respond to specific drugs is a fundamental obstacle in precision oncology, largely because of the vast molecular diversity observed across cancers. Deep learning approaches have demonstrated potential for modeling complex relationships between molecular profiles and drug responses, yet their adoption in clinical settings has been hindered by limited transparency and poor generalization across different datasets. In this work, we introduce Residual Attention Variational Autoencoder with Elastic Net (ResAttnVAE-EN), a deep generative model augmented with attention mechanisms that combines somatic genomic alterations (SGAs) and transcriptomic profiles to learn biologically interpretable cellular representations for predicting drug sensitivity. Leveraging large-scale pharmacogenomic datasets from the Genomics of Drug Sensitivity in Cancer (GDSC) database, we show that ResAttnVAE-EN consistently surpasses standard variational autoencoder and regression-based benchmarks across a broad panel of therapeutic compounds. The attention layers highlight genomically coherent pathway-level drivers, and the learned latent spaces encode cellular states that generalize across multiple cancer types for response prediction. Notably, models derived from cell line experiments effectively distinguish survival trajectories and treatment outcomes in independent The Cancer Genome Atlas (TCGA) lung cancer patient cohorts. These findings position attention-augmented deep generative approaches as a reliable and interpretable framework for clinically translatable drug sensitivity modeling.

Cite this Paper


BibTeX
@InProceedings{pmlr-v340-ren26a, title = {Attention-Enhanced Deep Generative Modeling Enables Interpretable Prediction of Cancer Drug Sensitivity}, author = {Ren, Shuangxia and QIU, AODONG and Lu, Mengyao and Lu, Xinghua}, booktitle = {Proceedings of the 11th Machine Learning for Healthcare Conference}, pages = {1667--1688}, year = {2026}, editor = {Krishnan, Rahul G. and van Amsterdam, Wouter A. C. and Chopra, Sumit and Overgaard, Shauna and Hughes, Michael and Ötleş, Erkin and Shen, Yiqiu and Shanmugam, Divya and Nayan, Madhur and Engelhard, Matthew and Fackler, Jim and Oberst, Michael}, volume = {340}, series = {Proceedings of Machine Learning Research}, month = {12--14 Aug}, publisher = {PMLR}, pdf = {https://raw.githubusercontent.com/mlresearch/v340/main/assets/ren26a/ren26a.pdf}, url = {https://proceedings.mlr.press/v340/ren26a.html}, abstract = {Predicting how tumors respond to specific drugs is a fundamental obstacle in precision oncology, largely because of the vast molecular diversity observed across cancers. Deep learning approaches have demonstrated potential for modeling complex relationships between molecular profiles and drug responses, yet their adoption in clinical settings has been hindered by limited transparency and poor generalization across different datasets. In this work, we introduce Residual Attention Variational Autoencoder with Elastic Net (ResAttnVAE-EN), a deep generative model augmented with attention mechanisms that combines somatic genomic alterations (SGAs) and transcriptomic profiles to learn biologically interpretable cellular representations for predicting drug sensitivity. Leveraging large-scale pharmacogenomic datasets from the Genomics of Drug Sensitivity in Cancer (GDSC) database, we show that ResAttnVAE-EN consistently surpasses standard variational autoencoder and regression-based benchmarks across a broad panel of therapeutic compounds. The attention layers highlight genomically coherent pathway-level drivers, and the learned latent spaces encode cellular states that generalize across multiple cancer types for response prediction. Notably, models derived from cell line experiments effectively distinguish survival trajectories and treatment outcomes in independent The Cancer Genome Atlas (TCGA) lung cancer patient cohorts. These findings position attention-augmented deep generative approaches as a reliable and interpretable framework for clinically translatable drug sensitivity modeling.} }
Endnote
%0 Conference Paper %T Attention-Enhanced Deep Generative Modeling Enables Interpretable Prediction of Cancer Drug Sensitivity %A Shuangxia Ren %A AODONG QIU %A Mengyao Lu %A Xinghua Lu %B Proceedings of the 11th Machine Learning for Healthcare Conference %C Proceedings of Machine Learning Research %D 2026 %E Rahul G. Krishnan %E Wouter A. C. van Amsterdam %E Sumit Chopra %E Shauna Overgaard %E Michael Hughes %E Erkin Ötleş %E Yiqiu Shen %E Divya Shanmugam %E Madhur Nayan %E Matthew Engelhard %E Jim Fackler %E Michael Oberst %F pmlr-v340-ren26a %I PMLR %P 1667--1688 %U https://proceedings.mlr.press/v340/ren26a.html %V 340 %X Predicting how tumors respond to specific drugs is a fundamental obstacle in precision oncology, largely because of the vast molecular diversity observed across cancers. Deep learning approaches have demonstrated potential for modeling complex relationships between molecular profiles and drug responses, yet their adoption in clinical settings has been hindered by limited transparency and poor generalization across different datasets. In this work, we introduce Residual Attention Variational Autoencoder with Elastic Net (ResAttnVAE-EN), a deep generative model augmented with attention mechanisms that combines somatic genomic alterations (SGAs) and transcriptomic profiles to learn biologically interpretable cellular representations for predicting drug sensitivity. Leveraging large-scale pharmacogenomic datasets from the Genomics of Drug Sensitivity in Cancer (GDSC) database, we show that ResAttnVAE-EN consistently surpasses standard variational autoencoder and regression-based benchmarks across a broad panel of therapeutic compounds. The attention layers highlight genomically coherent pathway-level drivers, and the learned latent spaces encode cellular states that generalize across multiple cancer types for response prediction. Notably, models derived from cell line experiments effectively distinguish survival trajectories and treatment outcomes in independent The Cancer Genome Atlas (TCGA) lung cancer patient cohorts. These findings position attention-augmented deep generative approaches as a reliable and interpretable framework for clinically translatable drug sensitivity modeling.
APA
Ren, S., QIU, A., Lu, M. & Lu, X.. (2026). Attention-Enhanced Deep Generative Modeling Enables Interpretable Prediction of Cancer Drug Sensitivity. Proceedings of the 11th Machine Learning for Healthcare Conference, in Proceedings of Machine Learning Research 340:1667-1688 Available from https://proceedings.mlr.press/v340/ren26a.html.

Related Material