Test-Time Adaptation via Orthogonal Meta-Learning for Medical Imaging
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 suboptimal 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 article proposes an orthogonal meta-learning (OML) framework for test-time adaptation (TTA) 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.
Authors
- Ziyuan Yang (ORCID: https://orcid.org/0000-0002-0275-4098)
- Zexin Lu (ORCID: https://orcid.org/0000-0001-5307-145X)
- Tao Wang (ORCID: https://orcid.org/0000-0002-5311-3821)
- Jingfeng Lu (ORCID: https://orcid.org/0000-0002-9391-1297)
- Yinyu Chen
- Zhiwen Wang (ORCID: https://orcid.org/0000-0002-9936-9347)
- Yi Zhang (ORCID: https://orcid.org/0000-0001-7201-2092)
- Hui Yu (ORCID: https://orcid.org/0000-0002-6604-7819)
- Zhongxian Wang
Institutions
- Sichuan University (CN)
- Chengdu University (CN)
Publication Details
- Journal
- IEEE Transactions on Radiation and Plasma Medical Sciences
- Published
- 2024-09-17
- DOI
- https://doi.org/10.1109/trpms.2024.3462542
- Citations
- 1
- Primary Topic
- Radiomics and Machine Learning in Medical Imaging
- Type
- article
- Field-Weighted Citation Impact
- 0.40