ImmunoDiff: A Diffusion Model for Immunotherapy Response Prediction in Lung Cancer

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bhattacharya, Moinak, Huang, Judy, Sher, Amna F., Singh, Gagandeep, Chen, Chao, Prasanna, Prateek
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913866114400256
author Bhattacharya, Moinak
Huang, Judy
Sher, Amna F.
Singh, Gagandeep
Chen, Chao
Prasanna, Prateek
author_facet Bhattacharya, Moinak
Huang, Judy
Sher, Amna F.
Singh, Gagandeep
Chen, Chao
Prasanna, Prateek
contents Accurately predicting immunotherapy response in Non-Small Cell Lung Cancer (NSCLC) remains a critical unmet need. Existing radiomics and deep learning-based predictive models rely primarily on pre-treatment imaging to predict categorical response outcomes, limiting their ability to capture the complex morphological and textural transformations induced by immunotherapy. This study introduces ImmunoDiff, an anatomy-aware diffusion model designed to synthesize post-treatment CT scans from baseline imaging while incorporating clinically relevant constraints. The proposed framework integrates anatomical priors, specifically lobar and vascular structures, to enhance fidelity in CT synthesis. Additionally, we introduce a novel cbi-Adapter, a conditioning module that ensures pairwise-consistent multimodal integration of imaging and clinical data embeddings, to refine the generative process. Additionally, a clinical variable conditioning mechanism is introduced, leveraging demographic data, blood-based biomarkers, and PD-L1 expression to refine the generative process. Evaluations on an in-house NSCLC cohort treated with immune checkpoint inhibitors demonstrate a 21.24% improvement in balanced accuracy for response prediction and a 0.03 increase in c-index for survival prediction. Code will be released soon.
format Preprint
id arxiv_https___arxiv_org_abs_2505_23675
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle ImmunoDiff: A Diffusion Model for Immunotherapy Response Prediction in Lung Cancer
Bhattacharya, Moinak
Huang, Judy
Sher, Amna F.
Singh, Gagandeep
Chen, Chao
Prasanna, Prateek
Image and Video Processing
Computer Vision and Pattern Recognition
Accurately predicting immunotherapy response in Non-Small Cell Lung Cancer (NSCLC) remains a critical unmet need. Existing radiomics and deep learning-based predictive models rely primarily on pre-treatment imaging to predict categorical response outcomes, limiting their ability to capture the complex morphological and textural transformations induced by immunotherapy. This study introduces ImmunoDiff, an anatomy-aware diffusion model designed to synthesize post-treatment CT scans from baseline imaging while incorporating clinically relevant constraints. The proposed framework integrates anatomical priors, specifically lobar and vascular structures, to enhance fidelity in CT synthesis. Additionally, we introduce a novel cbi-Adapter, a conditioning module that ensures pairwise-consistent multimodal integration of imaging and clinical data embeddings, to refine the generative process. Additionally, a clinical variable conditioning mechanism is introduced, leveraging demographic data, blood-based biomarkers, and PD-L1 expression to refine the generative process. Evaluations on an in-house NSCLC cohort treated with immune checkpoint inhibitors demonstrate a 21.24% improvement in balanced accuracy for response prediction and a 0.03 increase in c-index for survival prediction. Code will be released soon.
title ImmunoDiff: A Diffusion Model for Immunotherapy Response Prediction in Lung Cancer
topic Image and Video Processing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2505.23675