MSCA-UNet: Multi-Scale Convolutional Attention UNet for Automatic Cell Counting Using Density Regression

Abstract

The quantification of cell numbers in microscopy images plays a vital role in biomedical research and medical diagnosis. Presently, deep regression networks are widely employed to generate cell density maps, and the number of cells is obtained by integrating the density maps. However, automating cell counting remains challenging due to the variability in cell morphology, the diversity of cell types, and the interference of image backgrounds. This paper aims to address the central question: ‘Can we design a robust and efficient deep learning model that can effectively count cells in microscopy images, regardless of these challenges?’ To tackle this issue, we propose a novel multi-scale convolutional attention UNet (MSCA-UNet) based on density regression. Compared with other advanced density regression methods, our method introduces two key innovations. Firstly, we employ an MSCA block with multi-scale interaction ability as an encoder component, which, when combined with spatial attention, enhances the extraction of cell details and spatial information. Secondly, the design of the asymmetric UNet allows the encoder to extract more global information and better understand the image. In the meantime, using smaller convolutional kernels and strides in the decoder helps to restore image details and edge information, resulting in improved network performance. Our method outperformed other advanced methods on three publicly available benchmark cell datasets, including the synthetic bacterial (VGG) dataset, the modified bone marrow (MBM) dataset, and the human subcutaneous adipose tissue (ADI) dataset.

Keywords

• Automatic cell counting
• microscopy images
• density map
• multi-scale convolutional attention