Predicting complex quantitative traits with Bayesian neural networks: a case study with Jersey cows and wheat

  • Daniel Gianola1, 2, 3,

    Affiliated with

    • Hayrettin Okut1, 4Email author,

      Affiliated with

      • Kent A Weigel2 and

        Affiliated with

        • Guilherme JM Rosa1, 3

          Affiliated with

          BMC Genetics201112:87

          DOI: 10.1186/1471-2156-12-87

          Received: 22 July 2011

          Accepted: 7 October 2011

          Published: 7 October 2011



          In the study of associations between genomic data and complex phenotypes there may be relationships that are not amenable to parametric statistical modeling. Such associations have been investigated mainly using single-marker and Bayesian linear regression models that differ in their distributions, but that assume additive inheritance while ignoring interactions and non-linearity. When interactions have been included in the model, their effects have entered linearly. There is a growing interest in non-parametric methods for predicting quantitative traits based on reproducing kernel Hilbert spaces regressions on markers and radial basis functions. Artificial neural networks (ANN) provide an alternative, because these act as universal approximators of complex functions and can capture non-linear relationships between predictors and responses, with the interplay among variables learned adaptively. ANNs are interesting candidates for analysis of traits affected by cryptic forms of gene action.


          We investigated various Bayesian ANN architectures using for predicting phenotypes in two data sets consisting of milk production in Jersey cows and yield of inbred lines of wheat. For the Jerseys, predictor variables were derived from pedigree and molecular marker (35,798 single nucleotide polymorphisms, SNPS) information on 297 individually cows. The wheat data represented 599 lines, each genotyped with 1,279 markers. The ability of predicting fat, milk and protein yield was low when using pedigrees, but it was better when SNPs were employed, irrespective of the ANN trained. Predictive ability was even better in wheat because the trait was a mean, as opposed to an individual phenotype in cows. Non-linear neural networks outperformed a linear model in predictive ability in both data sets, but more clearly in wheat.


          Results suggest that neural networks may be useful for predicting complex traits using high-dimensional genomic information, a situation where the number of unknowns exceeds sample size. ANNs can capture nonlinearities, adaptively. This may be useful when prediction of phenotypes is crucial.


          Challenges in the study of associations between genomic variables (e.g., molecular markers) and complex phenotypes include the possible existence of cryptic relationships that may not be amenable to parametric statistical modeling, as well as the high dimensionality of the data, illustrated by the growing number of single nucleotide polymorphisms, now close to 10 million in humans http://​www.​genome.​gov/​11511175. These associations have been investigated primarily using naïve single-marker regressions and, more recently, with Bayesian linear regression models of various types [13] but that assume additive inheritance almost invariably, while typically ignoring interactions and non-linearity. Taking into account these phenomena may enhance the ability of predicting outcomes, and this is relevant in genome-assisted management of livestock and plants and in individualized medicine.

          There has been a growing interest in the use of non-parametric methods for prediction of quantitative traits based on reproducing kernel Hilbert spaces regressions on markers [2, 47] and radial basis functions models [8] or related approaches [9]. Artificial neural networks (ANN) provide an interesting alternative because these learning machines can act as universal approximators of complex functions [10, 11]. ANNs can capture non-linear relationships between predictors and responses and learn about functional forms in an adaptive manner, because a series of transformations called activation functions are driven by parameters. ANNs can be viewed as a computer based system composed of many processing elements (neurons) operating in parallel [12], and also as a schematic of Kolmogorov's theorem for representation of multivariate functions [13]. An ANN is determined by the network structure, represented by the number of layers and of neurons, by the strength of the connections (akin to non-parametric regression coefficients) between inputs, neurons and outputs, and by the type of processing performed at each neuron, represented by a linear or non-linear transformation: the activation function. Neural networks have the potential of accommodating complex relationships between input and response variables, as well as of difficult to model interactions among inputs. For these reasons, ANNs are interesting candidates for the analysis of complex traits affected by cryptic forms of gene X gene interaction, and many algorithms for training (fitting) such networks are now available [14].

          In this study we investigated the performance of several ANN architectures using Bayesian regularization (a method for coping with the "small n, large p" problem that arises in statistical models including a massive number of explanatory variables) when predicting milk production traits in a sample of Jersey cows or mean grain yield in hundreds of inbred wheat lines. The architectures considered differed in terms of number of neurons and activation functions used, and the input (predictor) variables were derived from pedigree and molecular marker information on the corresponding samples. The paper begins with a brief account of Bayesian regularized neural networks, of their connection with linear random regression models often used in quantitative genetics, and of how Bayesian regularization is made. Subsequently, it is shown how a neural network treatment of genomic data can enhance predictive ability over and above that using pedigree information (in Jerseys) or linear Bayesian regression on markers (in both cows and wheat), which is representative of a standard approach in quantitative genomics.


          For clarity of presentation the methodology is presented first, as the main objective of the paper was to cast neural networks in a quantitative genetics predictive context. Subsequently, a description of the two sets of data used to illustrate how the Bayesian neural networks were run is provided. As stated, the first data set consisted of milk, protein and fat yield in dairy cows. The second set represented 599 lines of wheat, with mean grain yield as target trait.

          Excursus: Feed-Forward Neural Networks

          To illustrate, consider a network with three layers, as shown in Figure 1 for the Jersey cow data. In the left-most layer, there are input variables, 297 in Figure 1, or transformations thereof (called features) that enter into the network as predictors. In the middle ("hidden") layer there is a varying number of neurons; 5 are shown in Figure 1, but the number used is a model selection issue, with this addressed via an evaluation of predictive performance. In the right-most layer, there is a single ("output") node, at least for quantitative response variables. Each input (or feature) connects to each neuron with a strength represented by an unknown coefficient w. The collected input into a given neuron can be transformed (or not, in which case one speaks of an identity or linear activation function), and this activated net input is emitted to the output layer with a strength represented by another unknown coefficient. A similar process takes place for every neuron.

          Figure 1

          Illustration of the neural networks used. In the Jersey data there were 297 elements of pedigree or genomic relationship matrices used as inputs (the p's) for each target trait. In the Figure, each p k (k = 1,2,...,297) is connected to 5 hidden neurons via coefficients w j,k (j denotes neuron, k denotes input). Each hidden and output neuron has a bias parameter , j denotes neuron, l denotes layer). The variable represents the trait predicted value for datum i.

          Algebraically, the process can be represented as follows. Let t i (the target phenotype) be a quantitative trait measured in individual i (i = 1,2,...,n) and let p i = {p ij } be a vector of inputs or explanatory variables, e.g., marker genotypes or any other covariate measured in each of such individuals, with allowance made for inclusion of a 1, corresponding to the indicator variable for an intercept in a regression model. Suppose there are S neurons in the hidden layer of the architecture. The input into neuron k (k = 1,2,...,S) prior to activation, as described subsequently, is the linear function w' k p i , where w' k ={w kj } is a vector of unknown connection strengths ("regressions") peculiar to neuron k, including an intercept (called "bias" in the machine learning literature) in w' k . This input is transformed ("activated") using some linear or non-linear function f(.), which can be neuron-specific or common to all neurons; this yields f k (w' k p i ) (k = 1,2,...,S). Subsequently, the so activated emission from neuron k is sent to the output layer, with the collection of emissions over all neurons being , where b is an overall bias parameter, c is a regression on an activated emission, g(.) is another activation function, possibly non-linear, and w 1 ,w 2 ,...,w S are regressions on each of the activated emissions f k (w'k pi). The link between the response variable (phenotype) and the inputs is provided by the model


          where e i ~ (0, σ 2) and σ 2 is a variance parameter. If g(.) is a linear or identity activation function, the model is a linear regression on the adaptive covariates f k (w' k p i ); if, further, f k (.), is also linear, the regression model is entirely linear. The term "adaptive" means that the covariates are functions of unknown parameters, the {w kj } connection strengths, so the networks can "learn" the relationship between explanatory variables and phenotypes, as opposed to posing it arbitrarily, as it is the case in standard regression models. In this manner, this type of neural network can also be viewed as a regression model, but with the extent of non-linearity dictated by the type of activation functions used. Since the number of parameters increases linearly with the number of neurons, and the number of predictors given by the length of p (e.g., the number of markers) can amply exceed sample size, it is necessary to treat the connection strengths as random effects in which case the Bayesian connection is immediate [15, 16]. This approach is called "Bayesian regularization".

          Fisher's infinitesimal model viewed as a neural network

          Let t represent an n × 1 vector of phenotypic values and u ~ (0, A σ 2 u ) be a vector of infinitesimal additive genetic effects, where σ 2 u is the additive genetic variance, A = CC' = {a ij } is the numerator relationship matrix and C is its lower triangular Cholesky factor decomposition. Fisher's linear model on additive genetic effects (ignoring an overall mean and nuisance fixed effects, for simplicity) admits at least three representations:

          I) t = u + e = Cz σ u + e = Cu* + e,

          where z is a vector of independent standard normal deviates, u* = z σ u ~ (0, I σ 2 u ) and e ~ (0, I σ 2) is a residual vector with σ 2 interpretable as environmental variance.

          II) t = AA -1 u + e = Au** + e,

          where u** = A -1 u ~ (0, A -1 σ 2 u ), and

          III) t = A -1 Au + e = A -1 u*** + e,

          where u*** = Au ~ (0, A 3 σ 2 u).

          In each of these formulations Fisher's model can be viewed as a neural network with a single neuron in the middle layer, where g(.) is an identity or linear activation function. The respective representations for the three models given above are


          Here, a bias parameter b is included for the sake of generality. Hence, the additive model can be viewed as a single-neuron network regression on either elements of the Cholesky decomposition of the numerator relationship matrix, on the relationships themselves or on the elements of the inverse of A, with the strengths of the connections represented by the corresponding entries of u*, u** and u***, respectively.

          Is it possible to exploit knowledge of relationships in a fuller manner? Since a neural network is a universal approximator, the predictive performance of the classical infinitesimal linear model can be enhanced, at least potentially, by taking a model on, say, S neurons, while effecting non-linear transformations simultaneously. The rationale is that Fisher's model holds under some assumptions which may be violated, such as linkage equilibrium, e.g., entries of the numerator relationship matrix are expected values in the absence of selection and under linkage equilibrium. For instance, using the second representation above one could write


          Here, the inputs are entries a ij of the relationship matrix, connecting individual i to all other individuals in the genealogy; the coefficient is the connection strength for input j in neuron k; b k is the bias parameter associated with neuron k; g k is an activation function peculiar to neuron k; w k is the connection strength between the activated emission from neuron k and the output layer, b is the outer layer bias parameter and g(.) is the outer activation function, which may be linear or non-linear, although it is typically taken as linear for quantitative responses. The nonlinear transformations modify the connection strengths between additive relationships and phenotypes in an adaptive manner, underlining the potential for an improvement in predictive ability.

          Given the availability of dense markers in humans and animals, an alternative or complementary source of input that can be used in equation (2) consists of the elements of a marker-based relationship matrix, as in [17]; in this case the a ij coefficients are replaced by g ij , i.e., elements of some genome or marker-derived relationship matrix G. As noted by [2], when G is proportional to XX', where X is the incidence matrix of a linear regression model on markers, this is equivalent to Bayesian ridge regression. Of course, nothing precludes using both pedigree-derived and marker-derived inputs in the construction of a neural network.

          Bayesian regularization

          The objective in ANNs is to arrive at some configuration that fits the training data well but that it also has a reasonable ability of predicting yet to be seen observations. This can be achieved by placing constraints on the size of the network connection strengths, e.g., via shrinkage, and the process is known as regularization. A natural way of attaining this compromise between goodness of fit and predictive ability is by means of Bayesian methods [2, 11, 15, 18]. In this section, an approach used often for Bayesian regularization in neural networks [18, 19] is presented along the lines of the hierarchical models employed by quantitative geneticists [15].

          Conditionally on m network parameters, the n phenotypes or outputs (represented as D for data) are assumed to be mutually independent, with density function (inputs p are omitted in the notation)

          where N(.) denotes a normal density; b is the outer bias parameter; w denotes all connection strength coefficients (including all neuron-specific biases); σ 2 is the residual variance and M represents a given neural network architecture (i.e., a choice of number of neurons and activation functions). The mean of this distribution is the conditional (given all regression coefficients) expectation function , i = 1,2,...,n. The bias parameter b can be eliminated simply by taking deviations from the mean, or assigned a flat prior; for simplicity the first of the two options was employed in this study. The Bayesian approach used in regularized neural networks software (e.g., MATLAB) assigns the same normal prior distribution to each of the connection strengths, assumed independent a priori, such that p(w| σ 2 w ) = N(0, I σ 2 w ), where σ 2 w is the variance of connection strengths. More general specifications can be posed, but currently available software (public or commercially) lacks flexibility for doing so. Assuming that the two variance parameters are known, the posterior density of the connection strengths is

          where the denominator is the marginal density of the data, that is

          For a neural network with a least one non-linear activation function, the integral is expressible as


          which does not have closed form, because of non-linearity. Recall that b can be set to 0 provided the observations are suitably centered.

          Although a Bayesian neural network can be fitted using Markov chain Monte Carlo sampling, the computations are taxing because of the enormous non-linearities present coupled with the high-dimensionality of w, such as it is the case with genomic data. An alternative approach is based on computing conditional posterior modes of connection strengths, given some likelihood-based estimates of the variance parameters, i.e., as in best linear unbiased prediction (when viewed as a posterior mode) coupled with restricted maximum likelihood (where estimates of variances are the maximizers of a marginal likelihood). The conditional (given σ 2 and ) log-posterior density of w is from equation (4)

          Let and (a standard notation in neural networks literature), and


          and E w = w'w. It follows that maximizing is equivalent to minimizing F(α, β). This function is often referred to as a "penalized" sum of squares, and it embeds a compromise between goodness of fit, given by the sum of squares of the residuals E D , and the degree of model complexity, given by the sum of squares of the network weights E w . The value w = wMAP that maximizes is the mode of the conditional (given the variances) posterior density of the connection strengths; MAP stands for "maximum a posteriori".

          If the additive infinitesimal model is represented as a neural network, the coefficient of heritability is given by . As it can be seen in equation (6), if α<<β, the fitting or training algorithm places more weight on goodness of fit. If α>>β, the algorithm emphasizes reduction in the values of w (shrinkage), which produces a less wiggly output function [20].

          Given α and β, the w = wMAP estimates can be found via any non-linear maximization algorithm as in, e.g., the threshold and survival analysis models of quantitative genetics [21].

          Tuning parameters α and β

          A standard procedure used in neural networks (and in the software employed here) infers α and β by maximizing the marginal likelihood of the data in equation (5); this corresponds to what is often known as empirical Bayes. Because (5) does not have a closed form (except in linear neural networks), the marginal likelihood is approximated using a Laplacian integration done in the vicinity of the current value w = wMAP, which depends in turn on the values of the tuning parameters at which the expansion is made. This type of approach for non-linear mixed models has been used in animal breeding for almost two decades [22].

          The Laplacian approximation to the marginal density in equation (5) leads to the representation

          where K is a constant and is the Hessian matrix. A grid search can be used to locate the α, β maximizers of the marginal likelihood in the training set. An alternative approach described by [18, 23] involves the iteration (updating is from right to left, with w MAP evaluated at the "old" values of the tuning parameters)


          These expressions, as well as (7), are similar to those that arise in maximum likelihood estimation of variance components [2426], which vary depending on the algorithm used. Since β is a positive parameter, it must be that . The quantity is called "effective number of parameters" in the network [20] and its value ranges from 0 (or 1, if an overall intercept b is fitted) to m, the total number of connection coefficients and bias parameters in the network. If γ is close to n over-fitting results, leading to poor generalization. It follows that the computations are similar to those used in the linear and non-linear models employed by quantitative geneticists, the salient aspect being that a neural network can be strongly non-linear.

          More details on computing procedures for neural networks are in [12, 14, 18, 23, 27, 28]. Typically, an algorithm proceeds as follows: 1) initialize α, β and all elements in w. 2) Take one step of the Levenberg-Marquardt algorithm to minimize the objective function F(α, β) and find the current value of w. 3) Compute γ, the effective number of parameters, using the Gauss-Newton approximation to the Hessian matrix in the Levenberg-Marquardt training algorithm. 4) Compute updates α new and β new ; and 5) iterate steps 2-4 until convergence.

          Neural Network Architectures Evaluated and Implementation

          A prototype of the networks considered is in Figure 1; as already noted, the architecture shown has five neurons in the hidden (middle) layer. The ANN examined had from 1 through 6 neurons in the hidden layer. In architectures with a single neuron, two variants were considered. In one, the activation functions were linear (identity) throughout. In this case, e.g., when additive relationships a ij are used as inputs, the network becomes a random regression on such relationships. If regularization were based on as prior distribution, this would yield the standard additive ("animal") model of quantitative genetics; this was not the case here because the MATLAB software used (see below) bases regularization on . The second single-neuron architecture was based on equation (1) with a single outer bias parameter but with a non-linear activation g of the emission made by the sole neuron in the architecture. The algebraic representation of this network is

          where c is the regression of t i on the activated emission . The objective here was to explore non-linearities between the inputs (additive or genomic relationships in the Jersey data, or markers genotypes in the wheat data) and the targets (phenotypes); the standard additive genetic model assumes that these relationships are linear. The activation function chosen was the hyperbolic tangent transformation , where ; here, x can take any value in the real line and g(x i ) is the neuron emission for cow or wheat line i, which takes values between -1 and 1. Given the inputs, the predicted phenotype or network output is

          In models with 2-6 neurons the emissions were always assigned a hyperbolic tangent activation (the choice of function can be based on, e.g., cross-validation); these activations were summed and collected linearly as shown in Figure 1. For example, with 2 neurons the predictions are obtained as

          where the coefficients are the estimated linear regressions of the traits on the activated emissions fired by each of the two neurons.

          MATLAB's neural networks toolbox [29] was used for fitting the architectures studied, using Bayesian regularization in all cases. As mentioned earlier, two combinations of activation functions were tried: 1) the hyperbolic tangent sigmoid function for activating emissions from each neuron in the hidden layer, plus a linear activation function from the hidden to the output layer, and 2) a linear activation throughout, this corresponding to a linear model with random regression coefficients. To avoid spurious effects caused by starting values in each iterative sequence, the networks were trained 20 times in the Jersey data and 50 times in the wheat data set, for each of the architectures. In Jerseys, each run randomly allocated 60% of the animals to a training set, 20% to a validation set and 20% to a testing set; results reported are the average of the 20 runs for each of the configurations. In wheat, the records were randomly partitioned into a training (480 lines) and a testing (119 lines) set. Each of the 50 random repeats matched exactly those of [28], to provide a comparison with the predictive ability of the Bayesian Lasso and of support vector regression models used with the wheat data set by [30].

          The neural networks were fitted to data in the training set, with the α and β parameters, connection strengths and biases modified iteratively. In the Jersey data, as parameters changed in the course of training, the predictive ability of the network was gauged in parallel in the validation set, which was expected to be similar in structure to the testing set, because they were randomly constructed. The same was done with the wheat data, except that there was no "intermediate" validation set. Once the mean squared error of prediction reached an optimal level, training stopped, and this led to the best estimates of the network coefficients. This estimated network was then used for predicting the testing set; predictive correlations (Pearson) and mean-squared errors were evaluated.

          Before processing, MATLAB rescales all input and output variables such that they reside in the [-1, +1] range, to enhance numerical stability; this is done automatically using the "mapminmax" function. To illustrate, consider the vector x' = [3, 6, 4] so that x min = 3 and x max = 6. If values are to range between A min = -1 and A max = +1, one sets temporarily x temp ' = [-1,1,4], so only x 3 = 4 needs to be rescaled. This is done via the formula

          MATLAB uses the Levenberg-Marquardt algorithm (based on linearization) for computing the posterior modes in Bayesian regularization, and back-propagation is employed to minimize the penalized residual sum of squares. The maximum number of iterations (called epochs) in back-propagation was set to 1000, and iteration stopped earlier if the gradient of the objective function was below a suitable level or when there were obvious problems with the algorithm [28, 29, 31]. Each of these settings corresponds to the default values provided by MATLAB.

          Jersey cows data

          Because of the high-dimensionality of the genotypic data, the neural networks used either additive or genome-derived relationships among cows as inputs (instead of SNP genotype codes), to make computations feasible in MATLAB. The rationale for this is based on the representation of the infinitesimal model as a regression on a pedigree, or as a regression on a matrix that is proportional to genomic relationships, as argued by [2] in the context of reproducing kernel Hilbert spaces regression. The neural networks had the form


          where p ij = a ij (additive relationship between cows i and j) or g ij (genome-derived relationships). Thus, for each cow the input vector p i had order 297 × 1.

          The expected additive genetic relationship matrix, A = {a ij }, was developed from the pedigree information; this is a standard metric for degree of kinship used in quantitative genetics. A realized genomic relationship matrix, G = {g ij }, was constructed from the marker data following [18], and calculated as follows: 1) estimate marker allelic frequencies and let μ i be the estimated frequency of allele "A" at locus i. 2) Construct a 297 × 35,798 matrix of marker genotype codes M, with typical element m ij corresponding to the genotype of individual i for marker j. 3) Calculate the expected frequency of m ij under Hardy-Weiberg equilibrium from the estimates of the allelic frequencies, and form the 297 × 35,798 matrix of expectations E. 4) Form the estimated genomic relationship matrix (assuming linkage equilibrium among markers) as

          The matrix Z = M-E contains "centered" codes, such that the mean of the values in any of its columns is null; Z can be used as in incidence matrix in marker assisted regression models [17, 32, 33]. Then is interpretable as a covariance matrix, analogous to in the infinitesimal model. The term holds under linkage equilibrium only, and cannot be construed as additive genetic variance of marker effects in the classical sense of [33]; its relationship to additive genetic variance in a finite locus or infinitesimal model is tenuous [16, 34].

          Wheat lines data

          There were 599 wheat lines, each genotyped with 1279 DArT markers (Diversity Array Technology) generated by Triticarte Pty. Ltd. (Canberra, Australia; http://​www.​triticarte.​com.​au). The DArT markers may take on two values, denoted by their presence or absence. In this data set, the overall mean frequency of the allele coded as "1" was 0.5607, with a minimum of 0.0083 and a maximum of 0.9866. Markers with a minor allele frequency lower than 0.05 were removed. Missing genotypes at locus j of line i were imputed using samples from the marginal distribution of marker genotypes, that is, , where is the estimated allele frequency computed from the non-missing genotypes [34]. The phenotype studied was average grain yield of each line. The data came from the International Maize and Wheat improvement Center, Mexico, and it can be downloaded from R package BLR http://​cran.​r-project.​org/​web/​packages/​BLR/​index.​html; more information can be found in [30, 35, 36]. The wheat data was partitioned randomly into a training set (480 lines) and a test set (119 lines), exactly as in [30].


          Degree of complexity

          The effective number of parameters (γ) associated with each of the networks examined in the Jersey data is presented in Table 1 and shown graphically in Figure 2, by trait and type of input considered, i.e., additive or genomic relationships. Clearly, use of genomic relationships resulted in a larger number of effective parameters than use of pedigree, for all traits and architectures. When using pedigree relationships, the average (over runs, but note the large standard errors) effective number of parameters ranged from 91 (fat yield, one-neuron model with non-linear activation), to 136 (protein yield, 6 neurons). This illustrates the impact of shrinkage, and of how regularized neural networks cope with the "curse of dimensionality. For example, a 6-neuron network has close to 1800 nominal parameters. Likewise, when using genomic relationships as inputs, the average effective number of parameters ranged from 127 to 166 (fat yield). Similar results were obtained in the wheat data (Table 2). The effective number of parameters ranged from 220 (nonlinear ANN with 4 neurons) to 299 (linear ANN).
          Table 1

          Effective number of parameters (± standard errors), by trait, in Jerseys.1


          Fat yield


          Fat yield


          Milk yield


          Milk yield


          Protein yield


          Protein yield



          123 ± 5.6

          166 ± 2.0

          124 ± 7.6

          162 ± 2.9

          118 ± 8.5

          151 ± 4.5

          1 neuron

          91 ± 4.9

          142 ± 2.0

          93 ± 5.8

          166 ± 2.0

          91 ± 10.3

          144 ± 2.5

          2 neurons

          104 ± 5.8

          128 ± 7.6

          122 ± 6.5

          145 ± 7.8

          114 ± 8.0

          136 ± 8.0

          3 neurons

          107 ± 5.8

          132 ± 5.7

          123 ± 5.1

          129 ± 6.0

          126 ± 6.9

          141 ± 4.9

          4 neurons

          108 ± 5.8

          129 ± 4.7

          112 ± 4.7

          131 ± 5.8

          129 ± 5.4

          138 ± 6.0

          5 neurons

          106 ± 4.9

          127 ± 4.9

          118 ± 4.8

          132 ± 5.4

          131 ± 4.9

          138 ± 5.6

          6 neurons

          114 ± 3.3

          128 ± 7.5

          122 ± 5.1

          132 ± 5.6

          136 ± 4.6

          137 ± 5.0

          1 Results are averages of 20 runs based on random partitions of the data

          Figure 2

          Effective number of parameters obtained from different network architectures in the Jersey data. Results shown are averages of 20 independent runs. "Linear" denotes a 1-neuron model with linear activation functions throughout.

          Table 2

          Effective number of parameters, predictive correlations, and mean squared errors of prediction: wheat.1

          ANN architectures


          1 neuron

          2 neurons

          3 neurons

          4 neurons



          Effective number of parameters

          299 ± 5.5

          260 ± 6.1

          253 ± 5.9

          238 ± 5.5

          220 ± 2.8

          Correlations in testing set

          0.48 ± 0.03

          0.54 ± 0.03

          056 ± 0.02

          0.57 ± 0.02

          0.59 ± 0.02

          Mean squared error in testing set

          0.99 ± 0.04

          0.77 ± 0.03

          0.74 ± 0.03

          0.71 ± 0.02

          0.72 ± 0.02

          1 The training-test partitions for this data were random and repeated 50 times; standard errors in parentheses

          The effective number of parameters behaved differentially with respect to model architecture and this depended on the input variables used. When using pedigrees in the Jersey data, the hyperbolic tangent activation function in the 1-neuron model reduced γ drastically, relative to the linear model (1 neuron with linear activation throughout). Then, an increment in number of neurons from 2 to 6 increased model complexity relative to that of the 1 neuron model with non-linear activation, but not beyond that attained with the linear model, save for protein yield. For this trait, γ was 118 for the linear model, and ranged from 126 to 136 in models with 3 through 6 neurons. When genomic relationships were used as inputs, γ was largest for the linear model for fat and protein yield, and for the 1-neuron model with a non-linear activation function in the case of milk yield. In wheat, the effective number of parameters decreased as architectures became more complex. There was large variation among runs in effective number of parameters for both data sets, but there was not a clear pattern in the variability.

          Predictive performance

          Results pertaining to predictive ability evaluated in the testing sets are shown in Table 2 for wheat and Tables 3 and 4 for the Jersey data. Figures 3 and 4 depict mean of squared errors of prediction and correlations coefficients in the Jersey cows.
          Table 3

          Prediction mean squared errors (± standard errors) by trait: Jerseys.1


          Fat yield (pedigree)

          Fat yield (genomic)

          Milk yield (pedigree)

          Milk yield (genomic)

          Protein yield (pedigree)

          Protein yield (genomic)


          1.19 ± 0.07

          0.86 ± 0.05

          1.09 ± 0.05

          0.88 ± 0.04

          1.00 ± 0.04

          0.75 ± 0.07

          1 neuron

          1.01 ± 0.04

          0.74 ± 0.03

          0.99 ± 0.04

          0.81 ± 0.03

          0.97 ± 0.04

          0.71 ± 0.04

          2 neurons

          0.93 ± 0.05

          0.70 ± 0.03

          0.96 ± 0.05

          0.76 ± 0.04

          1.02 ± 0.04

          0.72 ± 0.04

          3 neurons

          0.92 ± 0.04

          0.71 ± 0.03

          0.98 ± 0.02

          0.78 ± 0.04

          0.96 ± 0.06

          0.80 ± 0.04

          4 neurons

          0.99 ± 0.04

          0.84 ± 0.04

          0.98 ± 0.04

          0.72 ± 0.04

          0.90 ± 0.06

          0.70 ± 0.03

          5 neurons

          0.99 ± 0.04

          0.86 ± 0.04

          1.00 ± 0.05

          0.80 ± 0.04

          0.93 ± 0.04

          0.77 ± 0.04

          6 neurons

          0.95 ± 0.03

          0.77 ± 0.04

          1.02 ± 0.05

          0.79 ± 0.03

          0.95 ± 0.03

          0.76 ± 0.05

          1Results are the average of 20 runs based on random partitions on the data

          Table 4

          Correlation coefficients (± standard errors) in the Jersey testing data set, by trait.1


          Pedigree relationships

          Genomic relationships


          Fat yield

          Milk yield

          Protein yield

          Fat yield

          Milk yield

          Protein yield


          0.11 ± 0.04

          0.07 ± 0.03

          0.02 ± 0.02

          0.43 ± 0.02

          0.42 ± 0.03

          0.44 ± 0.02

          1 neuron

          0.23 ± 0.03

          0.10 ± 0.03

          0.09 ± 0.02

          0.51 ± 0.02

          0.45 ± 0.02

          0.44 ± 0.02

          2 neurons

          0.22 ± 0.03

          0.08 ± 0.01

          0.08 ± 0.03

          0.49 ± 0.02

          0.46 ± 0.03

          0.51 ± 0.02

          3 neurons

          0.22 ± 0.02

          0.13 ± 0.02

          0.10 ± 0.03

          0.53 ± 0.02

          0.52 ± 0.02

          0.47 ± 0.02

          4 neurons

          0.20 ± 0.02

          0.09 ± 0.02

          0.14 ± 0.02

          0.45 ± 0.03

          0.52 ± 0.02

          0.47 ± 0.03

          5 neurons

          0.23 ± 0.02

          0.13 ± 0.02

          0.15 ± 0.02

          0.42 ± 0.03

          0.50 ± 0.02

          0.47 ± 0.02

          6 neurons

          0.27 ± 0.02

          0.10 ± 0.03

          0.11 ± 0.02

          0.48 ± 0.04

          0.54 ± 0.02

          0.50 ± 0.03

          1Results are the average of 20 runs based on random partitions on the data

          Figure 3

          Prediction mean squared errors in the Jersey testing set (vertical axis) by network. Results are averages of 20 independent runs. "Linear" denotes a 1-neuron model with linear activation functions throughout.

          Figure 4

          Correlations between predictions and observations in the Jersey testing data set for the network considered. Results shown are averages of 20 independent runs. "Linear" denotes a 1-neuron model with linear activation functions throughout.

          The predictive correlations in wheat (Table 2) ranged from 0.48 with the linear ANN (equivalent to Bayesian ridge regression) to 0.59 for the nonlinear ANN with 4 neurons. Clear and significant differences between linear and nonlinear architectures were observed. The improvements over the linear ANN were 11.2, 14.3, 15.8 and 18.6% in predictive correlation for 1, 2, 3 and 4 neurons in the hidden layer, respectively. Mean squared errors were also 23-29% smaller than in the linear ANN.

          In the Jerseys, the large variability in mean squared error among runs (Table 3) precludes making strong statements about differences among architectures. However, predictive correlations (Table 4) were clearly larger for the non-linear ANN. For fat yield, the results with pedigrees employed as input suggest that a non-linear, adaptive use of additive relationships (as done in all networks with the hyperbolic tangent activation function) can improve predictive performance beyond that of the infinitesimal model. Further, use of genomic relationships led to more reliable prediction of phenotypes than use of pedigree information as measured by the predictive correlations in Table 4. The relative increase in strength of association, as measured by the correlation, is much larger than the ones that have been reported, e.g., in dairy cattle [32, 37], when prediction of breeding values of bulls was made from genomic information, as opposed to from pedigrees. Our result is encouraging and suggests that genomic data may also play an important role in prediction of individual outcomes (as opposed to breeding value), an issue of relevance in medicine [4].


          The distribution of connection strengths in a network gives an indication of the extent of regularization attained. Typically, weight values decrease with model complexity, in the same manner that estimates of marker effects become smaller in Bayesian regression models when p increases and training sample size is kept constant. Further, the distribution of weights is often linked to predictive ability; small values tend to lead to better generalization. Figure 5 depicts the distributions of weights for the linear models and for the nonlinear regularized networks that produced the largest predictive correlations for pedigree and genomic relationships in the Jersey data. The weights for the linear model were larger and more variable than for the nonlinear networks, where distributions were patently leptokurtic, indicating strong shrinkage towards 0. For example, the average (over runs) sum of squares of weights for the linear model and for the non-linear network with 6 neurons when using genomic relationships as predictors of milk yield were 7.5 and 8.5, respectively; however, the 7.5 for the linear model was the sum of squares of 297 weights whereas the 8.5 for the nonlinear model with 6 neurons was the sum of squares of 1782 weights (6 × 297). The same picture was observed in the wheat data (results are not reported).

          Figure 5

          Distribution of connection strengths(w kj ) in the linear and selected networks fitted to the Jersey data. The linear model has single neuron architecture with linear activation functions. a) Fat yield using pedigree relationships: linear model (above) and 6 neurons (below). b) Milk yield using pedigree relationships: linear model (above) and 6 neurons (below). c) Protein yield using pedigree relationships: linear model (above) and 5 neurons (below). d) Fat yield using genomic relationships: linear model (above) and 3 neurons (below), e) Milk yield using genomic relationships: linear model (above) and (below) and 6 neurons (below). f) Protein yield using genomic relationships: linear model (above) and 2 neurons (below).


          Models for prediction of fat, milk and protein yield in cows using pedigree and genomic relationship information as inputs, and wheat yield using molecular markers as predictor variables were studied. This was done using Bayesian regularized neural networks, and predictions were benchmarked against those from a linear neural network, which is a Bayesian ridge regression model. In the wheat data, the comparison was supplemented with results obtained by our group using RKHS or support vector methods. Different network architectures were explored by varying the number of neurons, and using a hyperbolic tangent sigmoid activation function in the hidden layer plus a linear activation function in the output layer. This combination has been shown to work well when extrapolating beyond the range of the training data [36]. The choice of number of neurons can be based on cross-validation, as in the present data, or on standard Bayesian metrics for model comparison [11, 15].

          The Levenberg-Marquardt algorithm, as implemented in MATLAB, was adopted to optimize weights and biases, as this procedure has been effective elsewhere [38]. Bayesian regularization was adopted to avoid over-fitting and to improve generalization, and cross-validation was used to assess predictive ability, as in [28, 39].

          Because Bayesian neural networks reduce the effective number of weights relative to what would be obtained without regularization, this helps to prevent over-fitting [40]. For the networks we examined, even though the total number of parameters, e.g., in Jerseys, ranged from 300 to 1795, the effective number of parameters varied from only 91 to 136 when pedigree relationships were used, and from 127 to 166 when genomic relationships were used as inputs, illustrating the extent of regularization. There were differences in predictive abilities of different architectures but the small sample used dictates a cautious interpretation. Nevertheless, the results seem to support networks with at least 2 neurons, which has been observed in several studies [20, 28, 4143]. This suggests that linear models based on pedigree or on genomic relationships may not provide an adequate approximation to genetic signals resulting from complex genetic systems. Because highly parameterized models are penalized in the Bayesian approach, we were able to explore complex architectures. However, there was evidence of over-fitting in the Jersey training set, where correlations between observed and predicted data in the training set were always larger than 0.90, sometimes exceeding 0.95. This explains why correlations were much lower in the testing set, which is consistent with what was observed in other studies with neural networks [42]. Although more parameters in a model can lead to smaller error in the training data, it cannot be overemphasized that this is not representative of prediction error in an independent data set, as shown by [43] working with human stature.

          Our results with ANN for wheat are at least as good as those obtained with the same data in two other studies. Crossa et al. [35] found cross-validation correlations with the following values when various methods were used: pedigree information (BLUP), 0.45; pedigree-based reproducing kernel Hilbert spaces regression (RKHS), 0.60; RKHS with both pedigree and markers, 0.61; Bayesian Lasso with markers, 0.46; Bayesian Lasso with markers and pedigree, 0.54, and Bayesian ridge regression on markers, 0.49. Long et al., [30] compared the Bayesian Lasso with four support-vector regression models consisting of the combination of two kernels and two loss functions. The predictive correlation for wheat yield (average of 50 random repeats of the cross-validation exercise) was 0.52 for the Bayesian Lasso, and ranged between 0.50 and 0.58 for the support vector implementations. Hence, it seems clear, at least for wheat yield in this data set, that the non-parametric methods can outperform a strong learner, the Bayesian Lasso, and that the neural networks are competitive with the highly regarded support vector methods [11].

          A question of importance in animal and plant breeding is how an estimated "breeding value", i.e., an estimate of the total additive genetic effect of an individual, can be arrived at from an ANN output. There are at least two ways in which this can be done. One is by posing architectures with a neuron in which the inputs enter in a strictly linear manner, followed by a linear activation function on this neuron; the remaining neurons in the architecture, receiving the same inputs, would be treated non-linearly. A second one, is obtained by observing that the infinitesimal model can be written as yi = z' i u + e, for some incidence row vector z' i peculiar to individual i. Here, the breeding value of the i th individual can be written as . Likewise, consider a linear regression model for p markers, , where β j is the substitution effect at marker locus j; x ij = 0,1,2 is the observed number of copies of a given allele at locus j on individual I, and and β = {β j } are row and column vectors, respectively, each with p elements. Here, the "marked breeding value" of individual i would be . Consider next a neural network with a hyperbolic tangent activation function throughout, that is

          Let p i = {p ij } be the vector of input covariates (e.g., genomic or additive relationships, marker genotype codes) observed on i. Adapting the preceding definitions to the ANN specification, one would have as breeding value (BV) of individual i




          Thus, the so defined breeding value of individual i depends on the values of the input covariates observed on this individual, on all connection strengths and bias parameters from inputs to neurons in the middle layer (the u's and the b's), and on all connection strengths from the middle layer to the output layer (the w's). In order to obtain an estimate of breeding value the unknown quantities would replaced by the corresponding maximum a posteriori (MAP) estimates or, say, by the estimate of their posterior expectation if a Markov chain Monte Carlo scheme is applied [44].

          Another issue is that of assessing the importance of an input relative to that of others. For example, in a linear regression model on markers, one could use a point estimate of the substitution effect or its "studentized" value (i.e., the point estimate divided by the corresponding posterior standard deviation), or some measure that involves estimates of substitution effects and of allelic frequencies. A discussion of some measures of relative importance of inputs in an ANN is in [28, 43], for example, the ratio between the absolute value of the estimate of a given connection strength, and the sum of the absolute values of all coefficients in the network.


          Non-linear neural networks tended to outperform benchmark linear models in predictive ability, and clearly so in the wheat data. Bayesian regularization allowed estimation of all connection strengths even when n<<p, and the effective number of parameters was much smaller than the corresponding nominal number. Although the study was based on small samples, and the differences found may be reflective of random variation, especially in the Jersey data, our results suggest that the neural networks may be useful for predicting complex traits using high-dimensional genomic information, a situation where the number of coefficients that need to be estimated exceeds sample size. Neural networks have the ability of capturing nonlinearities, and do so adaptively, which may be useful in the study of quantitative traits under complex gene action, and particularly when prediction of outcomes is crucial, such as in personalized medicine.

          In summary, predictive ability seemed to be enhanced by use of Bayesian neural networks. Due to small sample sizes no claim is made about superiority of any specific non-linear architecture. As it has been observed in many studies, the superiority of one predictive model over another depends on the species, trait and environment, and the same will surely hold for ANNs.



          artificial neural network


          Bayesian regularization


          Bayesian regularization artificial neural network


          Least absolute shrinkage and selection operator


          Maximum a posterior


          Neural network


          Reproducing kernel Hilbert spaces regression


          Single nucleotide polymorphism


          Sum of squared error.



          Research was supported by the Wisconsin Agriculture Experiment Station and by grants from Aviagen, Ltd., Newbridge, Scotland, and Igenity/Merial, Duluth, Georgia, USA.

          Authors’ Affiliations

          Dept. of Animal Sciences, University of Wisconsin
          Dept. of Dairy Science, University of Wisconsin
          Dept. of Biostatistics and Medical Informatics, University of Wisconsin
          Dept. of Animal Sciences, Biometry and Genetics Branch, University of Yuzuncu Yil


          1. Meuwissen THE, Hayes BJ, Goddard ME: Prediction of total genetic value using genome-wide dense marker maps. Genetics 2001, 157:1819–1829.PubMed
          2. de los Campos G, Gianola D, Rosa GJM: Reproducing kernel Hilbert spaces regression: A general framework for genetic evaluation. Journal of Animal Science 2009, 87:1883–1887.PubMedView Article
          3. de los Campos G, Gianola D, Allison DB: Predicting genetic predisposition in humans: the promise of whole genome markers. Nature Reviews Genetics 2010, 11:880–886.PubMedView Article
          4. de los Campos G, Gianola D, Rosa GJM, Weigel KA, Crossa J: Semi-parametric genomic-enabled prediction of genetic values using reproducing kernel Hilbert spaces methods. Genetics Research 2010, 92:295–308.PubMedView Article
          5. Gianola D, Fernando RL, Stella A: Genomic assisted prediction of genetic value with semi-parametric procedures. Genetics 2006, 173:1761–1776.PubMedView Article
          6. Gianola D, van Kaam JBCHM: Reproducing kernel hilbert spaces regression methods for genomic assisted prediction of quantitative traits. Genetics 2008, 178:2289–2303.PubMedView Article
          7. Gianola D, de los Campos G: Inferring genetic values for quantitative traits non-parametrically. Genetics Research 2008, 90:525–540.PubMedView Article
          8. Long N, Gianola D, Rosa GMJ, Weigel KA, Kranis A, González-Recio O: Radial basis function regression methods for predicting quantitative traits using SNP markers. Genetics Research 2010,92(3):209–25.PubMedView Article
          9. Ober U, Erbe M, Long N, Porcu E, Schlather M, Simianer H: Predicting genetic values: a kernel-based best linear unbiased prediction with genomic data. Genetics 2011,188(3):695–708.PubMedView Article
          10. Alados I, Mellado JA, Ramos F, Alados-Arboledas L: Estimating UV Erythema1 irradiance by means of neural networks. Photochemistry and Photobiology 2004, 80:351–358.PubMedView Article
          11. Bishop CM: Pattern Recognition and Machine Learning. Singapore: Springer; 2006.
          12. Lamontagne L, Marchand M: Advances in Artificial Intelligence. Berlin: Springer; 2006.View Article
          13. Pereira BDB, Rao CR: Data Mining using Neural Networks: A Guide for Statisticians. [http://​www.​po.​ufrj.​br/​basilio/​publicacoes/​livros/​2009_​datamining_​using_​neural_​networks.​pdf] 2009.
          14. Lampinen J, Vehtari A: Bayesian approach for neural networks review and case studies. Neural Networks 2001, 14:257–274.PubMedView Article
          15. Sorensen D, Gianola D: Likelihood, Bayesian and MCMC methods in quantitative genetics. New York: Springer; 2002.
          16. Gianola D, de los Campos G, Hill WG, Manfredi E, Fernando R: Additive genetic variability and the Bayesian alphabet. Genetics 2009, 183:347–363.PubMedView Article
          17. Van Raden PM: Efficient methods to compute genomic predictions. J Dairy Sci 2008, 91:4414–4423.View Article
          18. MacKay DJC: Baysian interpolation. Neural Computation 1992, 4:415–447.View Article
          19. Titterington DM: Bayesian methods for neural networks and related models. Statistical Science 2004, 19:128–139.View Article
          20. Foresee FD, Hagan MT: Gauss-Newton approximation to Bayesian learning. Proc IEEE Int Conf Neural Networks 1997, 1930–1935.
          21. Gianola D: Inferences from mixed models in quantitative genetics. In Handbook of Statistical Genetics. Third edition. Edited by: Balding DJ, Bishop M, Cannings C. West Sussex UK: John Wiley & Sons; 2007.
          22. Tempelman RJ, Gianola D: Marginal maximum likelihood estimation of variance components in Poisson mixed models using Laplace integration. Genetics, Selection, Evolution 1993, 25:305–319.View Article
          23. Xu M, Zengi G, Xu X, Huang G, Jiang R, Sun W: Application of Bayesian regularized BP neural network model for trend analysis, acidity and chemical composition of precipitation in North. Water, Air, and Soil Pollution 2006, 172:167–184.View Article
          24. Smith SP, Graser HU: Estimating variance components in a class of mixed models by restricted maximum likelihood. J Dairy Sci 1986, 69:1165.
          25. Graser HU, Smith SP, Tier B: A derivative-free approach for estimating variance components in animal models by restricted maximum likelihood. J Anim Sci 1987, 64:1362.
          26. Hassami M, Anctil F, Viau AA: Selection of an artificial neural network model for the post-calibration of weather radar rainfall estimation. Journal of Data Science 2004, 220:107–124.
          27. MacKay JCD: Information Theory, Inference and Learning Algorithms. Cambridge; Cambridge University Press; 2008.
          28. Okut H, Gianola D, Rosa GJM, Weigel KA: Prediction of body mass index in mice using dense molecular markers and a regularized neural network. Genetics Research 2011, 93:189–201.View Article
          29. Beal MH, Hagan MT, Demuth HB: Neural Network Toolbox' 6 User's Guide. The MathWorks, Inc; 2010.
          30. Long N, Gianola D, Rosa GJM, Weigel KA: Application of support vector regressions to genome-assisted prediction of quantitative traits. Theoretical and Applied Genetics 2011. (under review)
          31. Haykin S: Neural Networks: Comprehensive Foundation. New York USA: Prentice-Hall; 2008.
          32. Habier D, Fernando RL, Dekkers JCM: The impact of genetic relationship information on genome-assisted breeding values. Genetics 2007,177(4):2389–2397.PubMed
          33. Van Raden PM, Tooker ME, Cole JB: Can you believe those genomic evaluations for young bulls? Journal of Dairy Science 2009,92(E-Suppl 1):314.
          34. Falconer DS, McKay TFC: Introduction to Quantitative Genetics. Malaysia: Longmans Green; 1996.
          35. Crossa J, de los Campos G, Perez P, Gianola D, Dreisigacker S, Burgueño J, Araus JL, Makumb D, Yan J, Singh R, Arief V, Banzinger M, Braun HJ: Prediction of genetic values for quantitative traits in plant breeding using pedigree and molecular markers. Genetics 2010, 186:713–724.PubMedView Article
          36. Perez P, de los Campos G, Crossa J, Gianola D: Genomic-enabled prediction based on molecular markers and pedigree using the Bayesian Linear Regression package in R. The Plant Genome 2010, 3:106–116.PubMedView Article
          37. Hayes BJ, Bowman BJ, Chamberlain AJ, Goddard ME: Invited review: Genomic selection in dairy cattle: Progress and challenges. J Dairy Sci 2009, 92:433–443.PubMedView Article
          38. Maier HR, Dandy CG: Neural networks for the prediction and forecasting of water resources variables: a review of modelling issues and applications. Environmental Modelling & Software 2000, 15:101–124.View Article
          39. Demuth H, Beale M, Hagan M: Neural Network Toolbox™ 6 User's Guide. The MathWorks, Inc; 2009.
          40. Fernandez M, Caballero J: Ensembles of Bayesian-regularized genetic neural networks for modeling of acetylcholinesterase inhibition by huprines. Chem Biol Drug Des 2006, 68:201–212.PubMedView Article
          41. Winkler DA, Burden FR: Modelling blood-brain barrier partitioning using Bayesian neural nets. Journal of Molecular Graphics and Modelling 2004, 22:499–505.PubMedView Article
          42. Joseph H, Huang WL, Dickman M: Neural network modelling of coastal algal blooms. Ecol Model 2003, 159:179–201.View Article
          43. Sorich MJ, Miners JO, Ross AM, Winker DA, Burden FR, Smith PA: Comparison of linear and nonlinear classification algorithms for the prediction of drug and chemical metabolism by human UDP-Glucuronosyltransferase isoforms. J Chem Inf Comput Sci 2003, 43:2019–2024.PubMedView Article
          44. Makowski R, Pajewski NM, Klimentidis YC, Vazquez AI, Duarte CW, Allison DA, de los Campos G: Beyond missing heritability: prediction of complex traits. PLOS Genetics 2011, 7:1–9.


          © Gianola et al; licensee BioMed Central Ltd. 2011

          This is an Open Access article distributed under the terms of the Creative Commons Attribution License (http://​creativecommons.​org/​licenses/​by/​2.​0), which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.