Platform Science Use Cases Research Blog Request Access
Foundation Models

Why Plant Science Needs Its Own Foundation Model

Cyril Veran 8 min read
Abstract visualization representing foundation model pre-training on plant genomic sequences

The machine learning community spent a decade arguing about whether pre-training on large unlabeled corpora was worth the compute cost. That argument is over. But plant genomics has largely been absent from this conversation — and the reasons why illuminate exactly why a domain-specific foundation model is not optional, it is overdue.

The Transfer Problem Nobody Talks About

When a large language model trained on human text is fine-tuned for a narrow task — legal document summarization, say, or medical note extraction — it carries with it a vast prior about language structure, pragmatics, and world knowledge. That transfer is the thing that makes fine-tuning on 10,000 labeled examples competitive with models trained on millions.

Biologists have hoped for analogous transfer from protein language models like ESM-2 to plant genomics. The logic seems sound: DNA is a sequence, proteins are sequences, surely representations learned in one domain carry over. In practice, the transfer is shallow. Plant nuclear genomes differ fundamentally from anything a protein model sees: repetitive transposable element landscapes occupying 60–85% of the genome in polyploid species like Triticum aestivum, GC content variation between coding and intergenic regions, pervasive alternative splicing regulated by small RNAs that don't exist in the model's prior. A protein language model cannot represent what it has never seen, and it has never seen a grass genome intron.

The same limitation applies to general DNA foundation models trained primarily on human and vertebrate sequences. When we applied one of the best-available genomic sequence models to maize (Zea mays) variant annotation, its zero-shot performance on functional variant classification was barely above a k-mer baseline on the test set. The features it had learned were tuned to mammalian biology.

What Pre-Training on Plant Sequences Actually Looks Like

Our pre-training corpus draws from over 200 million plant genomic sequences spanning 47 species in Ensembl Plants and NCBI SRA, with additional accessions from curated GRIN holdings for wild relatives of major crops. This is not simply a matter of downloading FASTA files and running a tokenizer. The corpus required extensive preprocessing:

  • Repeat masking calibrated per-species (RepeatModeler outputs differ substantially between a grass and a legume; using a generic mask distorts k-mer distributions)
  • Assembly quality filtering — sequences from draft assemblies with N50 below 50 kb were excluded unless they came from species with no better reference
  • Taxonomic balance weighting to prevent the model from over-representing the three most-sequenced crops (Zea mays, Oryza sativa, Solanum lycopersicum) at the expense of minor species
  • Cross-species near-duplicate removal to prevent the model from memorizing syntenic blocks rather than learning generalizable sequence representations

The pre-training objective is masked sequence modeling — not masked language modeling in the BERT sense, but a variant that operates on contiguous subsequences and requires the model to reconstruct both the masked nucleotides and their positional context within the broader locus. This formulation forces the model to learn the positional grammar of regulatory elements, splice sites, and open reading frames simultaneously.

Why Species Diversity Matters More Than Depth

Early in our pre-training experiments, we ran a controlled comparison: train on 50 million sequences from five species at high depth versus 50 million sequences drawn evenly from 47 species. Downstream trait prediction accuracy on held-out species improved by 14 percentage points when trained on the broader species distribution. The depth-vs-breadth tradeoff in plant genomics strongly favors breadth, because the biology you want the model to generalize to — regulatory logic, promoter architecture, splice signal grammar — is conserved at the functional level even when the sequences themselves have diverged beyond alignment.

This is not unique to genomics. It echoes what multilingual NLP found when scaling language coverage: cross-lingual transfer improves not just on the added languages, but on the original ones too. The model is forced to learn more abstract representations when it cannot rely on surface-level sequence similarity.

What the Embeddings Encode

A 512-dimensional embedding vector per 512-token input window is the output we expose through the API. What do those dimensions encode? This is not a rhetorical question — we have spent considerable time probing them with linear classifiers trained on labeled subsets of Ensembl Plants annotations.

The short answer: the first few principal components of the embedding space track taxonomic distance, which is expected and somewhat uninteresting. What is interesting is that subspaces corresponding to functional annotations — promoter accessibility signals, exon-intron boundary patterns, transposable element family membership — emerge without any explicit labeling. We recover these annotations from the embedding space with linear classifiers at accuracies of 87–94% on held-out chromosomes. This is consistent with what protein language model probing studies found: the model learns biology because biology is the structure in the sequence.

The practical implication is that you do not need labeled functional annotation to get useful signal. A breeder who has whole-genome sequencing data for 800 lines but phenotype records for only 150 of them can embed all 800, then train a trait prediction head on the 150 labeled lines and predict the rest. The embedding has already done the hard work of representing sequence variation in a biologically meaningful way.

Where We Are Not Making Extravagant Claims

We should be direct about what this model is not. It is not a substitute for experimental biology. The correlation between embedding-derived predicted breeding values and observed field measurements is strong — around 0.87–0.91 on the traits we have evaluated internally — but that leaves 9–13% of variance unexplained. In a real breeding program, that unexplained variance often includes the most commercially consequential interactions: gene-by-environment effects at specific locations, epigenetic state differences between nursery and target environment, pathogen pressure variation that no sequence-based model can observe.

We are also not saying that pre-training on plant sequences renders species-specific fine-tuning unnecessary. Fine-tuning on a well-curated dataset of 2,000–5,000 labeled accessions for a target species consistently improves trait prediction accuracy by 8–15 percentage points compared to the zero-shot pre-trained model. The pre-training provides the representation; the fine-tuning calibrates it to the specific biology of the breeding population.

The Adapter Layer Architecture

Our deployment model uses species-specific adapter layers — lightweight LoRA-style modules that modify approximately 4% of the model parameters — rather than full fine-tuning. This has two practical advantages. First, it is computationally accessible: adapter training on a single A100 for a new crop species with 3,000 labeled accessions takes under six hours. Second, it preserves the generalist representations in the frozen pre-trained weights. Full fine-tuning on small datasets often causes catastrophic forgetting of cross-species patterns that were useful precisely because they encoded conserved biology.

For plant breeding organizations that have historical phenotype records from multi-environment trials but modest sequencing budgets, this means the path to a useful genomic prediction model is significantly shorter than building a species-specific model from scratch. The pre-trained backbone already understands the sequence-function vocabulary of plants. The adapter layer learns your specific population structure and target environment.

Next Questions

The open problems in this space are more interesting than the solved ones. How do you handle polyploidy correctly — specifically, homeologous chromosome sets in hexaploid wheat, where the same locus exists in three subgenomes with subtly different regulatory contexts? How do you incorporate epigenetic state into a sequence-only model without requiring bisulfite sequencing as a prerequisite? How do you generalize across ploidy levels when your training data is dominated by diploid reference genomes?

These are not rhetorical obstacles. They are the questions we are actively building datasets and experiments around. A plant genomics foundation model is not a product you finish — it is a scientific instrument you calibrate continuously as the reference genome landscape improves and as more phenotypic diversity becomes available in public archives like GRIN and CIMMYT's trial databases.

More from the lab