Block Diffusion: Interpolación entre Modelos de Lenguaje Autoregresivos y de DifusiónBlock Diffusion: Interpolating Between Autoregressive and Diffusion
Language Models
Los modelos de lenguaje basados en difusión ofrecen ventajas únicas sobre los modelos autorregresivos debido a su potencial para la generación paralelizada y la controlabilidad, aunque se quedan atrás en el modelado de verosimilitud y están limitados a la generación de longitud fija. En este trabajo, presentamos una clase de modelos de lenguaje de difusión por bloques que interpolan entre la difusión de eliminación de ruido discreta y los modelos autorregresivos. La difusión por bloques supera limitaciones clave de ambos enfoques al permitir la generación de longitud flexible y mejorar la eficiencia de inferencia con el almacenamiento en caché de KV y el muestreo paralelo de tokens. Proponemos una receta para construir modelos de difusión por bloques efectivos que incluye un algoritmo de entrenamiento eficiente, estimadores de varianza del gradiente y programaciones de ruido basadas en datos para minimizar la varianza. La difusión por bloques establece un nuevo estado del arte en el rendimiento entre los modelos de difusión en benchmarks de modelado de lenguaje y permite la generación de secuencias de longitud arbitraria. Proporcionamos el código, junto con los pesos del modelo y una entrada de blog en la página del proyecto: https://m-arriola.com/bd3lms/.