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/}
}