Enhanced Nuclei Segmentation and Classification via Category Descriptors in the SAM Model

Bioengineering (Basel). 2024 Mar 21;11(3):294. doi: 10.3390/bioengineering11030294.

Abstract

Segmenting and classifying nuclei in H&E histopathology images is often limited by the long-tailed distribution of nuclei types. However, the strong generalization ability of image segmentation foundation models like the Segment Anything Model (SAM) can help improve the detection quality of rare types of nuclei. In this work, we introduce category descriptors to perform nuclei segmentation and classification by prompting the SAM model. We close the domain gap between histopathology and natural scene images by aligning features in low-level space while preserving the high-level representations of SAM. We performed extensive experiments on the Lizard dataset, validating the ability of our model to perform automatic nuclei segmentation and classification, especially for rare nuclei types, where achieved a significant detection improvement in the F1 score of up to 12%. Our model also maintains compatibility with manual point prompts for interactive refinement during inference without requiring any additional training.

Keywords: domain alignment; long-tailed distribution; nuclei classification; nuclei segmentation; prompt guided segmentation.