将计数与定位分离,用简单模型实现更准的细胞计数。
Rethinking Cell Counting Methods: Decoupling Counting and Localization
- 分两步走:先粗略计数生成密度图,再精细定位
- 在4个数据集上超越现有方法,提升显著
- 适合医学图像分析和自动化检测场景
显微镜图像中的细胞计数在医学和生物学中至关重要,但手动操作极其繁琐。尽管近年来自动化方法有所进展,但顶尖方案趋向复杂模型设计。本文提出一种概念简单却高效的解耦学习框架,包含独立的计数网络和定位网络。不同于联合学习计数与密度图估计,我们发现解耦反而能提升效果。计数器基于中间特征图而非像素空间,利用全局上下文生成计数估计并输出粗略密度图;定位器则结合原始图像和计数器提供的粗密度图,重建高分辨率密度图以精确定位细胞。此外,引入全局消息传递模块整合跨区域模式,进一步提升计数精度。在四个数据集上的大量实验表明,尽管方法简洁,本方案仍显著超越现有方法,达到当前最优性能。关键洞察在于:解耦学习避免了直接在高分辨率密度图上学习计数,使模型能聚焦于对准确估计至关重要的全局特征。代码已开源:https://github.com/MedAITech/DCL。
原文摘要 · Abstract (English)
Cell counting in microscopy images is vital in medicine and biology but extremely tedious and time-consuming to perform manually. While automated methods have advanced in recent years, state-of-the-art approaches tend to increasingly complex model designs. In this paper, we propose a conceptually simple yet effective decoupled learning scheme for automated cell counting, consisting of separate counter and localizer networks. In contrast to jointly learning counting and density map estimation, we show that decoupling these objectives surprisingly improves results. The counter operates on intermediate feature maps rather than pixel space to leverage global context and produce count estimates, while also generating coarse density maps. The localizer then reconstructs high-resolution density maps that precisely localize individual cells, conditional on the original images and coarse density maps from the counter. Besides, to boost counting accuracy, we further introduce a global message passing module to integrate cross-region patterns. Extensive experiments on four datasets demonstrate that our approach, despite its simplicity, challenges common practice and achieves state-of-the-art performance by significant margins. Our key insight is that decoupled learning alleviates the need to learn counting on high-resolution density maps directly, allowing the model to focus on global features critical for accurate estimates. Code is available at https://github.com/MedAITech/DCL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。