Beyond Conditional Computation: Retrieval-Augmented Genomic Foundation Models with Gengram

Huinan Xu, Xuyang Feng, Junhong Chen, Junchen Liu, Kaiwen Deng, Kai Ding, Shengning Long, Jiaxue Shuai, Zhaorong li, Shiping Liu, Guirong Xue, Zhan Xiao


Abstract
Current genomic foundation models (GFMs) rely largely on extensive neural computation to implicitly infer conserved biological motifs from single-nucleotide inputs. We introduce Gengram, a conditional memory module that provides an explicit and efficient lookup primitive for multi-base motifs through a genome-specific hashing scheme. Integrated into state-of-the-art (SOTA) GFM backbones, Gengram achieves substantial improvements of up to 22.6% across functional genomics tasks and generalizes across diverse architectures. Further analysis of Gengram’s latent space reveals biologically meaningful representations aligned with established genomic principles. By introducing structured motif memory as a modeling primitive, Gengram improves empirical performance and mechanistic interpretability, offering a scalable and biology-aligned direction for next-generation GFMs. The code is available at https://github.com/zhejianglab/Gengram, and the model weights are available at https://huggingface.co/ZhejiangLab/Gengram.
Anthology ID:
2026.emnlp-main.1464
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
33480–33511
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1464/
DOI:
Bibkey:
Cite (ACL):
Huinan Xu, Xuyang Feng, Junhong Chen, Junchen Liu, Kaiwen Deng, Kai Ding, Shengning Long, Jiaxue Shuai, Zhaorong li, Shiping Liu, Guirong Xue, and Zhan Xiao. 2026. Beyond Conditional Computation: Retrieval-Augmented Genomic Foundation Models with Gengram. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 33480–33511, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Beyond Conditional Computation: Retrieval-Augmented Genomic Foundation Models with Gengram (Xu et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1464.pdf