描述
OmegaFold是一款尖端的AI模型,专为高分辨率的从头蛋白质结构预测而设计,直接从一级氨基酸序列进行预测。此发布代码提供了对论文“从一级序列进行高分辨率从头结构预测”中所述模型的访问。该项目致力于持续优化,重点在于降低推理长蛋白质的GRAM需求并发布更强大的模型。
阅读更多
OmegaFold的关键进展包括显著降低GRAM需求,使得在使用特定的`--subbatch_size`设置时,可以在具有80GB内存的NVIDIA A100 GPU上推理长达4096个残基的蛋白质序列。模型对`--subbatch_size`的敏感性得到了强调,它提供了计算时间和内存使用之间的权衡。用户可以调整此参数来管理GPU内存限制,建议在出现内存问题时将该值减半。
对于macOS用户,OmegaFold支持MPS(Apple Silicon)加速,前提是安装了最新版本的PyTorch nightly版本。当前代码要求macOS用户克隆存储库并使用`python main.py`运行模型。设置过程包括通过pip安装工具或克隆存储库并运行`python setup.py install`。输入格式是标准的FASTA文件,输出包括每个序列一个PDB文件,置信度值嵌入在PDB文件的b-factor列中。
OmegaFold在平衡预测质量和资源使用方面提供了灵活性。用户可以修改`--subbatch_size`来权衡计算时间和GRAM,或调整`--num_cycle`来权衡计算时间和平均预测质量。可以通过`omegifold --help`命令访问更多详细信息和选项。该项目鼓励引用相关研究论文以获取其效用。
开发人员致力于提高易用性并扩展模型的性能。预计未来的更新将进一步提高性能和可访问性,以适应更广泛的用户和计算环境。此开源发布促进了结构生物学和药物发现领域的研究与开发。
- 类别科学发现与实验室工具
- 定价模式开源
- 部署云端, 自托管
- 平台macOS
OmegaFold的核心功能
从一级序列进行高分辨率的从头蛋白质结构预测
显著降低推理长蛋白质的GRAM需求
支持NVIDIA A100 (80GB) 上长达4096个残基的蛋白质序列
可调的`--subbatch_size`以实现内存-时间权衡
支持macOS用户的MPS (Apple Silicon) 加速
在PDB b-factor列中提供置信度值
通过`--num_cycle`选项权衡计算时间和预测质量
用于运行预测的命令行界面
GitHub上提供开源发布代码
可通过`--model 2`命令行参数使用模型2
需要PyTorch和biopython以实现基本功能
OmegaFold入门
设置环境:通过pip安装OmegaFold或克隆存储库并运行setup.py
准备输入:创建包含蛋白质序列的FASTA文件
运行预测:执行`omegifold INPUT_FILE.fasta OUTPUT_DIRECTORY`或`python main.py INPUT_FILE.fasta OUTPUT_DIRECTORY`
配置参数:调整`--subbatch_size`以优化内存或`--num_cycle`以权衡质量
访问结果:从输出目录检索生成的PDB文件
利用macOS加速:安装最新的PyTorch nightly版本并使用`python main.py`
探索选项:运行`omegifold --help`以获取详细的命令行参数
OmegaFold的使用案例
- 蛋白质结构预测
- 结构生物学研究
- 药物发现
- 生物信息学分析
- 计算化学
- 基因组学研究






