AI RESEARCH

ROBUST-WT: Robust Uncertainty-aware Segmentation Transform via Whitening and Training Enhancements

arXiv CS.LG

ArXi:2606.03069v1 Announce Type: cross Generalized segmentation of medical images prevents performance degradation when different imaging devices and clinical protocols are used across multiple domains. The Whitening Transform-based Probabilistic Shape Regularization Extractor (WT-PSE), published in IEEE Transactions on Medical Imaging in 2024, addresses this challenge by employing feature decorrelation and Wasserstein distance-based knowledge distillation to achieve robust cross-domain segmentation. This study systematically examines improvements to the WT-PSE learning framework.