ElemedexEnciclopedia de bolsillo de arquitecturas de IA

Glosario (MT) › Entrenamiento y ajuste

MT139

Modelo de recompensa

También: modelo de preferencias

En inglés: Reward model

Modelo entrenado para dar un puntaje numérico a una respuesta según cuánto la preferirían evaluadores humanos; funciona como juez automático durante el RLHF.

En el RLHF hacen falta millones de evaluaciones, y ninguna empresa puede pagar a personas para leerlas todas. La solución es entrenar un segundo modelo que prediga el juicio humano (Alammar y Grootendorst, 2024, p. 379). Se toma una copia del modelo ajustado por instrucciones y se le cambia la «cabeza»: en lugar de probabilidades sobre el vocabulario, devuelve un solo número. InstructGPT usó modelos de recompensa de 6.000 millones de parámetros (Ouyang et al., 2022, sec. 3.5).

Los datos son comparaciones: una instrucción, una respuesta aceptada y otra rechazada (no necesariamente una buena y una mala; a veces ambas son buenas y una es mejor) (Alammar y Grootendorst, 2024, pp. 380–382). El entrenamiento empuja a que la aceptada puntúe más alto. La diferencia de puntajes se interpreta como una probabilidad de preferencia, al modo del modelo de Bradley y Terry (1952) o del sistema Elo del ajedrez (Christiano et al., 2017, sec. 2.2.3):

P(A preferida a B) = 1 / (1 + e^−(r_A − r_B)). Si r_A − r_B = 1, P ≈ 0,73.

El riesgo es que el modelo optimizado aprenda a complacer al juez en lugar de hacer lo que el juez intenta medir (la llamada ley de Goodhart). Es la versión lingüística del reward hacking que Buckner (2024) documenta en el aprendizaje por refuerzo: un agente que da vueltas en círculo para juntar bonificaciones sin terminar nunca la carrera (2024, p. 68). Por eso InstructGPT penaliza alejarse del modelo previo (Ouyang et al., 2022, sec. 3.5), y Llama 2 usa dos modelos de recompensa, uno de utilidad y otro de seguridad (Alammar y Grootendorst, 2024, p. 383).

Para un inferencialista hay una distinción que el procedimiento borra: la que separa tomar algo por correcto de que sea correcto. Un modelo de recompensa es un modelo de las aprobaciones de un grupo de evaluadores. ¿Puede un modelo de actitudes hacer las veces de un modelo de normas? Ver inferencialismo.

Fuentes

  • Alammar, J., & Grootendorst, M. (2024). Hands-on large language models: Language understanding and generation. O’Reilly Media. §
  • Bradley, R. A., & Terry, M. E. (1952). Rank analysis of incomplete block designs: I. The method of paired comparisons. Biometrika, 39(3–4), 324–345. https://doi.org/10.1093/biomet/39.3-4.324 §
  • Buckner, C. J. (2024). From deep learning to rational machines: What the history of philosophy can teach us about the future of artificial intelligence. Oxford University Press. https://doi.org/10.1093/oso/9780197653302.001.0001 §
  • Christiano, P. F., Leike, J., Brown, T., Martic, M., Legg, S., & Amodei, D. (2017). Deep reinforcement learning from human preferences. In Advances in Neural Information Processing Systems (Vol. 30). Curran Associates. https://proceedings.neurips.cc/paper_files/paper/2017/hash/d5e2c0adad503c91f91df240d0cd4e49-Abstract.html §
  • Ouyang, L., Wu, J., Jiang, X., Almeida, D., Wainwright, C., Mishkin, P., Zhang, C., Agarwal, S., Slama, K., Ray, A., Schulman, J., Hilton, J., Kelton, F., Miller, L., Simens, M., Askell, A., Welinder, P., Christiano, P. F., Leike, J., & Lowe, R. (2022). Training language models to follow instructions with human feedback. In Advances in Neural Information Processing Systems (Vol. 35, pp. 27730–27744). Curran Associates. https://doi.org/10.52202/068431-2011 §