Prism-GRPO: Faster VLA Policy Optimization via Splitting Same-outcome Groups
Prism-GRPO acelera la optimización de políticas de Visión-Lenguaje-Acción al aumentar las recompensas binarias de éxito con puntuaciones de calidad de ejecución para dividir los grupos de mismo resultado, recuperando así señales de entrenamiento de ejecuciones descartadas y logrando hasta un 56% menos de ejecuciones para alcanzar las tasas de éxito objetivo.
Artículo original bajo licencia CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). Esta es una explicación generada por IA del artículo a continuación. No ha sido escrita ni avalada por los autores. Para mayor precisión técnica, consulte el artículo original. Leer descargo de responsabilidad completo
Los robots que pueden ver, entender el lenguaje y moverse con un propósito ya no son ciencia ficción. Se están construyendo utilizando un tipo de inteligencia artificial llamada modelo de visión-lenguaje-acción. Estos sistemas actúan como un puente entre lo que el robot ve y lo que se le ordena hacer. Imagine un robot que escucha "recoge la taza roja" y ve una taza sobre una mesa; debe entonces calcular los movimientos precisos de su brazo para sujetarla. Para enseñar a estos robots a mejorar, los investigadores suelen utilizar un método llamado aprendizaje por refuerzo. En este proceso, el robot intenta una tarea y, si tiene éxito, recibe una recompensa. Si falla, no recibe nada. A lo largo de miles de intentos, el robot aprende qué acciones conducen al éxito. Sin embargo, este proceso de aprendizaje es increíblemente costoso. Cada vez que el robot intenta una tarea, consume tiempo y potencia de cálculo. Si el robot falla repetidamente, o si tiene éxito de una manera torpe, la computadora a menudo desecha ese intento, tratándolo como datos inútiles. Este desperdicio es un importante cuello de botella que ralentiza el desarrollo de robots que puedan trabajar de forma segura y eficiente en el mundo real.
Un equipo de investigadores ha desarrollado un nuevo enfoque para solucionar este desperdicio, convirtiendo lo que antes eran datos descartados en lecciones valiosas. Su método, llamado Prism-GRPO, cambia la forma en que el robot evalúa sus propios intentos. En la forma estándar de entrenamiento, se le da al robot una puntuación simple de aprobado o reprobado. Si un grupo de intentos tiene éxito, todos reciben la misma puntuación perfecta. Si todos fallan, todos reciben la misma puntuación de cero. Debido a que las puntuaciones son idénticas, la computadora no puede distinguir qué intento fue mejor que los otros, por lo que descarta todo el grupo para ahorrar tiempo. Los investigadores se dieron cuenta de que, incluso cuando un robot falla, o incluso cuando tiene éxito, a menudo existen diferencias sutiles en cómo se desempeñó. Un intento exitoso pudo haber sido suave y delicado, mientras que otro fue brusco y golpeó objetos cercanos. Un intento fallido pudo haber estado cerca de la meta, mientras que otro estuvo totalmente fuera de objetivo. Al ignorar estos matices, el método antiguo estaba desechando la información necesaria para enseñar al robot a ser más preciso.
El nuevo método resuelve esto añadiendo una segunda capa de retroalimentación. En lugar de solo preguntar "¿Funcionó?", el sistema también pregunta "¿Qué tan bien funcionó?". Asigna una puntuación de calidad basada en la ejecución física de la tarea, como la cantidad de fuerza utilizada, qué tan suaves fueron los movimientos o si el robot chocó accidentalmente con cosas que no debía tocar. Esta puntuación se combina con el resultado de aprobado o reprobado. Ahora, incluso si un grupo de intentos tiene éxito, el sistema puede ver que uno fue más limpio que los otros y recompensarlo en consecuencia. Del mismo modo, si un grupo falla, el sistema puede identificar qué fallo fue el menos perjudicial y utilizarlo como una señal de aprendizaje. Este simple cambio significa que casi no se desperdicia información. El robot aprende de cada intento, no solo de los casos raros donde algunos tienen éxito y otros no.
Los investigadores probaron esta idea en cuatro tareas robóticas diferentes, que iban desde levantar una olla con dos brazos hasta mover una lata y colocarla junto a una olla. Encontraron que su nuevo método permitió al robot alcanzar el mismo nivel de éxito utilizando hasta un 56 por ciento menos de intentos que el método estándar. Esto supone un ahorro significativo de tiempo y potencia de cálculo. Más importante aún, los robots entrenados con este nuevo método no solo aprendieron a tener éxito; aprendieron a tener éxito con gracia. Fueron menos propensos a desarrollar "atajos" o conductas erráticas. Por ejemplo, en una tarea en la que el robot debía levantar una lata y colocarla junto a una olla, el método antiguo a veces enseñaba al robot a simplemente empujar la olla hacia la lata en lugar de levantar la lata. Esto satisfacía la comprobación de éxito de la computadora, pero no era lo que el humano había pedido. El nuevo método, al penalizar el movimiento brusco de empuje, enseñó al robot a realizar la acción correcta de levantar y colocar con delicadeza.
Cuando los investigadores llevaron a los robots con mejor desempeño de la simulación a un robot físico real, los resultados se mantuvieron. Los robots entrenados con el nuevo método fueron más fiables y menos propensos a realizar los atajos torpes que habían funcionado en la simulación pero fallaron en el mundo real. El estudio demuestra que, al prestar atención a la calidad de una acción, y no solo al resultado final, podemos entrenar a los robots mucho más rápido y hacerlos más seguros. Los investigadores demostraron matemáticamente que este enfoque nunca perjudica la capacidad del robot para aprender la tarea principal, y sus experimentos confirmaron que mejora consistentemente tanto la velocidad como el comportamiento. Este trabajo sugiere un camino claro para lograr que los robots no solo sean capaces de hacer un trabajo, sino capaces de hacerlo con el cuidado y la precisión requeridos para trabajar junto a los humanos.
¿Ahogado en artículos de tu campo?
Recibe resúmenes diarios de los artículos más novedosos que coincidan con tus palabras clave de investigación — con resúmenes técnicos, en tu idioma.