Skip to main content

ORIGINAL RESEARCH article

Front. Oncol. , 06 February 2025

Sec. Gastrointestinal Cancers: Colorectal Cancer

Volume 15 - 2025 | https://doi.org/10.3389/fonc.2025.1496820

A pelvis MR transformer-based deep learning model for predicting lung metastases risk in patients with rectal cancer

Yin Li,,&#x;Yin Li1,2,3†Shuang Li,&#x;Shuang Li2,4†Ruolin Xiao,&#x;Ruolin Xiao5,6†Xi LiXi Li7Yongju Yi,Yongju Yi1,2Liangyou Zhang,Liangyou Zhang1,2You Zhou,You Zhou1,2Yun WanYun Wan7Chenhua WeiChenhua Wei3Liming Zhong,Liming Zhong5,6Wei Yang,*Wei Yang5,6*Lin Yao,,*Lin Yao2,3,4*
  • 1Department of Information, The Sixth Affiliated Hospital, Sun Yat-sen University, Guangzhou, China
  • 2Biomedical Innovation Center, The Sixth Affiliated Hospital, Sun Yat-sen University, Guangzhou, China
  • 3Department of Information, The Sixth Affiliated Hospital, Sun Yat-sen University Yuexi Hospital, Maoming, China
  • 4Department of General Practice, The Sixth Affiliated Hospital, Sun Yat-sen University, Guangzhou, China
  • 5School of Biomedical Engineering, Southern Medical University, Guangzhou, China
  • 6Guangdong Provincial Key Laboratory of Medical Image Processing, Guangzhou, China
  • 7Department of Radiology, The Sixth Affiliated Hospital, Sun Yat-sen University Yuexi Hospital, Maoming, China

Objective: Accurate preoperative evaluation of rectal cancer lung metastases (RCLM) is critical for implementing precise medicine. While artificial intelligence (AI) methods have been successful in detecting liver and lymph node metastases using magnetic resonance (MR) images, research on lung metastases is still limited. Utilizing MR images to classify RCLM could potentially reduce ionizing radiation exposure and the costs associated with chest CT in patients without metastases. This study aims to develop and validate a transformer-based deep learning (DL) model based on pelvic MR images, integrated with clinical features, to predict RCLM.

Methods: A total of 819 patients with histologically confirmed rectal cancer who underwent preoperative pelvis MRI and carcinoembryonic antigen (CEA) tests were enrolled. Six state-of-the-art DL methods (Resnet18, EfficientNetb0, MobileNet, ShuffleNet, DenseNet, and our transformer-based model) were trained and tested on T2WI and DWI to predict RCLM. The predictive performance was assessed using the receiver operating characteristic (ROC) curve.

Results: Our transformer-based DL model achieved impressive results in the independent test set, with an AUC of 83.74% (95% CI, 72.60%-92.83%), a sensitivity of 80.00%, a specificity of 78.79%, and an accuracy of 79.01%. Specifically, for stage T4 and N2 rectal cancer cases, the model achieved AUCs of 96.67% (95% CI, 87.14%-100%, 93.33% sensitivity, 89.04% specificity, 94.74% accuracy), and 96.83% (95% CI, 88.67%-100%, 100% sensitivity, 83.33% specificity, 88.00% accuracy) respectively, in predicting RCLM. Our DL model showed a better predictive performance than other state-of-the-art DL methods.

Conclusion: The superior performance demonstrates the potential of our work for predicting RCLM, suggesting its potential assistance in personalized treatment and follow-up plans.

Introduction

Colorectal cancer ranks third globally in terms of incidence among the most prevalent cancers, with rectal cancer (RC) accounting for approximates 30% of all cases within the colorectal cancer category (1, 2). Due to the unique venous drainage through the iliac vessels in the rectum, RC patients have a significantly higher incidence of lung metastasis than those with colon cancers (35). Surgical resection of lung metastasis is an optimal treatment method for RC patients to survive long-term (6), which increases the 5-year survival rate to approximately 50% (7, 8). Nevertheless, the prognosis of RC with metastasis remains poor without timely treatment (9). Thus, timely assessment of lung metastasis in patients with RC (RCLM) is important, which further influences the clinical personalized treatment and follow-up plans.

However, the evaluation of RCLM through long-term follow-up with chest computed tomography (CT) scans may present challenges. Radiologists may encounter difficulties in detecting early metastatic lesions due to their small size and the presence of various benign lesions, which will bring additional ionizing radiation exposure and the costs associated with chest CT in patients without lung metastases and may delay the treatment period (1012). Therefore, a new diagnostic method is needed for reducing radiation exposure and mitigating treatment delays in patients without lung metastases.

Pelvis magnetic resonance imaging (MRI), which has no radiation exposure, is a standard procedure for the detection and staging of RC (13, 14). Previous studies have highlighted the promising role of T2-weighted image (T2WI) in detecting distant metastasis (DM) in RC patients (15, 16). Besides, diffusion-weighted image (DWI), using differences in water molecules to generate image contrast, has shown improved accuracy in detecting RC patients with DM (17, 18). Artificial intelligence (AI) has shown great success in the detection of liver or lymph node metastasis (1922). However, few have attempted to evaluate of AI models for predicting RCLM based on pelvis MR images of primary tumor. Although these MRI-based AI methods can potentially predict lung metastasis risk, the inherent locality and the consecutive down-sampling operations in the convolutional neural networks limit the extraction of global spatial dependencies. Moreover, the performance of predicting RCLM based solely on pelvis MRI scans is modest.

In this study, we introduce a pelvis MR transformer-based deep learning (DL) model for predicting RCLM based on T2WI and DWI. Our DL model leverages pre-trained UniMiSS (23), which built upon the ViT framework and trained on medical images, as our primary feature extraction network. Numerous studies have shown that carcinoembryonic antigen (CEA) is a critical biomarker for monitoring recurrence and metastasis in RC patients (14, 32). Therefore, we incorporate clinical information such as CEA, age, and gender due to their stronger association with RCLM, in our DL model to improve the performance of predicting RCLM.

Materials and methods

Ethics statement

This single-center retrospective study received approval from our institutional review board and complied with ethical regulations. The requirement for informed consent was waived for this retrospective study of anonymized data.

Patients

With the approval of institution ethics committee, we collected two independent patient cohorts. The detailed inclusion and exclusion criteria are shown in Figure 1. The lung metastasis cohort of 157 RC patients with lung metastasis risk diagnosed between Jan 2018 and Jun 2023. Inclusion criteria were as follows: (a) pathological confirmation of rectal adenocarcinoma; (b) availability of pre-treatment pelvis MRI scans before the initiation of therapy; and (c) utilization of high-resolution contrast enhanced chest CT scans for lung metastasis diagnosis. The exclusion criteria included: (a) concurrent others primary malignant neoplasms or previous anticancer treatment; (b) missing MRI data or insufficient image quality; (c) missing or incomplete electronic medical records; and (d) simultaneous occurrence of other DM.

Figure 1
www.frontiersin.org

Figure 1. The flowcharts of patient selection.

Besides, we collected a non-overlapping cohort of 662 patients diagnosed with RC at the same institution between Jan 2018 and Jun 2023, forming a distinct non-metastasis cohort. Inclusion criteria (a) and (b) mirrored those of the lung metastasis cohort, while criteria (c) were omitted in the non-metastasis cohort due to the absence of metastasis in these patients. The exclusion criteria remained consistent with those applied to the lung metastasis cohort.

Clinical information, including age, gender, pre-treatment histologic grade (24), and CEA levels, was extracted from medical records. It is important to note that CEA levels were determined through routine blood tests conducted within one week before treatment.

MR imaging protocol

MRI was conducted using a 1.5-T MR system (Optimal 360, GE Healthcare, Waukesha, Wis) or a 3.0-T MR system (MR 750w, GE Medical System) with a phased-array body coil (eight-channel and sixteen-channel phased-array body coil). The standard procedure included axial oblique T2WI sequences and transverse DWI sequences. For the T2WI sequences, the in-plane pixel spacings ranged from 0.366 mm to 0.703 mm, with an average of 0.490 mm, and slice thicknesses ranged from 3.500 mm to 7.000 mm, with an average of 4.692 mm. For the DWI sequences, the in-plane pixel spacings ranged from 0.976 mm to 1.953 mm, with an average of 1.526 mm, and slice thicknesses ranged from 3.500 mm to 7.000 mm, with an average of 4.694 mm. Among the 819 patients in the lung metastasis cohort and non-metastasis cohort, images obtained from different scanners were randomly distributed in the training, validation, and independent test sets.

Image pre-processing and segmentation

Radiologists with over 10 years of experience in MRI manually delineated the entire RC tumor using ITK-SNAP 3.9 on pre-treatment T2WI and pre-treatment DWI at b=1000 s/mm2. The resulting tumor masks were cropped into image patches as the inputs of 3D networks. To mitigate the impact of variability in acquisition and sequence parameters, image pre-processing was implemented before analysis. All MR images were used the Simple-ITK toolkit to correct the bias field artifacts (25). Gray-level normalization was applied to harmonize the gray values of MR images, compensating for intensity variations across different MRI scanners.

Network details

The architecture of our proposed transformer-based DL model for predicting RCLM is illustrated in Figure 2. We used two pre-trained UniMiSS models, which were built upon the ViT framework, as our primary feature extraction network. T2WI and DWI scans underwent individual processing through each branch network, and the resulting features were fused through concatenation using a post-fusion approach. This fusion strategy engendered a comprehensive representation, combining complementary information from both T2WI and DWI. To distill spatial information, an adaptive max pooling operation was employed to reduce the feature length to 1000. These features were then concatenated with clinical information, including CEA level, age, and gender. The combined features were subsequently fed into a fully connected layers to effectively learn the imaging and clinical information for accurate predicting RCLM. This architecture enables the model to leverage both imaging and clinical data synergistically to enhance prediction performance.

Figure 2
www.frontiersin.org

Figure 2. Overview of the proposed transformer-based deep learning model for rectal cancer with lung metastases.

Data partitioning and network implementation

The performance of the models was evaluated on an independent test set, which was not included in the model training. The remaining data were randomly divided into training and validation sets for parameter tuning. Once optimal model parameters were determined using the training and validation sets, the models were evaluated on the independent test set. The models were implemented using PyTorch v1.7. Our models process 3D volumes, with a data batch size of 32 and input image dimensions of 128 (height) × 128 (width) × 16 (depth). In our preprocessing pipeline, we use manually annotated RC tumor masks to crop image patches from the 3D volume data to a specified size, while ensuring that the tumor region is fully retained. During the cropping process, the cropping area is dynamically adjusted based on the mask to ensure that the entire tumor is included within the cropped region. All MR images were linearly transformed to the range of [-1, 1] via gray-level normalization. Data augmentation techniques, including random rotation in the range of [-30, 30] and random scaling in the scale of [0.95, 1.05], were applied for model training to enhance generalization. For optimization, the Adam optimizer with an initial learning rate of 0.0002 was employed, and the models were trained for 100 epochs. Cross-entropy loss was used as the loss function for model training. To ensure a fair comparison, all compared models were trained using the same image size, learning rate, and number of iterations.

Among these, the model using only clinical characteristics employed the Logistic Regression model. The remaining five models of seven combinations of MRI sequences and clinical features used the pre-trained UniMiSS model (built on the ViT framework) to extract image features. Specifically, the “Images-only” model concatenated the DWI and T2WI images along the channel dimension before feeding them into the feature extraction network. When clinical features are included, a post-fusion method was applied to combine the image and clinical features. In addition, the two demographic characteristics (age and gender) and one CEA used in the comparison experiment were pre-treatment. Gender was labeled with 0 and 1, age was standardized with z-score, and CEA was scaled using minimum-maximum normalization. The latter two characteristics of information were subsequently transformed into one-dimensional representations via Word Embedding.

For a thorough evaluation, our method is compared with state-of-the-art deep learning models, including ResNet18, EfficientNetb0, MobileNet, ShuffleNet, and DenseNet. ResNet18 employs residual blocks with skip connections, enabling the network to learn residual functions. EfficientNetb0 uses a compound scaling method to balance depth, width, and resolution, optimizing performance and computational efficiency. MobileNet leverages depthwise separable convolutions to reduce complexity. ShuffleNet introduces a channel shuffle operation to enhance the efficiency of group convolutions while minimizing computational costs. DenseNet incorporates dense connections among all layers, fostering feature reuse and improving gradient flow, resulting in reduced parameters and enhanced training efficiency.

Statistical analysis

All well-trained models were evaluated on both validation and independent test sets. The performance of the prediction model was assessed using various metrics, including the area under the receiver operating characteristic curve (AUC), accuracy (ACC), sensitivity (SEN), specificity (SPE) and the wilcoxon rank-sum test for statistical comparison. The optimal threshold for the AUC value was determined by maximizing the sum of the sensitivity and specificity values. The 95% CIs were obtained using bootstrapping to assess variability, and p < 0.05 considered indicative of statistically significant difference.

Results

Patient characteristics

Applying specific inclusion and exclusion criteria, our study enrolled a total of 819 patients, comprising 524 males and 295 females, with an average age of 57.25 ± 12.14 years (ranging from 17 to 86 years). The average CEA level among the patients was 12.58 ng/mL (ranging from 0.50 to 99.62 ng/mL). Among these patients, 657 were randomly assigned to the training set, 81 to the validation set, and 81 to the independent test set. Of the total patients, 157 patients (19.17%) were RC with lung metastasis, while 662 patients (80.83%) were RC without metastasis. In terms of tumor staging, there were 19 patients at the T1 stage (with 1 lung metastasis), 122 patients at the T2 stage (with 9 lung metastasis), 485 patients at T3 stage (with 82 lung metastasis), and 193 patients at T4 stage (with 65 lung metastasis). Regarding the N stage, there were 271 patients at the N0 stage (with 31 lung metastasis), 282 patients at the N1 stage (with 54 lung metastasis), and 266 patients at N2 stage (with 72 lung metastasis). The details of the demographic and clinical characteristics of patients were presented in Table 1.

Table 1
www.frontiersin.org

Table 1. The details of the demographic and clinical characteristics of patients.

Model performance on the validation and independent test sets

We developed a transformer-based model for the prediction of RCLM. As shown in Table 2, the transformer-based model achieved an Area Under Curve (AUC) of 84.24% (95% CI, 73.87%-92.68%) on the validation set and 83.74% (95% CI, 72.60%-92.83%) on the independent test set. Our model outperformed the performance of ResNet18, EfficientNetb0, MobileNet, ShuffleNet, and DenseNet. Figures 3A, B illustrates the Receiver Operating Characteristic (ROC) curves for six deep learning models. These models yielded AUC values of 70.30% (95% CI, 56.18%-82.92%, p = 0.0196), 72.02% (95% CI, 58.73%-83.08%, p = 0.0186), 72.32% (95% CI, 56.36%-86.31%, p = 0.0158), 75.96% (95% CI, 61.84%-88.36%, p = 0.0495), 77.17% (95% CI, 64.26%-87.92%, p = 0.0412), and 83.74% (95% CI, 72.60%-92.83%) for ResNet18, EfficientNetb0, MobileNet, ShuffleNet, DenseNet and our method on independent test set, respectively.

Table 2
www.frontiersin.org

Table 2. RCLM prediction performance obtained by the different DL models on both the validation and independent test set.

Figure 3
www.frontiersin.org

Figure 3. ROC curves of six distinct deep learning-based detection models, including ResNet18, EfficientNetb0, MobileNet, ShuffleNet, DenseNet and the proposed method, for rectal cancer with lung metastases (RCLM) detection on the validation set (A) and independent test set (B); ROC curves of seven kinds of medical images and clinical features combination modes to detect RCLM on the validation set (C) and independent test set (D).

Table 3 presented the performance of our transformer-based model trained with seven combinations of MRI sequences and clinical features. These combinations include single image models (DWI-only or T2WI-only), models using a single image and clinical features (DWI-Clinical features or T2WI-Clinical features), a model using only clinical features (Clinical features-only), an image-only model with DWI and T2WI (Images-only), and a model using both DWI, T2WI, and clinical features (Images-Clinical features (Ours)). Incorporating two demographic characteristics (age and gender) and one clinicopathologic factors (CEA) into our model achieved the best performance. As outlined in Table 3, with increases of 2.86% in AUC, 23.07% in accuracy, and 30.00% in specificity compared to the original Image-only model on the independent test set, the combination of T2WI, DWI, and clinical features achieved the AUC of 83.74% (95% CI, 72.60%-92.83%) on the independent test set. Figures 3C, D illustrate the ROC curves for seven different combinations of image and clinical features. These combinations yielded AUC values of 56.57% (95% CI, 41.69%-71.22%, p = 0.0233), 61.62% (95% CI, 43.67%-78.57%, p = 0.0046), 72.73% (95% CI, 60.40%-83.75%, p = 0.0124), 69.70% (95% CI, 52.96%-84.55%, p < 0.001), 51.52% (95% CI, 37.73%-64.58%, p < 0.001), 81.41% (95% CI, 69.93%-90.72%, p = 0.0143), and 83.74% (95% CI, 72.60%-92.83%) for DWI-only, T2WI-only, DWI-clinical characteristics, DWI-clinical characteristics, clinical characteristics-only, images-only, and images-clinical characteristic on independent test set, respectively.

Table 3
www.frontiersin.org

Table 3. Detection performance of seven combination models on the validation and independent test set.

Table 4 and Figure 4 presented comprehensive results on both the validation set and the independent test set, focusing on different T and N stages. In Table 4, we provided a directly overview of various models’ performance at N stages. Furthermore, Figure 4 visually illustrated the AUC, accuracy (ACC), sensitivity (SEN), and specificity (SPE) for stages T3 and T4 in different models. Our method outperformed the aforementioned state-of-the-art DL methods in terms of AUC in general. Specifically, we achieved an AUC of 76.32% (95% CI, 59.80%-89.55%) in the T3 stage, an AUC of 96.67% (95% CI, 87.14%-100%) in the T4 stage, an AUC of 85.00% (95% CI, 66.67%-100%) in the N0 stage, an AUC of 54.46% (95% CI, 29.89%-82.76%) in the N1 stage, and an AUC of 96.83% (95% CI, 88.67%-100%) in the N2 stage on the independent test set.

Figure 4
www.frontiersin.org

Figure 4. Radar plots that summarize the detection performance of the various models on the validation and independent test set. Characteristics of Patients Stratified by T3 and T4 stage. To test the efficacy of each algorithm, we calculated ACC = Accuracy, SEN = Sensitivity, SPE = Specificity, AUC= Area Under ROC Curve.

Table 4
www.frontiersin.org

Table 4. Detection performance of the various models on the validation and independent test set.

Performance of our model vs. experts

Three experts (Li X, Li B, Wan Y, with 32 years, 10 years and 4 years of experience, respectively) dedicated to imaging diagnosis based on MRI data in the independent test set. They consecutively and independently evaluated the MRI data from the independent test set. The diagnostic performance of subjective evaluation by three radiologists was presented in Table 5. The results of our model had been binarized for fair comparison. Table 5 showed performance with AUCs of 56.52% (95% CI, 45.30%-69.65%, p < 0.001), 56.82% (95% CI,44.16%-70.89%, p < 0.001), 62.27% (95% CI, 47.88%-76.32%, p < 0.001) and 83.74% (95% CI,72.60%-92.83%) for three experts and our model. The diagnostic time for each case was 22.32s, 38.62s, 11.22s, and 0.67s for three experts and our model, respectively. As shown in Table 5, the experts’ results showed largely individual differences. In contrast, our model achieved the best performance in predicting RCLM.

Table 5
www.frontiersin.org

Table 5. Detection performance by radiologists on the independent test set.

Discussion

This research aimed to investigate the potential of utilizing DL for RCLM detection based on pelvis MRI, with the goal of improving clinical decision-making, reducing radiation exposure, and mitigating treatment delays in patients without lung metastases. Our developed transformer-based model showed improved performance compared to five state-of-the-art DL models. Furthermore, the combination of T2WI and DWI MRI sequences with clinical features model achieved the best performance in both validation and testing sets.

Endorectal ultrasound has been used for the preoperative staging of early RC. However, its accuracy is highly dependent on operator experience and tumor size, which limits its clinical applicability (33). Positron emission tomography (PET)/CT provides valuable metabolic information but has relatively low spatial resolution and poor soft tissue contrast, which limits its sensitivity for detecting small lesions or early metastases (34). While PET/MRI can generate high-resolution anatomical and functional data with promising results for RC staging, it requires longer acquisition times and is more expensive (35). In our study, the combination of T2WI and DWI was chosen based on their complementary strengths: T2WI provides detailed structural information critical for local staging, while DWI enhances the detection of DM by offering functional insights. This combination was selected to maximize sensitivity and accuracy in detecting RC and its metastases, as demonstrated by the variations in sensitivity observed in Table 3.

As shown in Table 1, our dataset only contained only one RC patients with lung metastasis at T1 stage and 9 positive cases at T2 stage. Specifically, 9 patients with lung metastasis were randomly divided in the training set, resulting 2 cases in validation set and one case in testing set. Consequently, traditional metrics such as AUC, SEN, and SPE become inadequate for evaluating model performance when there are very few positive samples. Thus, we chose to focus on presenting results for RC patients at T3-T4 stages in Figure 4. Although our model demonstrated strong performance at T3-T4 stages, the limited sample size of T1-T2 stages posed challenges, resulting in modest information capture and average performance (26). Moreover, the locally advanced RC (T3-T4) is an important risk factor supporting lung metastasis diagnosis (3). Our transformer-based model yielded superior performance with an AUC of 76.32% (95% CI, 59.80%-89.55%) in the T3 stage and an AUC of 96.67% (95% CI, 87.14%-100%) in the T4 stage.

Existing state-of-the-art classification models for RCLM prediction suffer from field-of-view limitations due to the local receptive fields inherent in convolutions. Relying solely on local feature learning is insufficient for capturing the complex characteristics needed to predict lung metastases from pelvic MRI scans. In contrast, our Transformer-based models can capture global dependencies across image regions and integrate clinical features, resulting in more robust RCLM predictions. Our Transformer-based model offers several advantages over CNN-based methods. Firstly, our model consistently outperformed other state-of-the-art models, demonstrating its potential for higher diagnostic accuracy. Secondly, by integrating clinical features such as CEA levels, age, and gender, our model benefits from a holistic approach to prediction, which may help in capturing patient-specific factors that are important for accurate diagnosis.

Timely detection and intervention of lung metastasis play an important role in guiding clinicians in determining clinical decision-making, ultimately leading to enhanced R0 resection rates, reduced postoperative recurrence risks, and improved overall survival rates (27, 28). The method developed in this study provides the ability to predict RCLM based on pelvis MR images primary tumor. With its reliable information on lung metastasis detection, the model has the potential to alleviate the burden on clinicians and improve the efficiency of decision-making. Moreover, the model’s capability to perform these tasks based solely on MR images streamlines workflow and reduces dependence on other procedures, making it applicable in various clinical settings where MRI is routinely performed (29). The study’s findings demonstrated that the predictive performance of our model, leveraging image and clinical features of the primary tumor, surpassed the subjective evaluation by radiologists (Table 5).

Improving the accuracy of predicting distant metastasis in rectal cancer is crucial for informed clinical decision-making (10, 11). Previous research has predominantly focused on developing DL methods for predicting liver metastasis (21, 22, 30, 31) or lymph node metastasis (12, 15, 18) of RC. Specifically, there studies indicate that the imaging features of preoperative CT or MRI scans of RC have predictive values for the risk of distant metastasis. For instance, Lee et al. (22) proposed a CNN-based model incorporating clinical information and CT scans to predict liver metastasis in colorectal cancer that obtained an AUC of 74.70% (95% CI: 71.10%-78.30%). Numerous studies have shown that CEA is an essential indicator of recurrence and metastasis in patients with RC (14, 32). However, few studies have simultaneously combined MRI sequence of DWI and T2WI with CEA for predicting RCLM. In this study, we compared the performance of DL models using different combinations of medical images and clinical features in both the validation and independent test sets. Our findings suggest that utilizing the combination of images and clinical features achieves superior performance compared to those relying on a single feature in detecting lung metastasis in patients with RC.

Our study has several limitations. Firstly, the dataset was derived from a single center, which may affect the model’s generalizability. Future studies should prioritize incorporating multi-center data and additional prognosis-related data to strengthen external validation and ensure robust performance across diverse clinical settings. Secondly, the issue of sample imbalance, particularly for early-stage cases (T1 and T2), may have impacted the model’s predictive performance in these stages. Addressing this imbalance through targeted data collection or augmentation, such as synthetic data generation, will be crucial for future work. Thirdly, despite the superior performance on validation and independent test sets, the predictive scope is currently limited to lung metastasis. In our future studies, we will develop a DL model to predict various other types of metastasis. Fourthly, most cases in our dataset are locally advanced RC, and further research is needed to accurately identify distant metastasis in patients with early-stage RC. Additionally, we focused on integrating specific clinical features, such as CEA levels, age, and sex, due to their stronger association with predicting RCLM. However, we acknowledge that histologic grade may also provide valuable predictive information. We plan to investigate its potential inclusion in future studies to further enhance predictive accuracy. Lastly, RC lesion annotation by radiologists is time-consuming, and future studies should explore efficient automated segmentation networks for RC segmentation.

Conclusions

In conclusion, we have developed a transformer-based DL model for predicting RCLM, relying on preoperative pelvis MRI scans and clinical features. Our model demonstrates superior performance compared to state-of-the-art DL models on both validation and independent test sets. It is anticipated that our model holds potential as a practical tool to reduce radiation exposure and mitigate treatment delays in patients without lung metastases, thereby supporting personalized treatment and follow-up plans.

Data availability statement

The original contributions presented in the study are included in the article/supplementary material. Further inquiries can be directed to the corresponding authors.

Ethics statement

This single-center retrospective study was approved by the Ethics Committee of the Sixth Affiliated Hospital, Sun Yat-sen University (2023ZSLYEC-662). The requirement for informed consent was waived (retrospective analysis of MR images) by the Institutional Ethics Review Board.

Author contributions

YL: Methodology, Software, Writing – original draft. SL: Writing – original draft. RX: Writing – original draft. XL: Writing – original draft. YY: Writing – original draft. LYZ: Writing – review & editing. YZ: Writing – review & editing. YW: Writing – original draft. CW: Writing – original draft. LMZ: Methodology, Validation, Writing – review & editing. WY: Conceptualization, Supervision, Validation, Writing – review & editing. LY: Writing – review & editing.

Funding

The author(s) declare financial support was received for the research, authorship, and/or publication of this article. This work was partially supported by the Science and Technology Program of Guangzhou (No. 2023B03J1277) and Guangdong Basic and Applied Basic Research Foundation, China (2024A1515220073).

Conflict of interest

The authors declare that the research was conducted in the absence of any commercial or financial relationships that could be construed as a potential conflict of interest.

Publisher’s note

All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article, or claim that may be made by its manufacturer, is not guaranteed or endorsed by the publisher.

Abbreviations

ROC, Receiver operating characteristic curve; AUC, Area under the ROC curve; CI, Confidence Intervals; ACC, Accuracy; SEN, Sensitivity; SPE, Specificity; RC, Rectal Cancer; RCLM, Rectal Cancer with Lung Metastases; DWI, Diffusion-Weighted Image; T2WI, T2-Weighted Image; MRI, Magnetic Resonance Imaging; CT, Computed Tomography; PET, Positron Emission Tomography; DM, Distant Metastasis; CEA, Carcinoembryonic Antigen; AI, Artificial intelligence; DL, Deep Learning.

References

1. Siegel RL, Wagle NS, Cercek A, Smith RA, Jemal A. Colorectal cancer statistics, 2023. Ca: A Cancer J Clin. (2023) 73:233–54. doi: 10.3322/caac.21772

PubMed Abstract | Crossref Full Text | Google Scholar

2. Xia C, Dong X, Li H, Cao M, Sun D, He S, et al. Cancer statistics in China and United States, 2022: profiles, trends, and determinants. Chin Med J. (2022) 135:584–90. doi: 10.1097/CM9.0000000000002108

PubMed Abstract | Crossref Full Text | Google Scholar

3. Mitry E, Guiu B, Cosconea S, Jooste VER, Faivre J, Bouvier A. Epidemiology, management and prognosis of colorectal cancer with lung metastases: a 30-year population-based study. Gut. (2010) 59:1383–8. doi: 10.1136/gut.2010.211557

PubMed Abstract | Crossref Full Text | Google Scholar

4. Nordholm-Carstensen A, Krarup P, Jorgensen LN, Wille-Jørgensen PA, Harling H, Danish Colorectal Cancer Group. Occurrence and survival of synchronous pulmonary metastases in colorectal cancer: a nationwide cohort study. Eur J Cancer. (2014) 50:447–56. doi: 10.1016/j.ejca.2013.10.009

PubMed Abstract | Crossref Full Text | Google Scholar

5. Tan KK, de Lima Lopes G Jr, Sim R. How uncommon are isolated lung metastases in colorectal cancer? A review from database of 754 patients over 4 years. J Gastrointestinal Surgery. (2009) 13:642–8. doi: 10.1007/s11605-008-0757-7

PubMed Abstract | Crossref Full Text | Google Scholar

6. Tampellini M, Ottone A, Bellini E, Alabiso I, Baratelli C, Bitossi R, et al. The role of lung metastasis resection in improving outcome of colorectal cancer patients: results from a large retrospective study. Oncologist. (2012) 17:1430–8. doi: 10.1634/theoncologist.2012-0142

PubMed Abstract | Crossref Full Text | Google Scholar

7. Ibrahim T, Tselikas L, Yazbeck C, Kattan J. Systemic versus local therapies for colorectal cancer pulmonary metastasis: what to choose and when? J Gastrointestinal Cancer. (2016) 47:223–31. doi: 10.1007/s12029-016-9818-4

PubMed Abstract | Crossref Full Text | Google Scholar

8. Ge Y, Lei S, Cai B, Gao X, Wang G, Wang L, et al. Incidence and prognosis of pulmonary metastasis in colorectal cancer: a population-based study. Int J Colorectal Disease. (2020) 35:223–32. doi: 10.1007/s00384-019-03434-8

PubMed Abstract | Crossref Full Text | Google Scholar

9. Beckers P, Berzenji L, Yogeswaran SK, Lauwers P, Bilotta G, Shkarpa N, et al. Pulmonary metastasectomy in colorectal carcinoma. J Thorac Disease. (2021) 13:2628. doi: 10.21037/jtd-2019-pm-14

PubMed Abstract | Crossref Full Text | Google Scholar

10. Lu L, Dercle L, Zhao B, Schwartz LH. Deep learning for the prediction of early on-treatment response in metastatic colorectal cancer from serial medical imaging. Nat Commun. (2021) 12:6654. doi: 10.1038/s41467-021-26990-6

PubMed Abstract | Crossref Full Text | Google Scholar

11. Liu X, Zhang D, Liu Z, Li Z, Xie P, Sun K, et al. Deep learning radiomics-based prediction of distant metastasis in patients with locally advanced rectal cancer after neoadjuvant chemoradiotherapy: A multicentre study. Ebiomedicine. (2021) 69:103442. doi: 10.1016/j.ebiom.2021.103442

PubMed Abstract | Crossref Full Text | Google Scholar

12. Huang YQ, Liang CH, He L, Tian J, Liang C S, Chen X, et al. Development and validation of a radiomics nomogram for preoperative prediction of lymph node metastasis in colorectal cancer. J Clin Oncol. (2016) 34:2157–64. doi: 10.1200/JCO.2015.65.9128

PubMed Abstract | Crossref Full Text | Google Scholar

13. Wang P, Deng C, Wu B. Magnetic resonance imaging-based artificial intelligence model in rectal cancer. World J Gastroenterology. (2021) 27:2122. doi: 10.3748/wjg.v27.i18.2122

PubMed Abstract | Crossref Full Text | Google Scholar

14. Jiang X, Zhao H, Saldanha OL, Nebelung S, Kuhl C, Amygdalos I, et al. An MRI deep learning model predicts outcome in rectal cancer. Radiology. (2023) 307:e222223. doi: 10.1148/radiol.222223

PubMed Abstract | Crossref Full Text | Google Scholar

15. Wan L, Hu J, Chen S, Zhao R, Peng W, Liu Y, et al. Prediction of lymph node metastasis in stage T1–2 rectal cancers with MRI-based deep learning. Eur Radiology. (2023) 33:3638–46. doi: 10.1007/s00330-023-09450-1

PubMed Abstract | Crossref Full Text | Google Scholar

16. Yang L, Liu D, Fang X, Wang Z, Xing Y, Ma L, et al. Rectal cancer: can T2WI histogram of the primary tumor help predict the existence of lymph node metastasis? Eur Radiology. (2019) 29:6469–76. doi: 10.1007/s00330-019-06328-z

PubMed Abstract | Crossref Full Text | Google Scholar

17. Schurink NW, Lambregts DM, Beets-Tan RG. Diffusion-weighted imaging in rectal cancer: current applications and future perspectives. Br J Radiology. (2019) 92:20180655. doi: 10.1259/bjr.20180655

PubMed Abstract | Crossref Full Text | Google Scholar

18. Zhao X, Xie P, Wang M, Li W, Pickhardt PJ, Xia W, et al. Deep learning–based fully automated detection and segmentation of lymph nodes on multiparametric-mri for rectal cancer: A multicentre study. Ebiomedicine. (2020) 56:102780. doi: 10.1016/j.ebiom.2020.102780

PubMed Abstract | Crossref Full Text | Google Scholar

19. Krogue JD, Azizi S, Tan F, Flament-Auvigne I, Brown T, Plass M, et al. Predicting lymph node metastasis from primary tumor histology and clinicopathologic factors in colorectal cancer using deep learning. Commun Med. (2023) 3:59. doi: 10.1038/s43856-023-00282-0

PubMed Abstract | Crossref Full Text | Google Scholar

20. Kudo S, Ichimasa K, Villard B, Mori Y, Misawa M, Saito S, et al. Artificial intelligence system to determine risk of T1 colorectal cancer metastasis to lymph node. Gastroenterology. (2021) 160:1075–84. doi: 10.1053/j.gastro.2020.09.027

PubMed Abstract | Crossref Full Text | Google Scholar

21. Kim K, Kim S, Han K, Bae H, Shin J, Lim JS. Diagnostic performance of deep learning-based lesion detection algorithm in CT for detecting hepatic metastasis from colorectal cancer. Korean J Radiology. (2021) 22:912. doi: 10.3348/kjr.2020.0447

PubMed Abstract | Crossref Full Text | Google Scholar

22. Lee S, Choe EK, Kim SY, Kim HS, Park KJ, Kim D. Liver imaging features by convolutional neural network to predict the metachronous liver metastasis in stage I-III colorectal cancer patients based on preoperative abdominal CT scan. BMC Bioinf. (2020) 21:1–14. doi: 10.1186/s12859-020-03686-0

PubMed Abstract | Crossref Full Text | Google Scholar

23. Xie Y, Zhang J, Xia Y, Wu Q. Unimiss: Universal medical self-supervised learning via breaking dimensionality barrier. In: Avidan S, Brostow G, Cissé M, Farinella GM, Hassner T, editors. Computer Vision–ECCV 2022. ECCV 2022. Lecture Notes in Computer Science, vol. 13681. Cham: Springer Nature Switzerland (2022). doi: 10.1007/978-3-031-19803-8_33

Crossref Full Text | Google Scholar

24. Fokas E, Liersch T, Fietkau R, Hohenberger W, Beissbarth T, Hess C, et al. Tumor regression grading after preoperative chemoradiotherapy for locally advanced rectal carcinoma revisited: updated results of the CAO/ARO/AIO-94 trial. J Clin Oncol. (2014) 32:1554–62. doi: 10.1200/JCO.2013.54.3769

PubMed Abstract | Crossref Full Text | Google Scholar

25. Tustison NJ, Avants BB, Cook PA, Zheng Y, Egan A, Yushkevich PA, et al. N4ITK: improved N3 bias correction. IEEE Trans On Med Imaging. (2010) 29:1310–20. doi: 10.1109/TMI.2010.2046908

PubMed Abstract | Crossref Full Text | Google Scholar

26. Julious SA. Sample sizes for clinical trials with normal data. Stat Med. (2004) 23:1921–86. doi: 10.1002/sim.1783

PubMed Abstract | Crossref Full Text | Google Scholar

27. Ho D, Tan IBH, Motani M. Predictive models for colorectal cancer recurrence using multi-modal healthcare data. In Proceedings of the Conference on Health, Inference, and Learning (CHIL '21). New York, NY, USA: Association for Computing Machinery (2021) p. 204–13. doi: 10.1145/3450439.3451868

Crossref Full Text | Google Scholar

28. Li J, Yuan Y, Yang F, Wang Y, Zhu X, Wang Z, et al. Expert consensus on multidisciplinary therapy of colorectal cancer with lung metastases. J Hematol \& Oncol. (2019) 12:1–11. doi: 10.1186/s13045-019-0702-0

PubMed Abstract | Crossref Full Text | Google Scholar

29. Sourbron SP, Buckley DL. Classic models for dynamic contrast-enhanced MRI. Nmr Biomedicine. (2013) 26:1004–27. doi: 10.1002/nbm.2940

PubMed Abstract | Crossref Full Text | Google Scholar

30. Rompianesi G, Pegoraro F, Ceresa CD, Montalti R, Troisi RI. Artificial intelligence in the diagnosis and management of colorectal cancer liver metastases. World J Gastroenterology. (2022) 28:108. doi: 10.3748/wjg.v28.i1.108

PubMed Abstract | Crossref Full Text | Google Scholar

31. Vorontsov E, Cerny M, Régnier P, Di Jorio L, Pal CJ, Lapointe R, et al. Deep learning for automated segmentation of liver lesions at CT in patients with colorectal cancer liver metastases. Radiology: Artif Intelligence. (2019) 1:180014. doi: 10.1148/ryai.2019180014

PubMed Abstract | Crossref Full Text | Google Scholar

32. Zhang W, Yin H, Huang Z, Zhao J, Zheng H, He D, et al. Development and validation of MRI-based deep learning models for prediction of microsatellite instability in rectal cancer. Cancer Med. (2021) 10:4164–73. doi: 10.1002/cam4.3957

PubMed Abstract | Crossref Full Text | Google Scholar

33. Benson AB, Venook AP, Al-Hawary MM, Cederquist L, Chen YJ, Ciombor KK, et al. NCCN clinical practice guidelines in oncology. J Natl Compr Cancer Network. (2018) 16:874–901. doi: 10.6004/jnccn.2018.0061

PubMed Abstract | Crossref Full Text | Google Scholar

34. Borgheresi A, De Muzio F, Agostini A, Ottaviani L, Bruno A, Granata V, et al. Lymph nodes evaluation in rectal cancer: where do we stand and future perspective. J Clin Med. (2022) 11:2599. doi: 10.3390/jcm11092599

PubMed Abstract | Crossref Full Text | Google Scholar

35. Crimi F, Valeggia S, Baffoni L, Stramare R, Lacognata C, Spolverato G, et al. FDG PET/MRI in rectal cancer. Ann Nucl Med. (2021) 35:281–90. doi: 10.1007/s12149-021-01580-0

PubMed Abstract | Crossref Full Text | Google Scholar

Keywords: rectal cancer, lung metastases, MRI, transformer, deep learning

Citation: Li Y, Li S, Xiao R, Li X, Yi Y, Zhang L, Zhou Y, Wan Y, Wei C, Zhong L, Yang W and Yao L (2025) A pelvis MR transformer-based deep learning model for predicting lung metastases risk in patients with rectal cancer. Front. Oncol. 15:1496820. doi: 10.3389/fonc.2025.1496820

Received: 15 September 2024; Accepted: 20 January 2025;
Published: 06 February 2025.

Edited by:

Zexian Liu, Sun Yat-sen University Cancer Center (SYSUCC), China

Reviewed by:

Chao Huang, University of Georgia, United States
Xingchen Peng, Sichuan University, China
Weicui Chen, Guangdong Provincial Hospital of Chinese Medicine, China
Riken Chen, Guangzhou Medical University, China

Copyright © 2025 Li, Li, Xiao, Li, Yi, Zhang, Zhou, Wan, Wei, Zhong, Yang and Yao. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) and the copyright owner(s) are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

*Correspondence: Wei Yang, d2VpeWFuZ2dtQGdtYWlsLmNvbQ==; Lin Yao, eWFvbGluQG1haWwuc3lzdS5lZHUuY24=

These authors have contributed equally to this work and share first authorship

Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.

Research integrity at Frontiers

Man ultramarathon runner in the mountains he trains at sunset

94% of researchers rate our articles as excellent or good

Learn more about the work of our research integrity team to safeguard the quality of each article we publish.


Find out more