Trace-WAM : apprendre à agir par la prédiction de mouvement 3D pour les modèles d'action du monde
Des chercheurs proposent Trace-WAM, une architecture de World Action Model (WAM) publiée en préprint sur arXiv. Un WAM apprend à piloter un robot en couplant la prédiction de vidéo future et la génération d'actions. Trace-WAM ajoute à cet apprentissage un objectif auxiliaire: prédire le mouvement 3D futur de points suivis dans la scène. Cette supervision combine la correspondance temporelle des points, leur déplacement dans le plan image et leur variation de profondeur. L'ensemble repose sur une architecture à trois branches de type Mixture-of-Transformer (Tri-MoT), qui entraîne conjointement un expert vidéo, un expert mouvement et un expert action. L'expert mouvement prédit les trajectoires à partir des features visuelles courantes, et l'expert action ne consomme pas ces trajectoires prédites. À l'inférence, le système garde donc un chemin de politique classique: une seule passe visuelle, suivie du débruitage de l'action. Les tests portent sur RoboCasa-GR1, LIBERO et LIBERO-Plus, avec de bons résultats annoncés en distribution et en généralisation hors distribution, surtout face à Fast-WAM quand les démonstrations sont rares ou que la distribution change.
L'intérêt tient à l'argument central: pour manipuler, il faut surtout savoir comment évoluent le robot et les objets pertinents, pas reconstruire l'apparence et l'arrière-plan de toute la scène. Si cela se confirme, on pourrait obtenir des représentations plus utiles pour l'action sans alourdir l'inférence, ce qui compte pour la latence en production. Le gain annoncé en régime de peu de démonstrations parle aux intégrateurs, car la collecte de données reste le principal coût de déploiement. Il faut toutefois rester prudent: le résumé ne donne aucun chiffre, et les évaluations semblent limitées à des benchmarks simulés, sans test sur robot réel ni mesure de temps de cycle.
Trace-WAM s'inscrit dans le passage des VLA (vision-language-action) vers des modèles qui intègrent une prédiction du monde, avec Fast-WAM comme point de comparaison direct. RoboCasa-GR1 évalue des tâches de type humanoïde en simulation, tandis que LIBERO-Plus teste la robustesse aux perturbations. La suite logique serait une validation sur matériel réel et une comparaison avec les grandes politiques généralistes actuelles, ce que le préprint n'annonce pas.
Pas d\'impact direct sur la France/UE
Dans nos dossiers




