用最优传输设计不对称图像翻译模型,提升病理切片核分割精度
Optimal Transport Driven Asymmetric Image-to-Image Translation for Nuclei Segmentation of Histological Images
- 基于最优传输构建可逆生成器,解决图像域间信息不对称问题
- 在公开数据集上优于现有方法,且网络结构更简洁
- 适合需要高效精准核分割的病理图像分析场景
从组织病理图像中分割细胞核区域可实现核结构的形态计量分析,有助于疾病检测与诊断。为开发适用于不同目标域表示的核分割算法,可采用对目标域图像表示不变的图像到图像翻译网络。现有模型在两域信息量不对称时表现不佳。本文提出一种新型深度生成模型,通过嵌入空间处理信息丰富的病理图像域与信息贫乏的分割图域之间的差异。结合最优传输与测度论思想,构建可逆生成器,实现低复杂度高效优化,自动消除显式的循环一致性损失。模型在可逆生成框架内引入空间约束压缩操作,保持图像块内空间连续性。相比具有复杂架构的现有模型,该方法在模型性能与网络复杂度之间取得更好平衡。在多个公开病理图像数据集上验证了所提模型的有效性,并与当前先进核分割方法进行了对比。
原文摘要 · Abstract (English)
Segmentation of nuclei regions from histological images enables morphometric analysis of nuclei structures, which in turn helps in the detection and diagnosis of diseases under consideration. To develop a nuclei segmentation algorithm, applicable to different types of target domain representations, image-to-image translation networks can be considered as they are invariant to target domain image representations. One of the important issues with image-to-image translation models is that they fail miserably when the information content between two image domains are asymmetric in nature. In this regard, the paper introduces a new deep generative model for segmenting nuclei structures from histological images. The proposed model considers an embedding space for handling information-disparity between information-rich histological image space and information-poor segmentation map domain. Integrating judiciously the concepts of optimal transport and measure theory, the model develops an invertible generator, which provides an efficient optimization framework with lower network complexity. The concept of invertible generator automatically eliminates the need of any explicit cycle-consistency loss. The proposed model also introduces a spatially-constrained squeeze operation within the framework of invertible generator to maintain spatial continuity within the image patches. The model provides a better trade-off between network complexity and model performance compared to other existing models having complex network architectures. The performance of the proposed deep generative model, along with a comparison with state-of-the-art nuclei segmentation methods, is demonstrated on publicly available histological image data sets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。