Spontaneous breaking of symmetry in overlapping cell instance segmentation using diffusion models

Biol Methods Protoc. 2024 Nov 9;9(1):bpae084. doi: 10.1093/biomethods/bpae084. eCollection 2024.

Abstract

Instance segmentation is the task of assigning unique identifiers to individual objects in images. Solving this task requires breaking the inherent symmetry that semantically similar objects must result in distinct outputs. Deep learning algorithms bypass this break-of-symmetry by training specialized predictors or by utilizing intermediate label representations. However, many of these approaches break down when faced with overlapping labels that are ubiquitous in biomedical imaging, for instance for segmenting cell layers. Here, we discuss the reason for this failure and offer a novel approach for instance segmentation based on diffusion models that breaks this symmetry spontaneously. Our method outputs pixel-level instance segmentations matching the performance of models such as cellpose on the cellpose fluorescent cell dataset, while also permitting overlapping labels.

Keywords: computer vision; deep learning; diffusion model; instance segmentation; overlap.