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
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.
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.
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.
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.