The Four Color Theorem for Cell Instance Segmentation

Abstract

Cell instance segmentation is critical to analyzing biomedical images, yet accurately distinguishing tightly touching cells remains a persistent challenge. Existing instance segmentation frameworks, including detection-based, contour-based, and distance mapping-based approaches, have made significant progress, but balancing model performance with computational efficiency remains an open problem. In this paper, we propose a novel cell instance segmentation method inspired by the four-color theorem. By conceptualizing cells as countries and tissues as oceans, we introduce a four-color encoding scheme that ensures adjacent instances receive distinct labels. This reformulation transforms instance segmentation into a constrained semantic segmentation problem with only four predicted classes, substantially simplifying the instance differentiation process. To solve the training instability caused by the non-uniqueness of four-color encoding, we design an asymptotic training strategy and encoding transformation method. Extensive experiments on various modes demonstrate our approach achieves state-of-the-art performance. The code is available at https://github.com/zhangye-zoe/FCIS.

Cite

Text

Zhang et al. "The Four Color Theorem for Cell Instance Segmentation." Proceedings of the 42nd International Conference on Machine Learning, 2025.

Markdown

[Zhang et al. "The Four Color Theorem for Cell Instance Segmentation." Proceedings of the 42nd International Conference on Machine Learning, 2025.](https://mlanthology.org/icml/2025/zhang2025icml-four/)

BibTeX

@inproceedings{zhang2025icml-four,
  title     = {{The Four Color Theorem for Cell Instance Segmentation}},
  author    = {Zhang, Ye and Zhou, Yu and Wang, Yifeng and Xiao, Jun and Wang, Ziyue and Zhang, Yongbing and Chen, Jianxu},
  booktitle = {Proceedings of the 42nd International Conference on Machine Learning},
  year      = {2025},
  pages     = {77194-77215},
  volume    = {267},
  url       = {https://mlanthology.org/icml/2025/zhang2025icml-four/}
}