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 Heterogeneous Medical Data

Abstract: Heterogeneity in medical imaging data is often tackled, in the context of machine learning, using domain invariance, i.e. deriving models that are robust to domain shifts, which can be both within domain (e.g. demographics) and across domains (e.g. scanner/protocol characteristics). However this approach can be detrimental to performance because it necessitates averaging across intra-class variability and reduces discriminatory power of learned models, in order to achieve better intra- and inter-domain generalization. This paper instead embraces the heterogeneity and treats it as a multi-task learning problem to explicitly adapt trained classifiers to both inter-site and intra-site heterogeneity. We demonstrate that the error of a base classifier on challenging 3D brain magnetic resonance imaging (MRI) datasets can be reduced by 2-3 times, in certain tasks, by adapting to the specific demographics of the patients, and different acquisition protocols. Learning the characteristics of domain shifts is achieved via auxiliary learning tasks leveraging commonly available data and variables, e.g. demographics. In our experiments, we use gender classification and age regression as auxiliary tasks helping the network weights trained on a source site adapt to data from a target site; we show that this approach improves classification accuracy by 5-30 % across different datasets on the main classification tasks, e.g. disease classification.
Comments: Accepted by Med-NeurIPS 2020
Subjects: Machine Learning (cs.LG); Image and Video Processing (eess.IV)
Cite as: arXiv:2103.12857 [cs.LG]
  (or arXiv:2103.12857v1 [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.