A comparative study of machine learning methods for time-to-event survival data for radiomics risk modelling

Research output: Contribution to journalResearch articleContributedpeer-review

Contributors

Abstract

Radiomics applies machine learning algorithms to quantitative imaging data to characterise the tumour phenotype and predict clinical outcome. For the development of radiomics risk models, a variety of different algorithms is available and it is not clear which one gives optimal results. Therefore, we assessed the performance of 11 machine learning algorithms combined with 12 feature selection methods by the concordance index (C-Index), to predict loco-regional tumour control (LRC) and overall survival for patients with head and neck squamous cell carcinoma. The considered algorithms are able to deal with continuous time-to-event survival data. Feature selection and model building were performed on a multicentre cohort (213 patients) and validated using an independent cohort (80 patients). We found several combinations of machine learning algorithms and feature selection methods which achieve similar results, e.g.

, MSR-RF: C-Index = 0.71 and BT-COX: C-Index = 0.70 in combination with Spearman feature selection. Using the best performing models, patients were stratified into groups of low and high risk of recurrence. Significant differences in LRC were obtained between both groups on the validation cohort. Based on the presented analysis, we identified a subset of algorithms which should be considered in future radiomics studies to develop stable and clinically relevant predictive models for time-to-event endpoints.

Details

Original languageEnglish
Article number13206
Pages (from-to)13206-
JournalScientific Reports
Volume7
Issue number1
Publication statusPublished - 16 Oct 2017
Peer-reviewedYes

External IDs

Scopus 85031807357
researchoutputwizard legacy.publication#79774
PubMed 29038455
PubMedCentral PMC5643429
researchoutputwizard legacy.publication#79576
ORCID /0000-0002-7017-3738/work/142253913
ORCID /0000-0003-4261-4214/work/146644850
ORCID /0000-0003-1776-9556/work/171065666

Keywords