A robust variational autoencoder using beta divergence. 2022

Haleh Akrami, and Anand A Joshi, and Jian Li, and Sergül Aydöre, and Richard M Leahy
Signal and Image Processing Institute, University of Southern California, Los Angeles, CA, USA.

The presence of outliers can severely degrade learned representations and performance of deep learning methods and hence disproportionately affect the training process, leading to incorrect conclusions about the data. For example, anomaly detection using deep generative models is typically only possible when similar anomalies (or outliers) are not present in the training data. Here we focus on variational autoencoders (VAEs). While the VAE is a popular framework for anomaly detection tasks, we observe that the VAE is unable to detect outliers when the training data contains anomalies that have the same distribution as those in test data. In this paper we focus on robustness to outliers in training data in VAE settings using concepts from robust statistics. We propose a variational lower bound that leads to a robust VAE model that has the same computational complexity as the standard VAE and contains a single automatically-adjusted tuning parameter to control the degree of robustness. We present mathematical formulations for robust variational autoencoders (RVAEs) for Bernoulli, Gaussian and categorical variables. The RVAE model is based on beta-divergence rather than the standard Kullback-Leibler (KL) divergence. We demonstrate the performance of our proposed β-divergence-based autoencoder for a variety of image and categorical datasets showing improved robustness to outliers both qualitatively and quantitatively. We also illustrate the use of our robust VAE for detection of lesions in brain images, formulated as an anomaly detection task. Finally, we suggest a method to tune the hyperparameter of RVAE which makes our model completely unsupervised.

UI MeSH Term Description Entries

Related Publications

Haleh Akrami, and Anand A Joshi, and Jian Li, and Sergül Aydöre, and Richard M Leahy
March 2021, Accident; analysis and prevention,
Haleh Akrami, and Anand A Joshi, and Jian Li, and Sergül Aydöre, and Richard M Leahy
September 2021, BMC bioinformatics,
Haleh Akrami, and Anand A Joshi, and Jian Li, and Sergül Aydöre, and Richard M Leahy
December 2024, ArXiv,
Haleh Akrami, and Anand A Joshi, and Jian Li, and Sergül Aydöre, and Richard M Leahy
December 2023, ArXiv,
Haleh Akrami, and Anand A Joshi, and Jian Li, and Sergül Aydöre, and Richard M Leahy
January 2021, Frontiers in genetics,
Haleh Akrami, and Anand A Joshi, and Jian Li, and Sergül Aydöre, and Richard M Leahy
December 2025, Scientific reports,
Haleh Akrami, and Anand A Joshi, and Jian Li, and Sergül Aydöre, and Richard M Leahy
October 2023, Medical engineering & physics,
Haleh Akrami, and Anand A Joshi, and Jian Li, and Sergül Aydöre, and Richard M Leahy
February 2025, IEEE transactions on neural networks and learning systems,
Haleh Akrami, and Anand A Joshi, and Jian Li, and Sergül Aydöre, and Richard M Leahy
March 2022, Communications chemistry,
Haleh Akrami, and Anand A Joshi, and Jian Li, and Sergül Aydöre, and Richard M Leahy
February 2024, Cell reports methods,
Copied contents to your clipboard!