Diffusion Model with Representation Alignment for Protein Inverse Folding

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Chenglin, Zhou, Yucheng, Zhai, Zijie, Shen, Jianbing, Zhang, Kai
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915061282373632
author Wang, Chenglin
Zhou, Yucheng
Zhai, Zijie
Shen, Jianbing
Zhang, Kai
author_facet Wang, Chenglin
Zhou, Yucheng
Zhai, Zijie
Shen, Jianbing
Zhang, Kai
contents Protein inverse folding is a fundamental problem in bioinformatics, aiming to recover the amino acid sequences from a given protein backbone structure. Despite the success of existing methods, they struggle to fully capture the intricate inter-residue relationships critical for accurate sequence prediction. We propose a novel method that leverages diffusion models with representation alignment (DMRA), which enhances diffusion-based inverse folding by (1) proposing a shared center that aggregates contextual information from the entire protein structure and selectively distributes it to each residue; and (2) aligning noisy hidden representations with clean semantic representations during the denoising process. This is achieved by predefined semantic representations for amino acid types and a representation alignment method that utilizes type embeddings as semantic feedback to normalize each residue. In experiments, we conduct extensive evaluations on the CATH4.2 dataset to demonstrate that DMRA outperforms leading methods, achieving state-of-the-art performance and exhibiting strong generalization capabilities on the TS50 and TS500 datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2412_09380
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Diffusion Model with Representation Alignment for Protein Inverse Folding
Wang, Chenglin
Zhou, Yucheng
Zhai, Zijie
Shen, Jianbing
Zhang, Kai
Machine Learning
Artificial Intelligence
Protein inverse folding is a fundamental problem in bioinformatics, aiming to recover the amino acid sequences from a given protein backbone structure. Despite the success of existing methods, they struggle to fully capture the intricate inter-residue relationships critical for accurate sequence prediction. We propose a novel method that leverages diffusion models with representation alignment (DMRA), which enhances diffusion-based inverse folding by (1) proposing a shared center that aggregates contextual information from the entire protein structure and selectively distributes it to each residue; and (2) aligning noisy hidden representations with clean semantic representations during the denoising process. This is achieved by predefined semantic representations for amino acid types and a representation alignment method that utilizes type embeddings as semantic feedback to normalize each residue. In experiments, we conduct extensive evaluations on the CATH4.2 dataset to demonstrate that DMRA outperforms leading methods, achieving state-of-the-art performance and exhibiting strong generalization capabilities on the TS50 and TS500 datasets.
title Diffusion Model with Representation Alignment for Protein Inverse Folding
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2412.09380