Search papers, labs, and topics across Lattice.
This paper introduces Critical Interval MSE (CI-MSE), an offline validation metric designed to enhance the correlation between validation loss on expert demonstrations and real-world performance for robot manipulation policies. By focusing error computation on task-critical segments and aligning actions with rollout-time behavior, CI-MSE significantly outperforms traditional raw MSE, achieving a Spearman's rank correlation of -0.87 compared to -0.61 for raw MSE. The robustness of CI-MSE across various hyperparameters and evaluation distribution shifts positions it as a valuable tool for accelerating policy iteration in robotics.
CI-MSE dramatically improves the correlation between offline validation and real-world performance, making it a game-changer for robot policy evaluation.
Real-world evaluation is the gold standard for robot policies because it tests them against the physical conditions and deployment challenges they are ultimately designed to handle. However, real-world evaluation is also the bottleneck for iterating on robot policies: it is costly, difficult to reproduce, and often too sparse to reliably compare nearby model variants. A straightforward proxy for performance is validation loss on expert demonstrations, but this proxy is often poorly correlated with real-world performance. In this paper, we introduce Critical Interval MSE (CI-MSE), an intuitively simple yet effective offline validation metric. CI-MSE restricts error computation to task-critical segments and pairs it with simple action-alignment procedures that better match rollout-time behavior. Across simulation and real-world experiments, CI-MSE yields a stronger correlation between validation error and rollout performance than raw MSE. Across a wide range of policy checkpoints, CI-MSE achieves a Spearman's rank correlation of $-0.87$, much closer to the ideal value of $-1$ than raw MSE's $-0.61$, demonstrating a significant improvement. We show through sensitivity analysis that our metric is robust to a wide range of hyperparameters. We further study the effectiveness of CI-MSE under evaluation distribution shifts and suggest design boundaries when using this metric. In summary, this paper provides a simple and reliable offline validation tool for accelerating policy iteration. Project webpage: https://ci-mse.github.io/