Predicting bone metastasis risk of colorectal tumors using radiomics and deep learning ViT model

Abstract

Background

Colorectal cancer is a prevalent malignancy with a significant risk of metastasis, including to bones, which severely impacts patient outcomes. Accurate prediction of bone metastasis risk is crucial for optimizing treatment strategies and improving prognosis.

Purpose

This study aims to develop a predictive model combining radiomics and Vision Transformer (ViT) deep learning techniques to assess the risk of bone metastasis in colorectal cancer patients using both plain and contrast-enhanced CT images.

Materials and methods

We conducted a retrospective analysis of 155 colorectal cancer patients, including 81 with bone metastasis and 74 without. Radiomic features were extracted from segmented tumors on both plain and contrast-enhanced CT images. LASSO regression was applied to select key features, which were then used to build traditional machine learning models, including Support Vector Machine (SVM), K-Nearest Neighbors (KNN), Random Forest, LightGBM, and XGBoost. Additionally, a dual-modality ViT model was trained on the same CT images, with a late fusion strategy employed to combine outputs from the different modalities. Model performance was evaluated using AUC-ROC, accuracy, sensitivity, and specificity, and differences were statistically assessed using DeLong’s test.

Results

The ViT model demonstrated superior predictive performance, achieving an AUC of 0.918 on the test set, significantly outperforming all traditional radiomics-based models. The SVM model, while the best among traditional models, still underperformed compared to the ViT model. The ViT model’s strength lies in its ability to capture complex spatial relationships and long-range dependencies within the imaging data, which are often missed by traditional models. DeLong’s test confirmed the statistical significance of the ViT model’s enhanced performance, highlighting its potential as a powerful tool for predicting bone metastasis risk in colorectal cancer patients.

Conclusion

The integration of radiomics with ViT-based deep learning offers a robust and accurate method for predicting bone metastasis risk in colorectal cancer patients. The ViT model’s ability to analyze dual-modality CT imaging data provides greater precision in risk assessment, which can improve clinical decision-making and personalized treatment strategies. These findings underscore the promise of advanced deep learning models in enhancing the accuracy of metastasis prediction. Further validation in larger, multicenter studies is recommended to confirm the generalizability of these results.