arXiv:2412.06314eess.IVcs.AI2024-12

用胶囊网络增强Unet,精准分割新冠肺部感染病灶。

CAD-Unet: A Capsule Network-Enhanced Unet Architecture for Accurate Segmentation of COVID-19 Lung Infections from CT Images

论文配图:CAD-Unet: A Capsule Network-Enhanced Unet Architecture for Accurate Segmentation of COVID-19 Lung Infections from CT Images
图 1 · 摘自论文原文
  • 将胶囊网络融入Unet,通过向量传递捕捉病灶空间结构。
  • 在四个公开数据集上实现优于现有方法的分割精度。
  • 适合需要高精度肺部病灶分割的医学影像研究者使用。

自2019年新冠疫情爆发以来,医学影像已成为诊断COVID-19肺炎的主要手段。在临床中,从计算机断层扫描(CT)图像中分割肺部感染区域,可实现快速、准确的量化与诊断。然而,由于磨玻璃样影边界模糊、对比度低,且渗出物、肺组织与肺壁形态相似,分割任务极具挑战。为此,本文提出一种新型深度网络架构CAD-Unet,将胶囊网络引入经典Unet框架。胶囊网络采用向量传递信息,更利于提取复杂病灶的空间特征。我们设计了胶囊编码路径,并在Unet编码器与胶囊编码器间建立耦合路径,实现两者的互补优势与高效信息融合。在四个公开数据集上开展的大量实验表明,所提模型在二分类与多类别分割任务中均表现出优越性能。代码已开源:https://github.com/AmanoTooko-jie/CAD-Unet。

原文摘要 · Abstract (English)

Since the outbreak of the COVID-19 pandemic in 2019, medical imaging has emerged as a primary modality for diagnosing COVID-19 pneumonia. In clinical settings, the segmentation of lung infections from computed tomography images enables rapid and accurate quantification and diagnosis of COVID-19. Segmentation of COVID-19 infections in the lungs poses a formidable challenge, primarily due to the indistinct boundaries and limited contrast presented by ground glass opacity manifestations. Moreover, the confounding similarity between infiltrates, lung tissues, and lung walls further complicates this segmentation task. To address these challenges, this paper introduces a novel deep network architecture, called CAD-Unet, for segmenting COVID-19 lung infections. In this architecture, capsule networks are incorporated into the existing Unet framework. Capsule networks represent a novel network architecture that differs from traditional convolutional neural networks. They utilize vectors for information transfer among capsules, facilitating the extraction of intricate lesion spatial information. Additionally, we design a capsule encoder path and establish a coupling path between the unet encoder and the capsule encoder. This design maximizes the complementary advantages of both network structures while achieving efficient information fusion. \noindent Finally, extensive experiments are conducted on four publicly available datasets, encompassing binary segmentation tasks and multi-class segmentation tasks. The experimental results demonstrate the superior segmentation performance of the proposed model. The code has been released at: https://github.com/AmanoTooko-jie/CAD-Unet.

医学影像分割胶囊网络COVID-19

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。