Worse is better? Performance and bias implications of feature selection in radiomics-based survival analysis
We present an extension of the minimum redundancy, maximum relevance (mRMR) feature selection algorithm to accommodate right-censored survival outcomes. We use a set of 197 preoperative CT scans from patients that underwent hepatic resection to treat colorectal liver metastases, with the intention of predicting overall survival based on radiomic analysis of the largest metastasis. Using selection of the K-best features as a baseline, mRMR is introduced as a correction of the ranking, balancing relevance with the correlation of each candidate feature with those previously selected. Harrell’s C-index is introduced as a replacement for the F-statistic to accommodate right-censoring. Two pre-processing operations are considered: first, features of low univariate significance are removed; and second, features with low reproducibility in an independent data set are removed. Each algorithm was tested in a repeated 10-fold cross-validation of a Cox proportional hazards model of overall survival. We found that use of mRMR was associated with a decrease in performance C-index of -0.016. While mRMR models had lower bias, univariate significance thresholding increased the bias of mRMR models by 0.02, producing the models with highest bias. More stable feature selection methods were associated with reduced bias and better overall performance. The baseline K-best method, with no accounting for redundancy or censoring, and no pre-processing, obtained the best performance (C-index=0.600 (0.591–0.607)) and a bias of 0.000 (-0.001– 0.004).
Authors
- Amber L. Simpson (ORCID: https://orcid.org/0000-0002-4387-8417)
- Yun Shin Chun (ORCID: https://orcid.org/0000-0003-1384-8927)
- Mohammad Hamghalam (ORCID: https://orcid.org/0000-0003-2543-0712)
- Hyunseon C. Kang (ORCID: https://orcid.org/0000-0001-5313-3526)
- Jacob Peoples (ORCID: https://orcid.org/0000-0003-0191-7446)
- Richard Kinh Gian (ORCID: https://orcid.org/0000-0002-6554-0310)
- Natalie Gangai (ORCID: https://orcid.org/0000-0003-0603-7165)
- Maida Wasim
- Imani James
- X. John Rong (ORCID: https://orcid.org/0000-0002-1169-5042)
- Joshua Virani-Wall
Institutions
- Islamic Azad University, Tehran (IR)
- Memorial Sloan Kettering Cancer Center (US)
- The University of Texas MD Anderson Cancer Center (US)
- Queen's University (CA)
Publication Details
- Published
- 2025-04-04
- DOI
- https://doi.org/10.1117/12.3047247
- Primary Topic
- Radiomics and Machine Learning in Medical Imaging
- Type
- article
- Field-Weighted Citation Impact
- 0.00