Learning Environment-Associated Marker Rankings with Attention-Based Graph Neural Networks

Abstract

Genotype-by-environment interactions can complicate plant breeding because genetic effects may vary across weather regimes, locations, and years. Although genomic prediction models can use dense marker data to predict complex crop traits, they often provide limited information about how marker relevance changes across environmental conditions. This study introduces a graph-based framework for estimating environment-associated marker rankings from multi-environment crop trial data. We introduce an attention-based graph neural network that jointly represents SNP markers and weather-based environment vectors as interacting nodes and is designed to learn environment-associated marker rankings from supervised yield data. The model is trained using yield prediction as the supervised objective, and the learned marker-level attention signals are aggregated to estimate marker rankings overall and within groups of similar environments. We evaluate the framework on two genotype-by-environment datasets. In both datasets, we examine whether learned marker rankings are stable across independent training runs and determine the overlap between high-ranked markers and external GWAS references. In a maize dataset with data from 212 different site-years, we rank markers separately within each weather-defined environment cluster and define a global marker set as the markers shared across all clusters. This provides a structured way to examine how marker rankings vary across weather-defined groups and which markers remain important across environments. We further encode each weather variable with a separate LSTM combined by attention, which makes the contribution of individual weather variables readable, and compare the resulting variable ranking against an independent SHAP-based attribution. Genes located near the top-ranked markers are also examined for functional enrichment. Overall, the results suggest that graph-based, environment-conditioned marker attribution may complement genomic prediction as an exploratory tool for studying context-dependent genomic signals in multi-environment trials.