Прорыв в скорости и эффективности
NVIDIA и Институт дизайна белков Университета Вашингтона представили RFD4-Proteina — новую модель для совместного дизайна последовательностей и атомарных структур белков. Ключевое отличие от предшественника RFD3 заключается в использовании латентного представления на уровне остатков (residue-level) вместо атомарного, а также в сокращении шагов диффузии с ~400 до 60. Это позволило достичь значительного прироста производительности без потери качества генерации.
Сравнительные метрики: RFD4-Proteina против RFD3
Бенчмарки, проведенные на последовательностях длиной 200 аминокислотных остатков, демонстрируют существенное улучшение характеристик новой модели:
| Метрика | RFD3 | RFD4-Proteina | Улучшение |
|---|---|---|---|
| Скорость генерации | Базовая | Ускорена | 3.8x |
| Пиковое потребление памяти | Базовое | Снижено | 84% |
| Шаги диффузии | ~400 | 60 | Оптимизация процесса |
Поиск с подкреплением на этапе инференса
Модель использует reward-guided inference-time search (поиск, управляемый вознаграждением, во время вывода). Это означает, что вычислительные ресурсы фокусируются только на перспективных кандидатах. Процесс включает:
- Декодирование: Автоэнкодер одновременно генерирует последовательность и атомарную структуру, устраняя необходимость в отдельном шаге inverse-folding (как в ProteinMPNN).
- Оракулы: Интеграция CUDA-ускоренных инструментов, таких как OpenFold3 и TMol, для оценки геометрии, взаимодействий и уверенности модели.
- Итеративный поиск: Модель отбраковывает слабые варианты и ветвит перспективные состояния на лету.
Практическое применение: Дизайн биндеров к PD-L1
В качестве демонстрации была проведена задача дизайна биндера к онкологической мишени PD-L1 (структура PDB 5O45). Модель успешно сгенерировала кандидаты длиной 100–150 остатков, ориентируясь на горячие точки (hotspots) Tyr56, Met115 и Tyr123. Пользователи могут задавать условия через аннотированные CIF-файлы или JSON-спецификации, фиксируя целевую структуру и позволяя модели варьировать последовательность нового биндера.
Доступность и инструменты
RFD4-Proteina доступна в двух форматах:
- Open Model: Для локального развертывания на Linux/CUDA системах с прямым доступом к коду сэмплирования.
- NIM Microservice: HTTP-интерфейс для быстрого развертывания инференса без необходимости установки модели локально.
Для автоматизации кампаний по дизайну белков NVIDIA также предоставляет BioNeMo Agent Toolkit, который координирует генерацию, поиск и оценку результатов.
Источник: NVIDIA dev blog ↗