基于双分支特征交互和跨样本特征融合的胸腹部医学图像分割
Thoracic and Abdominal Medical Image Segmentation Based on Dual-branch Feature Interaction and Cross-sample Feature Fusion
-
摘要: 胸腹部医学图像分割在临床医学早期病变的筛查与诊断中具有不可替代的重要意义. 因胸腹部医学图像通常存在的伪影干扰和边界模糊等因素, 导致现有模型易产生分割错误. 本文提出了一种新的语义分割网络MQUNet, 通过构建双分支特征交互机制并联合跨样本特征融合来解决上述因素带来的问题. MQUNet采用基于CNN-Transformer的双分支架构提取当前图像的局部特征与全局特征, 以尽可能全面地捕捉器官间的局部细节和空间关联, 并新增设一个基于改进的激活函数的负向关注分支融合这两种特征, 以同时兼顾特征间的相似性和差异性, 避免语义信息的丢失. 另外, 根据训练样本的类别标签建立记忆队列储存解剖先验特征, 并通过跨样本特征检索与加权融合, 将训练样本中的稳定解剖先验动态加入到当前分割过程, 有效校正由伪影和模糊情况导致的特征偏移. 该方法在公开的心脏肿瘤(ACDC)和腹部多器官(Synapse)数据集上的Dice系数指标分别达到92.53%和85.17%, 优于现有的分割方法, 表明MQUNet在胸腹部医学图像分割任务中展现出显著优势.
Abstract: Thoracic and abdominal medical image segmentation plays an irreplaceable role in the screening and diagnosis of early lesions in clinical medicine. Due to the presence of artifacts and blurred boundaries in thoracic and abdominal medical images, existing models are prone to segmentation errors. This paper proposes a new semantic segmentation network, MQUNet, which solves the problems caused by the above factors by constructing a dual-branch feature interaction mechanism and combining cross-sample feature fusion. MQUNet utilizes a dual branch architecture based on CNN-Transformer to extract local and global features of the current image, in order to capture local details and spatial correlations between organs as comprehensively as possible. It also integrates the two features by adding a negative attention branch based on an improved activation function, while taking into account both similarities and differences between features, avoiding the loss of semantic information. In addition, a memory queue is established based on the category labels of the training samples to store anatomical prior features. Through cross sample feature retrieval and weighted fusion, stable anatomical prior features from the training samples are dynamically added to the current segmentation process, effectively correcting feature shifts caused by artifacts and blurring. The Dice coefficient index of this method reached 92.53% and 85.17% on publicly available datasets of cardiac cancer (ACDC) and abdominal multi organ (Synapse), respectively, which is superior to existing segmentation methods. This indicates that MQUNet has significant advantages in chest and abdominal medical image segmentation tasks.
下载: