ImPartial: Partial Annotations for Cell Instance Segmentation

Abstract

Segmenting noisy multiplex spatial tissue images constitutes a challenging task, since the characteristics of both the noise and the biology being imaged differs significantly across tissues and modalities; this is compounded by the high monetary and time costs associated with manual annotations. It is therefore imperative to build algorithms that can accurately segment the noisy images based on a small number of annotations. Recently techniques to derive such an algorithm from a few scribbled annotations have been proposed, mostly relying on the refinement and estimation of pseudo-labels. Other techniques leverage the success of self-supervised denoising as a parallel task to potentially improve the segmentation objective when few annotations are available. In this paper, we propose a method that augments the segmentation objective via self-supervised multi-channel quantized imputation, meaning that each class of the segmentation objective can be characterized by a mixture of distributions. This approach leverages the observation that perfect pixel-wise reconstruction or denoising of the image is not needed for accurate segmentation, and introduces a self-supervised classification objective that better aligns with the overall segmentation goal. We demonstrate the superior performance of our approach for a variety of cancer datasets acquired with different highly-multiplexed imaging modalities in real clinical settings. Code for our method along with a benchmarking dataset is available at https://github.com/natalialmg/ImPartial.