【深度学习精通】第21章 | 多模态学习 - CLIP、DALL-E与GPT-4V
环境声明
- Python版本:Python 3.10+
- 深度学习框架:PyTorch 2.0+
- 推荐GPU:NVIDIA GPU with CUDA 11.8+
- 依赖库:transformers、torchvision、PIL、numpy
学习目标
通过本章学习,你将掌握:
- 多模态学习的基本概念和核心挑战
- CLIP模型的对比学习原理与Zero-shot分类
- DALL-E图像生成机制
- GPT-4V视觉语言能力
- 多模态融合策略
- 2025年最新多模态模型进展
摘要
多模态学习是深度学习领域最具前景的方向之一,它致力于让机器同时理解和处理多种类型的数据(文本、图像、音频、视频等)。从OpenAI的CLIP到GPT-4V,从Google的Gemini到DALL-E,多模态技术正在重塑人工智能的边界。本章将深入解析这些革命性模型的工作原理,并通过完整代码实现帮助你掌握多模态应用开发。
1. 多模态学习概述
1.1 什么是多模态学习
多模态学习(Multimodal Learning)是指机器同时处理和理解来自多个模态(如视觉、语言、听觉等)的信息,并在这些模态之间建立联系和转换的能力。
补充:人类认知本身就是多模态的。当我们看到一只猫的图片时,大脑会自动联想到"猫"这个词、猫的叫声、以及触摸猫毛的感觉。多模态学习的目标就是让AI具备类似的跨模态理解能力。
1.2 多模态学习的核心挑战
| 挑战类型 | 描述 | 示例 |
|---|---|---|
| 表示差异 | 不同模态的数据结构差异巨大 | 图像是像素矩阵,文本是离散符号序列 |
| 对齐困难 | 需要找到不同模态间的对应关系 | 图像中的物体与描述它的词语对齐 |
| 融合复杂 | 如何有效结合多模态信息 | 早期融合 vs 晚期融合的选择 |
| 数据稀缺 | 配对的多模态数据难以获取 | 图文配对数据集规模有限 |
| 计算成本 | 处理多模态需要大量计算资源 | 训练CLIP需要数千GPU小时 |
1.3 多模态学习的应用场景
- 图像描述生成:为图片自动生成文字描述
- 视觉问答:根据图片内容回答问题
- 跨模态检索:用文本搜索相关图片,或用图片搜索相关文本
- 多模态情感分析:结合文本、语音、面部表情分析情感
- 自动驾驶:融合摄像头、雷达、GPS等多源信息
2. CLIP:对比语言-图像预训练
2.1 CLIP的革命性意义
CLIP(Contrastive Language-Image Pre-training)是OpenAI于2021年发布的多模态模型,它首次证明了通过对比学习可以在大规模图文数据上训练出强大的视觉表示。
一句话总结:CLIP让图像和文本在同一个向量空间中"说同一种语言"。
2.2 对比学习原理
对比学习(Contrastive Learning)是自监督学习的核心方法之一。其核心思想是:
- 拉近正样本:将配对的图文样本在特征空间中拉近
- 推远负样本:将不配对的图文样本在特征空间中推远
CLIP使用InfoNCE损失函数:
import torch
import torch.nn as nn
import torch.nn.functional as F
class CLIPLoss(nn.Module):
"""
CLIP对比学习损失函数
同时优化图像到文本和文本到图像两个方向
"""
def __init__(self, temperature=0.07):
super().__init__()
self.temperature = temperature
def forward(self, image_features, text_features):
"""
Args:
image_features: [batch_size, embed_dim]
text_features: [batch_size, embed_dim]
Returns:
loss: 标量损失值
"""
# 归一化特征
image_features = F.normalize(image_features, dim=-1)
text_features = F.normalize(text_features, dim=-1)
# 计算相似度矩阵 [batch_size, batch_size]
logits = torch.matmul(image_features, text_features.T) / self.temperature
# 标签:对角线为正样本
batch_size = image_features.shape[0]
labels = torch.arange(batch_size, device=image_features.device)
# 对称损失:图像到文本 + 文本到图像
loss_i2t = F.cross_entropy(logits, labels)
loss_t2i = F.cross_entropy(logits.T, labels)
loss = (loss_i2t + loss_t2i) / 2
return loss
2.3 CLIP架构详解
CLIP包含两个核心编码器:
import torch
import torch.nn as nn
from transformers import CLIPTokenizer, CLIPTextModel
import torchvision.models as models
class ImageEncoder(nn.Module):
"""
图像编码器:使用Vision Transformer或ResNet
"""
def __init__(self, embed_dim=512, model_name='resnet50'):
super().__init__()
# 使用预训练的ResNet作为骨干网络
if model_name == 'resnet50':
backbone = models.resnet50(pretrained=True)
self.backbone = nn.Sequential(*list(backbone.children())[:-1])
self.feature_dim = 2048
# 投影头:将特征映射到嵌入空间
self.projection = nn.Sequential(
nn.Linear(self.feature_dim, embed_dim),
nn.ReLU(),
nn.Linear(embed_dim, embed_dim)
)
def forward(self, images):
"""
Args:
images: [batch_size, 3, 224, 224]
Returns:
features: [batch_size, embed_dim]
"""
features = self.backbone(images)
features = features.view(features.size(0), -1)
embeddings = self.projection(features)
return embeddings
class TextEncoder(nn.Module):
"""
文本编码器:使用Transformer
"""
def __init__(self, vocab_size=49408, embed_dim=512,
max_length=77, num_layers=12):
super().__init__()
self.token_embedding = nn.Embedding(vocab_size, embed_dim)
self.position_embedding = nn.Embedding(max_length, embed_dim)
# Transformer编码器
encoder_layer = nn.TransformerEncoderLayer(
d_model=embed_dim,
nhead=8,
dim_feedforward=2048,
dropout=0.1,
batch_first=True
)
self.transformer = nn.TransformerEncoder(
encoder_layer,
num_layers=num_layers
)
# 投影层
self.projection = nn.Linear(embed_dim, embed_dim)
def forward(self, text_tokens):
"""
Args:
text_tokens: [batch_size, seq_length]
Returns:
features: [batch_size, embed_dim]
"""
batch_size, seq_length = text_tokens.shape
# 词嵌入 + 位置编码
positions = torch.arange(seq_length, device=text_tokens.device)
positions = positions.unsqueeze(0).expand(batch_size, -1)
x = self.token_embedding(text_tokens)
x = x + self.position_embedding(positions)
# Transformer编码
x = self.transformer(x)
# 取[CLS] token或平均池化
features = x.mean(dim=1)
embeddings = self.projection(features)
return embeddings
class CLIPModel(nn.Module):
"""
完整的CLIP模型
"""
def __init__(self, embed_dim=512, temperature=0.07):
super().__init__()
self.image_encoder = ImageEncoder(embed_dim=embed_dim)
self.text_encoder = TextEncoder(embed_dim=embed_dim)
self.temperature = temperature
def forward(self, images, text_tokens):
"""
Args:
images: [batch_size, 3, 224, 224]
text_tokens: [batch_size, seq_length]
Returns:
image_features, text_features, loss
"""
image_features = self.image_encoder(images)
text_features = self.text_encoder(text_tokens)
# 计算对比损失
loss_fn = CLIPLoss(temperature=self.temperature)
loss = loss_fn(image_features, text_features)
return image_features, text_features, loss
2.4 Zero-shot图像分类
CLIP最惊人的能力之一是Zero-shot分类:无需针对特定任务训练,就能对未见过的类别进行分类。
import torch
import torchvision.transforms as transforms
from PIL import Image
class ZeroShotClassifier:
"""
基于CLIP的Zero-shot分类器
"""
def __init__(self, model, class_names, templates=None):
self.model = model
self.model.eval()
# 默认模板
if templates is None:
templates = [
"a photo of a {}",
"a picture of a {}",
"an image of a {}",
"this is a photo of a {}"
]
self.templates = templates
self.class_names = class_names
# 预计算文本特征
self.text_features = self._encode_texts()
# 图像预处理
self.transform = transforms.Compose([
transforms.Resize(224),
transforms.CenterCrop(224),
transforms.ToTensor(),
transforms.Normalize(
mean=[0.48145466, 0.4578275, 0.40821073],
std=[0.26862954, 0.26130258, 0.27577711]
)
])
def _encode_texts(self):
"""编码所有类别的文本描述"""
all_features = []
for class_name in self.class_names:
# 为每个类别生成多个描述
texts = [template.format(class_name)
for template in self.templates]
# 这里简化处理,实际应使用tokenizer
# text_tokens = tokenize(texts)
# features = self.model.text_encoder(text_tokens)
# all_features.append(features.mean(dim=0))
return torch.stack(all_features)
def predict(self, image_path):
"""
对单张图片进行Zero-shot分类
Args:
image_path: 图片路径
Returns:
预测的类别和概率
"""
# 加载和预处理图像
image = Image.open(image_path).convert('RGB')
image_tensor = self.transform(image).unsqueeze(0)
with torch.no_grad():
# 编码图像
image_features = self.model.image_encoder(image_tensor)
image_features = torch.nn.functional.normalize(
image_features, dim=-1
)
# 计算与所有类别的相似度
text_features = torch.nn.functional.normalize(
self.text_features, dim=-1
)
similarity = (image_features @ text_features.T) * 100
probs = similarity.softmax(dim=-1)
# 获取预测结果
pred_idx = probs.argmax(dim=-1).item()
pred_class = self.class_names[pred_idx]
confidence = probs[0, pred_idx].item()
return pred_class, confidence, probs[0]
# 使用示例
class_names = ["cat", "dog", "bird", "car", "tree"]
classifier = ZeroShotClassifier(model=None, class_names=class_names)
3. 其他多模态模型
3.1 ALIGN:大规模图文对齐
ALIGN是Google发布的另一款重要多模态模型,与CLIP的主要区别:
| 特性 | CLIP | ALIGN |
|---|---|---|
| 训练数据 | 4亿图文对 | 18亿图文对 |
| 数据质量 | 经过清洗筛选 | 原始噪声数据 |
| 图像编码器 | ResNet或ViT | EfficientNet |
| 文本编码器 | Transformer | BERT |
| 核心思想 | 对比学习 | 对比学习 + 噪声鲁棒性 |
ALIGN证明了即使使用噪声较大的网络数据,通过适当的设计也能训练出强大的多模态模型。
3.2 FLAVA:基础语言与视觉对齐
FLAVA是Meta发布的统一多模态模型,特点包括:
- 统一架构:单一模型处理多种任务
- 多任务学习:同时优化对比学习、掩码语言建模、掩码图像建模
- 更强的单模态能力:不仅擅长跨模态任务,单模态表现也很强
4. DALL-E与图像生成
4.1 DALL-E的工作原理
DALL-E是OpenAI发布的文本到图像生成模型,它结合了两种关键技术:
- VQ-VAE:将图像压缩为离散token
- GPT:自回归生成图像token
4.2 VQ-VAE:向量量化变分自编码器
import torch
import torch.nn as nn
import torch.nn.functional as F
class VectorQuantizer(nn.Module):
"""
向量量化层:将连续特征映射到离散码本
"""
def __init__(self, num_embeddings=8192, embedding_dim=512, commitment_cost=0.25):
super().__init__()
self.num_embeddings = num_embeddings
self.embedding_dim = embedding_dim
self.commitment_cost = commitment_cost
# 可学习的码本
self.embeddings = nn.Embedding(num_embeddings, embedding_dim)
self.embeddings.weight.data.uniform_(-1/num_embeddings, 1/num_embeddings)
def forward(self, inputs):
"""
Args:
inputs: [batch_size, embedding_dim, height, width]
Returns:
quantized: 量化后的特征
loss: VQ损失
encoding_indices: 离散编码索引
"""
# 转换维度 [B, D, H, W] -> [B*H*W, D]
inputs = inputs.permute(0, 2, 3, 1).contiguous()
input_shape = inputs.shape
flat_input = inputs.view(-1, self.embedding_dim)
# 计算与码本中所有向量的距离
distances = (torch.sum(flat_input**2, dim=1, keepdim=True)
+ torch.sum(self.embeddings.weight**2, dim=1)
- 2 * torch.matmul(flat_input, self.embeddings.weight.t()))
# 找到最近的码本向量
encoding_indices = torch.argmin(distances, dim=1).unsqueeze(1)
# One-hot编码
encodings = torch.zeros(encoding_indices.shape[0], self.num_embeddings,
device=inputs.device)
encodings.scatter_(1, encoding_indices, 1)
# 量化
quantized = torch.matmul(encodings, self.embeddings.weight)
quantized = quantized.view(input_shape)
# VQ损失
e_latent_loss = F.mse_loss(quantized.detach(), inputs)
q_latent_loss = F.mse_loss(quantized, inputs.detach())
loss = q_latent_loss + self.commitment_cost * e_latent_loss
# 直通估计器(Straight Through Estimator)
quantized = inputs + (quantized - inputs).detach()
# 转换回原始维度
quantized = quantized.permute(0, 3, 1, 2).contiguous()
return quantized, loss, encoding_indices.view(input_shape[0], input_shape[1], input_shape[2])
class VQVAE(nn.Module):
"""
完整的VQ-VAE模型
"""
def __init__(self, in_channels=3, embedding_dim=512, num_embeddings=8192):
super().__init__()
# 编码器
self.encoder = nn.Sequential(
nn.Conv2d(in_channels, 128, 4, 2, 1), # 1/2
nn.ReLU(),
nn.Conv2d(128, 256, 4, 2, 1), # 1/4
nn.ReLU(),
nn.Conv2d(256, embedding_dim, 3, 1, 1),
)
# 向量量化层
self.vq = VectorQuantizer(num_embeddings, embedding_dim)
# 解码器
self.decoder = nn.Sequential(
nn.Conv2d(embedding_dim, 256, 3, 1, 1),
nn.ReLU(),
nn.ConvTranspose2d(256, 128, 4, 2, 1), # 2x
nn.ReLU(),
nn.ConvTranspose2d(128, in_channels, 4, 2, 1), # 4x
nn.Tanh()
)
def forward(self, x):
z = self.encoder(x)
quantized, vq_loss, indices = self.vq(z)
x_recon = self.decoder(quantized)
# 重建损失
recon_loss = F.mse_loss(x_recon, x)
total_loss = recon_loss + vq_loss
return x_recon, total_loss, indices
def encode(self, x):
"""将图像编码为离散token"""
z = self.encoder(x)
_, _, indices = self.vq(z)
return indices
def decode(self, indices):
"""从离散token解码为图像"""
# 从索引获取量化向量
quantized = self.vq.embeddings(indices)
quantized = quantized.permute(0, 3, 1, 2).contiguous()
x_recon = self.decoder(quantized)
return x_recon
4.3 DALL-E的文本到图像生成
class DALLEModel(nn.Module):
"""
简化的DALL-E模型架构
"""
def __init__(self, vocab_size=16384, image_vocab_size=8192,
embed_dim=1024, num_layers=24, num_heads=16):
super().__init__()
# 文本和图像共享的token嵌入
self.token_embed = nn.Embedding(vocab_size + image_vocab_size, embed_dim)
self.pos_embed = nn.Embedding(1024, embed_dim) # 最大序列长度
# GPT风格的Transformer解码器
decoder_layer = nn.TransformerDecoderLayer(
d_model=embed_dim,
nhead=num_heads,
dim_feedforward=embed_dim * 4,
dropout=0.1,
batch_first=True
)
self.transformer = nn.TransformerDecoder(
decoder_layer,
num_layers=num_layers
)
# 输出头
self.lm_head = nn.Linear(embed_dim, vocab_size + image_vocab_size, bias=False)
def forward(self, text_tokens, image_tokens=None):
"""
Args:
text_tokens: [batch_size, text_length]
image_tokens: [batch_size, image_length] (训练时提供,推理时自回归生成)
Returns:
logits: [batch_size, seq_length, vocab_size]
"""
# 拼接文本和图像token
if image_tokens is not None:
# 训练时:文本token + 图像token
tokens = torch.cat([text_tokens, image_tokens], dim=1)
else:
tokens = text_tokens
batch_size, seq_length = tokens.shape
# 嵌入
x = self.token_embed(tokens)
positions = torch.arange(seq_length, device=tokens.device)
x = x + self.pos_embed(positions).unsqueeze(0)
# 因果掩码(自回归)
mask = torch.triu(torch.ones(seq_length, seq_length), diagonal=1)
mask = mask.masked_fill(mask == 1, float('-inf'))
# Transformer
x = self.transformer(x, x, tgt_mask=mask)
# 输出logits
logits = self.lm_head(x)
return logits
def generate(self, text_tokens, max_length=256, temperature=1.0, top_k=50):
"""
自回归生成图像token
Args:
text_tokens: 编码后的文本token
max_length: 最大生成长度
temperature: 采样温度
top_k: Top-k采样
Returns:
生成的图像token序列
"""
self.eval()
generated = text_tokens.clone()
with torch.no_grad():
for _ in range(max_length):
logits = self.forward(generated)
# 只取最后一个位置的logits
next_token_logits = logits[:, -1, :] / temperature
# Top-k采样
if top_k > 0:
indices_to_remove = next_token_logits < torch.topk(
next_token_logits, top_k
)[0][..., -1, None]
next_token_logits[indices_to_remove] = float('-inf')
probs = F.softmax(next_token_logits, dim=-1)
next_token = torch.multinomial(probs, num_samples=1)
generated = torch.cat([generated, next_token], dim=1)
# 检查是否生成结束token(简化处理)
if next_token.item() == 0: # 假设0是结束token
break
return generated[:, text_tokens.shape[1]:] # 返回图像部分
5. GPT-4V:视觉语言能力
5.1 GPT-4V的能力边界
GPT-4V(GPT-4 Vision)是OpenAI在2023年发布的多模态大模型,它在GPT-4的基础上增加了视觉理解能力。
核心能力:
- 图像描述与理解
- 图表和文档分析
- 视觉问答
- 多模态推理
5.2 视觉指令微调
GPT-4V采用了视觉指令微调(Visual Instruction Tuning)技术:
class VisualInstructionDataset:
"""
视觉指令微调数据集
"""
def __init__(self, data_path, image_processor, tokenizer):
self.data = self.load_data(data_path)
self.image_processor = image_processor
self.tokenizer = tokenizer
def load_data(self, data_path):
"""加载指令数据"""
# 数据格式示例:
# {
# "image": "path/to/image.jpg",
# "conversations": [
# {"from": "human", "value": "<image>\n描述这张图片"},
# {"from": "gpt", "value": "这是一张..."}
# ]
# }
import json
with open(data_path, 'r') as f:
return json.load(f)
def __getitem__(self, idx):
item = self.data[idx]
# 处理图像
image = Image.open(item['image']).convert('RGB')
pixel_values = self.image_processor(image)
# 处理对话
conversation = item['conversations']
# 构建输入文本
human_text = conversation[0]['value'].replace('<image>', '')
gpt_text = conversation[1]['value']
# 编码
input_text = f"Human: {human_text}\nAssistant: {gpt_text}"
tokens = self.tokenizer(input_text)
return {
'pixel_values': pixel_values,
'input_ids': tokens['input_ids'],
'labels': tokens['input_ids'] # 语言模型目标
}
6. 多模态融合策略
6.1 融合策略对比
| 融合策略 | 描述 | 优点 | 缺点 | 适用场景 |
|---|---|---|---|---|
| 早期融合 | 在特征提取前融合原始数据 | 信息损失少 | 计算复杂度高 | 模态高度相关 |
| 晚期融合 | 各模态独立处理后再融合 | 模块化程度高 | 可能丢失交互信息 | 模态相对独立 |
| 中间融合 | 在特征层进行融合 | 平衡性能和效率 | 需要设计融合机制 | 大多数场景 |
| 注意力融合 | 使用注意力机制动态融合 | 自适应权重 | 计算量大 | 复杂多模态任务 |
| 双线性融合 | 计算模态间的交互特征 | 捕获高阶交互 | 参数量大 | 细粒度理解 |
6.2 注意力融合实现
class CrossModalAttention(nn.Module):
"""
跨模态注意力:让一种模态关注另一种模态的信息
"""
def __init__(self, dim, num_heads=8, dropout=0.1):
super().__init__()
self.num_heads = num_heads
self.head_dim = dim // num_heads
self.scale = self.head_dim ** -0.5
self.q_proj = nn.Linear(dim, dim)
self.k_proj = nn.Linear(dim, dim)
self.v_proj = nn.Linear(dim, dim)
self.out_proj = nn.Linear(dim, dim)
self.dropout = nn.Dropout(dropout)
def forward(self, query, key_value, mask=None):
"""
Args:
query: [batch_size, q_len, dim] - 查询模态
key_value: [batch_size, kv_len, dim] - 被关注的模态
mask: 可选的注意力掩码
Returns:
output: [batch_size, q_len, dim]
attention_weights: [batch_size, num_heads, q_len, kv_len]
"""
batch_size, q_len, dim = query.shape
kv_len = key_value.shape[1]
# 投影
Q = self.q_proj(query)
K = self.k_proj(key_value)
V = self.v_proj(key_value)
# 分头 [batch_size, num_heads, len, head_dim]
Q = Q.view(batch_size, q_len, self.num_heads, self.head_dim).transpose(1, 2)
K = K.view(batch_size, kv_len, self.num_heads, self.head_dim).transpose(1, 2)
V = V.view(batch_size, kv_len, self.num_heads, self.head_dim).transpose(1, 2)
# 注意力计算
scores = torch.matmul(Q, K.transpose(-2, -1)) * self.scale
if mask is not None:
scores = scores.masked_fill(mask == 0, float('-inf'))
attention_weights = F.softmax(scores, dim=-1)
attention_weights = self.dropout(attention_weights)
# 加权求和
output = torch.matmul(attention_weights, V)
# 合并头
output = output.transpose(1, 2).contiguous().view(batch_size, q_len, dim)
output = self.out_proj(output)
return output, attention_weights
class MultiModalFusion(nn.Module):
"""
多模态融合模块:融合视觉和文本特征
"""
def __init__(self, visual_dim=512, text_dim=512, fusion_dim=512, num_layers=4):
super().__init__()
# 投影到统一维度
self.visual_proj = nn.Linear(visual_dim, fusion_dim)
self.text_proj = nn.Linear(text_dim, fusion_dim)
# 双向跨模态注意力层
self.cross_attn_layers = nn.ModuleList([
nn.ModuleDict({
'visual_to_text': CrossModalAttention(fusion_dim),
'text_to_visual': CrossModalAttention(fusion_dim),
'visual_self': CrossModalAttention(fusion_dim),
'text_self': CrossModalAttention(fusion_dim),
'visual_ffn': nn.Sequential(
nn.Linear(fusion_dim, fusion_dim * 4),
nn.GELU(),
nn.Linear(fusion_dim * 4, fusion_dim)
),
'text_ffn': nn.Sequential(
nn.Linear(fusion_dim, fusion_dim * 4),
nn.GELU(),
nn.Linear(fusion_dim * 4, fusion_dim)
),
'norm1': nn.LayerNorm(fusion_dim),
'norm2': nn.LayerNorm(fusion_dim),
'norm3': nn.LayerNorm(fusion_dim),
'norm4': nn.LayerNorm(fusion_dim),
})
for _ in range(num_layers)
])
# 最终融合
self.fusion_layer = nn.Sequential(
nn.Linear(fusion_dim * 2, fusion_dim),
nn.ReLU(),
nn.Dropout(0.1)
)
def forward(self, visual_features, text_features):
"""
Args:
visual_features: [batch_size, num_patches, visual_dim]
text_features: [batch_size, seq_len, text_dim]
Returns:
fused_features: [batch_size, fusion_dim]
"""
# 投影
V = self.visual_proj(visual_features)
T = self.text_proj(text_features)
# 跨模态注意力
for layer in self.cross_attn_layers:
# 视觉关注文本
V_attn, _ = layer['visual_to_text'](V, T)
V = layer['norm1'](V + V_attn)
V_ffn = layer['visual_ffn'](V)
V = layer['norm2'](V + V_ffn)
# 文本关注视觉
T_attn, _ = layer['text_to_visual'](T, V)
T = layer['norm3'](T + T_attn)
T_ffn = layer['text_ffn'](T)
T = layer['norm4'](T + T_ffn)
# 池化
V_pooled = V.mean(dim=1) # [batch_size, fusion_dim]
T_pooled = T.mean(dim=1) # [batch_size, fusion_dim]
# 融合
fused = torch.cat([V_pooled, T_pooled], dim=-1)
fused = self.fusion_layer(fused)
return fused
7. 2025年多模态学习最新进展
7.1 GPT-4o:全能多模态模型
2024-2025年,OpenAI发布了GPT-4o("o"代表omni,全能),这是首个真正的端到端多模态模型:
核心突破:
- 原生多模态:文本、图像、音频作为统一模态处理,而非分别编码
- 实时交互:支持实时语音对话,延迟低至232毫秒
- 统一表示:所有模态共享同一个神经网络架构
- 增强视觉:图像理解能力显著提升,支持详细描述和推理
7.2 Gemini 1.5:超长上下文多模态
Google的Gemini 1.5系列带来了革命性的长上下文能力:
| 特性 | Gemini 1.5 Pro | Gemini 1.5 Flash |
|---|---|---|
| 上下文窗口 | 200万token | 100万token |
| 多模态支持 | 文本、图像、音频、视频 | 文本、图像、音频、视频 |
| 视频理解 | 可处理1小时视频 | 可处理30分钟视频 |
| 应用场景 | 复杂推理、代码生成 | 快速响应、高效部署 |
MoE架构:Gemini 1.5采用混合专家(Mixture of Experts)架构,每次前向传播只激活部分参数,提高效率。
7.3 其他重要进展
- Claude 3 Opus:Anthropic发布的多模态模型,在视觉推理方面表现优异
- LLaVA-1.6:开源视觉指令微调模型,性能接近闭源模型
- Qwen-VL:阿里巴巴的开源多模态大模型,支持中文场景
- CogVLM:智谱AI的开源视觉语言模型,采用视觉专家模块
8. 多模态模型应用代码实战
8.1 使用Hugging Face Transformers加载CLIP
import torch
from PIL import Image
import requests
from transformers import CLIPProcessor, CLIPModel
# 加载预训练CLIP模型
model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
# 加载图像
url = "http://images.cocodataset.org/val2017/000000039769.jpg"
image = Image.open(requests.get(url, stream=True).raw)
# 准备文本描述
candidates = [
"a photo of a cat",
"a photo of a dog",
"a photo of a bird",
"a photo of a sofa"
]
# 处理输入
inputs = processor(
text=candidates,
images=image,
return_tensors="pt",
padding=True
)
# 前向传播
with torch.no_grad():
outputs = model(**inputs)
logits_per_image = outputs.logits_per_image
probs = logits_per_image.softmax(dim=1)
# 输出结果
for i, candidate in enumerate(candidates):
print(f"{candidate}: {probs[0][i].item():.4f}")
8.2 图像-文本检索系统
import torch
import torch.nn.functional as F
from transformers import CLIPProcessor, CLIPModel
from PIL import Image
import numpy as np
from typing import List
class ImageTextRetriever:
"""
基于CLIP的图像-文本检索系统
"""
def __init__(self, model_name="openai/clip-vit-base-patch32"):
self.device = "cuda" if torch.cuda.is_available() else "cpu"
self.model = CLIPModel.from_pretrained(model_name).to(self.device)
self.processor = CLIPProcessor.from_pretrained(model_name)
self.model.eval()
# 数据库
self.image_features = []
self.image_paths = []
self.text_features = []
self.texts = []
def encode_images(self, image_paths: List[str]):
"""
编码图像库
Args:
image_paths: 图像路径列表
"""
self.image_paths = image_paths
self.image_features = []
for path in image_paths:
image = Image.open(path).convert('RGB')
inputs = self.processor(images=image, return_tensors="pt")
with torch.no_grad():
features = self.model.get_image_features(**inputs)
features = F.normalize(features, dim=-1)
self.image_features.append(features.cpu())
self.image_features = torch.cat(self.image_features, dim=0)
print(f"已编码 {len(image_paths)} 张图像")
def encode_texts(self, texts: List[str]):
"""
编码文本库
Args:
texts: 文本列表
"""
self.texts = texts
self.text_features = []
for text in texts:
inputs = self.processor(text=text, return_tensors="pt", padding=True)
with torch.no_grad():
features = self.model.get_text_features(**inputs)
features = F.normalize(features, dim=-1)
self.text_features.append(features.cpu())
self.text_features = torch.cat(self.text_features, dim=0)
print(f"已编码 {len(texts)} 条文本")
def search_images_by_text(self, query_text: str, top_k: int = 5):
"""
用文本搜索相关图像
Args:
query_text: 查询文本
top_k: 返回最相关的k个结果
Returns:
最相关的图像路径和相似度分数
"""
# 编码查询文本
inputs = self.processor(text=query_text, return_tensors="pt", padding=True)
with torch.no_grad():
query_features = self.model.get_text_features(**inputs)
query_features = F.normalize(query_features, dim=-1)
# 计算相似度
similarities = torch.matmul(query_features, self.image_features.T)
similarities = similarities.squeeze(0)
# 获取top-k
top_scores, top_indices = torch.topk(similarities, min(top_k, len(self.image_paths)))
results = []
for score, idx in zip(top_scores, top_indices):
results.append({
'image_path': self.image_paths[idx],
'similarity': score.item()
})
return results
def search_texts_by_image(self, query_image_path: str, top_k: int = 5):
"""
用图像搜索相关文本
Args:
query_image_path: 查询图像路径
top_k: 返回最相关的k个结果
Returns:
最相关的文本和相似度分数
"""
# 编码查询图像
image = Image.open(query_image_path).convert('RGB')
inputs = self.processor(images=image, return_tensors="pt")
with torch.no_grad():
query_features = self.model.get_image_features(**inputs)
query_features = F.normalize(query_features, dim=-1)
# 计算相似度
similarities = torch.matmul(query_features, self.text_features.T)
similarities = similarities.squeeze(0)
# 获取top-k
top_scores, top_indices = torch.topk(similarities, min(top_k, len(self.texts)))
results = []
for score, idx in zip(top_scores, top_indices):
results.append({
'text': self.texts[idx],
'similarity': score.item()
})
return results
# 使用示例
retriever = ImageTextRetriever()
# 假设有一些图像和文本
# retriever.encode_images(["image1.jpg", "image2.jpg", "image3.jpg"])
# retriever.encode_texts(["a red car", "a blue sky", "a cute puppy"])
# 文本搜图
# results = retriever.search_images_by_text("a vehicle", top_k=3)
# 图搜文本
# results = retriever.search_texts_by_image("query_image.jpg", top_k=3)
8.3 多模态情感分析
import torch
import torch.nn as nn
from transformers import BertModel, ViTModel, BertTokenizer, ViTImageProcessor
class MultimodalSentimentAnalyzer(nn.Module):
"""
多模态情感分析模型:结合文本和图像进行情感判断
"""
def __init__(self, num_classes=3, dropout=0.3):
super().__init__()
# 文本编码器(BERT)
self.text_encoder = BertModel.from_pretrained('bert-base-chinese')
self.text_dim = 768
# 图像编码器(ViT)
self.image_encoder = ViTModel.from_pretrained(
'google/vit-base-patch16-224'
)
self.image_dim = 768
# 冻结部分层(可选)
for param in self.text_encoder.encoder.layer[:6].parameters():
param.requires_grad = False
for param in self.image_encoder.encoder.layer[:6].parameters():
param.requires_grad = False
# 融合层
self.fusion = nn.Sequential(
nn.Linear(self.text_dim + self.image_dim, 512),
nn.ReLU(),
nn.Dropout(dropout),
nn.Linear(512, 256),
nn.ReLU(),
nn.Dropout(dropout)
)
# 分类头
# 3分类:负面(0)、中性(1)、正面(2)
self.classifier = nn.Linear(256, num_classes)
def forward(self, text_inputs, image_inputs):
"""
Args:
text_inputs: BERT输入字典
image_inputs: ViT输入字典
Returns:
logits: [batch_size, num_classes]
"""
# 编码文本
text_outputs = self.text_encoder(**text_inputs)
text_features = text_outputs.pooler_output # [batch_size, 768]
# 编码图像
image_outputs = self.image_encoder(**image_inputs)
image_features = image_outputs.last_hidden_state[:, 0] # [CLS] token
# 融合
combined = torch.cat([text_features, image_features], dim=-1)
fused = self.fusion(combined)
# 分类
logits = self.classifier(fused)
return logits
# 训练代码
class MultimodalTrainer:
def __init__(self, model, device='cuda'):
self.model = model.to(device)
self.device = device
self.criterion = nn.CrossEntropyLoss()
self.optimizer = torch.optim.AdamW(
model.parameters(),
lr=2e-5,
weight_decay=0.01
)
def train_step(self, text_inputs, image_inputs, labels):
self.model.train()
# 移动数据到设备
text_inputs = {k: v.to(self.device) for k, v in text_inputs.items()}
image_inputs = {k: v.to(self.device) for k, v in image_inputs.items()}
labels = labels.to(self.device)
# 前向传播
logits = self.model(text_inputs, image_inputs)
loss = self.criterion(logits, labels)
# 反向传播
self.optimizer.zero_grad()
loss.backward()
torch.nn.utils.clip_grad_norm_(self.model.parameters(), 1.0)
self.optimizer.step()
return loss.item()
def evaluate(self, dataloader):
self.model.eval()
total_loss = 0
correct = 0
total = 0
with torch.no_grad():
for batch in dataloader:
text_inputs = {k: v.to(self.device) for k, v in batch['text_inputs'].items()}
image_inputs = {k: v.to(self.device) for k, v in batch['image_inputs'].items()}
labels = batch['labels'].to(self.device)
logits = self.model(text_inputs, image_inputs)
loss = self.criterion(logits, labels)
total_loss += loss.item()
predictions = logits.argmax(dim=-1)
correct += (predictions == labels).sum().item()
total += labels.size(0)
avg_loss = total_loss / len(dataloader)
accuracy = correct / total
return avg_loss, accuracy
# 数据预处理
def prepare_multimodal_data(texts, images, labels, max_length=128):
"""
准备多模态数据
Args:
texts: 文本列表
images: 图像路径列表
labels: 标签列表
max_length: 最大序列长度
"""
# 文本tokenizer
text_tokenizer = BertTokenizer.from_pretrained('bert-base-chinese')
# 图像processor
image_processor = ViTImageProcessor.from_pretrained(
'google/vit-base-patch16-224'
)
# 处理文本
text_inputs = text_tokenizer(
texts,
padding=True,
truncation=True,
max_length=max_length,
return_tensors='pt'
)
# 处理图像
from PIL import Image
pil_images = [Image.open(img).convert('RGB') for img in images]
image_inputs = image_processor(pil_images, return_tensors='pt')
labels = torch.tensor(labels)
return {
'text_inputs': text_inputs,
'image_inputs': image_inputs,
'labels': labels
}
9. 避坑小贴士
9.1 常见错误与解决方案
| 问题 | 原因 | 解决方案 |
|---|---|---|
| 显存溢出 | 批次过大或模型过大 | 减小batch_size、使用梯度累积、启用混合精度训练 |
| 对比损失不下降 | 温度参数设置不当 | 调整temperature(通常0.05-0.1) |
| Zero-shot效果差 | 提示词模板不合适 | 尝试多种prompt template,使用集成 |
| 多模态对齐失败 | 数据质量差或数量少 | 清洗数据、增加数据量、使用预训练模型 |
| 融合后性能下降 | 模态间冲突 | 使用门控机制、注意力权重调节 |
9.2 训练技巧
- 学习率设置:多模态模型通常需要较小的学习率(1e-5到5e-5)
- 预热策略:使用学习率预热避免早期训练不稳定
- 数据增强:对图像使用随机裁剪、翻转;对文本使用同义词替换
- 梯度裁剪:防止梯度爆炸,设置max_norm=1.0
- 混合精度:使用torch.cuda.amp加速训练并节省显存
from torch.cuda.amp import autocast, GradScaler
scaler = GradScaler()
# 训练循环中使用混合精度
with autocast():
logits = model(inputs)
loss = criterion(logits, labels)
scaler.scale(loss).backward()
scaler.step(optimizer)
scaler.update()
10. 本章小结
10.1 核心知识点回顾
- 多模态学习基础:理解不同模态的表示差异和融合挑战
- CLIP模型:掌握对比学习原理和Zero-shot分类实现
- 图像生成:了解VQ-VAE和DALL-E的文本到图像生成机制
- 视觉语言模型:理解GPT-4V和Gemini的多模态架构
- 融合策略:早期/晚期/中间融合的特点和适用场景
- 最新进展:GPT-4o、Gemini 1.5等2025年重要模型
10.2 学习路径建议
- 入门:从使用预训练CLIP模型开始,实现Zero-shot分类
- 进阶:尝试微调CLIP,适配特定领域任务
- 深入:实现完整的对比学习训练流程
- 拓展:探索更多模态(音频、视频)的融合
10.3 推荐资源
- OpenAI CLIP论文:Learning Transferable Visual Models From Natural Language Supervision
- VQ-VAE论文:Neural Discrete Representation Learning
- DALL-E论文:Zero-Shot Text-to-Image Generation
- Hugging Face Transformers文档:多模态模型部分
AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。
更多推荐


所有评论(0)