TaxDiff: Taxonomic-Guided Diffusion Model for Protein Sequence Generation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zongying, Lin, Hao, Li, Liuzhenghao, Lv, Bin, Lin, Junwu, Zhang, Yu-Chian, Chen Calvin, Li, Yuan, Yonghong, Tian
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909121159102464
author Zongying, Lin
Hao, Li
Liuzhenghao, Lv
Bin, Lin
Junwu, Zhang
Yu-Chian, Chen Calvin
Li, Yuan
Yonghong, Tian
author_facet Zongying, Lin
Hao, Li
Liuzhenghao, Lv
Bin, Lin
Junwu, Zhang
Yu-Chian, Chen Calvin
Li, Yuan
Yonghong, Tian
contents Designing protein sequences with specific biological functions and structural stability is crucial in biology and chemistry. Generative models already demonstrated their capabilities for reliable protein design. However, previous models are limited to the unconditional generation of protein sequences and lack the controllable generation ability that is vital to biological tasks. In this work, we propose TaxDiff, a taxonomic-guided diffusion model for controllable protein sequence generation that combines biological species information with the generative capabilities of diffusion models to generate structurally stable proteins within the sequence space. Specifically, taxonomic control information is inserted into each layer of the transformer block to achieve fine-grained control. The combination of global and local attention ensures the sequence consistency and structural foldability of taxonomic-specific proteins. Extensive experiments demonstrate that TaxDiff can consistently achieve better performance on multiple protein sequence generation benchmarks in both taxonomic-guided controllable generation and unconditional generation. Remarkably, the sequences generated by TaxDiff even surpass those produced by direct-structure-generation models in terms of confidence based on predicted structures and require only a quarter of the time of models based on the diffusion model. The code for generating proteins and training new versions of TaxDiff is available at:https://github.com/Linzy19/TaxDiff.
format Preprint
id arxiv_https___arxiv_org_abs_2402_17156
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TaxDiff: Taxonomic-Guided Diffusion Model for Protein Sequence Generation
Zongying, Lin
Hao, Li
Liuzhenghao, Lv
Bin, Lin
Junwu, Zhang
Yu-Chian, Chen Calvin
Li, Yuan
Yonghong, Tian
Machine Learning
Artificial Intelligence
Biomolecules
Designing protein sequences with specific biological functions and structural stability is crucial in biology and chemistry. Generative models already demonstrated their capabilities for reliable protein design. However, previous models are limited to the unconditional generation of protein sequences and lack the controllable generation ability that is vital to biological tasks. In this work, we propose TaxDiff, a taxonomic-guided diffusion model for controllable protein sequence generation that combines biological species information with the generative capabilities of diffusion models to generate structurally stable proteins within the sequence space. Specifically, taxonomic control information is inserted into each layer of the transformer block to achieve fine-grained control. The combination of global and local attention ensures the sequence consistency and structural foldability of taxonomic-specific proteins. Extensive experiments demonstrate that TaxDiff can consistently achieve better performance on multiple protein sequence generation benchmarks in both taxonomic-guided controllable generation and unconditional generation. Remarkably, the sequences generated by TaxDiff even surpass those produced by direct-structure-generation models in terms of confidence based on predicted structures and require only a quarter of the time of models based on the diffusion model. The code for generating proteins and training new versions of TaxDiff is available at:https://github.com/Linzy19/TaxDiff.
title TaxDiff: Taxonomic-Guided Diffusion Model for Protein Sequence Generation
topic Machine Learning
Artificial Intelligence
Biomolecules
url https://arxiv.org/abs/2402.17156