TRIAGE : une inférence vision-langage-action efficace exige à la fois réutilisation et élagage
Des chercheurs proposent TRIAGE, une méthode d'inférence sans réentraînement pour accélérer les modèles vision-langage-action (VLA), ces réseaux qui transforment images de caméra et instruction en langage naturel en actions de robot. Le problème visé concerne les robots non reliés à un poste de travail, comme les manipulateurs mobiles et les humanoïdes. Ils doivent exécuter ces modèles sur des cartes embarquées, où la latence de « prefill » du backbone langage croît avec les centaines de tokens visuels traités à chaque pas de contrôle. Les accélérateurs existants ne recalculent qu'un sous-ensemble de tokens visuels. Ils élaguent les autres ou réutilisent leurs états clé-valeur (KV) en cache, mais appliquent une seule option à tous les tokens ignorés. Sur LIBERO et SimplerEnv, TRIAGE reste à moins de 3 points de pourcentage du recalcul complet en taux de succès moyen. Il réduit la latence de prefill jusqu'à 48 % sur un Jetson AGX Orin. Avec seulement 7 % des tokens visuels recalculés, il dépasse de 24,4 points la meilleure méthode de référence sur LIBERO-Object.
Le point clé est un diagnostic au niveau du token. Le choix entre réutilisation et élagage dépend de la « dérive » du token, c'est-à-dire l'écart de sa valeur en couches superficielles depuis le calcul de son cache. Sur les deux backbones testés, réutiliser un token sous un certain seuil de dérive n'est pas pire que l'élaguer. Au-delà, la réutilisation produit jusqu'à 4,4 fois plus d'erreur d'action que l'élagage. Le même diagnostic permet de calibrer ce seuil pour chaque backbone. Pour un intégrateur, cela contredit l'idée qu'une seule stratégie de saut suffit, et cela donne une règle simple pour tenir un budget temps réel. Le budget de tokens recalculés fixe directement la latence de prefill, ce qui la rend prévisible. Les limites sont à garder en vue : les résultats viennent de benchmarks simulés, avec un gain de latence mesuré sur une seule plateforme embarquée. Aucune validation sur robot réel n'est annoncée.
TRIAGE tient un cache pour valide si sa dérive est sous le seuil et s'il n'est pas trop ancien. Il recalcule un budget fixe de tokens en priorisant les tokens importants dont le cache est invalide. Il réutilise tous les autres tokens à cache valide et élague le reste. Les méthodes qui élaguent seulement ou réutilisent seulement en sont les deux cas extrêmes. Ces travaux s'inscrivent dans la recherche d'inférence efficace pour VLA, portée par la contrainte des cartes comme le Jetson AGX Orin sur les plateformes mobiles. Il s'agit d'une prépublication arXiv, sans code ni calendrier de déploiement mentionnés. La suite logique serait une validation sur matériel réel et sur d'autres familles de VLA.
Dans nos dossiers




