Data Augmentation-Based Unsupervised Domain Adaptation In Medical Imaging

Sebastian Nørgaard Llambias,Mads Nielsen,Mostafa Mehdipour Ghazi
2023-08-09
Abstract:Deep learning-based models in medical imaging often struggle to generalize effectively to new scans due to data heterogeneity arising from differences in hardware, acquisition parameters, population, and artifacts. This limitation presents a significant challenge in adopting machine learning models for clinical practice. We propose an unsupervised method for robust domain adaptation in brain MRI segmentation by leveraging MRI-specific augmentation techniques. To evaluate the effectiveness of our method, we conduct extensive experiments across diverse datasets, modalities, and segmentation tasks, comparing against the state-of-the-art methods. The results show that our proposed approach achieves high accuracy, exhibits broad applicability, and showcases remarkable robustness against domain shift in various tasks, surpassing the state-of-the-art performance in the majority of cases.
Image and Video Processing,Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
This paper aims to solve the problem of insufficient generalization ability of deep - learning models in medical imaging. Specifically, when these models are applied to new scan data, data heterogeneity caused by differences in hardware, acquisition parameters, population, and artifacts makes it difficult for the models to generalize effectively. This limitation poses a major challenge to the adoption of machine - learning models in clinical practice. To this end, the authors propose an unsupervised domain adaptation method based on MRI - specific enhancement techniques to improve the model robustness and generalization ability in brain MRI segmentation tasks. The paper verifies the effectiveness of the proposed method through extensive experiments, which cover multiple datasets, modalities, and segmentation tasks and are compared with the existing state - of - the - art methods. The results show that this method not only outperforms existing methods in most cases but also exhibits high accuracy and wide applicability, especially in handling domain shifts in different tasks.