arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.13698cs.CV

用于医学图像分割的CNN-Transformer校正协同学习

CNN-Transformer Rectified Collaborative Learning for Medical Image Segmentation

School of Information and Communication Engineering, Dalian University of Technology(大连理工大学信息与通信工程学院);Key Laboratory for Ubiquitous Network and Service Software of Liaoning Province(辽宁省泛在网络与服务软件重点实验室);DUT-RU International School of Information Science and Engineering, Dalian University of Technology(大连理工大学-鲁东大学国际信息科学与工程学院);Affiliated Zhongshan Hospital of Dalian University(大连大学附属中山医院)

Lanhu Wu, Miao Zhang, Yongri Piao, Zhenyan Yao, Weibing Sun, Feng Tian, Huchuan Lu

AI总结:

本文提出CNN-Transformer校正协同学习(CTRCL)框架,通过双向知识转移结合CNN与Transformer优势,并设计RLCL和CFCL策略优化logit与特征空间的知识传递,在医学图像分割任务上取得优异性能。

摘要:

自动且精确的医学图像分割(MIS)对临床诊断和分析至关重要。当前的MIS方法主要依赖卷积神经网络(CNN)或自注意力机制(Transformer)进行特征建模。然而,基于CNN的方法因全局依赖受限而存在定位不准确的问题,而基于Transformer的方法因缺乏对局部细节的强调常呈现粗糙的边界。尽管设计了一些CNN-Transformer混合方法来综合互补的局部和全局信息以获得更好性能,但CNN和Transformer的结合引入了大量参数并增加了计算成本。为此,本文提出了一种CNN-Transformer校正协同学习(CTRCL)框架,通过两者之间的双向知识转移,为MIS任务学习更强的基于CNN和基于Transformer的模型。具体而言,我们提出了一种校正的基于logit的协同学习(RLCL)策略,该策略引入真实标签来自适应地选择并校正学生软标签中的错误区域,从而在logit空间中实现准确的知识转移。我们还提出了一种类别感知的基于特征的协同学习(CFCL)策略,通过赋予其中间特征相似的类别感知能力,在基于CNN和基于Transformer的模型之间实现特征空间中的有效知识转移。在三个流行的MIS基准上的大量实验表明,我们的CTRCL在不同评估指标下优于大多数最先进的协同学习方法。

英文摘要:

Automatic and precise medical image segmentation (MIS) is of vital importance for clinical diagnosis and analysis. Current MIS methods mainly rely on the convolutional neural network (CNN) or self-attention mechanism (Transformer) for feature modeling. However, CNN-based methods suffer from the inaccurate localization owing to the limited global dependency while Transformer-based methods always present the coarse boundary for the lack of local emphasis. Although some CNN-Transformer hybrid methods are designed to synthesize the complementary local and global information for better performance, the combination of CNN and Transformer introduces numerous parameters and increases the computation cost. To this end, this paper proposes a CNN-Transformer rectified collaborative learning (CTRCL) framework to learn stronger CNN-based and Transformer-based models for MIS tasks via the bi-directional knowledge transfer between them. Specifically, we propose a rectified logit-wise collaborative learning (RLCL) strategy which introduces the ground truth to adaptively select and rectify the wrong regions in student soft labels for accurate knowledge transfer in the logit space. We also propose a class-aware feature-wise collaborative learning (CFCL) strategy to achieve effective knowledge transfer between CNN-based and Transformer-based models in the feature space by granting their intermediate features the similar capability of category perception. Extensive experiments on three popular MIS benchmarks demonstrate that our CTRCL outperforms most state-of-the-art collaborative learning methods under different evaluation metrics.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑