A seguirPróximo guia
Modelos de Espaço de Estado e Mamba
Fundamentos
GUIA de IA de linguagem
Mamba é um modelo de sequência construído em modelos de espaço de estados (SSMs) que processa texto em tempo linear, oferecendo uma alternativa rápida à atenção quadrática do Transformer.
Ouça este guiaNarrado por IA · ~2 min
Seu truque principal é fazer o modelo decidir seletivamente o que lembrar e esquecer com base na própria entrada.
Mamba, introduzido por Albert Gu e Tri Dao no final de 2023, é construído em modelos estruturados de espaço de estado. Um SSM clássico comprime todo o histórico de uma sequência em um estado oculto de tamanho fixo e o atualiza passo a passo, como uma rede recorrente sofisticada. A inovação é a seletividade: o Mamba faz com que os parâmetros do SSM (quanto manter, quanto deixar entrar) dependam do token atual, para que o modelo possa se concentrar em palavras relevantes e ignorar o preenchimento. Isso permite que um estado de tamanho fixo atue como uma memória com reconhecimento de conteúdo. Como evita comparar cada token com qualquer outro token, o Mamba é dimensionado linearmente com o comprimento da sequência e permanece rápido em entradas muito longas, como genomas, áudio ou texto do tamanho de um livro.
Modelagem de sequências de DNA extremamente longas onde transformadores de um milhão de tokens são muito caros
Capacitando assistentes de linguagem de longo contexto que resumem livros inteiros sem truncamento
Geração de áudio em tempo real e modelagem de fala que processam formas de onda brutas com eficiência
Implantações no dispositivo ou na borda, onde um pequeno estado recorrente de tamanho fixo economiza memória em comparação com um cache de atenção crescente
Mamba e seu sucessor Mamba-2 estão adotando arquiteturas híbridas que intercalam algumas camadas de atenção com muitas camadas SSM, capturando os pontos fortes de ambas. Espere SSMs em assistentes de contexto longo, modelos no dispositivo onde a memória é restrita e domínios não textuais, como DNA e áudio. A pesquisa está investigando se os SSMs puros podem se equiparar aos Transformers em tarefas que necessitam de recuperação precisa e se eles podem ser dimensionados para modelos de tamanhos maiores.
Boletim informativo gratuito
Três histórias verificadas de IA a cada manhã de dia útil, escritas em linguagem simples. Gratuito para sempre, sem anúncios.
Um e-mail a cada dia útil. Cancele a inscrição com um clique. Nunca vendemos ou compartilhamos seu endereço.
Teste você mesmo
Feedback instantâneo sobre cada resposta e um certificado compartilhável com um ID verificável assim que você concluir um curso.
Apoie a educação gratuita em IA. O AI Understanding é uma organização sem fins lucrativos 501(c)(3) — sem anúncios, sem paywall, nunca. Faça uma doação
Mamba é um modelo de sequência construído em modelos de espaço de estados (SSMs) que processa texto em tempo linear, oferecendo uma alternativa rápida à atenção quadrática do Transformer. Seu principal truque é fazer com que o modelo decida seletivamente o que lembrar e esquecer com base na própria entrada.
O Mamba evita a comparação de tokens de todos os pares, portanto seu custo cresce linearmente com o comprimento da sequência, em vez de quadraticamente como a atenção.
Seletividade significa que parâmetros como B, C e delta tornam-se funções da entrada atual, fornecendo memória com reconhecimento de conteúdo.
SSMs são modelos de estilo recorrente que dobram o passado em um estado de tamanho fixo, semelhante a um RNN.
A visão de convolução requer parâmetros fixos (invariantes no tempo); parâmetros dependentes de entrada quebram isso, então o Mamba usa uma varredura paralela.
O Mamba usa uma varredura seletiva com reconhecimento de hardware que mantém estados intermediários em SRAM rápida para recuperar o rendimento.
Continue aprendendo
Mais guias escolhidos para este tópico
A seguirPróximo guia
Modelos de Espaço de Estado e Mamba
Fundamentos