Rank-Then-Act: Reward-Free Control from Frame-Order Progress
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
- Decouple reward learning from absolute calibration.
- Temporal ordering from visual semantics is key.
- Ordinal signals can suffice for policy learning.
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
- Use a single pretrained scorer across diverse tasks.
- Apply correlation-based rewards for stable transfer.
- Replace explicit reward design with video supervision.
Topics
- Rank-Then-Act
- Reward-Free Learning
- Vision-Language Models
- Policy Learning
- Video Demonstrations
- Ordinal Progress
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Robotics Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.