Wordle enseña a entrenar agentes con RL en SageMaker AI

La guía de AWS sobre multi-turn reinforcement learning en Amazon SageMaker AI explica cómo entrenar un agente con una única recompensa al final de un episodio de varios pasos, sin corregir cada decisión intermedia. El ejemplo elegido es Wordle: gpt-oss-20b pasó de resolver el 61 % de las palabras a resolver el 84 % en el conjunto de evaluación, con un coste de unos 339 dólares en tokens de entrenamiento. Es un caso pequeño, pero enseña el método con claridad.

Qué ha publicado AWS y por qué importa el multi-turn reinforcement learning

AWS describe una técnica llamada Multi-Turn Reinforcement Learning (MTRL), disponible en Amazon SageMaker AI. La idea central es que el modelo no recibe una puntuación tras cada respuesta, sino una sola recompensa cuando termina el episodio completo, es decir, la secuencia de pasos que el agente da hasta resolver la tarea o fallar. El modelo en entrenamiento participa en todo ese recorrido, toma decisiones, recibe respuestas del entorno y vuelve a decidir. Al final, el resultado se puntúa y esa señal sirve para ajustar el modelo.

Lo relevante para un equipo técnico es lo que no hace falta definir. Con aprendizaje supervisado hay que disponer de ejemplos de la trayectoria correcta: qué herramienta llamar primero, qué responder después, cuándo parar. Con MTRL basta con saber puntuar el resultado final. Si el agente resolvió la tarea, la recompensa es alta; si no, es baja. El camino concreto lo descubre el propio modelo mediante prueba y error durante el entrenamiento.

AWS posiciona esto como una vía para ajustar modelos pequeños y baratos a flujos de trabajo internos que el aprendizaje supervisado no cubre bien. Es una propuesta concreta: en lugar de pagar siempre por un modelo grande para tareas repetitivas y específicas de una empresa, se entrena uno más modesto para que las haga con fiabilidad.

El ejemplo con Wordle sirve como banco de pruebas porque es un problema de varios pasos con un resultado fácil de verificar. Cada intento aporta información que condiciona el siguiente, y al final la palabra se ha resuelto o no. Esa claridad en la recompensa es justo lo que hace que el juego funcione como demostración didáctica.

Cómo funciona el multi-turn reinforcement learning por dentro

La parte más práctica de la guía es la arquitectura. El agente, que es el código que orquesta los pasos y habla con el entorno, se aloja en Amazon Bedrock AgentCore. Quien ya tenga su agente desplegado en infraestructura propia, ya sea EKS, EC2 o Fargate, también puede usarlo: en ese caso se conecta mediante un reenviador AWS Lambda. En ambos escenarios, el agente llama al modelo que se está entrenando, de modo que lo que se optimiza es el comportamiento real del agente y no una simulación aparte.

Esto tiene una consecuencia importante: el entrenamiento se hace sobre el mismo flujo que luego se usará en producción. No hay que reescribir la lógica del agente en un formato especial para entrenar. La separación entre agente, modelo y puntuación permite cambiar una pieza sin tocar las demás, y reaprovechar el agente que ya existe.

Los resultados del ejemplo son concretos. Con el nivel de razonamiento por defecto, gpt-oss-20b subió del 61 % al 84 % de palabras resueltas en el conjunto de evaluación. Son 23 puntos porcentuales de mejora en un modelo de 20.000 millones de parámetros, y el coste en tokens de entrenamiento fue de unos 339 dólares. Conviene leer el dato con cuidado: es una sola tarea, con un entorno acotado y una recompensa muy nítida, y AWS no presenta esta cifra como garantía para cualquier flujo de trabajo.

También hay una limitación implícita en el método: toda la calidad del entrenamiento depende de cómo se puntúe el resultado. Si la función de recompensa es ambigua o fácil de esquivar, el modelo optimizará lo que se mide, no lo que se quería. En Wordle esto apenas es un problema; en un proceso interno de una empresa, definir la puntuación buena suele ser la parte difícil.

Como pueden aplicar esto las empresas hoy

Para una PYME, lo primero es una advertencia sensata: no todo flujo de trabajo necesita reinforcement learning. Si ya tienes ejemplos fiables de cómo debe resolverse una tarea, el aprendizaje supervisado sigue siendo más sencillo. El multi-turn reinforcement learning tiene sentido cuando el proceso tiene varios pasos, el camino correcto no está documentado o admite varias vías, y sí puedes decidir con claridad si el resultado final es bueno o malo.

Un primer ejercicio útil no requiere lanzar ningún entrenamiento: identifica un flujo interno de varios pasos que hoy resuelve un modelo grande y caro, y comprueba si puedes escribir una regla automática que puntúe el resultado. Si puedes puntuarlo sin intervención humana en cada caso, tienes un candidato razonable. Si la evaluación depende de criterio subjetivo, el método se complica.

Sobre el coste, la referencia de unos 339 dólares en tokens de entrenamiento para el ejemplo de Wordle ayuda a dimensionar el orden de magnitud, pero no es extrapolable. Un flujo de trabajo real tendrá episodios más largos, entornos más complejos y más iteraciones. Antes de comprometer presupuesto, conviene hacer una prueba acotada con una tarea sencilla y medir cuánto mejora el modelo frente a su punto de partida.

Si ya tienes agentes en EKS, EC2 o Fargate, la vía del reenviador AWS Lambda evita migrarlos. Si partes de cero, Amazon Bedrock AgentCore es la opción que describe AWS para alojar el agente. En cualquier caso, reserva un conjunto de evaluación separado, como hace el ejemplo, para comprobar que la mejora no es solo memorización del entrenamiento. Y define de antemano qué porcentaje de mejora justifica el gasto, para decidir con datos y no por entusiasmo.

Analisis Blixel

El valor de esta guía no está en Wordle, sino en el cambio de pregunta que propone. Hasta ahora, adaptar un modelo a un proceso interno exigía reunir ejemplos de cómo se hace bien, y en muchas empresas ese material no existe: el conocimiento está en la cabeza de dos personas y en correos sueltos. Puntuar un resultado es mucho más fácil que describir el camino perfecto, y ahí está el atractivo real del multi-turn reinforcement learning.

Dicho esto, conviene bajar expectativas. Un salto del 61 % al 84 % en un juego con reglas cerradas es una buena demostración didáctica, no una promesa para tu facturación o tu soporte al cliente. Los procesos reales tienen excepciones, datos sucios y criterios de éxito que nadie ha escrito. La mayor parte del trabajo estará en diseñar la recompensa, y esa parte no la resuelve ninguna plataforma.

Mi posición es clara: esta técnica es interesante para equipos con perfil técnico y un flujo de trabajo repetitivo, medible y de volumen suficiente como para que ahorrar en un modelo pequeño compense. Para el resto, es mejor esperar a que haya más casos publicados con procesos de empresa y no con juegos. Que AWS permita usar el agente existente, en lugar de obligar a reconstruirlo, reduce la fricción de probar. Pero probar con un piloto barato y una métrica honesta es una cosa; apostar la estrategia de IA a una única cifra de un ejemplo, otra muy distinta.

Quieres aplicar esto en tu empresa? En Blixel.ai te ayudamos a integrar IA con sentido comun. Hablemos.

Newsletter IA · gratis

Recibe IA práctica cada semana en tu bandeja

Casos reales de automatización y agentes IA aplicados a empresas españolas. Sin relleno, sin spam — solo lo que de verdad puedes usar el lunes por la mañana. Cancela cuando quieras.

✓ Suscripción confirmada

Comentarios

Deja una respuesta

Tu dirección de correo electrónico no será publicada. Los campos obligatorios están marcados con *