Video wird geladen...
Video konnte nicht geladen werden
Can vision-language-action (VLA) models generalize to diverse OOD tasks and align with customized objectives? 🤔 🚀 We introduce GRAPE, a plug-and-play algorithm to generalize robot policies via preference alignment. GRAPE unfolds three benefits to boost the generalizability of VLAs: 👉1. GRAPE aligns VLAs on a trajectory level and endows... show more
19,988 Aufrufe • vor 1 Jahr •via X (Twitter)
7 Kommentare

[2/N] Detailed Method 1️⃣ Trajectory-wise Preference Optimization: GRAPE scales up step-wise VLAs and trains with a trajectory-wise objective, aligning policies globally by learning from both successes and failures. 2️⃣Customized Preference Synthesis: GRAPE breaks down complex tasks into stages, guided by spatiotemporal constraints from VL models. Flexibly aligns for arbitrary objectives, such as safety, efficiency, or task success. 3️⃣ Iterative Online Alignment: GRAPE refines the alignment process through iterative cycles of 1) online sample collection, 2) synthetic preference ranking, and 3) trajectory-wise preference optimization.

[3/N] Empirical Takeaway 1: Stronger generalizability on a wide array of OOD tasks. 1️⃣ Real-world OOD tasks GRAPE crushes OpenVLA-SFT in generalization: - Visual (new visual environments) 🌆: +20.7% - Subject (unseen objects) 🔍: +27.5% - Action (unseen actions)🏃: +10.0% - Semantic (unseen prompts)🧠: +5.0% - Language grounding (objects in unseen spatial positions)🌍: +26.7% 2️⃣ Simulation OOD tasks In Simpler-Env, GRAPE shines: - Subject (unseen objects) 🔍: +8.0% - Physical (unseen object sizes/shapes) 🏗️: +12.3% - Semantic (unseen prompts)🧠: +19.0%

[4/N] Empirical Takeaway 2: Versatility to align towards customized alignment objectives. GRAPE excels at aligning robot policies with diverse natural language goals: ✅ Task completion ✅ Safety ✅ Cost-efficiency Results: - 🚧 Safer policies: -44.31% collisions - ⏳ Efficient policies: -11.15% rollout lengths

[5/N] Nice work, @ZijianZhangNLP , Kyle Zheng, and nice collab. w/ @ZRChen_AISafety , @jang_yoel , @Yi_Li_UW , @chaoqi_w , @dingmyu , @fox_dieter17849

Cool work!Thank Prof. Yao and our nice collab!

what is the required training resources?

GRAPE seems like a promising leap for VLA models in robotics! The trajectory-level preference alignment and reward modeling are particularly intriguing for enabling safer, more efficient, and task-diverse applications. How scalable is GRAPE to real-world multi-agent environments?
