We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Machine Learning

Title: Embracing the Disharmony in Medical Imaging: A Simple and Effective Framework for Domain Adaptation

Abstract: Domain shift, the mismatch between training and testing data characteristics, causes significant degradation in the predictive performance in multi-source imaging scenarios. In medical imaging, the heterogeneity of population, scanners and acquisition protocols at different sites presents a significant domain shift challenge and has limited the widespread clinical adoption of machine learning models. Harmonization methods which aim to learn a representation of data invariant to these differences are the prevalent tools to address domain shift, but they typically result in degradation of predictive accuracy. This paper takes a different perspective of the problem: we embrace this disharmony in data and design a simple but effective framework for tackling domain shift. The key idea, based on our theoretical arguments, is to build a pretrained classifier on the source data and adapt this model to new data. The classifier can be fine-tuned for intra-site domain adaptation. We can also tackle situations where we do not have access to ground-truth labels on target data; we show how one can use auxiliary tasks for adaptation; these tasks employ covariates such as age, gender and race which are easy to obtain but nevertheless correlated to the main task. We demonstrate substantial improvements in both intra-site domain adaptation and inter-site domain generalization on large-scale real-world 3D brain MRI datasets for classifying Alzheimer's disease and schizophrenia.
Subjects: Machine Learning (cs.LG); Image and Video Processing (eess.IV)
Journal reference: Medical Image Analysis 76:102309 (2022)
DOI: 10.1016/j.media.2021.102309
Cite as: arXiv:2103.12857 [cs.LG]
  (or arXiv:2103.12857v3 [cs.LG] for this version)

Submission history

From: Rongguang Wang [view email]
[v1] Tue, 23 Mar 2021 21:36:39 GMT (142kb,D)
[v2] Tue, 27 Apr 2021 21:35:53 GMT (336kb,D)
[v3] Mon, 20 Dec 2021 16:23:40 GMT (355kb,D)

Link back to: arXiv, form interface, contact.