Clustering Multimodal Ensemble Learning for Predicting Gastric Cancer Neoadjuvant Chemotherapy Efficacy
Accurately predicting the response to neoadjuvant chemotherapy (NCT) is crucial for gastric cancer treatment planning. Multimodal diagnostic methods enhance prediction accuracy by integrating images and clinical features, but they often overlook intra-class differences, causing feature entanglement, and single models struggle to capture dataset diversity. To address these issues, we propose Clustering Multimodal Ensemble Learning (CMEL) framework, which aligns and fuses image features from different domains with clinical features and performs decision fusion using an ensemble model. Specifically, we design the clustering process using Siamese network and memory bank to partition images in feature space and introduce contrastive clustering loss to enhance clustering properties and reduce feature entanglement across domains. We design Hierarchical Feature Alignment Fusion (HFAF) module to generate multi-level image features at different depths and then separately align and fuse them with clinical features, providing diverse features to cover variations in the dataset. We design Dynamic Multi-Classifier Decision Fusion (DMDF) module to predict on fused features using a gating mechanism supervised by pseudo-labels for dynamic weights, enhancing prediction across different domains. On our collected the Gastric Cancer Chemotherapy Response (GCCR) dataset, CMEL achieves an AUC of 76.09%, accuracy of 72.00%, PPV of 86.02%. These results demonstrate the feasibility of joint image and clinical data in predicting NCT response and provide valuable insights for gastric cancer treatment. The code is available at https://github.com/WHX0259/CMEL.
Authors
- Yuehua Gong (ORCID: https://orcid.org/0000-0002-0146-2165)
- Zelan Li (ORCID: https://orcid.org/0000-0003-0156-6084)
- Jianning Chi (ORCID: https://orcid.org/0000-0002-9748-5619)
- Yujin Shi
- X. Sun (ORCID: https://orcid.org/0000-0003-4409-4574)
- Huixuan Wu (ORCID: https://orcid.org/0000-0003-1999-7114)
- Lu Wang (ORCID: https://orcid.org/0000-0002-1799-8984)
- Zitian Zhang (ORCID: https://orcid.org/0000-0002-0488-5083)
Institutions
- Dalian University of Technology (CN)
- First Hospital of China Medical University (CN)
- Liaoning Cancer Hospital & Institute (CN)
- China Medical University (CN)
- Northeastern University (CN)
Publication Details
- Published
- 2024-12-03
- DOI
- https://doi.org/10.1109/bibm62325.2024.10821805
- Primary Topic
- Gastric Cancer Management and Outcomes
- Type
- article
- Field-Weighted Citation Impact
- 0.00