PixCLIP:通过任意粒度像素-文本对齐学习实现细粒度视觉语言理解

研究方向:Image Captioning

论文全名:《PixCLIP: Achieving Fine-grained Visual Language Understanding via
Any-granularity Pixel-Text Alignment Learning》

1. 论文介绍

提升CLIP模型细粒度图像-文本对齐能力仍是一个活跃的研究焦点。为此,大多数现有工作采用

1)增加视觉信息处理粒度的策略(例如,结合视觉提示引导模型关注图像内特定局部区域)。

2)用长且详细的文本描述进行训练提高模型的细粒度视觉-语言对齐。

然而,CLIP文本编码器无法处理嵌入在长文本序列中的更细粒度文本信息。

为了协同利用增强视觉和文本内容处理粒度的优势,我们提出了PixCLIP,一个旨在同时容纳视觉提示输入和处理冗长文本描述的新颖框架。

首先建立一个自动化注释流程,能够为图像生成像素级的局部化、长篇形式的文本描述。利用这个流程,我们构建了LongGRIT,一个包含近150万个样本的高质量数据集。

其次用LLM替换CLIP的原始文本编码器,并提出了一个三分支的图像-文本对齐学习框架,促进图像区域与任意粒度下对应的文本描述之间的细粒度对齐。

2. 方法介绍

2.1 细粒度数据生成

LongGRiT数据集流程建立在GRIT-20M数据集(图像区域-文本对,每张图像中特定区域(由边界框或掩码定义)与对应的文本描述)之上,包括三个阶段:

对象级别表达注释:使用GRIT-20M的分割掩码隔离出每个物体。裁剪后的物体及其掩码被输入到InternVL2-76B中,生成一个初步的、关注视觉属性(例如,形状、颜色、纹理)的物体级标题。

为确保语义一致性,我们采用Qwen2-72B来验证标题与原始图像的一致性。任何被标记为不一致(例如,描述不存在的特征)或在语义上无效(例如,"一个红色的钟"对于一个蓝色花瓶)的标题都被丢弃,而有效的标题则作为物体级注释被保留。

上下文级别表达注释:将原始图像和物体的边界框输入到Griffion-G-26B(专门用于空间推理的视觉语言模型)。该模型生成一个描述物体位置(例如,"壁炉台上最左边的")及其与周围物体关系的上下文级标题。

细粒度表达注释:通过DeepSeek-R1-70B整合对向级和上下文级标题,该模型将两种输入合成为统一的细粒度表达。

例如,如果物体级标题陈述"装饰灯笼",而上下文级标题指出"壁炉台最左边的物体",DeepSeek-R1-70B将它们合并为:"这件物品是一盏装饰灯笼,上面有精美的雕刻,它被放置在木制壁炉架的最左侧,两侧是装裱好的肖像画。"

这种层次化的方法确保描述既精确又具有上下文联系。

2.2 模型结构

引入了一个并行的掩码patch嵌入层,用于接收单通道的二维输入。分别通过各自的patch嵌入层处理输入图像和掩码,然后将它们的输出相加。

输入图像I和掩码M,特征F:

和 是图像和掩码块嵌入层,En​是视觉编码器,P 表示位置编码

2.3 多粒度训练方法

PixCLIP 采用三种主要的对齐策略:全局图像-文本对齐、局部掩码-文本对齐和多尺度特征增强。

输入:​​​​​​​,原始图像 、掩码 和描述被掩蔽区域的长文本(LongGRiT), ​ 的裁剪图像 的掩码​。

文本编码器:LLAMA3-8B

适配器:来自LLM2Vec的适配器(一个简单的投影器)

掩码文本对比学习:

:掩码视觉嵌入,:长文本嵌入,计算余弦相似度

模型通过最大化与相应文本和图像嵌入的余弦相似度,同时最小化与批次中其他非对应嵌入的余弦相似度,来强制学习视觉和文本嵌入。

精细裁剪对齐:

:裁剪图像​和对应裁剪掩码​的视觉嵌入。

​:与该裁剪区域对应的文本嵌入

局部-全局表征增强:

输入原始图像与全1掩码,从全局密集表示中,我们再次提取局部表示。

根据掩码 M 所占区域,对视觉密集特征 {fᵢv}ᵢ≤P 进行池化,使用平均池化提取区域视觉表示:

总损失:

相关推荐
百度Geek说2 分钟前
AI 内容核查怎么做:智能内容 AI 核查系统实践
人工智能
武子康27 分钟前
Qwen-Drive 到底是什么:4B 级 VLM 如何成为驾驶策略底座
人工智能·算法·机器学习
落798.41 分钟前
《启元 PrimeSim 如何用生成视频评估机器人策略》
人工智能·机器人·启元·具身应用
老纪的技术唠嗑局1 小时前
token 暴降75%!减少长程 Agent 任务中的重复开销,PowerContext v1.5.0 发布
人工智能
智塑未来1 小时前
明源云AI走播机器人,打造智能直播新体验
人工智能·机器人
brave10251 小时前
LLMGuide 大模型面试指南
人工智能·ai·面试·ai编程
机 _ 长2 小时前
05 智算运维|8 张卡怎么才算「捆成一张」:NCCL 实测与拓扑
运维·人工智能·算法·智算运维
用户4356782936372 小时前
五线并行认知架构:从算力内卷到思考秩序的底层革命
人工智能
一 铭2 小时前
WrenAI 详解:给 AI Agent 的开源 GenBI 引擎
人工智能·开源
say_fall2 小时前
【AI大模型】MCP 模型上下文协议:客户端-服务器架构、JSON-RPC 消息与能力协商、工具发现与调用链路、与函数调用的分工
服务器·人工智能·架构