arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.09079cs.CVcs.AIcs.RO
收藏

GST-VLA: 结构化高斯空间标记用于3D深度感知视觉-语言-动作模型

GST-VLA: Structured Gaussian Spatial Tokens for 3D Depth-Aware Vision-Language-Action Models

Yeungnam University(延世大学);Korea Advanced Institute of Science and Technology, KAIST(韩国科学技术院)

Md Selim Sarowar, Omer Tariq, Sungho Kim

AI总结:

GST-VLA通过结构化高斯空间标记和深度感知链式思维提升3D视觉-语言-动作模型的精度和性能。

摘要:

VLA模型将视觉观测编码为无内在几何结构的2D补丁标记。我们介绍了GST-VLA,其有两个贡献。首先,高斯空间标记器(GST)将冻结的密集深度和冻结的语义补丁特征转换为$N_g{=}128$个各向异性3D高斯原始体,每个参数化为度量残差均值$μ\in \mathbb{R}^3$、对数尺度协方差$\log σ\in \mathbb{R}^3$和学习的不透明度$α\in (0,1)$。协方差的特征结构编码局部表面方向,不透明度提供每个原始体的几何置信度,这些信息无法从标量深度中获得。空间注意力池化与学习的查询将固定的标记预算集中在几何显著区域而不是均匀分布。其次,3D深度感知链式思维(DA-CoT)推理监督四个结构化的中间空间思维,涵盖3D物体定位、抓取可及性接触几何、成对度量距离和粗略SE(3)路径点,作为训练损失中的显式生成目标。在每个VLM变换器块中的交叉注意力子层提供直接访问DA-CoT生成过程中的原始256个原始体高斯场。一个具有3亿参数的流匹配动作专家,带有专家混合前馈子层,通过条件ODE积分解码7自由度的delta动作块,条件于VLM隐藏状态和DA-CoT输出通过双交叉注意力。通过跨三个渐进阶段的复合$\mathcal{L}_\mathrm{flow} + \mathcal{L}_\mathrm{CoT} + \mathcal{L}_\mathrm{depth}$训练,GST-VLA在LIBERO上达到96.4%(+2.0%),在SimplerEnv上达到80.2%(+5.4%)。消融实验隔离了每个GST组件、每个DA-CoT思维和每个训练阶段的贡献,确认了在精度需求任务中独立和协同增益的集中。

英文摘要:

VLA models encode visual observations as 2D patch tokens with no intrinsic geometric structure. We introduce GST-VLA with two contributions. First, the Gaussian Spatial Tokenizer (GST) converts frozen dense depth and frozen semantic patch features into $N_g{=}128$ anisotropic 3D Gaussian primitives, each parameterized by a metric residual mean $μ\in \mathbb{R}^3$, log-scale covariance $\log σ\in \mathbb{R}^3$, and learned opacity $α\in (0,1)$. The covariance eigenstructure encodes local surface orientation, and opacity provides per-primitive geometric confidence, both inaccessible from scalar depth. Spatial attention pooling with learned queries concentrates the fixed token budget on geometrically salient regions rather than distributing uniformly. Second, 3D Depth-Aware Chain-of-Thought (DA-CoT) reasoning supervises four structured intermediate spatial thoughts, covering 3D object grounding, grasp affordance contact geometry, pairwise metric distances, and coarse SE(3) waypoints, as explicit generation targets in the training loss. A cross-attention sublayer at every VLM transformer block provides direct access to the raw 256-primitive Gaussian field during DA-CoT generation. A 300M-parameter flow-matching action expert with mixture-of-experts feedforward sublayers decodes 7-DoF delta action chunks via conditional ODE integration, conditioned on both VLM hidden states and DA-CoT outputs through dual cross-attention. Trained with composite $\mathcal{L}_\mathrm{flow} + \mathcal{L}_\mathrm{CoT} + \mathcal{L}_\mathrm{depth}$ across three progressive stages, GST-VLA achieves 96.4% on LIBERO (+2.0%), and 80.2% on SimplerEnv (+5.4%). Ablations isolate the contribution of each GST component, each DA-CoT thought, and each training stage, confirming independent and synergistic gains concentrated on precision demanding tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑