【论文阅读】3D-VisTA:具身智能中统一多种下游任务的预训练Transformer模型

概述

本文发表于2023年。聚焦3D-Visual Language Grounding(视觉-语言接地),提出了一种用于3D视觉与文本对齐的已经预训练过的Transformer(1),并构建了首个用于3D-VL预训练的大规模3D场景-文本数据集scanscribe数据集(2)。scanscribe包括来自原有大型数据集scannet和3rscan数据集,并由gpt生成场景的描述。

3D-VisTA通过掩码语言/对象建模和场景-文本匹配,在ScanScribe上进行与训练。

引言

难点:大多数为3D视觉语言接地开发的模型仅仅专注于任务中的一两个(如仅关注视觉语言特征/视觉定位/密集captioning/问答/情景推理/语法学习),缺乏统一。

Dense Captioning:在一张图片(或视频帧)中,*同时定位多个区域(Region Proposals)并为每个区域生成独立的文字描述(caption)的任务。

涉及的3D-VL任务及其对应的数据集上的结果:

视觉接地(ScanRefer提升8.1%、Nr3D/Sr3D提升3.6%)

问答(ScanQA提升10.1%)

情境推理(SQA3D提升1.9%)

3D-VisTA

输入:场景点云+句子---(通过场景编码模块处理点云)--->文本/3D对象标记----->通过一个多模态融合模块进行融合,捕捉3D物体和文本之间的对应关系。

1. 场景编码

给定三维场景的点云 ,使用分割掩码将场景分解为对象。

分割掩码获取方式:

(1)真实标签

(2)实例分割模型

对每个对象,采样1024个点,将其坐标归一化到单位球内。

然后将点云 输入得到PointNet++中,获取其点特征 、语义类别。

组合点特征、语义类别、嵌入、三维位置,作为对象标记的表示,通过将物体令牌注入一个4层Transformer捕捉物体之间的交互。将物体的成对的空间关系明确编码到空间Transformer中。

2. 实验设置

epoch:30

batchsize:128

learning rate:1e-4

预热步数:3000,采用余弦衰减,AdamW优化器,β1=0.9 β2=0.98

相关推荐
feasibility.1 天前
AI 解牛:刀刃不再需要手——技艺、判断与一次学习机制的重构
人工智能·职场发展·具身智能·skill·道·术
天涯明月19932 天前
世界模型:原理、范式与工程实践
大数据·人工智能·大模型·具身智能·世界模型
深蓝学院2 天前
给机器人直接加「触觉」,真是“多此一举”。。。
机器人·具身智能
深蓝学院3 天前
CoRL 2026 | 在线 VLM 记忆已成瓶颈?MIT×CMU 给出机器人闭环控制解法
机器人·具身智能
深蓝学院3 天前
新加坡国立大学:Show‑Harness,一个接口,直接让VLM操控实体机器人
机器人·具身智能·世界模型
技术狂人1684 天前
OmniBot-System多任务机器人:任务调度器路由设计与Jetson Thor端侧部署实践
具身智能·vla·任务调度器·jetson thor·多任务机器人
一直在努力的小宁5 天前
【阅读笔记】具身智能的真机数采,到了分水岭
人工智能·深度学习·机器学习·agent·具身智能·vlm·vln
一直在努力的小宁5 天前
【阅读笔记】具身操作的数采方案概览
后端·json·restful·具身智能·vla·vlm
技术狂人1685 天前
WAM-Trainer世界动作模型训练实战:IDM/FDM模块化架构与DeepSpeed ZeRO-2多卡训练
pytorch·具身智能·deepspeed·世界动作模型·idm逆动力学
数据堂官方账号5 天前
数据堂联合华为共筑 AI 数据价值新引擎
人工智能·华为·数据采集·具身智能·数据标注