首页|期刊导航|陆军军医大学学报|环形特征融合网络(RFFNet):一种融合CNN与Transformer并校正内镜畸变实现炎症性肠病高精度实时诊断的深度学习模型

环形特征融合网络(RFFNet):一种融合CNN与Transformer并校正内镜畸变实现炎症性肠病高精度实时诊断的深度学习模型OA

Ring-Feature Fusion Network(RFFNet):a deep learning model integrating CNN and transformer with endoscopic distortion correction for high-accuracy real-time diagnosis of inflammatory bowel disease

中文摘要英文摘要

目的 借助深度学习技术辅助内镜医师对克罗恩病(Crohn's disease,CD)、溃疡性结肠炎(ulcerative colitis,UC)与正常(Normal)图像肠镜进行准确诊断.方法 收集2018年1月至2020年11月陆军军医大学大坪医院消化内科与浙江大学医学院附属邵逸夫医院消化内科1 309例受试者共24 492幅肠镜图像,其中包括4 729张CD图像(424例)、7 074张UC图像(605例)与12 689张正常(Normal)图像(280例).每类按照7∶1∶2的比例将病例随机划分为训练集915例17 136张图像,内部测试集131例2 626张图像,外部验证集263例4 730张图像,分别用于对模型进行训练、测试及验证.针对卷积神经网络(Convolutional Neural Network,CNN)与Transformer网络特征融合问题及肠镜图像畸变问题,本研究基于ResNeSt50与MViTv2构建环形特征融合网络(Ring-Feature Fusion Network,RFFNet).该网络通过双向通道-空间注意力实现跨阶段特征融合,有效结合CNN局部特征提取与Transformer全局建模优势;引入环形特征融合机制,减轻肠镜图像桶形畸变与景深现象,提升模型对远端肠腔几何特性的适应能力.与现有5种深度学习模型进行对比,通过混淆矩阵展示各模型在外部验证集上的识别结果,利用准确率、灵敏度、特异度、F1值与曲线下面积(area under the curve,AUC)等各项指标验证RFFNet的性能优势;通过消融实验,在外部验证集上利用准确率与F1值等指标验证RFFNet中各改进点的有效性与必要性;通过梯度加权类激活映射(gradient-weighted class activation mapping,Grad-CAM)算法生成类激活映射图,直观地展示模型关注区域.结果 RFFNet在外部验证集上总体准确率为95.68%(95%CI:94.14%~98.58%),总体AUC为0.987(95%CI:0.985~0.990).CD、UC和正常三类别的AUC分别为0.982(95%CI:0.979~0.986)和0.981(95%CI:0.977~0.984)和0.997(95%CI:0.995~0.998);灵敏度分别为 93.35%(95%CI:91.67%~94.71%)、94.00%(95%CI:92.60%~95.16%)和 98.56%(95%CI:97.99%~98.97%);特异度分别为 94.58%(95%CI:93.81%~95.27%)、94.20%(95%CI:93.36%~94.94%)和 98.61%(95%CI:98.05%~99.01%);F1 值分别为 87.81%(95%CI:86.28%~89.25%)、90.05%(95%CI:89.00%~91.25%)和98.58%(95%CI:98.22%~98.92%),表明RFFNet实现了较高精度诊断,与单独使用 CNN、Transformer 模型相比,RFFNet的总体准确率提升显著,差异具有统计学意义(P<0.05).消融实验证实,模型能够通过动态空间注意力机制,深度融合CNN的局部细粒度特征提取能力与Transformer的全局上下文建模优势,与只使用CNN模型相比,CNN-Transformer架构的总体准确率提升了0.11%;能够通过环形特征校正肠镜光学畸变与景深衰减,增强对远端肠腔几何结构的建模能力,使用环形特征后RFFNet总体准确率提升了0.57%.类激活映射图表明改进的模型具有自适应捕获肠镜结构特征的能力.结论 RFFNet通过双骨干架构深度耦合CNN局部纹理感知与Transformer全局依赖建模,并以环形特征显式校正肠镜畸变,实现了CD、UC与正常黏膜的高精度实时分型,实现了CD、UC和正常图像的高精度实时诊断.

Objective To leverage deep learning technology to assist endoscopists in accurately diagnosing colonoscopy images of Crohn's disease(CD),ulcerative colitis(UC),and Normal tissue.Methods A total of 24 492 colonoscopy images from 1 309 subjects were collected from Department of Gastroenterology,Daping Hospital,Army Medical University,and Sir Run Run Shaw Hospital between January 2018 and November 2020.The dataset included 4 729 CD images from 424 cases,7 074 UC images from 605 cases,and 12 689 Normal images from 280 cases.The images were randomly split by case in a 7∶1∶2 ratio into a training set(915 cases,17 136 images),an internal test set(131 cases,2 626 images),and an external validation set(263 cases,4 730 images).To address feature fusion between CNN and Transformer architectures and correct endoscopic image distortion,we constructed a Ring-Feature Fusion Network(RFFNet)based on ResNeSt50 and MViTv2.The network employs bidirectional channel-spatial attention for cross-stage feature fusion,effectively combining the local feature extraction strengths of CNN with the global modeling capabilities of Transformer.An innovative ring-feature mechanism was introduced to handle barrel distortion and depth-of-field phenomena in colonoscopy images,enhancing the model's adaptability to the geometric properties of the distal intestinal lumen.RFFNet was compared with five existing deep learning models.Recognition results on the external validation set were visualized using confusion matrices,and performance was evaluated using accuracy,sensitivity,specificity,F1-score,and area under the curve(AUC).Ablation studies were conducted on the external validation set using metrics like accuracy and F1-score to validate the effectiveness and necessity of each improvement in RFFNet.Gradient-weighted class activation mapping(Grad-CAM)was used to generate heatmaps,visually demonstrating the model's focus areas and improving interpretability.Results On the external validation set,RFFNet achieved an overall accuracy of 95.68%(95%CI:94.14%to 98.58%)and an overall AUC of 0.987(95%CI:0.985 to 0.990).For CD,UC,and Normal classes,the AUCs were 0.982(95%CI:0.979 to 0.986),0.981(95%CI:0.977 to 0.984),and 0.997(95%CI:0.995 to 0.998),respectively;sensitivities were 93.35%(95%CI:91.67%to 94.71%),94.00%(95%CI:92.60%to 95.16%),and 98.56%(95%CI:97.99%to 98.97%),respectively;specificities were 94.58%(95%CI:93.81%to 95.27%),94.20%(95%CI:93.36%to 94.94%),and 98.61%(95%CI:98.05%to 99.01%),respectively;F1-scores were 87.81%(95%CI:86.28%to 89.25%),90.05%(95%CI:89.00%to 91.25%),and 98.58%(95%CI:98.22%to 98.92%),respectively.These results indicate that RFFNet achieved high diagnostic precision.Compared to using CNN or Transformer models alone,RFFNet's overall accuracy improvement was statistically significant(P<0.05).Ablation studies confirmed that the model,through a dynamic spatial attention mechanism,deeply integrates CNN's local fine-grained feature extraction with Transformer's global contextual modeling advantages.The CNN-Transformer architecture improved overall accuracy by 0.11%compared to using CNN alone,respectively.The ring-feature mechanism corrected endoscopic optical distortion and depth-of-field attenuation,enhancing modeling of the distal lumen geometry,and improved RFFNet's overall accuracy by 0.57%.Grad-CAM heatmaps demonstrated the model's adaptive ability to capture structural features in colonoscopy images.Conclusion RFFNet,through its dual-backbone architecture that deeply couples CNN's local texture perception with Transformer's global dependency modeling,and explicit correction of endoscopic distortion via ring-features,enables high-accuracy real-time classification of CD,UC,and normal mucosa.

李华龙;魏艳玲;阮广聪;孟薇;吴毅;李颖;唐嘉杰;刘静静;粘永健

陆军军医大学(第三军医大学)生物医学工程与影像医学系数字医学教研室,重庆陆军军医大学(第三军医大学)大坪医院消化内科,重庆陆军军医大学(第三军医大学)大坪医院消化内科,重庆陆军军医大学(第三军医大学)大坪医院消化内科,重庆陆军军医大学(第三军医大学)生物医学工程与影像医学系数字医学教研室,重庆陆军军医大学(第三军医大学)生物医学工程与影像医学系数字医学教研室,重庆陆军军医大学(第三军医大学)第二附属医院信息科,重庆陆军军医大学(第三军医大学)生物医学工程与影像医学系数字医学教研室,重庆陆军军医大学(第三军医大学)生物医学工程与影像医学系数字医学教研室,重庆

医药卫生

炎症性肠病深度学习溃疡性结肠炎克罗恩病

inflammatory bowel diseasedeep learningulcerative colitis,the Crohn's disease

《陆军军医大学学报》 2026 (4)

479-492,14

重庆市技术创新与应用发展专项重点项目(CSTB2022TIAD-KPX0161)重庆市自然科学基金创新发展联合基金项目(CSTB2024NSCQ-LZX0141)重庆英才·创新创业领军人才项目(CQYC20220303576) Supported by the Chongqing Technology Innovation and Application Development Special Key Project(CSTB2022TIAD-KPX0161),the Chongqing Natural Science Foundation Innovation and Development Joint Fund(CSTB2024NSCQ-LZX0141)and the Chongqing Yingcai·Innovation and Entrepreneurship Leading Talent Project(CQYC20220303576).

10.16016/j.2097-0927.202510069

评论