用原型学习提升细胞计数模型可解释性,让结果看得懂。
CountXplain: Interpretable Cell Counting with Prototype-Based Density Map Estimation
- 通过原型层学习细胞和背景的典型视觉模式
- 在两个公开数据集上保持高精度计数且可解释
- 适合需要透明决策过程的生物医学研究者
生物医学图像中的细胞计数对多种临床应用至关重要,但深度学习模型的可解释性仍是重大挑战。我们提出一种基于原型的密度图估计方法,用于可解释的细胞计数。该方法在密度估计网络中引入原型层,使模型能够学习细胞与背景伪影的代表性视觉模式。通过生物学家问卷调查验证,所学原型与实际生物学特征高度相关,进一步证实了模型的可解释性。通过生成与原型最相似的输入区域,该方法清晰揭示模型识别和计数细胞的依据。在两个公开数据集上的大量实验表明,本方法在不牺牲计数效果的前提下实现了高可解释性。该工作为研究人员和临床医生提供了一种透明可靠的细胞计数工具,有助于提升对深度学习的信任并加速其在关键生物医学应用中的采纳。代码已开源:https://github.com/NRT-D4/CountXplain。
原文摘要 · Abstract (English)
Cell counting in biomedical imaging is pivotal for various clinical applications, yet the interpretability of deep learning models in this domain remains a significant challenge. We propose a novel prototype-based method for interpretable cell counting via density map estimation. Our approach integrates a prototype layer into the density estimation network, enabling the model to learn representative visual patterns for both cells and background artifacts. The learned prototypes were evaluated through a survey of biologists, who confirmed the relevance of the visual patterns identified, further validating the interpretability of the model. By generating interpretations that highlight regions in the input image most similar to each prototype, our method offers a clear understanding of how the model identifies and counts cells. Extensive experiments on two public datasets demonstrate that our method achieves interpretability without compromising counting effectiveness. This work provides researchers and clinicians with a transparent and reliable tool for cell counting, potentially increasing trust and accelerating the adoption of deep learning in critical biomedical applications. Code is available at https://github.com/NRT-D4/CountXplain.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。