SelfAdapt: Unsupervised Domain Adaptation of Cell Segmentation Models

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Reith, Fabian H., Franzen, Jannik, Palli, Dinesh R., Rumberger, J. Lorenz, Kainmueller, Dagmar
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913993326592000
author Reith, Fabian H.
Franzen, Jannik
Palli, Dinesh R.
Rumberger, J. Lorenz
Kainmueller, Dagmar
author_facet Reith, Fabian H.
Franzen, Jannik
Palli, Dinesh R.
Rumberger, J. Lorenz
Kainmueller, Dagmar
contents Deep neural networks have become the go-to method for biomedical instance segmentation. Generalist models like Cellpose demonstrate state-of-the-art performance across diverse cellular data, though their effectiveness often degrades on domains that differ from their training data. While supervised fine-tuning can address this limitation, it requires annotated data that may not be readily available. We propose SelfAdapt, a method that enables the adaptation of pre-trained cell segmentation models without the need for labels. Our approach builds upon student-teacher augmentation consistency training, introducing L2-SP regularization and label-free stopping criteria. We evaluate our method on the LiveCell and TissueNet datasets, demonstrating relative improvements in AP0.5 of up to 29.64% over baseline Cellpose. Additionally, we show that our unsupervised adaptation can further improve models that were previously fine-tuned with supervision. We release SelfAdapt as an easy-to-use extension of the Cellpose framework. The code for our method is publicly available at https: //github.com/Kainmueller-Lab/self_adapt.
format Preprint
id arxiv_https___arxiv_org_abs_2508_11411
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle SelfAdapt: Unsupervised Domain Adaptation of Cell Segmentation Models
Reith, Fabian H.
Franzen, Jannik
Palli, Dinesh R.
Rumberger, J. Lorenz
Kainmueller, Dagmar
Computer Vision and Pattern Recognition
Machine Learning
Deep neural networks have become the go-to method for biomedical instance segmentation. Generalist models like Cellpose demonstrate state-of-the-art performance across diverse cellular data, though their effectiveness often degrades on domains that differ from their training data. While supervised fine-tuning can address this limitation, it requires annotated data that may not be readily available. We propose SelfAdapt, a method that enables the adaptation of pre-trained cell segmentation models without the need for labels. Our approach builds upon student-teacher augmentation consistency training, introducing L2-SP regularization and label-free stopping criteria. We evaluate our method on the LiveCell and TissueNet datasets, demonstrating relative improvements in AP0.5 of up to 29.64% over baseline Cellpose. Additionally, we show that our unsupervised adaptation can further improve models that were previously fine-tuned with supervision. We release SelfAdapt as an easy-to-use extension of the Cellpose framework. The code for our method is publicly available at https: //github.com/Kainmueller-Lab/self_adapt.
title SelfAdapt: Unsupervised Domain Adaptation of Cell Segmentation Models
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2508.11411