Test-Time Adaptation Via Orthogonal Meta-Learning for Medical Imaging

Zhiwen Wang,Zexin Lu,Tao Wang,Ziyuan Yang,Hui Yu,Zhongxian Wang,Yinyu Chen,Jingfeng Lu,Yi Zhang
DOI: https://doi.org/10.1109/trpms.2024.3462542
2024-01-01
IEEE Transactions on Radiation and Plasma Medical Sciences
Abstract:Deep learning (DL) models, which have significantly promoted medical imaging, typically assume that training and testing data come from the same domain and distribution. However, these models struggle with unseen testing variations, like different imaging scanners or protocols, leading to sub-optimal results from distribution mismatches between training and testing data. Despite extensive research, the issue of distribution mismatch in DL-based medical imaging has been largely overlooked in current literature. To improve the performance with mismatched testing data, this paper proposes an orthogonal meta-learning (OML) framework for test-time adaptation in medical imaging. Specifically, during training, we develop supervised meta-training reconstruction tasks to guide the self-supervised meta-testing task. Additionally, we introduce an orthogonal learning strategy to enforce orthogonality of pretrained parameters during training, which accelerates convergence during TTA and enhances performance. During the testing stage, the fine-tuned meta-learned parameters effectively reconstruct new, unseen testing data. Extensive experiments on magnetic resonance imaging and computed tomography datasets were conducted to validate our method’s effectiveness against other state-of-the-art methods, including supervised ones, in various mismatch scenarios.
What problem does this paper attempt to address?