Voxel heatmap action heads for low-data manipulation fine-tuning
Small end-effector errors still decide many manipulation rollouts. ActionMap gives VLA teams a concrete retrofit to test: replace the native continuous action decoder with a voxel heatmap head over translation, rotation, and gripper commands, then decode continuous actions with top-k soft argmax.
The build is narrow enough for an ablation inside an existing OpenVLA-OFT or pi0.5 training run. Keep the backbone and dataset fixed, train the heatmap head against Gaussian blobs over the action grid, and compare grasp-position error plus task success against the current L1 or flow-matching head. The ActionMap results justify that check: on LIBERO with OpenVLA-OFT, the average rose from 89.1% to 97.3% at matched training steps. With only 43 LIBERO-Spatial demonstrations, it reached 93.2% against 67.2% for the L1 head. On real Franka tasks, it completed 20 of 30 trials against 7 of 30 for the regression head.
This is most useful for teams blocked by scarce demonstrations or millimeter-scale miss rates on grasping, sweeping, and insertion. The cheap validation is a per-task head swap on the current policy, with no backbone change, and a real-robot check that reports both success rate and end-effector error.