A continuaciónSiguiente guía
Modelos de espacio de estados y Mamba
Fundamentos
GUÍA de IA en idiomas
Mamba es un modelo de secuencia construido sobre modelos de espacio de estados (SSM) que procesa texto en tiempo lineal, ofreciendo una alternativa rápida a la atención cuadrática del Transformer.
Escuche esta guíaNarrado por IA · ~2 min
Su truco clave es hacer que el modelo decida selectivamente qué recordar y olvidar en función de la entrada misma.
Mamba, presentada por Albert Gu y Tri Dao a finales de 2023, se basa en modelos estructurados de espacio de estados. Un SSM clásico comprime todo el historial de una secuencia en un estado oculto de tamaño fijo y lo actualiza paso a paso, como una red recurrente sofisticada. El gran avance es la selectividad: Mamba hace que los parámetros del SSM (cuánto conservar, cuánto dejar entrar) dependan del token actual, por lo que el modelo puede centrarse en palabras relevantes e ignorar el relleno. Esto permite que un estado de tamaño fijo actúe como una memoria consciente del contenido. Debido a que evita comparar cada token con todos los demás, Mamba escala linealmente con la longitud de la secuencia y se mantiene rápido en entradas muy largas como genomas, audio o texto de la longitud de un libro.
Modelado de secuencias de ADN extremadamente largas donde los Transformers de un millón de tokens son demasiado caros
Impulsando asistentes de lenguaje de contexto largo que resumen libros completos sin truncarlos
Generación de audio y modelado de voz en tiempo real que procesan formas de onda sin procesar de manera eficiente
Implementaciones en el dispositivo o en el borde donde un pequeño estado recurrente de tamaño fijo ahorra memoria en comparación con un caché de atención cada vez mayor.
Mamba y su sucesor Mamba-2 están avanzando hacia arquitecturas híbridas que entrelazan algunas capas de atención con muchas capas SSM, capturando las fortalezas de ambas. Espere SSM en asistentes de contexto prolongado, modelos en dispositivos donde la memoria está limitada y dominios sin texto como ADN y audio. La investigación está investigando si los SSM puros pueden igualar a los Transformers en tareas que necesitan una recuperación precisa y si se escalan a los tamaños de modelo más grandes.
Boletín gratuito
Tres noticias verificadas sobre IA cada mañana de lunes a viernes, explicadas con claridad. Gratis para siempre y sin anuncios.
Un correo cada día laborable. Date de baja con un clic. Nunca vendemos ni compartimos tu dirección.
Ponte a prueba
Corrección inmediata en cada respuesta y un certificado para compartir, con un ID verificable, cuando apruebas un curso.
Apoya la educación gratuita sobre IA. AI Understanding es una organización sin fines de lucro 501(c)(3): sin anuncios ni muros de pago, nunca. Hacer una donación
Mamba es un modelo de secuencia construido sobre modelos de espacio de estados (SSM) que procesa texto en tiempo lineal, ofreciendo una alternativa rápida a la atención cuadrática del Transformer. Su truco clave es hacer que el modelo decida selectivamente qué recordar y olvidar en función de la entrada misma.
Mamba evita la comparación de tokens de todos los pares, por lo que su costo crece linealmente con la longitud de la secuencia en lugar de cuadráticamente como la atención.
La selectividad significa que parámetros como B, C y delta se convierten en funciones de la entrada actual, lo que proporciona una memoria consciente del contenido.
Los SSM son modelos de estilo recurrente que combinan el pasado en un estado de tamaño fijo, similar a un RNN.
La vista de convolución requiere parámetros fijos (invariantes en el tiempo); Los parámetros dependientes de la entrada rompen eso, por lo que Mamba usa un escaneo paralelo en su lugar.
Mamba utiliza un escaneo selectivo con reconocimiento de hardware que mantiene los estados intermedios en una SRAM rápida para recuperar el rendimiento.
sigue aprendiendo
Más guías seleccionadas para este tema.
A continuaciónSiguiente guía
Modelos de espacio de estados y Mamba
Fundamentos