中文

LPD:基于多样性正则化的可学习原型用于弱监督组织病理切片分割

计算机视觉与模式识别 2025-12-08 v1

摘要

Weakly supervised semantic segmentation (WSSS) in histopathology reduces pixel-level labeling by learning from image-level labels, but it is hindered by inter-class homogeneity, intra-class heterogeneity, and CAM-induced region shrinkage (global pooling-based class activation maps whose activations highlight only the most distinctive areas and miss nearby class regions). Recent works address these challenges by constructing a clustering prototype bank and then refining masks in a separate stage; however, such two-stage pipelines are costly, sensitive to hyperparameters, and decouple prototype discovery from segmentation learning, limiting their effectiveness and efficiency. We propose a cluster-free, one-stage learnable-prototype framework with diversity regularization to enhance morphological intra-class heterogeneity coverage. Our approach achieves state-of-the-art (SOTA) performance on BCSS-WSSS, outperforming prior methods in mIoU and mDice. Qualitative segmentation maps show sharper boundaries and fewer mislabels, and activation heatmaps further reveal that, compared with clustering-based prototypes, our learnable prototypes cover more diverse and complementary regions within each class, providing consistent qualitative evidence for their effectiveness.

关键词

引用

@article{arxiv.2512.05922,
  title  = {LPD: Learnable Prototypes with Diversity Regularization for Weakly Supervised Histopathology Segmentation},
  author = {Khang Le and Anh Mai Vu and Thi Kim Trang Vo and Ha Thach and Ngoc Bui Lam Quang and Thanh-Huy Nguyen and Minh H. N. Le and Zhu Han and Chandra Mohan and Hien Van Nguyen},
  journal= {arXiv preprint arXiv:2512.05922},
  year   = {2025}
}

备注

Note: Khang Le and Anh Mai Vu contributed equally