Etiqueta: reward model

  • Como funciona el RLHF que entrena a ChatGPT

    Como funciona el RLHF que entrena a ChatGPT

    El RLHF para alinear modelos de lenguaje es la pieza que convierte un modelo que solo predice la siguiente palabra en un asistente que sigue instrucciones, evita respuestas peligrosas y suena util. Detras de ChatGPT, Claude o Gemini hay un mismo esquema de tres fases que combina aprendizaje supervisado y aprendizaje por refuerzo guiado por preferencias humanas. No es magia: es una tuberia con sus costuras, sus inestabilidades y sus alternativas mas baratas. Entender como encaja cada parte ayuda a juzgar que esperar de estos sistemas y donde estan sus limites reales.

    Que es el RLHF y por que se convirtio en estandar

    El RLHF para alinear modelos de lenguaje (Reinforcement Learning from Human Feedback) parte de un problema concreto: un modelo preentrenado sobre billones de tokens sabe completar texto, pero no sabe que respuesta prefiere una persona. El pipeline tipico se divide en tres fases. Primero, el preentrenamiento del modelo de lenguaje sobre grandes corpus. Segundo, un fine-tuning supervisado (SFT) con ejemplos escritos por humanos que muestran como deberia responder ante una instruccion. Tercero, una etapa de refuerzo donde entra un reward model aprendido a partir de comparaciones humanas.

    Ese reward model es la clave. En lugar de pedir a una persona que puntue cada respuesta en una escala absoluta, se le muestran varias salidas del modelo y se le pide ordenarlas de mejor a peor. A partir de esos rankings se entrena un modelo que predice una puntuacion de preferencia. Despues, se usa ese reward model como senal para ajustar la politica del modelo de lenguaje mediante aprendizaje por refuerzo. Esta separacion entre recoger preferencias y optimizar contra ellas es lo que permitio escalar el alineamiento a modelos enormes sin necesidad de evaluacion humana en cada paso de entrenamiento.

    De los policy gradients a PPO: la mecanica del refuerzo

    La tercera fase reutiliza conceptos clasicos de aprendizaje por refuerzo: funciones de valor, policy gradients y arquitecturas actor-critic. El algoritmo dominante aqui es PPO (Proximal Policy Optimization). La idea es tratar el reward model como entorno: el modelo genera una respuesta, el reward model la puntua, y PPO ajusta la politica para maximizar esa puntuacion. Pero hay una restriccion fundamental. Si se optimiza sin freno, el modelo se aleja del comportamiento inicial y empieza a producir texto degenerado que engaña al reward model sin ser realmente bueno.

    Para evitarlo se añade un termino de divergencia KL que penaliza alejarse demasiado del modelo de referencia (el SFT). Es un equilibrio: maximizar la recompensa mientras se mantiene la coherencia linguistica del modelo base. Aqui aparecen los problemas practicos del RLHF para alinear modelos de lenguaje. PPO es inestable, sensible a hiperparametros y caro de afinar. Existe ademas el riesgo de overoptimization: cuanto mas se optimiza contra el reward model, mas se explotan sus errores en lugar de las preferencias reales. Por eso han ganado terreno alternativas mas simples basadas en aprendizaje por preferencias directo, como DPO (Direct Preference Optimization), que prescinde del reward model explicito y del bucle de refuerzo, ajustando la politica directamente desde las comparaciones.

    Cuando y para quien es relevante dominar esto

    El RLHF no es una tecnica que la mayoria de empresas vaya a implementar desde cero: requiere infraestructura de entrenamiento, equipos de anotacion y experiencia en aprendizaje por refuerzo. Su relevancia inmediata es para laboratorios y equipos de investigacion que entrenan modelos base, y para grupos que hacen fine-tuning serio de modelos abiertos. Para ellos, entender la diferencia entre RLHF clasico con PPO y metodos tipo DPO marca decisiones de coste y estabilidad muy concretas.

    Para el resto del mercado el horizonte es indirecto pero importante. Quien integra un LLM en producto consume el resultado del RLHF para alinear modelos de lenguaje sin tocarlo, pero entender que la alineacion es un proceso de optimizacion imperfecto explica fenomenos del dia a dia: por que un modelo se vuelve excesivamente prudente, por que evade preguntas legitimas o por que un cambio de version altera su tono. A medio plazo, las tecnicas de preferencias directas se estan abaratando lo suficiente como para que equipos medianos las apliquen sobre modelos abiertos. Quien quiera personalizar comportamiento sin depender de un proveedor cerrado deberia seguir de cerca DPO y sus variantes.

    Analisis Blixel

    Hay una idea incomoda que conviene asumir antes de fascinarse con estos pipelines: alinear un modelo no es enseñarle la verdad, es enseñarle a parecer util para un grupo concreto de anotadores. El reward model no captura preferencias humanas universales, captura las de quien etiqueto los datos, con sus sesgos y sus prisas. Eso explica por que distintos asistentes tienen personalidades tan distintas pese a usar la misma receta tecnica. La alineacion es, en buena parte, una decision editorial disfrazada de matematica.

    El segundo punto practico es la migracion silenciosa de PPO hacia DPO. PPO funciona, pero es caro de domar y fragil. Que metodos mas directos den resultados comparables con una fraccion del esfuerzo de ingenieria no es un detalle academico: democratiza el ajuste fino. En dos o tres años, personalizar el comportamiento de un modelo abierto podria estar al alcance de equipos que hoy ni se lo plantean, sin granjas de GPU dedicadas al bucle de refuerzo.

    La advertencia es el overoptimization. Cuanto mas se aprieta la recompensa, mas se explotan los fallos del evaluador. Es la version tecnica de optimizar la metrica equivocada hasta romperla. Cualquiera que afine modelos deberia tratar el reward model como un proxy imperfecto, no como un oraculo. La buena alineacion no es la que maximiza una puntuacion, sino la que sabe cuando dejar de optimizar.

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