CoTinyVLA: Chain-of-Thought Distillation for a Sub-Billion-Parameter Vision-Language-Action Model
CoTinyVLA is een Vision-Language-Action-model met minder dan een miljard parameters dat een state-of-the-art robuustheid bereikt op de LIBERO-Plus benchmark door gebruik te maken van gestructureerde supervisietechnieken — waaronder dual-view temporele inputs, hiërarchische chain-of-thought distillatie en parafrase-augmentatie — in plaats van de modelgrootte te vergroten.
Oorspronkelijk artikel gelicentieerd onder CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). Dit is een AI-gegenereerde uitleg van het onderstaande artikel. Het is niet geschreven of goedgekeurd door de auteurs. Raadpleeg het oorspronkelijke artikel voor technische nauwkeurigheid. Lees de volledige disclaimer
Stel je een wereld voor waarin robots niet alleen zielloze machines zijn die rigide codes volgen, maar behulpzame metgezellen die onze gesproken woorden begrijpen en kunnen uitzoeken hoe ze hun eigen armen moeten bewegen om taken uit te voeren. Dit is de droom van "embodied AI", een veld waar computers leren de wereld te zien via camera's, menselijke taal te begrijpen en die twee zaken te vertalen naar fysieke acties. Een lange tijd hadden de slimste robots enorme hersenen nodig—computermodellen met miljarden "neuronen"—om dit te kunnen afhandelen. Deze gigantische modellen waren als supercomputers die enorme, dure servers nodig hadden om te draaien, waardoor het onmogelijk was om ze in een kleine robot te passen die door een huis kan rollen of kan helpen in een keuken. De grote vraag die onderzoekers zich hebben gesteld is: Kunnen we een robot maken die net zo slim en betrouwbaar is, maar klein genoeg om in een rugzak te passen?
Dit artikel introduceert een nieuwe robothersenen genaamd CoTinyVLA. Beschouw dit als een kleine, superefficiënte robothersenen die erin slaagt om de enorme, zware modellen die momenteel het veld leiden, te overtreffen, terwijl ze slechts een fractie van het geheugen gebruiken. De onderzoekers hebben de hersenen niet alleen gekrompen; ze hebben de robot ook een betere manier van denken geleerd. In plaats van te proberen elke mogbare situatie te memoriseren, gaven ze de robot een "spiekbriefje" voor redeneren. Ze leerden de robot om een grote taak op te splitsen in een simpel plan (zoals een to-do lijstje) en vervolgens over elke kleine stap na te denken voordat hij beweegt. Ze gaven de robot ook een speciaal paar ogen—één die van een afstand kijkt om de hele kamer te zien, en één op de pols om precies te zien wat de hand aanraakt—en leerden hem om een korte video van de laatste paar seconden te bekijken om beweging te begrijpen. Het resultaat is een robotmodel met slechts 0,9 miljard parameters (minuscuul vergeleken met de reuzen van 7 miljard parameters) dat complexe, rommelige situaties in de echte wereld beter kan afhandelen dan modellen die acht keer zo groot zijn.
Het Probleem: Grote Hersenen, Kleine Robots
Jarenlang waren de beste robots voor het opvolgen van instructies als enorme, zware tanks. Om een commando als "pak de blauwe beker en zet hem op de tafel" te begrijpen, gebruiken deze robots enorme computermodellen met 3 tot 7 miljard parameters. Hoewel deze modellen ongelooflijk slim zijn, zijn ze ook enorm. Ze vereisen ongeveer 20 gigabyte aan computergeheugen om te draaien. Dat is alsof je een complete bibliotheek probeert te proppen in een rugzak; het past er simpelweg niet in op de kleine, batterijgestuurde computers die in mobiele robots of hulpmiddelen worden gevonden.
De onderzoekers wilden weten: Hebben we echt zo'n gigantische hersenpan nodig om slim te zijn? Of kunnen we een kleine, efficiënte robot bouwen die net zo goed is in het omgaan met de onverwachte rommeligheid van de echte wereld?
De Oplossing: Een Kleine Hersenen met een Grote Strategie
Het team bouwde CoTinyVLA, een robothersenen met slechts 0,9 miljard parameters. Om dit kleine model even sterk te maken als de reuzen, hebben ze niet alleen meer data gevoerd; ze hebben de manier waarop het leert en waar het op let, veranderd. Ze gebruikten drie belangrijke trucs, die ze "gestructureerd toezicht" noemen, om de robot beter te leren denken.
1. De "Twee-Ogen" Tijdsmachine
Stel je voor dat je een bal probeert te vangen. Als je alleen één foto van de bal ziet, weet je niet of hij naar je toe komt of van je af beweegt. Je moet een korte video zien om de beweging te begrijpen.
CoTinyVLA wordt geleerd om naar de wereld te kijken door twee verschillende "ogen" tegelijkertijd:
- Het Derde-Persoonsperspectief: Een camera die naar de hele kamer kijkt om te zien waar alles is.
- Het Polsperspectief: Een camera op de hand van de robot om precies te zien wat de grijper aanraakt.
In plaats van alleen naar het huidige moment te kijken, bekijkt de robot een korte "film" van de laatste 16 frames (8 van elk oog). Dit geeft het een gevoel van tijd en beweging, wat hels bij de robot helpt om te begrijpen hoe snel dingen bewegen en waar ze naartoe gaan.
2. Het "Plannen en Denken" Spiekbriefje
Dit is de belangrijkste truc. Voordat de robot beweegt, wordt hem geleerd om een verhaal op te schrijven.
- Het Plan: Aan het begin van een taak schrijft de robot een hoogwaardige "to-do lijst" (bijv. "Eerst de beker pakken. Tweede, de beker optillen. Derde, de beker op de tafel zetten"). Dit plan blijft hetzelfde voor de hele taak.
- Het Denken: Voordat elke kleine beweging wordt gemaakt, schrijft de robot een korte notitie over wat hij op dit moment doet (bijv. "Ik sluit mijn grijper" of "Ik til de beker op").
De robot leert dit door te kijken naar een veel grotere, slimmere robot (een "leraar" met 35 miljard parameters) die dezelfde taken oplost. De kleine robot kopieert het denkproces van de leraar. Dit helpt de kleine robot om te begrijpen waarom hij iets doet, en niet alleen wat hij moet doen.
3. Het "Parafraseren" Spelletje
Robots raken vaak in de war als je hetzelfde op een andere manier zegt. Als een robot alleen getraind is op de zin "pak de beker", kan hij vastlopen als je zegt "grijp de mok".
Om dit op te lossen, hebben de onderzoekers de robot getraind met 40 basisinstructies en deze uitgebreid naar 800 verschillende versies. Ze vervingen woorden (zoals "pakken" veranderen in "grijpen" of "optillen"), veranderden de beschrijvingen van objecten (zoals "zwarte kom" naar "donkere kom") en voegden zelfs beleefde zinnen toe zoals "Zou je alsjeblieft...". Dit leerde de robot om de betekenis van het commando te begrijpen, en niet alleen de specifieke woorden.
De Resultaten: Klein Wint Groot
De onderzoekers testten CoTinyVLA op LIBERO-Plus, een uitdagende testsuite die ontworpen is om te zien hoe goed robots omgaan met veranderingen in de wereld. De test bevat meer dan 10.000 verschillende scenario's waarin dingen mis zijn: de robot begint in een vreemde positie, de verlichting verandert, de achtergrond is anders, of de instructies zijn vreemd geformuleerd.
De resultaten waren verrassend. CoTinyVLA, met zijn kleine 0,9 miljard parameters, versloeg de sterkste 7-miljard-parameter modellen op alle vier de belangrijke testcategorieën:
- Ruimtelijk (Spatial): 90,8% succes (verslaat de grote modellen met 4,7 punten).
- Object: 87,3% succes (verslaat de grote modellen met 2,8 punten).
- Doel (Goal): 86,6% succes (verslaat de grote modellen met maar liefst 15,9 punten).
- Lang (Long): 80,7% succes (verslaat de grote modellen met 3,0 punten).
De grootste overwinning was in de categorie "Doel", waar de kleine robot bijna 16 punten beter was dan de beste reusachtige robot. Dit is vooral indrukwekkend omdat de kleine robot draait op een computer die slechts 2,25 GiB aan geheugen gebruikt, wat klein genoeg is om op veel echte robots te passen.
Waarom het ertoe doet
Dit artikel laat zien dat je geen enorme hersenpan nodig hebt om een slimme robot te zijn. Door een kleine robot te leren om op een gestructureerde manier te "denken" (plannen maken en zijn stappen controleren) en door hem de juiste visuele en talige training te geven, kan hij de rommelige, onvoorspelbare echte wereld beter aan dan veel grotere modellen.
De onderzoekers ontdekten ook dat het "Plan"-gedeelte van het denkproces cruciaal is. Als je de mogelijkheid van de robot wegneemt om een plan te schrijven, daalt het succespercentage met 40 tot 45 punten. Dit bewijst dat de robot niet zomaar gokt; hij gebruikt het plan daadwerkelijk om zijn acties te sturen.
Kortom, CoTinyVLA bewijst dat met de juiste onderwijsmethoden een kleine, efficiënte robot net zo capabel kan zijn als een enorme supercomputer, wat ons een stap dichter bij behulpzame, intelligente robots in onze huizen en werkplaatsen brengt.
Verdrinkt u in papers in uw vakgebied?
Ontvang dagelijkse digests van de nieuwste papers die bij uw onderzoekswoorden passen — met technische samenvattingen, in uw taal.