arXiv:2409.20332eess.IVcs.CV2024-09被引 8

用局部感知扩散模型生成高保真腹部CT,助力无监督器官分割

Devil is in Details: Locality-Aware 3D Abdominal CT Volume Generation for Self-Supervised Organ Segmentation

  • 设计局部损失与先验条件提取器,精准生成腹部解剖结构
  • 在AbdomenCT-1K上FID降至0.0002,接近真实数据质量
  • 生成数据显著提升自监督分割性能,适合医学影像合成研究

在医学图像分析中,自监督学习(SSL)虽能缓解标注需求,但仍面临因资源要求高和隐私限制导致的训练数据稀缺问题。现有方法多采用生成模型跨模态、跨解剖区域生成高质量未标注3D体数据,但腹部结构复杂且相似,生成难度远高于其他部位。为此,本文提出局部感知扩散模型(Lad),通过设计局部损失优化关键解剖区域,并引入腹部先验条件提取器,实现无需额外标签或报告即可生成大量高质量腹部CT数据。所生成数据在AbdomenCT-1K数据集上FID分数从0.0034降至0.0002,高度逼近真实数据,优于当前主流方法。大量实验表明,该合成数据可有效提升自监督器官分割性能,在两个腹部数据集上均显著提高平均Dice分数,验证了合成数据对医学图像自监督学习的推动潜力。

原文摘要 · Abstract (English)

In the realm of medical image analysis, self-supervised learning (SSL) techniques have emerged to alleviate labeling demands, while still facing the challenge of training data scarcity owing to escalating resource requirements and privacy constraints. Numerous efforts employ generative models to generate high-fidelity, unlabeled 3D volumes across diverse modalities and anatomical regions. However, the intricate and indistinguishable anatomical structures within the abdomen pose a unique challenge to abdominal CT volume generation compared to other anatomical regions. To address the overlooked challenge, we introduce the Locality-Aware Diffusion (Lad), a novel method tailored for exquisite 3D abdominal CT volume generation. We design a locality loss to refine crucial anatomical regions and devise a condition extractor to integrate abdominal priori into generation, thereby enabling the generation of large quantities of high-quality abdominal CT volumes essential for SSL tasks without the need for additional data such as labels or radiology reports. Volumes generated through our method demonstrate remarkable fidelity in reproducing abdominal structures, achieving a decrease in FID score from 0.0034 to 0.0002 on AbdomenCT-1K dataset, closely mirroring authentic data and surpassing current methods. Extensive experiments demonstrate the effectiveness of our method in self-supervised organ segmentation tasks, resulting in an improvement in mean Dice scores on two abdominal datasets effectively. These results underscore the potential of synthetic data to advance self-supervised learning in medical image analysis.

3D生成自监督学习腹部CT扩散模型

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