College of Electronic and Information Engineering, Hebei University, Hebei 071002, People's Republic of China.
Hebei Key Laboratory of Precise Imaging of Inflammation Related Tumors, Hebei, 071000, People's Republic of China.
Biomed Phys Eng Express. 2024 Nov 5;11(1). doi: 10.1088/2057-1976/ad8acb.
With the development of deep learning in the field of medical image segmentation, various network segmentation models have been developed. Currently, the most common network models in medical image segmentation can be roughly categorized into pure convolutional networks, Transformer-based networks, and networks combining convolution and Transformer architectures. However, when dealing with complex variations and irregular shapes in medical images, existing networks face issues such as incomplete information extraction, large model parameter sizes, high computational complexity, and long processing times. In contrast, models with lower parameter counts and complexity can efficiently, quickly, and accurately identify lesion areas, significantly reducing diagnosis time and providing valuable time for subsequent treatments. Therefore, this paper proposes a lightweight network named MCI-Net, with only 5.48 M parameters, a computational complexity of 4.41, and a time complexity of just 0.263. By performing linear modeling on sequences, MCI-Net permanently marks effective features and filters out irrelevant information. It efficiently captures local-global information with a small number of channels, reduces the number of parameters, and utilizes attention calculations with exchange value mapping. This achieves model lightweighting and enables thorough interaction of local-global information within the computation, establishing an overall semantic relationship of local-global information. To verify the effectiveness of the MCI-Net network, we conducted comparative experiments with other advanced representative networks on five public datasets: X-ray, Lung, ISIC-2016, ISIC-2018, and capsule endoscopy and gastrointestinal segmentation. We also performed ablation experiments on the first four datasets. The experimental results outperformed the other compared networks, confirming the effectiveness of MCI-Net. This research provides a valuable reference for achieving lightweight, accurate, and high-performance medical image segmentation network models.
随着深度学习在医学图像分割领域的发展,各种网络分割模型已经被开发出来。目前,医学图像分割中最常见的网络模型大致可以分为纯卷积网络、基于 Transformer 的网络以及卷积和 Transformer 架构结合的网络。然而,在处理医学图像中的复杂变化和不规则形状时,现有的网络面临着信息提取不完整、模型参数规模大、计算复杂度高和处理时间长等问题。相比之下,参数数量和复杂度较低的模型可以高效、快速和准确地识别病变区域,显著缩短诊断时间,并为后续治疗提供宝贵的时间。因此,本文提出了一个名为 MCI-Net 的轻量级网络,其参数数量仅为 5.48M,计算复杂度为 4.41,时间复杂度仅为 0.263。MCI-Net 通过对序列进行线性建模,永久标记有效特征并过滤掉无关信息。它通过使用少量通道高效地捕获局部-全局信息,减少参数数量,并利用注意力计算进行交换值映射。这实现了模型轻量化,并在计算中充分交互局部-全局信息,建立了局部-全局信息的整体语义关系。为了验证 MCI-Net 网络的有效性,我们在五个公共数据集上与其他先进的代表性网络进行了对比实验:X 射线、肺部、ISIC-2016、ISIC-2018 和胶囊内窥镜和胃肠道分割。我们还在前四个数据集上进行了消融实验。实验结果优于其他比较网络,证实了 MCI-Net 的有效性。这项研究为实现轻量级、准确和高性能的医学图像分割网络模型提供了有价值的参考。