English

AMix-1: A Pathway to Test-Time Scalable Protein Foundation Model

Biomolecules 2025-08-12 v3 Artificial Intelligence

Abstract

We introduce AMix-1, a powerful protein foundation model built on Bayesian Flow Networks and empowered by a systematic training methodology, encompassing pretraining scaling laws, emergent capability analysis, in-context learning mechanism, and test-time scaling algorithm. To guarantee robust scalability, we establish a predictive scaling law and reveal the progressive emergence of structural understanding via loss perspective, culminating in a strong 1.7-billion model. Building on this foundation, we devise a multiple sequence alignment (MSA)-based in-context learning strategy to unify protein design into a general framework, where AMix-1 recognizes deep evolutionary signals among MSAs and consistently generates structurally and functionally coherent proteins. This framework enables the successful design of a dramatically improved AmeR variant with an up to 50×50\times activity increase over its wild type. Pushing the boundaries of protein engineering, we further empower AMix-1 with an evolutionary test-time scaling algorithm for in silico directed evolution that delivers substantial, scalable performance gains as verification budgets are intensified, laying the groundwork for next-generation lab-in-the-loop protein design.

Keywords

Cite

@article{arxiv.2507.08920,
  title  = {AMix-1: A Pathway to Test-Time Scalable Protein Foundation Model},
  author = {Changze Lv and Jiang Zhou and Siyu Long and Lihao Wang and Jiangtao Feng and Dongyu Xue and Yu Pei and Hao Wang and Zherui Zhang and Yuchen Cai and Zhiqiang Gao and Ziyuan Ma and Jiakai Hu and Chaochen Gao and Jingjing Gong and Yuxuan Song and Shuyi Zhang and Xiaoqing Zheng and Deyi Xiong and Lei Bai and Wanli Ouyang and Ya-Qin Zhang and Wei-Ying Ma and Bowen Zhou and Hao Zhou},
  journal= {arXiv preprint arXiv:2507.08920},
  year   = {2025}
}
R2 v1 2026-07-01T03:57:13.305Z