arXiv:2504.06950cs.CV2025-04被引 4

用扩散模型提取病理图像特征,提升分割精度

PathSegDiff: Pathology Segmentation using Diffusion model representations

  • 用病理专用扩散模型做特征提取,替代传统预训练网络
  • 在BCSS和GlaS数据集上显著优于传统方法
  • 适合需要高精度病理图像分割的研究者

图像分割在计算病理学中至关重要,广泛应用于疾病诊断、分型及预后预测。传统分割模型依赖预训练特征提取器和成对的图像-掩码标注数据,通过轻量级预测模型将特征映射为像素级类别。特征提取器的选择直接影响最终性能,近期研究集中于寻找合适的预训练任务。本文提出PathSegDiff,一种利用潜空间扩散模型(LDM)作为预训练特征提取器的新方法。该方法基于自监督编码器引导的病理特异性LDM,从H&E染色病理图像中提取丰富的语义信息,并采用简单的全卷积网络处理这些特征以生成分割掩码。实验表明,在BCSS和GlaS数据集上,该方法显著优于传统方法,证明了领域特定扩散预训练在捕捉复杂组织结构和提升病理图像分割精度方面的有效性。

原文摘要 · Abstract (English)

Image segmentation is crucial in many computational pathology pipelines, including accurate disease diagnosis, subtyping, outcome, and survivability prediction. The common approach for training a segmentation model relies on a pre-trained feature extractor and a dataset of paired image and mask annotations. These are used to train a lightweight prediction model that translates features into per-pixel classes. The choice of the feature extractor is central to the performance of the final segmentation model, and recent literature has focused on finding tasks to pre-train the feature extractor. In this paper, we propose PathSegDiff, a novel approach for histopathology image segmentation that leverages Latent Diffusion Models (LDMs) as pre-trained featured extractors. Our method utilizes a pathology-specific LDM, guided by a self-supervised encoder, to extract rich semantic information from H\&E stained histopathology images. We employ a simple, fully convolutional network to process the features extracted from the LDM and generate segmentation masks. Our experiments demonstrate significant improvements over traditional methods on the BCSS and GlaS datasets, highlighting the effectiveness of domain-specific diffusion pre-training in capturing intricate tissue structures and enhancing segmentation accuracy in histopathology images.

病理分割扩散模型图像分割

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