Machine-learning driven strategies for adapting immunotherapy in metastatic NSCLC
Immune checkpoint inhibitors (ICIs), either as monotherapy (ICI-Mono) or combined with chemotherapy (ICI-Chemo), improves survival in advanced non-small cell lung cancer (NSCLC). However, prospective guidance for choosing between these options remains limited, and single-feature biomarkers like PD-L1 prove inadequate. We develop a machine learning model using clinicogenomic data from four cohorts (MD Anderson n = 750; Mayo Clinic n = 80; Dana-Farber n = 1077; Stand Up To Cancer n = 393) to predict individual benefit from adding chemotherapy. Benefit scores are calculated using five distinct functions derived from 28 genomic and 6 clinical features. Our integrated model, A-STEP (Attention-based Scoring for Treatment Effect Prediction), estimates heterogeneous treatment effects and achieves the largest reduction in 3-month progression risk, improving weighted risk reduction by 13-23% over stand-alone models. A-STEP recommends treatment changes for over 50% of patients, most often favoring ICI-Chemo. In simulation on external cohort, patients treated in accordance with A-STEP recommendations show improved 2-year progression-free survival (HR = 0.60 for ICI-Mono treatment arm; HR = 0.58 for ICI-Chemo treatment arm). Predictive features include FBXW7, APC, and PD-L1. In this study, we demonstrate how machine learning can fill critical gaps in immunotherapy selection for NSCLC, by modeling treatment heterogeneity with real-world clinicogenomic data, driving precision medicine beyond conventional biomarker boundaries.
Authors
- Percy P. Lee
- Carol C. Wu (ORCID: https://orcid.org/0000-0003-1005-0995)
- Jia Wu (ORCID: https://orcid.org/0000-0001-8392-8338)
- Xiuning Le (ORCID: https://orcid.org/0000-0002-8554-1185)
- David A. Jaffray (ORCID: https://orcid.org/0000-0001-7378-7543)
- Joe Y. Chang (ORCID: https://orcid.org/0000-0002-8435-2083)
- Federica Pecci (ORCID: https://orcid.org/0009-0000-9289-2007)
- Don L. Gibbons (ORCID: https://orcid.org/0000-0003-2362-3094)
- Simon Heeke (ORCID: https://orcid.org/0000-0002-5916-534X)
- Vivek Verma (ORCID: https://orcid.org/0000-0001-8129-4140)
- Brett W. Carter
- Ignacio I. Wistuba (ORCID: https://orcid.org/0000-0003-3365-6340)
- Lingzhi Hong (ORCID: https://orcid.org/0000-0001-6553-3279)
- Biagio Ricciuti (ORCID: https://orcid.org/0000-0002-0651-2678)
- Mehmet Altan (ORCID: https://orcid.org/0000-0001-9229-156X)
- Milena Petranović (ORCID: https://orcid.org/0000-0002-6412-7428)
- Daniel Boiarsky (ORCID: https://orcid.org/0009-0007-5740-9244)
- Maliazurina B. Saad (ORCID: https://orcid.org/0000-0002-2895-4186)
- Girish S. Shroff (ORCID: https://orcid.org/0000-0001-9488-7439)
- John Victor Heymach (ORCID: https://orcid.org/0000-0001-9068-8942)
- Ara Asadur Vaporciyan (ORCID: https://orcid.org/0000-0002-2060-3584)
- Cara Haymaker (ORCID: https://orcid.org/0000-0002-1317-9287)
- Shenduo Li (ORCID: https://orcid.org/0009-0000-1295-1960)
- Tina Cascone (ORCID: https://orcid.org/0000-0003-3008-5407)
- Ajay Sheshadri (ORCID: https://orcid.org/0000-0002-8091-0180)
- Qasem Al-Tashi (ORCID: https://orcid.org/0000-0001-7208-693X)
- Alessandro Di Federico (ORCID: https://orcid.org/0000-0001-8877-4315)
- Yasir Y. Elamin (ORCID: https://orcid.org/0000-0002-5312-909X)
- Justin F. Gainor (ORCID: https://orcid.org/0000-0001-8697-4081)
- Yanyan Lou (ORCID: https://orcid.org/0000-0001-6899-7124)
- Mark M. Awad (ORCID: https://orcid.org/0000-0003-0928-5244)
- Zhongxing Liao (ORCID: https://orcid.org/0000-0002-2224-9478)
- Natalie I. Vokes (ORCID: https://orcid.org/0000-0002-3766-5335)
- Seyedali Mirjalili (ORCID: https://orcid.org/0000-0002-1443-9458)
- Jianjun Zhang (ORCID: https://orcid.org/0000-0001-7872-3477)
- J. Jack Lee (ORCID: https://orcid.org/0000-0001-5469-9214)
- Wentao Li (ORCID: https://orcid.org/0000-0002-6826-6636)
Institutions
- The University of Texas MD Anderson Cancer Center (US)
- Harvard University (US)
- VSB - Technical University of Ostrava (CZ)
- Yale University (US)
- Massachusetts General Hospital (US)
- HOPE Clinic (US)
- Dana-Farber Cancer Institute (US)
- Torrens University Australia (AU)
- Mayo Clinic in Florida (US)
Publication Details
- Journal
- Nature Communications
- Published
- 2025-07-24
- DOI
- https://doi.org/10.1038/s41467-025-61823-w
- Citations
- 5
- Primary Topic
- Cancer Immunotherapy and Biomarkers
- Type
- article
- Field-Weighted Citation Impact
- 5.23
Funders
- Cancer Prevention and Research Institute of Texas
- National Institutes of Health
- University of Texas MD Anderson Cancer Center