Predicting geographic location from genetic variation with deep neural networks

  1. CJ Battey  Is a corresponding author
  2. Peter L Ralph
  3. Andrew D Kern
  1. University of Oregon, Institute of Ecology and Evolution, United States
9 figures and 4 additional files

Figures

Conceptual schematic of our approach.

Regions of the genome reflect correlated sets of genealogical relationships (A), each of which represents a set of ancestors with varying spatial positions back in time. We extract genotypes from …

Figure 2 with 2 supplements
Validation error for Locator runs on simulations with varying dispersal rates.

Simulations were on a 50 × 50 landscape and error is expressed in map units. (A) True and predicted locations by population mean dispersal rate and number of SNPs. 450 randomly-sampled individuals …

Figure 2—figure supplement 1
Validation error for Locator runs on simulations with varying dispersal distance, expressed in generations of mean dispersal (test error divided by mean dispersal distance per generation).

(A) Error for runs with 100,000 SNPs and varying numbers of training samples. (B) Error for runs with 450 training samples and varying number of SNPs.

Figure 2—figure supplement 2
Example training and validation loss histories.

The first three epochs (with very high loss) were excluded from the plot to improve axis scaling.

Figure 3 with 1 supplement
Test error and run times for Locator and SPASIBA on simulated data with dispersal distance equal to 0.45 map units/generation (top; 450 randomly sampled training samples) and empirical data from the ag1000g phase one dataset (bottom; 612 training samples from 14 sampling localities).
Figure 3—figure supplement 1
Predicted (colored points) and true (black circles) locations for Locator and SPASIBA on the ag1000g dataset.

Number of SNPs per run is shown on the right. Both methods were run on randomly selected SNPs with minor allele count >2 from the first five million base pairs of chromosome 2L.

Figure 4 with 1 supplement
Predicted and true locations for eight individuals simulated in a population with mean per-generation dispersal 0.45 (roughly 1% of the landscape width).

Black points are predictions from 2Mbp windows, blue points are training sample locations, and the red point is the true location for each individual. Contours show the 95%, 50%, and 10% quantiles …

Figure 4—figure supplement 1
Predicted and true locations for eight individuals simulated in a population with an expected dispersal rate of 0.63 map units/generation, using a set of 10,000 randomly sampled SNPs.

Here, we generate predictions (black points) from bootstrap samples of the complete genotype matrix (in contrast to using separate sets of SNPs extracted from windows as used for figures in the main …

Figure 5 with 3 supplements
Top – Predicted locations for 153 Anopheles gambiae/coluzzii genomes from the AG1000G panel, using 612 training samples and a 2Mbp window size.

The geographic centroid of per-window predictions for each individual is shown in black points, and lines connect predicted to true locations. Sample localities are colored by the mean test error …

Figure 5—figure supplement 1
Comparison of cross-validation performance on the ag1000g dataset using SNPs from chromosome 3R, under varying network architectures and numbers of SNPs.

Boxplots show the distribution of Euclidean distance between the true and predicted locations of validation samples across 10 replicate training runs. Network shapes are described on the horizontal …

Figure 5—figure supplement 2
Performance on 10,000 SNPs from chromosome 2L in the ag1000g phase one dataset when all samples from localities in the true country are dropped from the training set.
Figure 5—figure supplement 3
Performance on 10,000 SNPs from chromosome 2L in the ag1000g phase one dataset when all samples from the true locality are dropped from the training set.
Figure 6 with 1 supplement
Top – Predicted locations for 881 Plasmodium falciparum from the Plasmodium falciparum Community Project (Pearson et al., 2019) (5% of samples for each collecting locality), using 5084 training samples and a 500Kbp window size.

The geographic centroid of per-window predictions for each individual is shown in black points, and lines connect predicted to true locations. Sample localities are colored by the mean test error …

Figure 6—figure supplement 1
Centroid prediction error as a function of within-host diversity (FWS) for the Plasmodium falciparum dataset.

FWS scales from 0 (maximum complexity) to 1 (minimum complexity). The blue line shows a linear regression (p<2.2e-16,R2=0.006,slope=148.1). High within-host diversity does not appear to explain outliers in Locator’s prediction …

Figure 7 with 1 supplement
Top – Predicted locations for 162 individuals from the HGDP panel, using 773 training samples and a 10Mbp window size.

The geographic centroid of per-window predictions for each individual is shown in black points, and lines connect predicted to true locations. Sample localities are colored by the mean test error …

Figure 7—figure supplement 1
Outliers in windowed Locator analyses identify genomic regions enriched for admixed ancestry.

(A) Windowed Locator predictions for Maya sample HGDP00871. (B) PCAs of all HGDP samples run on SNPs extracted from windows with predicted locations in western Europe (left) and west Africa (right). …

Figure 8 with 4 supplements
Per-window test error and mean recombination rate for human populations in the HGDP dataset.

The top 2% of windows by test error were excluded from this analysis. The slope of the least-squares linear fit is −99.9723 km/(cM/Mbp) and has adjusted R2=0.2704.

Figure 8—figure supplement 1
Mean test error for HGDP samples in 10-megabase windows.

Triangles show approximate centromere locations.

Figure 8—figure supplement 2
Mean test error for HGDP samples in 10-centimorgan windows.

Triangles show approximate centromere locations.

Figure 8—figure supplement 3
Distributions of centroid prediction error across samples.

Despite differences in error among genomic windows (Figure 8—figure supplements 1 and 2), error in the mean genome-wide predicted location is very similar when using megabase (top) or centimorgan …

Figure 8—figure supplement 4
Distributions of prediction error across windows when using megabase- versus centimorgan-based windows.
Predicted locations for HGDP samples from central Asia using a model trained on SNPs within 100 kb of EDAR.

Black points show sampling locations. Arrows are colored by genotype at variant rs3827760 and point towards the predicted location. Frequency of the A allele by longitude is shown below the map.

Additional files

Supplementary file 1

Validation error in terms of map units and generations of mean population dispersal for Locator runs in simulations with 450 training samples and 100,000 SNPs.

Note that while absolute error increases along with dispersal rate, it is roughly constant when expressed in terms of generations of mean dispersal.

https://cdn.elifesciences.org/articles/54507/elife-54507-supp1-v2.csv
Supplementary file 2

Mean and median prediction error for Locator and SPASIBA run on simulations and Anopheles data as shown in Figure 3.

Error is in terms of map units for simulated data (total landscape width = 50).

https://cdn.elifesciences.org/articles/54507/elife-54507-supp2-v2.csv
Supplementary file 3

Test error for windowed analyses of empirical datasets using the location with highest kernel density and the centroid of per-window predictions, as median (90% interval).

https://cdn.elifesciences.org/articles/54507/elife-54507-supp3-v2.csv
Transparent reporting form
https://cdn.elifesciences.org/articles/54507/elife-54507-transrepform-v2.docx

Download links