# Cell Instance Segmentation via Multi-Task Image-to-Image Schr"odinger Bridge

arXiv: https://arxiv.org/abs/2604.12318 · HTML: https://mlsanity.com/paper/2604.12318/

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Builds on 10 of its 16 references that are in the map.
- First posted on arXiv 14 Apr 2026.

## Builds on (most cited first, 10 of 10)

- [High-Resolution Image Synthesis with Latent Diffusion Models](https://mlsanity.com/paper/2112.10752/) (2021)
- [Mask R-CNN](https://mlsanity.com/paper/1703.06870/) (2017)
- [HoVer-Net: Simultaneous Segmentation and Classification of Nuclei in Multi-Tissue Histology Images](https://mlsanity.com/paper/1812.06499/) (2018)
- [Cell Detection with Star-convex Polygons](https://mlsanity.com/paper/1806.03535/) (2018)
- [PanNuke Dataset Extension, Insights and Baselines](https://mlsanity.com/paper/2003.10778/) (2020)
- [CellViT: Vision Transformers for Precise Cell Segmentation and Classification](https://mlsanity.com/paper/2306.15350/) (2023)
- [Ambiguous Medical Image Segmentation using Diffusion Models](https://mlsanity.com/paper/2304.04745/) (2023)
- [MEDIAR: Harmony of Data-Centric and Model-Centric for Multi-Modality Microscopy](https://mlsanity.com/paper/2212.03465/) (2022)
- [Generative Medical Segmentation](https://mlsanity.com/paper/2403.18198/) (2024)
- [Cyto R-CNN and CytoNuke Dataset: Towards reliable whole-cell segmentation in bright-field histological images](https://mlsanity.com/paper/2401.15638/) (2024)

## Abstract (arXiv)

Existing cell instance segmentation pipelines typically combine deterministic predictions with post-processing, which imposes limited explicit constraints on the global structure of instance masks. In this work, we propose a multi-task image-to-image Schr"odinger Bridge framework that formulates instance segmentation as a distribution-based image-to-image generation problem. Boundary-aware supervision is integrated through a reverse distance map, and deterministic inference is employed to produce stable predictions. Experimental results on the PanNuke dataset demonstrate that the proposed method achieves competitive or superior performance without relying on SAM pre-training or additional post-processing. Additional results on the MoNuSeg dataset show robustness under limited training data. These findings indicate that Schr"odinger Bridge-based image-to-image generation provides an effective framework for cell instance segmentation.