A distributed feature selection pipeline for survival analysis using radiomics in non-small cell lung cancer patients

Benedetta Gottardelli, Varsha Gouthamchand, Carlotta Masciocchi*, Luca Boldrini, Antonella Martino, Ciro Mazzarella, Mariangela Massaccesi, René Monshouwer, Jeroen Findhammer, Leonard Wee, Andre Dekker, Maria Antonietta Gambacorta, Andrea Damiani

*Corresponding author for this work

Research output: Contribution to journalArticleAcademicpeer-review

Abstract

Predictive modelling of cancer outcomes using radiomics faces dimensionality problems and data limitations, as radiomics features often number in the hundreds, and multi-institutional data sharing is ()often unfeasible. Federated learning (FL) and feature selection (FS) techniques combined can help overcome these issues, as one provides the means of training models without exchanging sensitive data, while the other identifies the most informative features, reduces overfitting, and improves model interpretability. Our proposed FS pipeline based on FL principles targets data-driven radiomics FS in a multivariate survival study of non-small cell lung cancer patients. The pipeline was run across datasets from three institutions without patient-level data exchange. It includes two FS techniques, Correlation-based Feature Selection and LASSO regularization, and Cox Proportional-Hazard regression with Overall Survival as endpoint. Trained and validated on 828 patients overall, our pipeline yielded a radiomic signature comprising "intensity-based energy" and "mean discretised intensity". Validation resulted in a mean Harrell C-index of 0.59, showcasing fair efficacy in risk stratification. In conclusion, we suggest a distributed radiomics approach that incorporates preliminary feature selection to systematically decrease the feature set based on data-driven considerations. This aims to address dimensionality challenges beyond those associated with data constraints and interpretability concerns.
Original languageEnglish
Article number7814
JournalScientific Reports
Volume14
Issue number1
DOIs
Publication statusPublished - 1 Dec 2024

Keywords

  • Distributed learning
  • Feature selection
  • NSCLC
  • Radiomics

Cite this