A Multimodal Ensemble Driven by Multiobjective Optimisation to Predict Overall Survival in Non-Small-Cell Lung Cancer

J Imaging. 2022 Nov 2;8(11):298. doi: 10.3390/jimaging8110298.

Abstract

Lung cancer accounts for more deaths worldwide than any other cancer disease. In order to provide patients with the most effective treatment for these aggressive tumours, multimodal learning is emerging as a new and promising field of research that aims to extract complementary information from the data of different modalities for prognostic and predictive purposes. This knowledge could be used to optimise current treatments and maximise their effectiveness. To predict overall survival, in this work, we investigate the use of multimodal learning on the CLARO dataset, which includes CT images and clinical data collected from a cohort of non-small-cell lung cancer patients. Our method allows the identification of the optimal set of classifiers to be included in the ensemble in a late fusion approach. Specifically, after training unimodal models on each modality, it selects the best ensemble by solving a multiobjective optimisation problem that maximises both the recognition performance and the diversity of the predictions. In the ensemble, the labels of each sample are assigned using the majority voting rule. As further validation, we show that the proposed ensemble outperforms the models learning a single modality, obtaining state-of-the-art results on the task at hand.

Keywords: convolutional neural networks; medical imaging; multiexpert systems; multimodal deep learning; oncology; optimisation; precision medicine; tabular data.

Grants and funding

This work was partially founded by: (i) Università Campus Bio-Medico di Roma under the programme “University Strategic Projects 2018 call” within the project “a CoLlAborative multi-sources Radiopathomics approach for personalised Oncology in non-small cell lung cancer (CLARO)”, (ii) “University-Industry Educational Centre in Advanced Biomedical and Medical Informatics (CEBMI)” (Grant agreement no. 612462-EPP-1-2019-1-SK-EPPKA2-KA, Educational, Audiovisual and Culture Executive Agency of the European Union), (iii) the project n. F/130096/01-05/X38-Fondo per la Crescita Sostenibile-ACCORDI PER L’INNOVAZIONE DI CUI AL D.M. 24 MAGGIO 2017—Ministero dello Sviluppo Economico (Italy), (iv) Programma Operativo Nazionale (PON) “Ricerca e Innovazione” 2014–2020 CCI2014IT16M2OP005 Azione IV.4, (v) Regione Lazio PO FSE 2014-2020 Avviso Pubblico “Contributi per la permanenza nel mondo accademico delle eccellenze” Asse III—Istruzione e formazione-Priorità di investimento 10 ii)-Obiettivo specifico 10.5 Azione Cardine 21.