RT Journal Article
SR Electronic
T1 Visualizing Population Structure with Variational Autoencoders
JF bioRxiv
FD Cold Spring Harbor Laboratory
SP 2020.08.12.248278
DO 10.1101/2020.08.12.248278
A1 C. J. Battey
A1 Gabrielle C. Coffing
A1 Andrew D. Kern
YR 2020
UL http://biorxiv.org/content/early/2020/10/19/2020.08.12.248278.abstract
AB Dimensionality reduction is a common tool for visualization and inference of population structure from genotypes, but popular methods either return too many dimensions for easy plotting (PCA) or fail to preserve global geometry (t-SNE and UMAP). Here we explore the utility of variational autoencoders (VAEs) – generative machine learning models in which a pair of neural networks seek to first compress and then recreate the input data – for visualizing population genetic variation. VAEs incorporate non-linear relationships, allow users to define the dimensionality of the latent space, and in our tests preserve global geometry better than t-SNE and UMAP. Our implementation, which we call popvae, is available as a command-line python program at github.com/kr-colab/popvae. The approach yields latent embeddings that capture subtle aspects of population structure in humans and Anopheles mosquitoes, and can generate artificial genotypes characteristic of a given sample or population.Competing Interest StatementThe authors have declared no competing interest.