用通用Transformer实现蛋白折叠,性能媲美顶尖模型。
SimpleFold: Folding Proteins is Simpler than You Think
- 仅用标准Transformer块和流匹配训练,无专用模块。
- 30亿参数模型在基准测试中达领先水平,集成预测表现强。
- 架构简单高效,可在消费级硬件上快速部署。
蛋白折叠模型通常依赖领域知识设计复杂结构,如三角更新、显式配对表示和多目标训练。本文提出SimpleFold,首个基于流匹配的通用蛋白折叠模型,仅使用标准Transformer块与自适应层,并通过生成式流匹配目标结合结构项训练。模型规模达30亿参数,在约900万条蒸馏蛋白结构及实验PDB数据上训练。在标准折叠基准上,SimpleFold-3B性能媲美当前最优模型;同时在集成预测任务中表现突出,这通常是确定性重建模型难以实现的。由于其通用架构,模型具备部署效率优势,可在消费级硬件上高效推理。该工作挑战了蛋白折叠对复杂专用架构的依赖,为未来研究开辟新方向。
原文摘要 · Abstract (English)
Protein folding models have achieved groundbreaking results typically via a combination of integrating domain knowledge into the architectural blocks and training pipelines. Nonetheless, given the success of generative models across different but related problems, it is natural to question whether these architectural designs are a necessary condition to build performant models. In this paper, we introduce SimpleFold, the first flow-matching based protein folding model that solely uses general purpose transformer blocks. Protein folding models typically employ computationally expensive modules involving triangular updates, explicit pair representations or multiple training objectives curated for this specific domain. Instead, SimpleFold employs standard transformer blocks with adaptive layers and is trained via a generative flow-matching objective with an additional structural term. We scale SimpleFold to 3B parameters and train it on approximately 9M distilled protein structures together with experimental PDB data. On standard folding benchmarks, SimpleFold-3B achieves competitive performance compared to state-of-the-art baselines, in addition SimpleFold demonstrates strong performance in ensemble prediction which is typically difficult for models trained via deterministic reconstruction objectives. Due to its general-purpose architecture, SimpleFold shows efficiency in deployment and inference on consumer-level hardware. SimpleFold challenges the reliance on complex domain-specific architectures designs in protein folding, opening up an alternative design space for future progress.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。