Development of a CT radiomics prognostic model for post renal tumor resection overall survival based on transformer enhanced K-means clustering

Abstract

Background

Kidney tumors, common in the urinary system, have widely varying survival rates post-surgery. Current prognostic methods rely on invasive biopsies, highlighting the need for non-invasive, accurate prediction models to assist in clinical decision-making.

Purpose

This study aimed to construct a K-means clustering algorithm enhanced by Transformer-based feature transformation to predict the overall survival rate of patients after kidney tumor resection and provide an interpretability analysis of the model to assist in clinical decision-making.

Methods

This study was based on a publicly available C4KC-KiTS-2019 dataset from the TCIA database, including preoperative computed tomography (CT) images and survival time data of 210 patients. Initially, the radiomics features of the kidney tumor area were extracted using the 3D slicer software. Feature selection was then conducted using ICC, mRMR algorithms, and LASSO regression to calculate radiomics scores. Subsequently, the selected features were input into a pre-trained Transformer model for feature transformation to obtain a higher-dimensional feature set. Then, K-means clustering was performed using this feature set, and the model was evaluated using receiver operating characteristic (ROC) and Kaplan–Meier curves. Finally, the SHAP interpretability algorithm was used for the feature importance analysis of the K-means clustering results.

Results

Eleven important features were selected from 851 radiomics features. The K-means clustering model after Transformer feature transformation showed AUCs of 0.889, 0.841, and 0.926 for predicting 1-, 3-, and 5-year overall survival rates, respectively, thereby outperforming both the K-means model with original feature inputs and the radiomics score method. A clustering analysis revealed survival prognosis differences among different patient groups, and a SHAP analysis provided insights into the features that had the most significant impacts on the model predictions.

Conclusions

The K-means clustering algorithm enhanced by the Transformer feature transformation proposed in this study demonstrates promising accuracy and interpretability in predicting the overall survival rate after kidney tumor resection. This method provides a valuable tool for clinical decision-making and contributes to improved management and treatment strategies for patients with kidney tumors.