Generalizable AI predicts immunotherapy outcomes across cancers and treatments

· Source: Machine learning : nature.com subject feeds · Field: Science & Research — Artificial Intelligence & Machine Learning, Health & Medical Research, Life Sciences & Biology · Depth: Expert, extended

Summary

COMPASS, a pan-cancer foundation model, accurately predicts immunotherapy response from bulk tumor transcriptomes using a concept bottleneck transformer. Trained on 10,184 tumors across 33 cancer types, COMPASS encodes gene expression into 44 biologically grounded immune concepts. It outperforms 22 existing methods across 16 clinical cohorts spanning seven cancers and six immune checkpoint inhibitors (ICIs), achieving an 8.5% improvement in accuracy and a 15.7% increase in area under the precision-recall curve. The model generalizes to unseen cancer types and treatments, and in survival analyses, patients predicted as responders had significantly longer overall survival (hazard ratio = 4.7, P < 0.0001). Personalized response maps offer mechanistic insights, identifying programs like TGFβ signaling and B cell deficiency in non-responders.

Key takeaway

For AI Scientists and Research Scientists developing precision oncology tools, COMPASS offers a robust, interpretable framework for predicting immunotherapy outcomes. You should integrate concept bottleneck models and multi-stage fine-tuning. This enhances generalizability and mechanistic insight, especially when working with diverse cancer types or limited clinical data. This approach can improve patient stratification and biomarker discovery in future trials.

Key insights

COMPASS uses a concept bottleneck transformer to predict ICI response from transcriptomes, offering interpretable mechanistic insights.

Principles

Method

COMPASS pretrains on 10,184 TCGA transcriptomes using self-supervised contrastive learning, then fine-tunes on clinical cohorts with parameter-efficient strategies (e.g., partial fine-tuning, linear probing) to predict ICI response.

In practice

Topics

Code references

Best for: AI Scientist, Research Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Machine learning : nature.com subject feeds.