Rank-Then-Act: Reward-Free Control from Frame-Order Progress

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Robotics & Autonomous Systems · Depth: Advanced, quick

Summary

The Rank-Then-Act (RTA) framework enables learning control policies from expert video demonstrations without relying on environment rewards. RTA trains an offline Vision-Language Model (VLM) as a progress-based ordinal scorer, utilizing a Group Relative Policy Optimization (GRPO) objective on shuffled frame sequences to ensure temporal order recovery from visual semantics. Instead of direct scalar reward use, RTA employs a correlation-based reward function for reinforcement learning, computing the Spearman rank correlation between predicted progress rankings and true temporal indices within each interaction window. This approach provides a bounded, scale-invariant learning signal, decoupling reward learning from absolute calibration and facilitating stable transfer across tasks. Evaluated on discrete control benchmarks like PyBoy (Catrap, Kirby) and continuous control tasks (PointMaze, MetaWorld), RTA consistently matches or surpasses existing video-based reward learning methods and rank-based baselines, demonstrating robust cross-task reuse of a single pretrained scorer.

Key takeaway

For Machine Learning Engineers developing agents from expert demonstrations, Rank-Then-Act (RTA) provides a robust alternative to explicit reward engineering. By utilizing correlation-based rewards derived from video frame order, you can achieve stable policy learning and strong cross-task reuse of a single pretrained scorer. Consider implementing RTA's approach to decouple reward learning from absolute calibration, potentially simplifying complex task setups and improving transferability.

Key insights

Correlation-structured supervision from video-derived ordinal signals enables reward-free policy learning.

Principles

Method

Train a VLM with GRPO on shuffled video frames to predict temporal order. Compute Spearman rank correlation between predicted and true order for RL reward.

In practice

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Robotics Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.