MedShift: Implicit Conditional Transport for X-Ray Domain Adaptation

Caetano, Francisco, Viviers, Christiaan, de With, Peter H. H., van der Sommen, Fons

arXiv.org Artificial Intelligence 

Synthetic medical data offers a scalable solution for training robust models, but significant domain gaps limit its gen-eralizability to real-world clinical settings. This paper addresses the challenge of cross-domain translation between synthetic and real X-ray images of the head, focusing on bridging discrepancies in attenuation behavior, noise characteristics, and soft tissue representation. W e propose Med-Shift, a unified class-conditional generative model based on Flow Matching and Schr odinger Bridges, which enables high-fidelity, unpaired image translation across multiple domains. Unlike prior approaches that require domain-specific training or rely on paired data, MedShift learns a shared domain-agnostic latent space and supports seamless translation between any pair of domains seen during training. W e introduce X-DigiSkull, a new dataset comprising aligned synthetic and real skull X-rays under varying radiation doses, to benchmark domain translation models. Experimental results demonstrate that, despite its smaller model size compared to diffusion-based approaches, Med-Shift offers strong performance and remains flexible at inference time, as it can be tuned to prioritize either perceptual fidelity or structural consistency, making it a scalable and generalizable solution for domain adaptation in medical imaging.