← 返回 2026-04-14

TAIHRI:面向近距人机交互的任务感知3D人体关键点定位 TAIHRI: Task-Aware 3D Human Keypoints Localization for Close-Range Human-Robot Interaction

Ao Li, Yonggen Ling, Yiyang Lin, Yuji Wang, Yong Deng, Yansong Tang 📅 2026-04-10 👍 2 2026-07-13 08:36
3D人体姿态估计 GRPO 人机交互 任务感知 强化学习微调 视觉-语言模型

首个面向近距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 inference procedure overview
Fig. 4: TAIHRI inference procedure overview

实验结果

核心发现可总结为三点。**发现一: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。

Quantitative comparison on Harmony4D-Egocentric and EgoBody test samples
Table 1: Quantitative comparison on Harmony4D-Egocentric and EgoBody test samples
Comparison with mainstream VLMs and vision foundation models
Table 2: Comparison with mainstream VLMs and vision foundation models
Ablation studies on camera intrinsics injection and training strategy
Table 3: Ablation studies on camera intrinsics injection and training strategy
Visualization comparison with SOTA 3D human pose estimation methods
Fig. 5: Visualization comparison with SOTA 3D human pose estimation methods
Demonstration of TAIHRI applications
Fig. 6: Demonstration of TAIHRI applications
查看结构化数据
任务指标本文基线提升
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条指令未公开,复现者只能随机采样类似模板,效果可能下降。