RAGP: a retrieval-augmented deep learning model for genomic prediction in crop breeding
- 1School of Computer Science and Engineering, Nanjing University of Science and Technology, Nanjing, 210094 China
- 2Agricultural Information Institute of CAAS, Beijing, 10081 China
- 3National Agriculture Science Data Center, Beijing, 10081 China
Abstract
Genomic prediction (GP) plays a pivotal role in expediting genetic gains for crop breeding. Recently, deep learning models has attracted growing attention in this field due to its superior performance over traditional models. However, most advanced models rely on simplified numerical representations of genomic variants and treat individuals in isolation, failing to capture the full complexity of genetic variation and the genetic relatedness among individuals. In response, we propose RAGP, a deep learning model driven by a retrieval-augmented mechanism composed of two synergistic components: (1) an embedding-based retrieval module that identifies a group of nearest neighbors highly relevant to the target sample as references, employs a phenotype-aware encoder to refine the representation space and capture informative genetic structures; (2) an augmentation module that integrates the retrieved references to enhance the target representation. The enriched representation is then propagated through a regression network for final prediction. This retrieval-guided strategy enables more informed and fine-grained utilization of genomic data and strengthens the model’s capacity to capture individual-level genetic relationships. Extensive experiments reveal that RAGP achieves consistently superior predictive performance compared to both conventional methods and state-of-the-art deep learning models. By incorporating retrieval-augmented mechanisms, RAGP achieves substantial improvements in predictive accuracy and offers a promising direction for advancing genomic selection in crop breeding. Our code is available on https://github.com/l00907l/RAGP.
Related
- QProtFuse: Query-Guided Layer Fusion for Retrieval-Augmented Protein Function Prediction
- RA-PLA: Retrieval-augmented graph convolutional networks for protein-ligand binding affinity prediction
- Optimizing progeny allocation strategies in breeding schemes while updating genomic prediction models.
- From long-term progeny trials to genomic selection: empirical prediction and simulation-guided redesign of Scots pine breeding in Germany
- Integrating Envirotyping and Phenomics for AI-Enabled Multi-Environment Genomic Prediction in Crop Breeding
BibTeX
@article{zuo2026ragp,
title = {{RAGP: a retrieval-augmented deep learning model for genomic prediction in crop breeding}},
author = {Lingling Zuo and Xinger Li and Xiaoli Wang and Rui Man and Yang Yang},
journal = {Plant Methods},
year = {2026},
month = aug,
doi = {10.1186/s13007-026-01574-x},
url = {https://doi.org/10.1186/s13007-026-01574-x},
note = {CC BY-NC-ND 4.0},
}
RIS
TY - JOUR
TI - RAGP: a retrieval-augmented deep learning model for genomic prediction in crop breeding
AU - Lingling Zuo
AU - Xinger Li
AU - Xiaoli Wang
AU - Rui Man
AU - Yang Yang
PY - 2026
DA - 2026/08/09
JO - Plant Methods
DO - 10.1186/s13007-026-01574-x
UR - https://doi.org/10.1186/s13007-026-01574-x
AB - Genomic prediction (GP) plays a pivotal role in expediting genetic gains for crop breeding. Recently, deep learning models has attracted growing attention in this field due to its superior performance over traditional models. However, most advanced models rely on simplified numerical representations of genomic variants and treat individuals in isolation, failing to capture the full complexity of genetic variation and the genetic relatedness among individuals. In response, we propose RAGP, a deep learning model driven by a retrieval-augmented mechanism composed of two synergistic components: (1) an embedding-based retrieval module that identifies a group of nearest neighbors highly relevant to the target sample as references, employs a phenotype-aware encoder to refine the representation space and capture informative genetic structures; (2) an augmentation module that integrates the retrieved references to enhance the target representation. The enriched representation is then propagated through a regression network for final prediction. This retrieval-guided strategy enables more informed and fine-grained utilization of genomic data and strengthens the model’s capacity to capture individual-level genetic relationships. Extensive experiments reveal that RAGP achieves consistently superior predictive performance compared to both conventional methods and state-of-the-art deep learning models. By incorporating retrieval-augmented mechanisms, RAGP achieves substantial improvements in predictive accuracy and offers a promising direction for advancing genomic selection in crop breeding. Our code is available on https://github.com/l00907l/RAGP.
SN - 1746-4811
ER -