Skip to content

CRISP: Constrained Refinement via Iterative Squeezing Process for Robust Medical Image Segmentation under Domain Shift

Jul 2026 · arXiv.org · Vol abs/2607.15231 · 0 citations · 20 references
Computer Science

TL;DR

CRISP is a model-agnostic framework that requires no test-time parameter updates and no target-domain data--a target-free, plug-in refinement framework that segments with frozen weights, exploiting the stability of probability rankings under distribution shift to derive robust spatial priors.

Abstract

Distribution shift in medical imaging remains a central bottleneck for the clinical translation of medical AI. Failure to address it can lead to severe performance degradation in unseen environments and exacerbate health inequities. Existing methods for domain adaptation are inherently limited by exhausting predefined possibilities through simulated shifts or pseudo-supervision. Such strategies struggle in the open-ended and unpredictable real world, where distribution shifts are effectively infinite. To address this challenge, we adopt the"Rank Stability of Positive Regions"as a working assumption under distribution shift, and use it to derive robust spatial hints for source-only segmentation. Guided by this assumption, we propose CRISP, a model-agnostic framework that, unlike deployment-time adaptation, requires no test-time parameter updates and no target-domain data--a target-free, plug-in refinement framework that segments with frozen weights. Rather than using ranking to directly output masks, CRISP exploits the stability of probability rankings under distribution shift to derive robust spatial priors. Via latent feature perturbation, perturbation-invariant high-grade regions define a high-precision (HP) core, while voxels that remain potentially foreground under at least one perturbation define a high-recall (HR) support; these dual priors are then recursively refined under perturbation. We then design an iterative training framework that progressively squeezes HP and HR toward the final segmentation. Extensive evaluations on multi-center cardiac MRI and CT-based lung vessel segmentation demonstrate CRISP's superior robustness, significantly outperforming state-of-the-art methods with striking HD95 reductions of up to 0.14 (7.0% improvement), 1.90 (13.1% improvement), and 8.39 (38.9% improvement) pixels across multi-center, demographic, and modality shifts, respectively.

View source

Similar papers

Preprint Aug 2026

SymmAdapt: Symmetrical Flow Matching for Source-Free Domain Adaptation in Medical Image Segmentation

Domain shift across imaging modalities and acquisition sites remains a significant barrier to the clinical deployment of segmentation models. Source-free unsupervised domain adaptation (SFUDA) addresses this by adapting a pretrained model to an unlabeled target domain without requiring access to sensitive source data. We introduce a novel SFUDA framework built on Symmetrical Flow Matching, a unified generative model that segments an input image and synthesizes a source-like image from a mask within the same learned flow. By initializing inference from a domain-agnostic Gaussian origin, the model preserves structural consistency across domains and grounds predictions in learned anatomy rather than shifted texture statistics. Our pipeline leverages this symmetry to generate reliable pseudo-labels and corresponding source-like synthetic images from unlabeled target data, creating a generative replay buffer that anchors source knowledge during a generative self-training stage that fine-tunes on a joint set of real target and synthetic source-like images. We evaluate on abdominal multi-organ and cardiac segmentation, covering cross-modality MRI<->CT shifts, and multi-site prostate segmentation. Our approach outperforms SFUDA baselines and is competitive with conventional UDA methods.

T. Grossman, N. Cahan, H. Greenspan · 0 citations
#artificial intelligence Preprint Aug 2026

MedSegBenchmarker: A Raw-Count-First Framework for Controlled 2D Medical Image Segmentation Benchmarks

Despite rapid advances in MIS, fair and reproducible comparisons of segmentation models remain challenging due to heterogeneous datasets, inconsistent evaluation protocols, and rapidly evolving architectures. In particular, comparisons often implicitly assume that model rankings are invariant to data partitioning, preprocessing, metric aggregation, uncertainty estimation, and computational constraints. The lack of extensible and unified evaluation frameworks further limits systematic investigation of new models, datasets, and training paradigms. We present MEDSEGBENCHMARKER (MSB), a configuration-driven framework for controlled benchmarking of 2D MIS. It integrates duplicate and near-duplicate image detection, group-aware data splitting, YAML study specifications, resumable training, hyperparameter optimization, cross-validation, and checkpoint-based evaluation. Rather than retaining only aggregate performance measures, MSB exports sample- and class-level pixel counts and predictions together with the evaluation context. These elementary artifacts enable post-hoc analyses without repeated inference. We demonstrate MSB in a case study involving three heterogeneous 2D datasets and multiple MIS and general-purpose vision models evaluated at 256- and 512-pixel input resolutions. Reaggregation of identical predictions changes the top-ranked architecture in three of six dataset-resolution settings, despite high rank correlations between aggregation strategies. Increasing input resolution produces model- and dataset-dependent performance gains and losses that must be considered alongside empirically measured inference complexity. These results show that seemingly minor choices in evaluation and experimental setup can affect benchmark conclusions. MSB, available at GitHub, provides a practical and extensible basis for making benchmark conditions and evaluation choices explicit and reproducible.

Vanessa Borst, Lukas Horn, Daniel Grillmeyer et al. · 0 citations
#diffusion models Open access Aug 2026

Robust unsupervised domain adaptation for medical image segmentation via frequency-conditioned graph diffusion

Cross-domain variability in medical imaging, arising from differences in scanners, acquisition protocols, and patient populations, remains a major challenge for reliable semantic segmentation. Existing unsupervised domain adaptation (UDA) methods predominantly rely on image-level transformations or feature alignment, which often fail to preserve anatomical consistency under large domain shifts. In this work, we propose a novel structured latent UDA framework that performs domain alignment in a topology-aware representation space rather than directly modifying image appearance. Specifically, we introduce a Frequency-Conditioned Graph Diffusion paradigm, where convolutional features are transformed into anatomical graphs to explicitly capture structural relationships. A latent diffusion process then progressively refines these graph embeddings, guided by frequency-aware contextual cues, enabling robust cross-domain alignment. To further enhance generalization, we integrate structural consistency regularization with adversarial latent alignment, eliminating the need for labeled target data. A dedicated decoder reconstructs dense segmentation maps, while stochastic diffusion sampling provides uncertainty estimates for improved potential clinical reliability. Extensive experiments on multiple public medical imaging benchmarks demonstrate that our method consistently outperforms state-of-the-art UDA approaches, achieving superior segmentation accuracy and robustness under significant domain shifts. These results highlight the effectiveness of structured latent modeling and diffusion-based learning for robust domain-adaptive segmentation.

Usman Ahmad Usmani, Arunava Roy, Junzo Watada · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.