DIRECT: When and Where Should You Allocate Test-Time Compute in Embodied Planners?
About
Vision-Language Models (VLMs) are increasingly deployed as high-level planners for embodied agents, with an emerging strategy of scaling test-time compute to improve capability. However, we observe that doing so increases latency, token usage, and FLOPs while yielding uneven, often diminishing gains in downstream success, limiting where embodied agents can be deployed. We argue that choosing when and where to spend test-time compute is central to bringing frontier performance to the real world. We introduce DIRECT, a routing framework that uses multimodal scene context to allocate compute per prompt, improving the success--cost Pareto frontier over fixed model selection. Across three dominant scaling axes, namely chain-of-thought depth, model size, and memory history, our experiments on VLABench and RoboMME show that test-time compute is not a uniform lever: different axes yield qualitatively distinct capability gains. We validate these insights on a physical Franka arm in a DROID setup spanning zero-shot manipulation and long-horizon chaining, where our router matches or exceeds a stronger model's success rate at up to 65% lower average latency. Ultimately, our results show that naively scaling test-time compute is wasteful, and that DIRECT can provide frontier-level embodied planning in robotic systems at a fraction of the cost. Project page can be found at jadee-dao.github.io/direct/.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Language-conditioned Manipulation Planning | VLABench | Success Rate64.83 | 35 | |
| High-level Planning | VLABench vlm_evaluation 1.0 (test) | AUC59.57 | 25 | |
| Physical Routing | DROID memory axis (physical evaluation) | Q (%)100 | 10 | |
| Physical Routing | Physical routing skill command model-size axis | Q (%)93.1 | 10 | |
| Physical task routing | Physical Suite Intelligence/Thinking axis (test) | Success Rate (Q)60.87 | 10 | |
| Multi-step planning and manipulation | Multi-step grocery bagging | Success Rate95.24 | 3 |