TAIHRI:面向近距人机交互的任务感知3D人体关键点定位 TAIHRI: Task-Aware 3D Human Keypoints Localization for Close-Range Human-Robot Interaction
首个面向近距HRI的VLM,任务感知地定位3D人体关键点
前置知识
视觉-语言模型 (VLM)
VLM是结合视觉编码器和语言模型的多模态架构,能同时理解图像与文本,并通过next token prediction方式自回归生成文本输出。TAIHRI基于Qwen3-VL 4B/2B构建,把3D坐标预测转化为token序列生成问题。
本文核心是把3D关键点定位建模为token预测问题,必须理解VLM的输入输出机制和自回归生成流程。
SMPL/SMPL-X人体参数化模型
SMPL/SMPL-X是用形状参数 $\beta$ 和姿态参数 $\theta$ 驱动的3D人体网格统计模型,能从低维参数生成完整mesh顶点。文中用AMASS+SMPL-X在Blender中渲染normal map以构造CloseHRI训练数据。
数据集构建依赖此模型,且下游human mesh recovery任务也用SAM 3D Body等SMPL族模型输出mesh。
GRPO (Group Relative Policy Optimization)
GRPO是DeepSeek提出的强化学习算法,对同一prompt采样 $K$ 个response,用组内归一化优势 $\hat{A}_i = r_i - \bar{r}$ 做策略优化,比PPO更省显存。TAIHRI在RFT阶段用它做端到端微调。
文中RFT阶段的核心算法,决定奖励信号如何转化为策略更新,是TAIHRI超越纯SFT的关键。
相机内参 (Camera Intrinsics)
相机内参 $K_{int}$ 包含焦距 $f$ 和主点 $(c_x, c_y)$,用于把图像像素反投影到相机坐标系下的3D射线。TAIHRI把焦距统一为1000并用random crop模拟主点偏移,实现intrinsic-conditioned augmentation。
近距HRI要求在camera space定位,必须把内参信息显式注入模型,否则深度估计会严重退化。
体素离散化 (Voxel Discretization)
把连续3D空间划分为体素网格,每轴量化到0-999的离散token,便于VLM做next token prediction。文中交互空间定义为机器前方cuboid (W, H, D),$X_i=\lfloor x_i/W \times 999 \rfloor$ 即为token index。
这是连接连续3D坐标和离散token预测的桥梁,是TAIHRI能把3D定位转化为VLM可学习任务的关键。
Next Token Prediction与Chain-of-Thought
自回归地一个token一个token生成输出,每个token基于前面所有token的条件概率分布 $p(t_i|t_{<i})$。TAIHRI创新性地用CoT思想,把3D预测拆成'关键点名→2D像素坐标→3D voxel token'三段式输出。
是VLM输出3D坐标的核心范式,本文通过2D reasoning作为CoT改进它,Table 3显示可带来33mm精度提升。
研究动机
现有3D人体姿态估计方法(CameraHMR、PromptHMR、SAM 3D Body)侧重于以根关节为参考的整体重建质量,在root-centric坐标系下评估。但在近距HRI场景中,相机装在机器人或人身上,距离0.5-3米,常出现严重截断、遮挡和视角变化。此时机器人更需要task-relevant身体部位(如手、腋下、肩膀)的metric-scale精确3D定位。Table 1显示在Harmony4D-Egocentric的Upper body任务上,CameraHMR G-MPJPE高达167.50mm,PromptHMR为158.70mm,SAM 3D Body-DINOv3为124.91mm。手腕、踝关节等远端部位因深度模糊误差极大(Figure 5显示CameraHMR在手腕上误差达939.5mm)。即使是Gemini-2.5 Pro和Qwen3-VL-235B等通用VLM也分别达到436.9mm和1298.3mm的误差,远达不到机器人安全接触所需的精度(通常需<50mm)。
本文的目标是本文目标是构建首个专为近距HRI感知设计的VLM——TAIHRI——给定机器人egocentric RGB图像和人类任务指令(如'give the person a hug'),直接在camera坐标系下预测任务相关3D关键点的metric-scale坐标。模型需具备四项能力:(1) 任务感知,能根据prompt动态选择关键点子集;(2) 鲁棒性,能处理截断、遮挡、深度模糊;(3) 高精度,G-MPJPE降至100mm以下;(4) 通用性,可用于自然语言控制、global mesh recovery、机器人IK等下游任务。具体在Harmony4D和EgoBody两个egocentric benchmark上达到SOTA。
与已有工作不同的是,现有工作各有局限:通用HMR方法精度不足且是root-relative坐标系;通用VLM(GPT-5.2、Gemini-2.5 Pro)缺乏3D几何建模;depth estimation方法(DepthLM、Depth Anything 3)未建模人体articulated structure。TAIHRI的独特切入角度是首次把'task-aware + close-range + camera space'三个维度同时建模:用CloseHRI数据集(120万张0.5-3m的egocentric合成图)训练、用discretized interaction space体素化适配VLM、2D reasoning→3D prediction的chain-of-thought降低深度难度、SFT+GRPO两阶段训练。与SAM 3D Body等'full-body reconstruction'思路不同,本文聚焦于'task-critical parts in camera space',用交互指令驱动注意力分配,是首个把VLM范式成功落地到近距HRI的工作。
核心方法
TAIHRI的核心思路是'把3D关键点定位转化为VLM擅长的token预测问题'。方法流程分四步:(1) 把图像通过resize统一焦距到 $f=1000$,实现intrinsic-conditioned augmentation;(2) Qwen3-VL 4B/2B backbone融合图像+任务prompt+相机内参编码;(3) 自回归生成三段式输出——关键点名→2D像素坐标→3D voxel token;(4) 把3D voxel token解码回metric camera坐标。模型在CloseHRI数据集(约120万张近距egocentric图)上先做SFT(5 epoch, lr=2e-5, batch=4),再用GRPO做强化微调(rollout size=8, KL系数 $\beta=0.01$, lr=1e-6, 5 epoch, batch=32, 1万张采样)。整个pipeline的设计哲学是'用2D几何作为3D推理的中间表征',这模仿人类视觉先定位平面位置、再推断深度的认知过程。
三个核心创新点区别于已有方法。第一,task-aware prompt机制:构建6000+交互指令组成的prompt bank,覆盖'握手'、'轮椅转移'、'肩膀按摩'等场景,模型能根据指令动态选择关键点子集(1-6个),而非一次性输出全身22个点。第二,discretized interaction space:把机器前方cuboid量化成1000×1000×1000的voxel token,把连续3D定位问题转化为next token prediction,让VLM可以end-to-end学习。第三,2D reasoning as chain-of-thought:先输出2D像素坐标 `<283><764>...`,再输出3D voxel token `...`,用2D几何线索降低深度估计难度。Table 3显示去掉2D reasoning后G-MPJPE从93.83mm升至126.67mm(Upper,35%退化),去掉RFT后升至101.82mm。
方法步骤详情
完整方法分六阶段。**Stage 1 数据准备**:从AMASS采样motion sequence,用SMPL-X在Blender渲染normal map,相机距离0.5-3m随机化;用SDXL把normal map合成photo-realistic图像;用SAM3过滤IoU<0.9的样本,用VitPose过滤2D reprojection error>15像素的样本,得到CloseHRI(约100万张)。再融合BEDLAMv1/v2、PDhuman数据集,过滤平均深度>3m的样本,最终训练集约120万张。**Stage 2 输入处理**:图像resize使焦距统一为 $f=1000$,prompt从prompt bank中50%采样+50%默认prompt(让模型定位1-6个关键点),相机内参 $(f, c_x, c_y)$ 作为文本token输入。**Stage 3 SFT**:Qwen3-VL 4B/2B微调,batch=4,lr=2e-5,5 epoch,token-level交叉熵损失。**Stage 4 RFT**:GRPO rollout size=8,KL系数 $\beta=0.01$,batch=32,lr=1e-6,5 epoch,奖励 $r=\lambda\exp(-\tau d) + (1-\lambda)\frac{1}{|\mathcal{V}|}\sum_{j\in\mathcal{V}}\mathbb{I}(d_j<\kappa)$。**Stage 5 推理**:图像+VLM编码→autoregressive生成关键点名+2D像素+3D voxel→反量化到metric camera坐标。**Stage 6 下游**:3D关键点可作为anchor做global mesh recovery,或直接喂给IK做机器人控制。
技术新颖性
技术新颖性体现在四个层面。**层面一:任务感知范式**。SAM 3D Body等现有方法无差别输出全身22点,本文通过prompt动态选择1-6个task-relevant点,是首个把'任务条件'显式建模进3D关键点定位的工作。**层面二:近距场景适配**。通过焦距统一 + 主点random crop的intrinsic-conditioned augmentation(参考DepthLM),避免训出相机参数adapter的过拟合问题,让单一模型适配不同内参。**层面三:2D-as-CoT推理**。把2D keypoint作为3D推理的intermediate reasoning step,Table 3验证可带来33mm的G-MPJPE下降,这是把chain-of-thought从NLP迁移到3D vision的成功案例。**层面四:GRPO奖励设计**。奖励组合Huber-style平滑项 $\rho_\delta(d)$ 和PCK-style成功项 $\mathbb{I}(d_j<\kappa)$,比直接MSE更适合RL训练(Table 3显示用RMSE会导致G-MPJPE飙升至795.24mm,因为MSE对outlier敏感导致biased gradient)。这套奖励设计让TAIHRI在4B参数规模下超越235B的Qwen3-VL。
实验结果
核心发现可总结为三点。**发现一:TAIHRI全面超越SOTA 3D HMR方法**。Table 1显示在Harmony4D-Egocentric的四个body-part配置(Upper/Lower/L-Upper/R-Upper)上,TAIHRI-4B分别达到93.83/114.98/107.81/98.23mm,比CameraHMR(167.50/165.72/179.22/169.26mm)平均降低约44%,比SAM 3D Body-DINOv3(124.91/127.80/143.13/123.58mm)平均降低约20%。在EgoBody上也全面领先:Upper 75.77mm vs CameraHMR 94.92mm(提升20%),R-Upper 81.27mm vs CameraHMR 99.45mm(提升18%)。**发现二:比通用VLM和depth方法有数量级优势**。Table 2显示TAIHRI Full Model在Upper body任务上仅97.2mm,而Qwen3-VL-235B-A22B-Instruct高达1298.3mm(13倍差距),Gemini-2.5 Pro为436.9mm,VP+Depth Anything 3为352.2mm,VP+DepthLM为282.3mm。GPT-5.2无法输出3D坐标。**发现三:消融验证各模块必要性**。Table 3显示w/o Cam Int.导致Upper G-MPJPE升至425.13mm(4.5倍退化),w Ray Emb.升至380.29mm,w/o 2D reasoning升至126.67mm,w/o RFT升至101.82mm,证明每个组件都不可或缺。Figure 5可视化进一步展示TAIHRI在手腕等远端关键点上误差仅15-50mm,而CameraHMR等方法动辄400-900mm。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| Harmony4D-Egocentric Upper body关键点定位 | G-MPJPE (mm),越低越好 | 93.83 | 167.50 (CameraHMR) | 44%↓ |
| Harmony4D-Egocentric Lower body | G-MPJPE (mm) | 114.98 | 165.72 (CameraHMR) | 31%↓ |
| Harmony4D-Egocentric R-Upper(远端手臂) | G-MPJPE (mm) | 98.23 | 169.26 (CameraHMR) | 42%↓ |
| EgoBody Upper body | G-MPJPE (mm) | 75.77 | 94.92 (CameraHMR) | 20%↓ |
| EgoBody R-Upper | G-MPJPE (mm) | 81.27 | 99.45 (CameraHMR) | 18%↓ |
| 通用VLM对比(Upper body) | G-MPJPE (mm) | 97.2 | 436.9 (Gemini-2.5 Pro) | 78%↓ (4.5x) |
| vs Qwen3-VL-235B-A22B-Instruct | G-MPJPE (mm) | 97.2 | 1298.3 | 13x↓ |
| vs DepthLM (VitPose+DepthLM) | G-MPJPE (mm) | 97.2 | 282.3 | 65%↓ |
局限与改进
局限性分三层。**作者承认的局限**:(1) 仅在近距(≤3m)场景评估,对远距离或多人场景未验证;(2) 关键点集合限于SMPL-X预定义的人体骨骼,未涵盖手部22个关节或面部表情细节;(3) 强化微调只在1万张采样图像上进行,理论上更大规模RFT可能进一步提升性能;(4) 推理阶段需prompt支持,无prompt时只能用默认模板定位1-6个关键点。**我的额外观察**:(1) 模型规模仅4B,相比Qwen3-VL-235B等大模型小两个数量级却精度远超,这说明task-specific微调比scaling更有效,但也意味着对训练分布外长尾场景的泛化能力存疑;(2) Table 2的VLM对比仅在50个样本上做,对Gemini-2.5 Pro的测评可能因prompt engineering不足而偏高估计其误差;(3) 推理速度未报告,VLM的autoregressive生成在实时机器人控制中可能成为瓶颈(通常需≥10Hz);(4) CloseHRI数据集基于SDXL合成,sim-to-real gap未充分量化;(5) 评测只用了Upper/Lower/L-Upper/R-Upper四个body-part配置,未评估单手22点等更细粒度任务。
独立分析的弱点
独立分析的核心弱点有四个。**弱点一:自回归生成延迟**。TAIHRI需要依次生成关键点名→2D→3D token,假设每个响应50-100个token,按Qwen3-VL 4B推理速度估算单帧约500ms-1s,远达不到机器人30Hz控制频率。改进方向:用teacher forcing+parallel decoding,或把多关键点的3D预测并行化。**弱点二:prompt bank覆盖面有限**。6000条交互指令虽多,但实际HRI场景的指令空间是开放的,遇到训练集外措辞(如'扶我起来' vs 'assist to stand')可能退化。改进方向:引入instruction tuning或在prompt中加入few-shot example。**弱点三:close-range专项的代价**。模型在>3m范围未训练,对于机器人在大空间导航+局部操作的混合任务可能失效。改进方向:远-近混合数据训练或多尺度fusion。**弱点四:sim-to-real gap**。CloseHRI基于Blender+SDXL合成,合成图的纹理、光照、衣物褶皱与真实机器人相机捕获有差距,文中缺乏sim-to-real消融实验。改进方向:在真实场景中finetune或加domain randomization。**弱点五:缺少多人/遮挡处理**。真实HRI场景常涉及多人交互或物体遮挡,但训练集CloseHRI主要是单人在0.5-3m视角下的合成图,模型在多人场景下的关键点分配('left elbow'对应哪个人)能力未验证。
未来方向
未来研究方向可分两类。**作者暗示的方向**:(1) 扩展到包含手部22点+面部关键点的全身精细定位,支撑更精细的manipulation任务;(2) 把TAIHRI的3D预测直接作为机器人manipulation policy的input,构建end-to-end perception-action model;(3) 探索更大backbone(如Qwen3-VL 8B/32B)以扩展能力上限;(4) 把交互指令从prompt bank扩展到open-vocabulary。**基于成果可延伸的方向**:(1) 把discretized interaction space从单一cuboid扩展到multi-zone(如机器人的reach space + 安全区域 + 协作区);(2) 把GRPO的pose-aware reward替换为下游任务reward(如抓取成功率、按摩接触力),实现真正的end-to-end optimization;(3) 引入temporal modeling,把单帧扩展到video-level的keypoint tracking,应对遮挡补全和motion prediction;(4) 探索TAIHRI的zero-shot generalization到不同机器人形态(mobile manipulator vs humanoid vs 灵巧手);(5) 与HOI(human-object interaction)结合,理解人在操作工具时的3D姿态以辅助协作任务;(6) 把任务感知的注意力机制可视化,研究VLM是否真正学到了human anatomy的semantic grounding。
复现评估
复现评估分四维。**代码与数据**:作者声明代码开源于https://github.com/Tencent/TAIHRI,但论文写作时仓库内容待确认;CloseHRI数据集未明确是否公开(仅说'to be released',120万张SDXL合成图加多源数据集的体量可能涉及授权问题)。**算力门槛**:训练用4块NVIDIA H20 GPU(每块约96GB显存,H20是华为昇腾衍生品,主要在国内可获取),SFT 5 epoch + RFT 5 epoch,预计总训练时间7-14天。H20对海外研究者复现成本较高,理论上可用A100/H100替代但成本上升。**复现难度**:中高。技术栈涉及Qwen3-VL微调、GRPO实现、SDXL图像合成管线、VitPose+SAM3质量过滤、Blender渲染SMPL-X normal map等多个组件,每个环节都有调参空间,特别是intrinsic unification的resize规则需严格对齐。**潜在风险**:(1) GRPO的奖励权重 $\lambda$ 和阈值 $\delta, \kappa$ 未明确报告,文中仅给出框架;(2) VitPose的15像素过滤阈值也可能引入数据分布偏差;(3) 强化学习训练随机性较大,论文未报告多次运行的方差;(4) prompt bank的6000条指令未公开,复现者只能随机采样类似模板,效果可能下降。
论文图表
展示了TAIHRI在4个典型HRI任务上的3D关键点定位结果:shake right hand(握右手)、give shoulder massage(肩膀按摩)、catch right arm(接住右臂)、lift from underarms(从腋下抬起)。每个子图标注了预测的3D坐标(如(-0.15,-0.16,0.67))和GT误差(error=15.79mm等),直观展示了模型在近距离task-relevant关键点上的高精度。
这张图是论文teaser,把模型能力凝缩到4个具体场景,是理解论文motivation和method capability的第一入口。
对比图左侧的'large-scale full body'传统HMR范式(root joint评估)和右侧的'close-up local part' HRI感知范式(camera coordinate评估)。展示了视角从第三视角大场景到机器人在胸口/头部第一人称视角的转变,以及评估坐标系从根相对到camera space的转变。
精确定义了本文要解决的问题场景,是problem statement的视觉化版本,对理解motivation至关重要。
数据生成pipeline:从AMASS采样motion sequence → 用SMPL-X在Blender渲染normal map(相机距离0.5-3m)→ 输入SDXL合成photo-realistic图像(带prompt如'a person wearing red shirt')→ 用SAM3过滤IoU<0.9的低质量样本 → 用VitPose过滤2D reprojection error>15像素的样本。最终产出100万张CloseHRI数据集。
展示了训练数据的来源和构建流程,对理解method的数据基础、sim-to-real gap的讨论、以及复现工作至关重要。