Track trials, findings, and clinical evidence as it lands.
Search papers, labs, and topics across Lattice.
This study investigates the effectiveness and durability of pretraining data filtering for protein language models, specifically the ESM-2 family, by removing viral proteins. Results show that filtering viral proteins significantly impairs performance on viral protein tasks, especially for larger models and human-infecting viruses, while largely maintaining performance on non-viral proteins. Crucially, this performance degradation is not durable, as fine-tuning on the removed viral sequences restores lost capabilities with modest computational cost, even for large-scale models.
Pretraining data filtering for protein models is surprisingly brittle, with performance on targeted viral proteins easily restored via fine-tuning for under $100.
Pretraining filtering is one of the most common safeguards for open-weight language models. Here we study its effectiveness, durability and scalability using the ESM-2 family of protein language models. We find that filtering viral proteins from the pretraining data meaningfully degrades masked language modelling, mutation effect prediction and long-range contact prediction on viral proteins, while largely preserving performance on non-viral proteins. These effects are concentrated among proteins from human-infecting viruses, and grow as we scale models from 8M to 3B parameters. As models increase in size, we find that stringent filtering becomes more effective. However, fine-tuning on sequences from the removed viral families restores the lost performance. Although the cost to recover performance increases with model scale, even for the largest models it costs about 80 USD of H100 time. This work establishes pretraining filtering as moderately effective at selectively degrading performance on viral proteins, but not durable against fine-tuning with modest computational resources and access to relevant data.