Abstract
Machinelearning techniques have helped solve a broad range of prediction problems, yet are not widely used to build polygenic risk scores for the prediction of complex traits. We propose a novel heuristic based on machinelearning techniques (GraBLD) to boost the predictive performance of polygenic risk scores. Gradient boosted regression trees were first used to optimize the weights of SNPs included in the score, followed by a novel regional adjustment for linkage disequilibrium. A calibration set with sample size of ~200 individuals was sufficient for optimal performance. GraBLD yielded prediction R ^{2} of 0.239 and 0.082 using GIANT summary association statistics for height and BMI in the UK Biobank study (N = 130 K; 1.98 M SNPs), explaining 46.9% and 32.7% of the overall polygenic variance, respectively. For diabetes status, the area under the receiver operating characteristic curve was 0.602 in the UK Biobank study using summarylevel association statistics from the DIAGRAM consortium. GraBLD outperformed other polygenic score heuristics for the prediction of height (p < 2.2 × 10^{−16}) and BMI (p < 1.57 × 10^{−4}), and was equivalent to LDpred for diabetes. Results were independently validated in the Health and Retirement Study (N = 8,292; 688,398 SNPs). Our report demonstrates the use of machinelearning techniques, coupled with summarylevel data from large genomewide metaanalyses to improve the prediction of polygenic traits.
Introduction
The advent of precision medicine depends in large part on the availability of accurate and highly predictive polygenic risk scores. While progress has been made identifying genetic determinants of polygenic traits, the amount of phenotypic variance explained by polygenic risk scores derived from genomewide significant associations remains modest. On the other hand, moderate to high narrowsense heritability has been established for many human traits. It has been proposed that weak, yet undetected, associations underlie polygenic trait heritability^{1}. Consistent with this hypothesis, polygenic risk scores that include both strongly and weakly associated variants are vastly superior than those including only genomewide significant variants. For example, a recent study by Abraham et al. showed that a polygenic risk score incorporating 49,310 variants had a discrimination ability that was similar and complementary to the widely used clinical Framingham risk score for the prediction of coronary artery disease (CAD)^{2}. Thus, there is a need for polygenic risk score methods that can integrate a large number of genetic variants.
The most popular heuristic for polygenic risk score is based on linkage disequilibrium (LD) pruning of SNPs, prioritizing the most significant associations up to an empirically determined pvalue threshold, and pruning the remaining SNPs based on LD^{3}. This “pruning and thresholding” (P+T) approach has the advantage of being simple and computationally efficient, but discards some information because of LD pruning. To remediate this issue, a novel method, LDpred, which uses LD information from an external reference panel, was recently proposed to infer the mean causal effect size using a Bayesian approach^{4}. While the latter method has been shown to improve prediction R ^{2}, it depends on estimates of polygenic heritability and causal fraction, and can be sensitive to the misspecification of LD. We hypothesized that a further gain in prediction R ^{2} could be made by tuning the weights of SNPs included in polygenic risk scores using principles of machinelearning.
Machinelearning encompasses a wideranging class of algorithms widely used to solve complex prediction problems. It has proven particularly useful when prediction is dependent on the integration of a large number of predictors, including higherorder interactions, and when sizeable training datasets are available for model fitting. In particular, gradient boosted regression trees are powerful and versatile methods for continuous outcome prediction^{5}, and thus, are ideal for updating the SNP weights in polygenic risk scores. Treebased models partition the predictor space according to simple rules by identifying regions having the most homogeneous responses to predictors and fitting the mean response for observations in that region. Gradient boosting^{6} is an efficient algorithm that sequentially combines a large number of weakly predictive models to optimize performance.
We propose to leverage the large number of SNPs and the available summarylevel statistics from genomewide association studies (GWAS) to calibrate the weights of SNPs contributing to the polygenic risk score, adjusting for LD instead of pruning. Our heuristic, gradient boosted and LD adjusted (GraBLD; https://github.com/GMELab/GraBLD), involves two steps and uses the univariate regression coefficients from external metaanalysis^{7,8,9} summary association statistics as the starting point (see Fig. 1 and Methods). First, the external univariate regression coefficients were updated with respect to a target population by the gradient boosted regression tree models. Second, the updated weights were corrected for LD to produce the final polygenic risk score.
Results
We applied our machinelearning heuristic for height predictions using a calibration set of 10,000 participants, as well as an independent validation set of 130,215, both from the UK Biobank (UKB). The inputs for the gradient boosted regression trees were obtained from the Genetic Investigation of Anthropometric Traits (GIANT) consortium summary association statistics^{10,11} of 1.98 M SNPs. Since the UKB is not part of the GIANT consortium, the initial weights were assumed to be independent of the target population. As recently proposed^{12}, principal components were added to the model and included in the prediction R ^{2}. The prediction R ^{2} of our GraBLD polygenic risk score that included all SNPs was 0.239, corresponding to 46.9% of the total polygenic genetic variance estimated at 0.509 in the UKB using variance component models^{13}. This compared advantageously to the optimal prediction R ^{2} obtained with P+T (0.220; 177 K SNPs), LDpred (0.207), or an unadjusted polygenic risk score (0.165) (p < 2.2 × 10^{−16} for all pairwise comparisons with GraBLD; Figs 2 and 3).
We also tested the performance of GraBLD for the prediction of body mass index (BMI) and diabetes in the UKB using summary association statistics from the GIANT consortium for BMI, and the DIAbetes Genetics Replication And Metaanalysis (DIAGRAM) consortium^{14} for diabetes, respectively. The resulting score for BMI had a prediction R ^{2} of 0.082, outperforming the prediction R ^{2} of the unadjusted polygenic risk score (0.069), P+T (0.069), and LDpred (0.074) (p < 1.6 × 10^{−4} for all pairwise comparisons with GraBLD; Fig. 2). The GraBLD polygenic risk score accounted for 32.7% of the total polygenic variance, which was estimated at 0.251 for BMI in the UKB using variance component models. For diabetes, the area under the receiver operator curve (AUC) was 0.602, which was not statistically different from LDpred (0.613; p = 0.06), and compared favorably to the unadjusted polygenic risk score (0.583), as well as P+T (0.576) (p < 10^{−5} for comparisons with GraBLD; Figs 2 and 3). For sensitivity analyses, we tested the influence of the number of folds used to fit SNP weights, the number of SNPs used for LD adjustments, and the interaction depth on polygenic score performance (Supplementary Figure S2). We also illustrated the relationship between weights updated by gradient boosted regression trees and the external regression coefficients from consortia (Supplementary Figure S3).
Calibration, or the ability of a gene score to accurately predict real observations, is as important as predictiveness when gene scores are used to infer unobserved traits. To evaluate calibration, we calculated the average absolute difference between the predicted trait and the actual trait for height and BMI in the validation set. For all methods, polygenic risk scores were first calibrated in the training set through the use of a linear regression model (along with the top principal components). The average absolute difference was the smallest for GraBLD for height (0.690 SD) and BMI (0.742 SD), compared to other polygenic scores (p < 10^{−52} for all pairwise comparisons with GraBLD). We tested for calibration for diabetes using the HosmerLemeshow test^{15}, partitioning the UKB validation set by deciles of the predicted trait (Fig. 4). There was no evidence of mismatch between the predicted and observed event rates (p > 0.05).
The set of participants used to calibrate GraBLD can theoretically be the test set since the univariate regression coefficient of each SNP in the target population is not used to tune its own polygenic score weight. However, doing so presents practical challenges when one wants to predict a trait unobserved in the target population. In such cases, a smaller training sample size is advantageous. Therefore, we explored the effect of the size of the calibration set on GraBLD performance by subsampling an increasing proportion of our calibration set for tuning. We determined that a calibration set as small as 200 was adequate to provide a high prediction R ^{2} for height and BMI (Fig. 5). For diabetes, we selected an increasing number of casecontrol pairs, and 100 pairs were sufficient for adequate performance.
For any given SNP, the regression coefficient observed in the UKB was not used to determine its own weight in the polygenic risk score. Nonetheless, regression coefficients of other SNPs in the UKB were used, raising the issue of transferability to other populations. Hence, we tested GraBLD derived from the UKB in Health and Retirement Study (HRS) participants of European descent (N = 8,292). Only directly genotyped SNPs were used for this analysis and 683 K SNPs overlapped with both the UKB and consortia associations. For each method, the optimal GraBLD derived in the UKB calibration set was tested in the HRS without any further fitting or adjustment. Consistent with the UKB results, our machinelearning heuristic produced superior polygenic risk scores for height and BMI, compared to all others methods, and was a close second to LDpred for diabetes (Figs 2 and 3).
Discussion
Our proposed machinelearning heuristic led to significant improvements of polygenic risk scores in prediction R ^{2}, compared to existing methods. Furthermore, we showed that GraBLD risk scores were well calibrated, requiring only a small “tuning” set sample size (N~200) to achieve satisfactory performance. This latter characteristic makes our method advantageous for the prediction of unobserved traits, and stems from the fact that our heuristic leverages the large number of genetic variants reported in GWAS to train gradient boosted regression tree models through genome partitioning. Overall, our results demonstrate that machinelearning techniques coupled with summarylevel data from large genomewide metaanalysis improve the prediction of polygenic traits.
The regression trees approach we used can capture nonlinear effects and higherorder interactions, while the gradient boosting algorithm combines individually weak predictors to produce a strong classifier that enables a better prediction of genetic effects. The gradient boosted regression trees adaptively reweight the contribution of each SNP in order to maximize the prediction R ^{2} in a target population. Summary association statistics obtained from large external metaanalyses are implicitly assumed to provide the best initial estimates and regression trees “adapt” them to the regression coefficients observed in the target population. To avoid overfitting, SNPs were divided into five distinct contiguous sets (thus circumventing potential LD spillover) and the weights of SNPs in each set were calculated using the prediction models trained on the remaining four sets. For example, the first set comprised SNPs from chromosomes 1, 2, and part of 3 such that SNPs from the remaining part of chromosome 3, as well as those on chromosomes 4 to 22 were used to derive prediction models for SNPs in the first set. Thus, the observed regression coefficients of any single SNP in the target population was never used directly or indirectly to derive its own weight in the polygenic score. In addition, we used a small learning rate for the boosting algorithm to reduce the risk of overfitting as it has been suggested that boosting is quite robust to overfit^{16}. We also explored alternative machinelearning techniques to tune the SNP weights, with bagging being a close second to gradient boosted regression trees in terms of prediction R ^{2} (0.229 for height and 0.080 for BMI) as it is based on a similar principle of subsampling. Neural net produced inferior results and slower computation. Support vector machine and random forest proved to be computationally prohibitive with run times exceeding 7 days for the same analyses done in 8.25 hours by GraBLD.
It is advantageous to correct the derived weights for LD when including multiple SNPs in a score, unless SNPs were first LD pruned. The novel correction we propose is based on the sum of pairwise LD r ^{2} of each SNP over neighboring SNPs. The polygenic risk score weights of each SNP were divided by the corresponding sum of r ^{2}. To illustrate with a simple example, if five SNPs are in perfect LD (r ^{2} = 1) with each other, but in linkage equilibrium with all other SNPs (r ^{2} = 0), then the polygenic score weights of those five SNPs would be divided by five. Since all five SNPs are included in the score and the effect of all five SNPs are summed, the corrected weight contributions are equivalent to including a single SNP without correction. Thus, it is necessary to apply the LD correction only after adjusting SNP weights with gradient boosted regression trees as otherwise important information on the strength of association of individual SNPs would be lost.
LD is only summed over SNPs included in the polygenic risk score such that our correction is specific to the set of SNPs included in a given score. When the genetic effects were strictly additive (i.e., no haplotype or interaction effect), the resulting polygenic score provided an unbiased estimate of the underlying genetic variance although at a tradeoff of increased polygenic score variance as compared to the “true” unobserved genetic model (see Methods). It can be shown that the variance explained by the polygenic risk score R ^{2} = R ^{2} _{true} in simple cases where the pairwise r ^{2} LD is either 0 or 1 and the summary association statistics are derived from an asymptotically large sample. In more common scenarios with partial LD, the variance explained by R ^{2} < R ^{2} _{true} reflects the loss of information when, for example, two SNPs are in partial LD and have true genetic effects with opposite directions. Using simulations, we estimated this loss of information at ~12% as the prediction R ^{2} explained ~88% of the true genetic variance, on average (see Methods).
A few limitations are worth mentioning. First, our method was based on the premise that SNPs contribute additively to genetic variance. While empirical evidence suggests this holds true in most cases, our method is not expected to perform well in genomic regions where strong genetic interactions are present (e.g. HLA). In such situations, alternative methods such as LDpred might be better suited^{4}. Second, there is a possibility that the polygenic risk scores derived using our method are inherently populationspecific. However, with the exception of unadjusted polygenic risk scores, all methods require a determination of parameters in the target population and ours is no exception. Furthermore, if the genetic architecture varies between populations, then no polygenic risk score will perform universally well and it will be beneficial to tailor gene scores to each population. The observation that our heuristic performed equally well in the HRS, compared to other methods, suggests this might not be the case. Moreover, the small calibration sample size required by GraBLD is an advantage over other gene score methods. Third, our correction for LD yielded advantageous results yet is expected to lead to some loss of information when truly associated SNPs are in partial LD. Nonetheless, our method has several benefits over other methods, including its simplicity, use of summary association statistics, and intrinsic robustness to minor misspecification of LD or association strength.
In summary, we propose a novel heuristic based on machinelearning concepts to improve the prediction of polygenic traits using gene scores. Our results show that for the classic polygenic traits, height and BMI, 46.9% and 32.7% of the estimated polygenic genetic variance was captured by our GraBLD gene scores. These results demonstrate the potential of machinelearning methods to harness the considerable amount of information available from local GWAS and external genomewide metaanalyses. This is made possible through partitioning of the genome, enabling training of regression trees over large numbers of observations. Indeed, a small training sample size (~200) was sufficient to greatly improve the predictiveness of polygenic risk scores. As with other prediction problems involving machinelearning techniques, incremental improvements are to be expected with increased sample size, the inclusion of additional predictors, and the availability of more precise summary association statistics.
Methods
Datasets
Summary association statistics were used to tune the weight of SNPs in polygenic risk scores according to the target population. Univariate regression coefficients for height and body mass index (BMI) were downloaded from the Genetic Investigation of Anthropometric Traits (GIANT) consortium^{4,10,11,17} at http://portals.broadinstitute.org/collaboration/giant/index.php/GIANT_consortium_data_files. Univariate coefficients for diabetes were obtained from DIAbetes Genetics Replication And Metaanalysis (DIAGRAM) consortium^{14} at www.diagramconsortium.org/.
The UK Biobank^{18} (UKB) is a large populationbased study from the United Kingdom. A total of 152,249 participants were genotyped using either the UK BiLEVE or the UK Biobank Affymetrix Axiom arrays, and a subset of 140,215 participants of European (British and Irish) Caucasian ancestry were used in the analyses. Genotypes were imputed using the UK10K reference panel using IMPUTE2, resulting in ~72 M SNPs. Height and BMI were adjusted for age and sex in all analyses. To mitigate the effects of outliers, values outside the 1^{st} and 99^{th} percentile were removed. All analyses were adjusted for the first 15 genetic principal components unless stated otherwise. The final sample size for height and BMI was 130,215. The UKB is not part of the GIANT metaanalysis of height and BMI^{19,20}, nor of the DIAGRAM consortium for diabetes^{14}. There are 6,746 individuals with prevalent diabetes in the subset of the UKB included in the current report. We randomly selected 6,746 individuals without diabetes as paired controls on a 1:1 ratio. We then randomly sampled 1,000 casecontrol pairs as the calibration set, with the remaining 5,746 pairs forming the validation set.
The Health and Retirement Study (HRS) is a longitudinal study conducted on Americans over age 50. We downloaded publicly available genomewide data that are part of the HRS (dbGaP Study Accession: phs000428.v1.p1) and were generated using the Illumina Human Omni2.5Quad BeadChip. The following HRS quality control criteria were used to filter genotype and phenotype data: (1) SNPs and individuals with missingness higher than 2% were excluded, (2) related individuals were excluded, (3) only participants with selfreported European ancestry and genetically confirmed by principal component analysis were included, (4) individuals for whom the reported sex does not match their genetic sex were excluded, (5) SNPs with HardyWeinberg equilibrium p < 1×10^{−6} were excluded, and (6) SNPs with minor allele frequencies lower than 0.02 were removed. The final dataset included 8,292 European participants genotyped for 688,398 SNPs. Height and BMI was adjusted for age and sex in all analyses, and to mitigate the effect of outliers, values outside the 1^{st} and 99^{th} percentile range were removed. All analyses were adjusted for the first 20 genetic principal components unless stated otherwise. The final sample sizes for height and BMI were 8,291 and 8,262, respectively. There were 1,815 individuals with diabetes and 6,477 controls. HRS was not part of the GIANT metaanalysis of height and BMI^{19,20}, nor of the DIAGRAM consortium for diabetes^{14}.
Polygenic risk scores
The genotypes for n individuals at m SNPs in the target population are given by a matrix
with each column vector \({{\boldsymbol{x}}}_{{\bf{1}}},\,{{\boldsymbol{x}}}_{{\bf{2}}},\,\ldots ,\,{{\boldsymbol{x}}}_{{\boldsymbol{n}}}\) representing the coded genotypes for an individual. Without loss of generality, we assume each column of X (i.e., genotypes for a single SNP) to be standardized to have mean 0 and variance 1. For a standardized quantitative trait y with mean 0 and variance 1, the underlying linear model can be expressed as:
where β is a vector of true genetic effects that are fixed across individuals, but random across SNPs, with mean 0 and covariance matrix σ ^{2}I such that the total expected genetic variance is:
and ε the error term with mean 0 and covariance \((1m{\sigma }^{2}){\boldsymbol{I}}\), so that the covariance of y is I. Given x _{i}, the genotypes of m SNPs for the i ^{th} individual, the gradient boosted and LD adjusted (GraBLD) polygenic risk score g(x _{i}) is defined as:
where w is an mdimensional vector of boosted weights and C is an m × m diagonal matrix with entries, \((\frac{1}{{\eta }_{1}},\,\frac{1}{{\eta }_{2}},\ldots ,\frac{1}{{\eta }_{m}})\) adjusting for LD. For quantitative traits, the performance of the polygenic risk score was measured by the coefficient of determination (i.e., the prediction R ^{2}), and for binary traits, performance was measured using the area under the receiver operator characteristic (ROC) curve.
Gradient boosted regression trees
Gradient boosted regression trees are powerful and versatile methods that combine otherwise weak classifiers to produce a strong learner for continuous outcome prediction^{5}. They are ideally suited for improving SNP weights (w) in the polygenic risk score, without requiring individuallevel genotypes since they can be used to predict continuous outcomes and can model nonlinear relationships without feature selection. We also tested support vector machine (SVM), bagging, neural net, and random forest. SVM (“e1071” R package) and random forest (“randomForest” R package) took an inordinate amount of time to complete and were deemed impractical. Gradient boosted regression trees gave the best results when compared to bagging (“caret” R package) and neural net (“nnet” R package) using default parameters. Thus, all analyses were performed using gradient boosted regression trees. The fitted \(\hat{w}\) gave the contribution of individual SNPs to the final polygenic risk score. The weights used in gene scores (\(\hat{w})\) were defined by the following:
where β _{ext} refers to the univariate regression coefficients obtained as summarylevel association statistics from the external consortium (assumed to have been standardized for reference allele frequency), and \(\hat{{\boldsymbol{d}}}\) is derived to reflect the amount of deviation towards the null hypothesis of no association in the target population (β _{obs}) with respect to the externally derived estimates of summary association statistics (β _{ext}). When d = 0 then β _{obs} = β _{ext}, implicitly assuming regression coefficients from large metaanalyses provide the best initial weights. While some information is lost because of this construct, the fitted weights are more robust and expected to improve the overall performance of polygenic risk scores.
The dependent variable used in gradient boosted regression trees is constructed as:
and the fitted deviation \(\hat{{\boldsymbol{d}}}\) can be found by minimizing the squarederror loss function
where \(f\) is a regression function of trees with input variables \({\boldsymbol{Z}}=({Z}_{1},{Z}_{2},\ldots {Z}_{k})\). The gradient boost algorithm aims to iteratively minimize the expected square error loss, with respect to \(f\), on weighted versions of the training data (Z, d). While multiple SNP annotations could be included as inputs (i.e. \({Z}_{1},{Z}_{2},\ldots {Z}_{k}\)), we only included the absolute value of the SNP regression coefficient for the target trait from the external consortium to reflect the strength of association, irrespective of the direction of effect. Importantly, SNPs were divided into 5 distinct sets of contiguous SNPs (to avoid LD spillover), and the fitted \(\hat{{\boldsymbol{d}}}\) derived using the regression trees models trained on the remaining 4 sets was used to calculate the actual polygenic risk scores. The observed regression coefficient (β _{obs}) of an individual SNP is therefore, never used directly or indirectly to derive its own weight. Furthermore, the SNP annotations used in the regression trees model should be independent of the population in which the polygenic risk score is applied.
Gradient boosted regression trees models were fitted using the “GBM” R package (https://CRAN.Rproject.org/package=gbm) with a squared error loss function. A total of 2,000 trees were fitted with an interaction depth of 5, a shrinkage parameter of 0.001, and a bag fraction of 0.5. The final number of trees used for modeling was selected as per GBM package instructions. All other parameters were set to their default values. The run time for each of the 5 folds was 8.25 hours when performed on a single 3 GHz core.
LD Adjustment for SNP weights
We propose a simple method to correct weights for LD in such a way that all SNPs can be included in a gene score, irrespective of LD. Let r ^{2} _{j,k} denote the pairwise linkage disequilibrium (r ^{2}) between the j ^{th} and k ^{th} SNPs. The LD adjustment (η_{j}) for the j ^{th} SNP is defined by the sum of r ^{2} between the j ^{th} SNP and the 100 SNPs upstream and downstream:
with a distance of 100 SNPs assumed sufficient to ensure linkage equilibrium (other values may be used). Including only SNPs that are part of the polygenic risk score in the calculation of η_{j}, the LDcorrected weights are given by:
where w _{j} is the weight for the j ^{th} SNP.
Prediction R ^{2} of polygenic risk score
The prediction R ^{2} of the gene score in the target population is expressed as:
and the expected value can be approximated by:
and further simplified to
by deriving the following relations: (1) \({\rm{E}}[{\rm{Cov}}(g(X),{\boldsymbol{y}})]={R}_{{\rm{true}}}^{2}\), implying the covariance between the gene score and the trait is an unbiased estimator of the true genetic variance, and (2) \(E[\mathrm{Var}(g(X))] > \,{R}_{{\rm{true}}}^{2}\); thus, E[R ^{2}]<R ^{2} _{treu}, implying the expected prediction R ^{2} must be bounded above by the true genetic variance. We derive these two relations in the following subsections and further verify with simulations (Supplementary Figure S1).
An Unbiased Estimator of the True Genetic Variance
The sample covariance of the gene score with the observed y in the target sample is given by:
where e ^{*} and e are the residual error in the unobserved population used to derive summary association statistics and the target population, respectively. The reported b ^{*} in GWAS metaanalyses are constructed to estimate the univariate regression coefficients from the otherwise unobserved genotype matrix X ^{*} _{N×m}, and quantitative trait y ^{*}:
Assuming the target population is independent of the metaanalysis (i.e., e ^{*} and e are independent), we establish the expected value of the quadratic forms in (eq. 12):
This equality holds for all positive definite matrices of the form \(\frac{{{\boldsymbol{X}}}^{\ast {\boldsymbol{T}}}{{\boldsymbol{X}}}^{\ast }}{N}\,C\,\frac{{{\boldsymbol{X}}}^{{\boldsymbol{T}}}{\boldsymbol{X}}}{{\rm{n}}}\), assuming the LD structure in the two populations is identical. Thus, Cov(g(X), y) is an unbiased estimator of the true genetic variance.
Variance of the polygenic risk score
The denominator in (eq. 11), E[Var(g(X))], can be shown to be greater than R ^{2} _{true}:
while
And thus:
which leads to
From the above inequality, we can conclude that E[Var(g(X))] is biased and will always be greater than or equal to the true genetic variance.
All analyses were conducted in R statistical software, the scripts for gradient boosted regression trees and LD adjustments can be found at https://github.com/GMELab/GraBLD.
Simulations to assess the effect of LD adjustment on polygenic risk score bias and variance
We performed simulations to confirm the effect of LD adjustment on bias and polygenic risk score variance. A total of 5,000 individuals were simulated for 450 contiguous SNPs using phased haplotypes from the 1000 Genomes Project [19]. The genetic effect of each SNP was randomly selected from a normal distribution according to a predefined, unobserved, true regional genetic variance that assumed genomewide heritability varying from 0 to 0.5. For each genetic variance setpoint, 1,000 simulations were completed and a polygenic risk score incorporating LD correction was derived. The average (±SD) gene score prediction R ^{2}, and the gene score variance and covariance between the gene score and the true (unobserved) genetic effect was calculated (Supplementary Figure S3). Based on these simulations, we confirmed that (1) LDcorrected gene scores were unbiased estimators of true genetic variance (i.e. E[Cov(g(X), y)] = R ^{2} _{true}, and (2) the variance of the gene score was indeed higher than the true genetic variance. We further estimated the loss of information at ~12%, or in other words, the polygenic risk score prediction, R ^{2}, explained ~88% of the true genetic effect variance, on average.
Pruning and thresholding polygenic risk score, LDpred and other methods
Pruning and thresholding (P+T) polygenic scores were derived using the “clump” function of PLINK^{21} with an LD r ^{2} threshold of 0.2 and testing pvalue thresholds in a continuous manner from the most to the least significant association. LDpred adjusts GWAS summary statistics for the effects of linkage disequilibrium, providing reweighted effect estimates that are then used in polygenic risk scores^{4}. LDpred was run as recommended by the authors, and included data synchronization and LDpred steps. LDpred requires a specification for the fraction of SNPs assumed to be causal. For each model, we tested causal fractions of 1 (infinitesimal), 0.3, 0.1, 0.03, 0.01, 0.003, 0.001, 0.0003, and 0.0001, as recommended. The results are presented using the causal fraction of the best results. A heritability estimate was also required by the algorithm and was estimated from the summary association statistics from LDpred. As a sensitivity analysis, we additionally used heritability estimates given by the variance component models in the UKB. The results were consistent and only the default option is shown. Polygenic genetic variance (i.e., narrow sense heritability) was estimated for height and BMI in the UKB using the variance components implemented in GCTA^{13}. All LD measures or related estimates used throughout the manuscript were derived from the UKB calibration set genotypes.
Change history
20 November 2019
An amendment to this paper has been published and can be accessed via a link at the top of the paper.
References
 1.
Yang, J. et al. Common SNPs explain a large proportion of the heritability for human height. Nature genetics 42, 565–569, https://doi.org/10.1038/ng.608 (2010).
 2.
Abraham, G. et al. Genomic prediction of coronary heart disease. Eur Heart J 37, 3267–3278, https://doi.org/10.1093/eurheartj/ehw450 (2016).
 3.
International Schizophrenia, C. et al. Common polygenic variation contributes to risk of schizophrenia and bipolar disorder. Nature 460, 748–752, https://doi.org/10.1038/nature08185 (2009).
 4.
Vilhjalmsson, B. J. et al. Modeling Linkage Disequilibrium Increases Accuracy of Polygenic Risk Scores. Am J Hum Genet 97, 576–592, https://doi.org/10.1016/j.ajhg.2015.09.001 (2015).
 5.
Hastie, T., Tibshirani, R. & Friedman, J. H. The elements of statistical learning: data mining, inference, and prediction. 2nd edn, (Springer, 2009).
 6.
Schapire, R. E. & Freund, Y. Boosting: Foundations and algorithms. (MIT Press, 2012).
 7.
Evangelou, E. & Ioannidis, J. P. Metaanalysis methods for genomewide association studies and beyond. Nat Rev Genet 14, 379–389, https://doi.org/10.1038/nrg3472 (2013).
 8.
Lohmueller, K. E., Pearce, C. L., Pike, M., Lander, E. S. & Hirschhorn, J. N. Metaanalysis of genetic association studies supports a contribution of common variants to susceptibility to common disease. Nature genetics 33, 177–182, https://doi.org/10.1038/ng1071 (2003).
 9.
Zeggini, E. & Ioannidis, J. P. Metaanalysis in genomewide association studies. Pharmacogenomics 10, 191–201, https://doi.org/10.2217/14622416.10.2.191 (2009).
 10.
Locke, A. E. et al. Genetic studies of body mass index yield new insights for obesity biology. Nature 518, 197–206, https://doi.org/10.1038/nature14177 (2015).
 11.
Wood, A. R. et al. Defining the role of common variation in the genomic and biological architecture of adult human height. Nature genetics 46, 1173–1186, https://doi.org/10.1038/ng.3097 (2014).
 12.
Chen, C. Y., Han, J., Hunter, D. J., Kraft, P. & Price, A. L. Explicit Modeling of Ancestry Improves Polygenic Risk Scores and BLUP Prediction. Genet Epidemiol 39, 427–438, https://doi.org/10.1002/gepi.21906 (2015).
 13.
Yang, J., Lee, S. H., Goddard, M. E. & Visscher, P. M. GCTA: a tool for genomewide complex trait analysis. Am J Hum Genet 88, 76–82, https://doi.org/10.1016/j.ajhg.2010.11.011 (2011).
 14.
Morris, A. P. et al. Largescale association analysis provides insights into the genetic architecture and pathophysiology of type 2 diabetes. Nature genetics 44, 981–990, https://doi.org/10.1038/ng.2383 (2012).
 15.
Hosmer, D. W., Lemeshow, S. & Sturdivant, R. X. Applied logistic regression. Third edition/edn, (Wiley, 2013).
 16.
Friedman, J., Hastie, T. & Tibshirani, R. Additive logistic regression: a statistical view of boosting (With discussion and a rejoinder by the authors). The Annals of Statistics 28, 337–407 (2000).
 17.
Speliotes, E. K. et al. Association analyses of 249,796 individuals reveal 18 new loci associated with body mass index. Nature genetics 42, 937–948, https://doi.org/10.1038/ng.686 (2010).
 18.
Sudlow, C. et al. UK biobank: an open access resource for identifying the causes of a wide range of complex diseases of middle and old age. PLoS Med 12, e1001779, https://doi.org/10.1371/journal.pmed.1001779 (2015).
 19.
Berndt, S. I. et al. Genomewide metaanalysis identifies 11 new loci for anthropometric traits and provides insights into genetic architecture. Nature genetics 45, 501–512, https://doi.org/10.1038/ng.2606 (2013).
 20.
Lango Allen, H. et al. Hundreds of variants clustered in genomic loci and biological pathways affect human height. Nature 467, 832–838, https://doi.org/10.1038/nature09410 (2010).
 21.
Purcell, S. et al. PLINK: a tool set for wholegenome association and populationbased linkage analyses. Am J Hum Genet 81, 559–575, https://doi.org/10.1086/519795 (2007).
Acknowledgements
Guillaume Pare is supported by the Canada Research Chair in Genetic and Molecular Epidemiology and the CISCO Professorship in Integrated Health Systems. We are thankful to all the participants of the UK Biobank and Health Retirement Study, and to all the consortia for making their data publically available.
Author information
Affiliations
Contributions
G.P. designed the experiments; G.P. and W.Q.D. wrote the manuscript; S.M. analyzed the data and prepared the tables and figures; all authors reviewed the manuscript.
Corresponding author
Ethics declarations
Competing Interests
The authors are listed as inventors on patent disclosures owned by McMaster University and related to trait prediction using genetic data.
Additional information
Publisher's note: Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Electronic supplementary material
Rights and permissions
Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons license, and indicate if changes were made. The images or other third party material in this article are included in the article’s Creative Commons license, unless indicated otherwise in a credit line to the material. If material is not included in the article’s Creative Commons license and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this license, visit http://creativecommons.org/licenses/by/4.0/.
About this article
Cite this article
Paré, G., Mao, S. & Deng, W.Q. A machinelearning heuristic to improve gene score prediction of polygenic traits. Sci Rep 7, 12665 (2017). https://doi.org/10.1038/s41598017130561
Received:
Accepted:
Published:
Further reading

The emerging landscape of health research based on biobanks linked to electronic health records: Existing resources, statistical challenges, and potential opportunities
Statistics in Medicine (2020)

Estimating the heritability of psychological measures in the Human Connectome Project dataset
PLOS ONE (2020)

Unravelling the complex genetics of common kidney diseases: from variants to mechanisms
Nature Reviews Nephrology (2020)

A Polygenic Risk Score as a Risk Factor for Medication‐Associated Fractures
Journal of Bone and Mineral Research (2020)

Systematic Review of Polygenic Risk Scores for Type 1 and Type 2 Diabetes
International Journal of Molecular Sciences (2020)
Comments
By submitting a comment you agree to abide by our Terms and Community Guidelines. If you find something abusive or that does not comply with our terms or guidelines please flag it as inappropriate.