Harmonization with Flow-based Causal Inference
arXiv:2106.06845
Abstract
Heterogeneity in medical data, e.g., from data collected at different sites and with different protocols in a clinical study, is a fundamental hurdle for accurate prediction using machine learning models, as such models often fail to generalize well. This paper leverages a recently proposed normalizing-flow-based method to perform counterfactual inference upon a structural causal model (SCM), in order to achieve harmonization of such data. A causal model is used to model observed effects (brain magnetic resonance imaging data) that result from known confounders (site, gender and age) and exogenous noise variables. Our formulation exploits the bijection induced by flow for the purpose of harmonization. We infer the posterior of exogenous variables, intervene on observations, and draw samples from the resultant SCM to obtain counterfactuals. This approach is evaluated extensively on multiple, large, real-world medical datasets and displayed better cross-domain generalization compared to state-of-the-art algorithms. Further experiments that evaluate the quality of confounder-independent data generated by our model using regression and classification tasks are provided.
References in corpus (10)
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- Auto-Encoding Variational Bayes
- Normalizing Flows for Probabilistic Modeling and Inference
- Invariant Risk Minimization
- Detect and Correct Bias in Multi-Site Neuroimaging Datasets
- Causality for Machine Learning
- Deep Structural Causal Models for Tractable Counterfactual Inference
- Image-level Harmonization of Multi-Site Data using Image-and-Spatial Transformer Networks
- Medical Image Harmonization Using Deep Learning Based Canonical Mapping: Toward Robust and Generalizable Learning in Imaging
- Invertible Generative Modeling using Linear Rational Splines