Su Yue, Xia Xianwu, Sun Rong, Yuan Jianjun, Hua Qianjin, Han Baosan, Gong Jing, Nie Shengdong
School of Health Science and Engineering, University of Shanghai for Science and Technology, Shanghai, 200093, China.
Department of Oncology Intervention, Municipal Hospital Affiliated of Taizhou University, Zhejiang, Taizhou, 318000, China.
J Imaging Inform Med. 2024 Dec;37(6):2883-2894. doi: 10.1007/s10278-024-01149-z. Epub 2024 Jun 11.
This study aims to develop a CT-based hybrid deep learning network to predict pathological subtypes of early-stage lung adenocarcinoma by integrating residual network (ResNet) with Vision Transformer (ViT). A total of 1411 pathologically confirmed ground-glass nodules (GGNs) retrospectively collected from two centers were used as internal and external validation sets for model development. 3D ResNet and ViT were applied to investigate two deep learning frameworks to classify three subtypes of lung adenocarcinoma namely invasive adenocarcinoma (IAC), minimally invasive adenocarcinoma and adenocarcinoma in situ, respectively. To further improve the model performance, four Res-TransNet based models were proposed by integrating ResNet and ViT with different ensemble learning strategies. Two classification tasks involving predicting IAC from Non-IAC (Task1) and classifying three subtypes (Task2) were designed and conducted in this study. For Task 1, the optimal Res-TransNet model yielded area under the receiver operating characteristic curve (AUC) values of 0.986 and 0.933 on internal and external validation sets, which were significantly higher than that of ResNet and ViT models (p < 0.05). For Task 2, the optimal fusion model generated the accuracy and weighted F1 score of 68.3% and 66.1% on the external validation set. The experimental results demonstrate that Res-TransNet can significantly increase the classification performance compared with the two basic models and have the potential to assist radiologists in precision diagnosis.
本研究旨在开发一种基于CT的混合深度学习网络,通过将残差网络(ResNet)与视觉Transformer(ViT)相结合来预测早期肺腺癌的病理亚型。从两个中心回顾性收集的总共1411个经病理证实的磨玻璃结节(GGN)用作模型开发的内部和外部验证集。应用3D ResNet和ViT分别研究两种深度学习框架,以对肺腺癌的三种亚型进行分类,即浸润性腺癌(IAC)、微浸润性腺癌和原位腺癌。为了进一步提高模型性能,通过将ResNet和ViT与不同的集成学习策略相结合,提出了四种基于Res-TransNet的模型。本研究设计并开展了两项分类任务,分别是从非IAC中预测IAC(任务1)和对三种亚型进行分类(任务2)。对于任务1,最优的Res-TransNet模型在内部和外部验证集上的受试者操作特征曲线下面积(AUC)值分别为0.986和0.933,显著高于ResNet和ViT模型(p<0.05)。对于任务2,最优融合模型在外部验证集上的准确率和加权F1分数分别为68.3%和66.1%。实验结果表明,与两种基本模型相比,Res-TransNet可以显著提高分类性能,并且有潜力协助放射科医生进行精准诊断。