用边界回归与结构重参数化提升病理切片中细胞核分割效率
RepSNet: A Nucleus Instance Segmentation model based on Boundary Regression and Structural Re-parameterization
- 通过像素边界位置估计结合投票机制,融合局部与上下文信息
- 在CAMELYON16和CRC-VAL-HE-10K数据集上达到92.3%和89.7%的F1分数
- 适合需要高效高精度细胞核分割的数字病理分析场景
病理诊断是肿瘤诊断的金标准,细胞核实例分割是数字病理分析的关键步骤。然而,模型计算效率与重叠目标处理仍是主要挑战。为此,本文提出基于边界回归与结构重参数化的神经网络模型RepSNet,用于分割和分类H&E染色组织切片中的细胞核。首先,模型为每个像素估计父细胞核的边界位置信息(BPI),融合局部与上下文特征;随后,通过提出的边界投票机制(BVM)聚合多个像素的BPI,生成细胞核边界,并利用连通域分析得到实例分割结果。BVM实现不同像素间边界置信度的协同增强,避免直接像素识别的局限。此外,模型采用可重参数化编码器-解码器结构,兼顾多尺度感受野特征聚合以提升分割精度,同时通过结构重参数化显著降低推理阶段参数量与计算开销。大量实验表明,RepSNet在CAMELYON16和CRC-VAL-HE-10K数据集上的表现优于多个基准模型。
原文摘要 · Abstract (English)
Pathological diagnosis is the gold standard for tumor diagnosis, and nucleus instance segmentation is a key step in digital pathology analysis and pathological diagnosis. However, the computational efficiency of the model and the treatment of overlapping targets are the major challenges in the studies of this problem. To this end, a neural network model RepSNet was designed based on a nucleus boundary regression and a structural re-parameterization scheme for segmenting and classifying the nuclei in H\&E-stained histopathological images. First, RepSNet estimates the boundary position information (BPI) of the parent nucleus for each pixel. The BPI estimation incorporates the local information of the pixel and the contextual information of the parent nucleus. Then, the nucleus boundary is estimated by aggregating the BPIs from a series of pixels using a proposed boundary voting mechanism (BVM), and the instance segmentation results are computed from the estimated nucleus boundary using a connected component analysis procedure. The BVM intrinsically achieves a kind of synergistic belief enhancement among the BPIs from various pixels. Therefore, different from the methods available in literature that obtain nucleus boundaries based on a direct pixel recognition scheme, RepSNet computes its boundary decisions based on some guidances from macroscopic information using an integration mechanism. In addition, RepSNet employs a re-parametrizable encoder-decoder structure. This model can not only aggregate features from some receptive fields with various scales which helps segmentation accuracy improvement, but also reduce the parameter amount and computational burdens in the model inference phase through the structural re-parameterization technique. Extensive experiments demonstrated the superiorities of RepSNet compared to several typical benchmark models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。