Gene Swin transformer: New deep learning method for colorectal cancer prognosis using transcriptomic data

Abstract

Transcriptome sequencing has become essential in clinical tumor research, providing in-depth insights into the biology and functionality of tumor cells. However, the vast amount of data generated and the complex relationships between gene expressions make it challenging to effectively identify clinically relevant information. In this study, we developed a method called Gene Swin Transformer to address these challenges. This approach converts transcriptomic data into Synthetic Image Elements (SIEs). We utilized data from 12 datasets, including GSE17536-GSE103479 datasets (n = 1771) and The Cancer Genome Atlas (n = 459), to generate SIEs. These elements were then classified based on survival time using deep learning algorithms to predict colorectal cancer prognosis and build a reliable prognostic model. We trained and evaluated four deep learning models-BeiT, ResNet, Swin Transformer, and ViT Transformer-And compared their performance. The enhanced Swin-T model outperformed the other models, achieving weighted precision, recall, and F1 scores of 0.708, 0.692, and 0.705, respectively, along with area under the curve values of 80.2%, 72.7%, and 76.9% across three datasets. This model demonstrated the strongest prognostic prediction capabilities among those evaluated. Additionally, the PEX10 gene was identified as a key prognostic marker through both visual attention matrix analysis and bioinformatics methods. Our study demonstrates that the Gene Swin model effectively transforms Ribonucleic Acid (RNA) sequencing data into SIEs, enabling prognosis prediction through attention-based algorithms. This approach supports the development of a data-driven, unified, and automated model, offering a robust tool for classification and prediction tasks using RNA sequencing data. This advancement presents a novel clinical strategy for cancer treatment and prognosis forecasting.