LazyLLM: Poda Dinámica de Tokens para una Inferencia Eficiente de LLM de Contexto LargoLazyLLM: Dynamic Token Pruning for Efficient Long Context LLM Inference
La inferencia de los modelos de lenguaje grandes basados en transformadores consta de dos etapas secuenciales: 1) una etapa de precarga para calcular la caché KV de las indicaciones y generar el primer token, y 2) una etapa de decodificación para generar tokens subsecuentes. Para indicaciones largas, la caché KV debe calcularse para todos los tokens durante la etapa de precarga, lo que puede aumentar significativamente el tiempo necesario para generar el primer token. En consecuencia, la etapa de precarga puede convertirse en un cuello de botella en el proceso de generación. Queda abierta la pregunta de si todos los tokens de la indicación son esenciales para generar el primer token. Para responder a esto, presentamos un método novedoso, LazyLLM, que calcula selectivamente el KV para los tokens importantes para la predicción del siguiente token en ambas etapas, tanto la de precarga como la de decodificación. A diferencia de los enfoques estáticos de poda que podan la indicación de una vez, LazyLLM permite a los modelos de lenguaje seleccionar dinámicamente diferentes subconjuntos de tokens del contexto en diferentes pasos de generación, aunque puedan haber sido podados en pasos anteriores. Experimentos exhaustivos en conjuntos de datos estándar en diversas tareas demuestran que LazyLLM es un método genérico que puede integrarse fácilmente con modelos de lenguaje existentes para acelerar significativamente la generación sin necesidad de ajustes finos. Por ejemplo, en la tarea de pregunta-respuesta multi-documento, LazyLLM acelera la etapa de precarga del modelo LLama 2 7B en un 2.34x manteniendo la precisión.