一次训练搞定病理图像分割,速度提升2-5倍且更准
Single-Stage Hierarchical Rectification for Weakly Supervised Histopathology Segmentation

- 用深层语义信息实时修正浅层特征中的噪声
- 在两个数据集上优于现有最先进方法,精度更高
- 适合需要快速部署的医学影像分析场景
现有计算病理学中的弱监督语义分割方法普遍采用多阶段流程:生成类激活图(CAM),离线伪标签优化,再进行全监督微调。该解耦范式不仅计算开销大,还存在错误传播问题——浅层卷积网络中的局部纹理偏差会生成假阳性伪标签,后续优化步骤难以纠正。为此,我们提出单阶段分层校正(SSHR)框架,不被动后处理CAM,而是主动在前向传播中净化中间特征表示。引入分层特征校正模块(HFRM),利用深层全局语义上下文过滤浅层局部异常,直接在单次训练循环内生成高保真激活图。在LUAD-HistoSeg和BCSS数据集上的实验表明,SSHR显著优于现有最先进多阶段方法。同时,训练时间缩短2至5倍,有效降低计算开销,加速大规模病理图像分析的临床应用。代码已开源:https://github.com/trongduc-nguyen/SSHR
原文摘要 · Abstract (English)
Existing weakly supervised semantic segmentation (WSSS) methods in computational pathology rely on a multi-stage paradigm: class activation map (CAM) generation, offline pseudo-mask refinement, and fully supervised retraining. While established, this decoupled approach presents fundamental limitations. The multi-stage process not only incurs high computational training costs but also suffers from error propagation: local texture biases in shallow CNN layers generate false-positive artifacts that subsequent refinement steps often fail to correct. To address these persistent challenges through a simple yet highly effective approach, we propose the Single-Stage Hierarchical Rectification (SSHR) framework. Rather than passively refining CAMs post-hoc, our method proactively purifies intermediate feature representations during the forward pass. We introduce a Hierarchical Feature Rectification Module (HFRM) that utilizes deep global semantic context to filter out local anomalies in shallow layers. This mechanism generates high-fidelity activation maps directly within a single training loop. Experiments on the LUAD-HistoSeg and BCSS datasets demonstrate that SSHR outperforms state-of-the-art multi-stage methods. Furthermore, SSHR reduces training duration by 2 to 5 times. This efficiency minimizes computational overhead and accelerates clinical translation for large-scale histopathology workflows. The code is available at: https://github.com/trongduc-nguyen/SSHR
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。