Linear regression

Gene

Elastic net regularization

Biology

Statistics

Support vector machine

Trait

Selection (genetic algorithm)

Computer science

Single-nucleotide polymorphism

Genetics

Regression

Mathematics

Genotype

Feature selection

Artificial intelligence

Best linear unbiased prediction

Programming language

Genomic selection

Machine learning

Frontiers in Genetics

Machine learning (ML) is perhaps the most useful tool for the interpretation of large genomic datasets. However, the performance of a single machine learning method in genomic selection (GS) is currently unsatisfactory. To improve the genomic predictions, we constructed a stacking ensemble learning framework (SELF), integrating three machine learning methods, to predict genomic estimated breeding values (GEBVs). The present study evaluated the prediction ability of SELF by analyzing three real datasets, with different genetic architecture; comparing the prediction accuracy of SELF, base learners, genomic best linear unbiased prediction (GBLUP) and BayesB. For each trait, SELF performed better than base learners, which included support vector regression (SVR), kernel ridge regression (KRR) and elastic net (ENET). The prediction accuracy of SELF was, on average, 7.70% higher than GBLUP in three datasets. Except for the milk fat percentage (MFP) traits, of the German Holstein dairy cattle dataset, SELF was more robust than BayesB in all remaining traits. Therefore, we believed that SEFL has the potential to be promoted to estimate GEBVs in other animals and plants.

A Stacking Ensemble Learning Framework for Genomic Prediction