ProteinMPNN 中 tied_featurize 函数介绍
ProteinMPNN 是一种专为蛋白质设计任务开发的神经网络模型，广泛用于蛋白质序列生成与结构预测任务。本文将深入介绍其核心函数之一——tied_featurize，结合代码示例、详细解析与图解，帮助你理解该函数的作用、实现及在 ProteinMPNN 中的关键地位。
1. tied_featurize 的作用
在 ProteinMPNN 中，tied_featurize 主要负责将输入的蛋白质序列和结构信息转化为模型可处理的特征张量。该函数的主要功能包括：
- 将序列和结构信息进行编码。
- 保证特征向量的长度和顺序与输入保持一致。
- 生成的特征张量可以直接输入模型进行后续处理。
2. 函数结构概览
以下是 tied_featurize 的核心代码结构：
def tied_featurize(sequence, structure):
"""
将蛋白质序列和结构特征进行绑定编码，生成模型输入特征张量。
参数:
- sequence: 蛋白质序列 (str)
- structure: 蛋白质结构信息 (dict)
返回:
- features: 特征张量 (numpy 或 PyTorch 张量)
"""
# 步骤 1: 序列编码
seq_features = encode_sequence(sequence)
# 步骤 2: 结构编码
struct_features = encode_structure(structure)
# 步骤 3: 特征绑定 (Tied)
features = bind_features(seq_features, struct_features)
return features3. 核心步骤解析
3.1 序列编码
序列编码将氨基酸序列转化为数值化的特征表示。例如，每个氨基酸可以表示为固定维度的向量。
代码示例：
def encode_sequence(sequence):
"""
将氨基酸序列转化为数值特征表示。
"""
amino_acids = 'ACDEFGHIKLMNPQRSTVWY'
one_hot = {aa: idx for idx, aa in enumerate(amino_acids)}
seq_features = [one_hot.get(aa, -1) for aa in sequence] # 用 -1 表示未知氨基酸
return np.array(seq_features)图解：
- 输入序列：
ACDE - One-hot 编码后：
[0, 1, 2, 3]
3.2 结构编码
结构编码提取蛋白质的空间构象信息，例如每个氨基酸的原子坐标、键长、二面角等。
代码示例：
def encode_structure(structure):
"""
编码蛋白质结构特征，例如位置、二面角等。
"""
positions = structure['positions'] # 每个氨基酸的空间坐标
dihedrals = structure['dihedrals'] # 二面角信息
struct_features = np.hstack((positions, dihedrals))
return struct_features图解：
每个氨基酸的空间特征可能包含：
- ( x, y, z )：原子坐标。
- (\phi, \psi, \omega)：主链二面角。
- 结果特征矩阵：
\[
\text{Feature} =
\begin{bmatrix}
x_1 & y_1 & z_1 & \phi_1 & \psi_1 & \omega_1 \\
x_2 & y_2 & z_2 & \phi_2 & \psi_2 & \omega_2 \\
\vdots & \vdots & \vdots & \vdots & \vdots & \vdots \\
\end{bmatrix}
\]
3.3 特征绑定（Tied）
绑定特征是指将序列特征和结构特征结合起来，形成统一的输入特征张量。
代码示例：
def bind_features(seq_features, struct_features):
"""
将序列特征与结构特征绑定。
"""
# 假设序列和结构特征长度一致
features = np.concatenate((seq_features[:, np.newaxis], struct_features), axis=1)
return features图解：
- 序列特征：
[0, 1, 2, 3] - 结构特征（简化表示）：
\[
\text{Structure} =
\begin{bmatrix}
x_1 & y_1 & z_1 \\
x_2 & y_2 & z_2 \\
\vdots & \vdots & \vdots \\
\end{bmatrix}
\]
- 绑定后特征矩阵：
\[
\text{Tied Features} =
\begin{bmatrix}
0 & x_1 & y_1 & z_1 \\
1 & x_2 & y_2 & z_2 \\
2 & x_3 & y_3 & z_3 \\
\end{bmatrix}
\]
4. 应用场景
4.1 用于蛋白质序列设计
ProteinMPNN 的核心目标是基于已知结构生成最可能的蛋白质序列。tied_featurize 提供了统一的输入表示，为后续的深度学习模型提供高质量的特征。
4.2 结合深度学习模型
生成的特征可以直接输入 Transformer 或其他序列模型：
import torch
# 转换为 PyTorch 张量
features_tensor = torch.tensor(features, dtype=torch.float32)
# 模型输入
output = model(features_tensor)5. 总结
5.1 关键点
tied_featurize将蛋白质序列和结构信息结合，生成统一的特征张量。- 包含三个主要步骤：序列编码、结构编码、特征绑定。
- 是 ProteinMPNN 输入处理的核心部分。
5.2 优势
- 高效：简化了特征工程过程。
- 通用：适用于不同的蛋白质设计任务。
- 灵活：支持多种编码方式和特征扩展。
通过本文的讲解，希望你对 tied_featurize 函数的原理和实现有了深入理解，可以灵活应用到蛋白质序列设计和结构分析中！