Variational AutoEncoder For Regression: Application to Brain Aging Analysis

04/11/2019
by   Qingyu Zhao, et al.
0

While unsupervised variational autoencoders (VAE) have become a powerful tool in neuroimage analysis, their application to supervised learning is under-explored. We aim to close this gap by proposing a unified probabilistic model for learning the latent space of imaging data and performing supervised regression. Based on recent advances in learning disentangled representations, the novel generative process explicitly models the conditional distribution of latent representations with respect to the regression target variable. Performing a variational inference procedure on this model leads to joint regularization between the VAE and a neural-network regressor. In predicting the age of 245 subjects from their structural Magnetic Resonance (MR) images, our model is more accurate than state-of-the-art methods when applied to either region-of-interest (ROI) measurements or raw 3D volume images. More importantly, unlike simple feed-forward neural-networks, disentanglement of age in latent representations allows for intuitive interpretation of the structural developmental patterns of the human brain.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
09/05/2019

Independent Subspace Analysis for Unsupervised Learning of Disentangled Representations

Recently there has been an increased interest in unsupervised learning o...
research
12/11/2019

A Closer Look at Disentangling in β-VAE

In many data analysis tasks, it is beneficial to learn representations w...
research
04/22/2020

Polarized-VAE: Proximity Based Disentangled Representation Learning for Text Generation

Learning disentangled representations of real world data is a challengin...
research
06/12/2017

Channel-Recurrent Variational Autoencoders

Variational Autoencoder (VAE) is an efficient framework in modeling natu...
research
07/09/2020

Modelling the Distribution of 3D Brain MRI using a 2D Slice VAE

Probabilistic modelling has been an essential tool in medical image anal...
research
05/26/2020

Unsupervised Brain Abnormality Detection Using High Fidelity Image Reconstruction Networks

Recent advances in deep learning have facilitated near-expert medical im...
research
10/19/2020

Learning Optimal Conditional Priors For Disentangled Representations

A large part of the literature on learning disentangled representations ...

Please sign up or login with your details

Forgot password? Click here to reset