Investigating Data Pruning for Pretraining Biological Foundation Models at Scale

Read full text
Authors: Association for Artificial Intelligence 2026, Jiayang Chen, He Hu, Jiyue Jiang, Cheng Jin, Yu Li, Yiqi Wang, Yifan Wu, Yitao Xu, Xichen Ye, Jiao Yuan, Weizhong Zhang
+1 more Chang Zhou
Cited by: 0

Abstract

Biological foundation models (BioFMs), pretrained on large-scale biological sequences, have recently shown strong potential in providing meaningful representations for diverse downstream bioinformatics tasks. However, such models often rely on millions to billions of training sequences and billions of parameters, resulting in prohibitive computational costs and significant barriers to reproducibility and accessibility—particularly for academic labs. To address these challenges, we investigate the feasibility of data pruning for BioFM pretraining and propose a post-hoc influence-guided data pruning framework tailored to biological domains. Our approach first introduces a subset-based self-influence formulation that enables efficient estimation of sample importance at low computational cost. Built upon this, we propose two simple yet effective selection strategies: Top-$k$ Influence (Top I) and Coverage-Centric Influence (CCI). Then, we empirically validate our method on two representative BioFMs: RNA-FM and ESM-C. For RNA, our framework consistently outperforms random selection baselines under an extreme pruning rate of over 99\%, which displays our framework's effectiveness. Furthermore, we demonstrate the generalizability of our framework on protein-related tasks using ESM-C. In specific, our coreset even outperforms random $10\times$ subsets in both RNA and protein settings, revealing substantial redundancy in biological sequence dataset. These findings underscore the potential of influence-guided data pruning to substantially reduce the computational cost of BioFM pretraining, paving the way for more efficient, accessible, and sustainable biological AI research. The code and a technical appendix for better digital viewing are included as supplementary materials and scheduled to be open-sourced upon publication.

Fields of Study & Subjects

Machine Learning in Bioinformatics Machine Learning and Data Classification Genomics and Phylogenetic Studies

Publication details

Journal / source
Underline Science Inc.
Publication date
7 January 2026
Language
EN