从‘相似’到‘原型’：深入对比Siamese Network和Prototypical Network，教你为电影分类任务选对模型

从‘相似’到‘原型’：深入对比Siamese Network和Prototypical Network，教你为电影分类任务选对模型

当面对电影分类这类需要快速适应新类别的任务时，传统深度学习模型往往因样本不足而束手无策。小样本学习技术中的Siamese Network（连体网络）和Prototypical Network（原型网络）正是为解决这一痛点而生。本文将带您深入两种模型的运作机理，通过电影评论分类的实战案例，揭示它们在不同场景下的性能差异与选型策略。

1. 核心思想：两种截然不同的学习范式

1.1 Siamese Network：相似度对比的艺术

Siamese Network的核心在于成对样本比较。想象两位电影评论家各自审阅一条评论，网络的任务是判断这两条评论是否属于同一情感类别（正面/负面）。其典型结构包含：

• 共享权重的双胞胎网络：两个完全相同的子网络并行处理输入样本
• 距离度量层：计算两个输出向量的相似度（常用余弦相似度或欧式距离）
• 对比损失函数：最小化同类样本距离，最大化异类样本距离

# 简化版Siamese Network核心代码 class SiameseNetwork(nn.Module): def __init__(self): super().__init__() self.encoder = nn.Sequential( nn.Linear(100, 64), nn.ReLU(), nn.Linear(64, 32) ) def forward(self, x1, x2): out1 = self.encoder(x1) out2 = self.encoder(x2) return F.cosine_similarity(out1, out2)

提示：Siamese Network特别适合类别边界模糊的场景，如区分"积极"与"非常积极"的影评

1.2 Prototypical Network：类原型的引力模型

Prototypical Network采用中心引力思维——每个类别在特征空间都有一个"引力中心"（原型），新样本根据距离最近的原型判定类别。其工作流程分为三步：

1. 通过嵌入函数将支持集样本映射到特征空间
2. 计算每个类别所有样本的特征均值作为原型
3. 新样本通过softmax概率分配到最近的原型类别

2. 电影分类实战：代码层面的差异解剖

2.1 数据处理的特异性

两种网络对数据准备有本质区别：

Siamese Network需要成对样本：

# 生成正负样本对 pairs = [] labels = [] for i in range(len(positive_samples)): # 正样本对 pairs.append([positive_samples[i], positive_samples[(i+1)%len(positive_samples)]]) labels.append(1) # 负样本对 pairs.append([positive_samples[i], negative_samples[i%len(negative_samples)]]) labels.append(0)

Prototypical Network需要支持集/查询集划分：

def split_episode(data, n_way=5, k_shot=3): """划分支持集和查询集""" support = [] query = [] for class_id in range(n_way): samples = data[class_id] selected = np.random.choice(len(samples), k_shot+5, False) support.extend(samples[selected[:k_shot]]) query.extend(samples[selected[k_shot:]]) return support, query

2.2 损失函数的哲学差异

• Siamese Network使用对比损失：

  def contrastive_loss(out1, out2, label, margin=1.0): distance = F.pairwise_distance(out1, out2) loss = (1-label) * distance.pow(2) + \ label * F.relu(margin - distance).pow(2) return loss.mean()

• Prototypical Network使用负对数似然：

  def prototypical_loss(prototypes, queries, labels): distances = torch.cdist(queries, prototypes) log_p_y = F.log_softmax(-distances, dim=1) loss = -log_p_y.gather(1, labels.unsqueeze(1)).mean() return loss

3. 关键选型因素：何时选择哪种模型？

3.1 计算资源考量

• 资源紧张选Prototypical：原型网络前向传播只需计算样本到各类原型的距离
• GPU充足可考虑Siamese：并行计算能缓解成对比较的计算压力

3.2 数据特性分析

3.3 准确率与效率的权衡

在IMDb电影评论数据集上的对比实验：

4. 进阶技巧：提升电影分类性能的实战策略

4.1 针对Siamese Network的优化

• 困难样本挖掘：自动识别分类困难的影评对

  def hard_negative_mining(embeddings, labels, top_k=5): pairwise_dist = torch.cdist(embeddings, embeddings) mask = labels.unsqueeze(0) != labels.unsqueeze(1) hard_negatives = torch.topk(pairwise_dist[mask], top_k).values return hard_negatives.mean()

• 动态边际调整：根据训练进度自动调整对比损失的边际值

4.2 增强Prototypical Network的方案

• 原型精炼：在推理阶段用支持集样本微调原型

  def refine_prototype(prototype, support_embeddings, alpha=0.3): distances = torch.cdist(support_embeddings, prototype.unsqueeze(0)) weights = F.softmax(-distances, dim=0) return alpha*prototype + (1-alpha)*(weights*support_embeddings).sum(0)

• 注意力增强：为不同词语分配重要性权重

  class AttentionEmbedding(nn.Module): def __init__(self, vocab_size, embed_dim): super().__init__() self.embed = nn.Embedding(vocab_size, embed_dim) self.attention = nn.Sequential( nn.Linear(embed_dim, 32), nn.Tanh(), nn.Linear(32, 1) ) def forward(self, x): embeddings = self.embed(x) # [seq_len, embed_dim] attn_weights = F.softmax(self.attention(embeddings), dim=0) return (attn_weights * embeddings).sum(0)

在实际电影分类项目中，当遇到类别定义模糊（如区分"悬疑"和"惊悚"）时，Siamese Network的细粒度对比优势明显；而当需要快速适应新类型（如新增"元宇宙电影"类别）时，Prototypical Network只需计算新类原型即可实现零样本分类。