Time-to-Goal : des métriques temporelles apprises pour la planification à long horizon avec les modèles du monde JEPA
Des chercheurs proposent Time-to-Goal (T2G), une métrique temporelle apprise qui corrige un défaut de planification des modèles du monde de type JEPA (Joint-Embedding Predictive Architectures). Ces architectures planifient en comparant les états latents prédits à un but encodé. Pour que cette comparaison guide la progression, les états proches du but dans le temps doivent recevoir un coût plus faible. Or l'entraînement des JEPA, centré sur la prédiction et la régularisation de la distribution, n'impose pas cet ordre. Les auteurs analysent LeWorldModel (LeWM) dans sa version plate, et identifient des conditions où des états situés à des décalages temporels longs et différents reçoivent des coûts euclidiens moyens similaires, ce qui efface leur distinction. T2G est une métrique légère entraînée sur des triplets latents issus de trajectoires hors ligne, avec une perte de classement à marge (margin ranking loss) qui favorise des distances plus faibles pour les états temporellement plus proches. Elle ne demande ni récompenses ni étiquettes de politique. Elle remplace le coût euclidien lors de la planification, pendant que l'encodeur et le prédicteur JEPA restent gelés. Sur quatre benchmarks de navigation et de manipulation, le taux de succès moyen passe de 32,7 % (LeWM plat) à 63,0 %.
Pour la robotique, le résultat montre qu'une partie des échecs de planification à long horizon vient de la fonction de coût et non de la représentation apprise. Corriger le coût, sans réentraîner le modèle du monde, presque double le succès moyen. Les diagnostics sur AntMaze et FourRoom confirment un meilleur contraste temporel et un meilleur classement entre états proches et lointains à grands décalages. T2G atteint ou dépasse par endroits les références hiérarchiques H-JEPA, ce qui suggère qu'un modèle plat peut rivaliser avec des architectures plus complexes sur ces tâches. Pour un ingénieur ou un intégrateur, l'intérêt est la modularité : un module léger et sans récompense se greffe sur un modèle pré-entraîné et gelé, à partir de trajectoires déjà collectées. Les limites sont nettes. Il s'agit de travaux de recherche évalués sur des benchmarks simulés, sans robot réel, sans données de latence ni de temps de planification. Le chiffre moyen masque aussi la variabilité entre tâches, et il reste à voir si le gain tient sur des scènes visuelles réelles et bruitées.
Ces travaux s'inscrivent dans le courant des modèles du monde prédictifs en espace latent, popularisé par les JEPA, qui prédisent des représentations plutôt que des pixels. La planification par comparaison à un but encodé hérite de cette approche, et les variantes hiérarchiques (H-JEPA) ont été proposées pour traiter les longs horizons en décomposant la tâche. T2G prend la voie inverse : garder un modèle plat et rendre le coût temporellement cohérent. Il se place ainsi face aux méthodes hiérarchiques, mais aussi face aux politiques VLA entraînées de bout en bout, qui évitent la planification explicite. Les suites logiques sont des tests sur des modèles du monde plus grands, sur des tâches de manipulation plus longues et sur du matériel réel. L'article, publié sur arXiv (2611.05847v1), est une première version qui n'a pas encore été relue par des pairs.
Dans nos dossiers




