Saltar al contenido
AI Atlas

Paper #15 · Razonamiento

Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameters

Escalar de forma óptima el cómputo en inferencia puede ser más efectivo que escalar los parámetros

Autores:
Snell, Lee, Xu y Kumar
Institución:
UC Berkeley · Google DeepMind
Año:
2024 · arXiv:2408.03314

En pocas palabras

Si se permite que un modelo «piense más» al responder, ya sea generando y verificando muchas soluciones o revisando la suya, puede superar a un modelo 14 veces más grande. La clave es ajustar cuánto y cómo se gasta ese cómputo según la dificultad de cada pregunta.

Por qué importa hoy: Un cambio conceptual enorme: la inteligencia no solo escala en el entrenamiento; también puedes gastar más cómputo al responder.

El problema

Hasta entonces, la forma principal de mejorar era entrenar modelos más grandes. Pero una persona dedica más tiempo a los problemas difíciles. ¿Cuánto puede mejorar un modelo fijo si se le da más cómputo en el momento de la inferencia, y cuándo conviene eso más que un modelo más grande?

Ideas clave

  1. Dos palancas

    Buscar: generar muchas soluciones y elegir con un verificador por pasos (best-of-N, beam search, lookahead). Revisar: que el modelo corrija iterativamente su propia respuesta.

  2. Depende de la dificultad

    En preguntas fáciles conviene revisar secuencialmente la respuesta; en las difíciles, explorar en paralelo muchas alternativas. Ninguna estrategia gana siempre.

  3. Estrategia óptima por prompt

    Estimando la dificultad de cada pregunta y eligiendo la estrategia en consecuencia, se obtiene el mismo rendimiento con unas 4 veces menos cómputo que best-of-N.

  4. Intercambio con el preentrenamiento

    Con los mismos FLOPs totales, en problemas de dificultad baja o media sale más a cuenta un modelo pequeño que piensa más; en los más difíciles sigue ganando el preentrenamiento.

Resultados

  • Más de 4 veces de mejora en eficiencia frente a best-of-N con la estrategia óptima por prompt.
  • Con FLOPs equivalentes, un modelo pequeño con más cómputo en inferencia supera a uno 14 veces más grande en problemas donde el pequeño ya tiene cierto éxito.

Legado

Dio fundamento empírico a la nueva ley de escala en inferencia que popularizaron o1, DeepSeek-R1 y los demás modelos de razonamiento: el presupuesto de «pensamiento» es ahora una perilla más, junto al tamaño del modelo.

Conceptos para leerlo

Explicación original en español elaborada por AI Atlas; no es una traducción del paper. El texto completo pertenece a sus autores.