Prism-GRPO: Faster VLA Policy Optimization via Splitting Same-outcome Groups
Prism-GRPO versnelt de optimalisatie van Vision-Language-Action-beleid door binaire succesbeloningen aan te vullen met scores voor de kwaliteit van de uitvoering om groepen met hetzelfde resultaat te splitsen, waardoor trainingssignalen worden teruggewonnen uit weggegooide rollouts en tot 56% minder rollouts nodig zijn om de beoogde succespercentages te bereiken.
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
Robots die kunnen zien, taal begrijpen en met een doelgerichtheid bewegen, zijn niet langer sciencefiction. Ze worden gebouwd met behulp van een type kunstmatige intelligentie dat een vision-language-action model wordt genoemd. Deze systemen fungeren als een brug tussen wat de robot ziet en wat er tegen hem gezegd wordt. Stel je een robot voor die hoort: "pak de rode beker op" en een tafel met een beker ziet; de robot moet vervolgens de precieze bewegingen van zijn arm berekenen om deze vast te grijpen. Om deze robots beter te laten worden, gebruiken onderzoekers vaak een methode genaald reinforcement learning (versterkingsleren). In dit proces probeert de robot een taak, en als hij slaagt, krijgt hij een beloning. Als hij faalt, krijgt hij niets. Over duizenden pogingen leert de robot welke acties leiden tot succes. Dit leerproces is echter ongelooflijk duur. Elke keer dat de robot een taak probeert, verbruikt het tijd en rekenkracht. Als de robot herhaaldelijk faalt, of als hij op een onhandige manier slaagt, gooit de computer die poging vaak weg, waarbij het als nutteloze data wordt beschouwd. Deze verspilling is een grote flessenhals die de ontwikkeling van robots die veilig en efficiënt in de echte wereld kunnen werken, vertraagt.
Een team van onderzoekers heeft een nieuwe aanpak ontwikkeld om deze verspilling aan te pakken, waarbij ze wat voorheen weggegooide data was, veranderen in waardevolle lessen. Hun methode, genaamd Prism-GRPO, verandert de manier waarop de robot zijn eigen pogingen evalueert. In de standaardmanier van trainen krijgt een robot een eenvoudige score van geslaagd of niet geslaagd. Als een groep pogingen allemaal slaagt, krijgen ze allemaal dezelfde perfecte score. Als ze allemaal falen, krijgen ze allemaal dezelfde nulscore. Omdat de scores identiek zijn, kan de computer niet zien welke poging beter was dan de andere, waardoor de computer de hele groep wegwerpt om tijd te besparen. De onderzoekers realiseerden zich dat zelfs wanneer een robot faalt, of zelfs wanneer hij slaagt, er vaak subtiele verschillen zijn in hoe hij presteerde. Eén succesvolle poging was misschien vloeiend en zacht, terwijl een andere schokkerig was en objecten in de buurt omverstootte. Eén mislukte poging was misschien dicht bij het doel, terwijl een andere er wild naast zat. Door deze nuances te negeren, gooide de oude methode precies de informatie weg die nodig is om de robot nauwkeuriger te laten worden.
De nieuwe methode lost dit op door een tweede laag feedback toe te voegen. In plaats van alleen te vragen "Is het gelukt?", vraagt het systeem ook: "Hoe goed is het gelukt?". Het wijst een kwaliteitsscore toe op basis van de fysieke uitvoering van de taak, zoals hoeveel kracht er werd gebruikt, hoe vloeiend de bewegingen waren, of of de robot per ongeluk dingen raakte die hij niet had mogen aanraken. Deze score wordt gecombineerd met het geslaagd-of-niet-geslaagd resultaat. Nu, zelfs als een groep pogingen allemaal slaagt, kan het systeem zien dat de ene schoner was dan de andere en dit dienovereenkomstig belonen. Op dezelfde manier, als een groep allemaal faalt, kan het systeem identificeren welke mislukking de minst schadelijke was en deze als een leersignaal gebruiken. Deze eenvoudige verandering betekent dat bijna geen enkele data verloren gaat. De robot leert van elke poging, niet alleen van de zeldzame gevallen waarin sommigen slagen en anderen niet.
De onderzoekers testten dit idee op vier verschillende robotische taken, variërend van het tillen van een pot met twee armen tot het verplaatsen van een blikje en het plaatsen ervan naast een pot. Ze ontdekten dat hun nieuwe methode de robot in staat stelde om hetzelfde succesniveau te bereiken met tot wel 56 procent minder pogingen dan de standaardmethode. Dit is een aanzienlijke besparing in tijd en rekenkracht. Belangrijker nog, de robots die met deze nieuwe methode werden getraind, leerden niet alleen te slagen; ze leerden om gracieus te slagen. Ze waren minder geneigd om "shortcuts" of sluiproutes te ontwikkelen. Bijvoorbeeld, in een taak waarbij de robot een blikje moest optillen en naast een pot moest plaatsen, leerde de oude methode de robot soms om de pot simpelweg naar het blikje te duwen in plaats van het blikje op te tillen. Dit voldeed aan de succescontrole van de computer, maar was niet wat de mens had gevraagd. De nieuwe methode, door de ruwe duwbeweging te bestraffen, leerde de robot de juiste, zachte til-en-plaatsactie uit te voeren.
Toen de onderzoekers de best presterende robots uit de simulatie op een echte fysieke robot plaatsten, hielden de resultaten stand. De robots getraind met de nieuwe methode waren betrouwbaarder en minder geneigd om de onhandige sluiproutes uit te voeren die in de simulatie hadden gewerkt maar in de echte wereld faalden. De studie laat zien dat door aandacht te besteden aan de kwaliteit van een actie, en niet alleen aan de uiteindelijke uitkomst, we robots veel sneller kunnen trainen en ze veiliger kunnen maken. De onderzoekers bewezen wiskundig dat deze aanpak het vermogen van de robot om de hoofdtaken te leren nooit schaadt, en hun experimenten bevestigden dat het consequent zowel de snelheid als het gedrag verbetert. Dit werk suggereert een duidelijke weg vooruit om robots te maken die niet alleen in staat zijn om een taak uit te voeren, maar ook in staat zijn om dit te doen met de zorg en precisie die vereist is om naast mensen te werken.
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.