NastępnyNastępny poradnik
Modele przestrzeni stanów i Mamba
Podstawy
PRZEWODNIK Językowy AI
Mamba to model sekwencji zbudowany na modelach przestrzeni stanów (SSM), który przetwarza tekst w czasie liniowym, oferując szybką alternatywę dla uwagi kwadratowej Transformera.
Posłuchaj tego przewodnikaOpowiadane przez sztuczną inteligencję · ~2 min
Its key trick is making the model selectively decide what to remember and forget based on the input itself.
Mamba, wprowadzona przez Alberta Gu i Tri Dao pod koniec 2023 roku, jest zbudowana na ustrukturyzowanych modelach przestrzeni stanów. Klasyczny SSM kompresuje całą historię sekwencji do ukrytego stanu o stałym rozmiarze i aktualizuje ją krok po kroku, jak wyrafinowana sieć rekurencyjna. Przełomem jest selektywność: Mamba sprawia, że parametry SSM (ile zachować, ile wpuścić) zależą od aktualnego tokena, dzięki czemu model może skupić się na odpowiednich słowach i zignorować wypełniacze. Dzięki temu jeden stan o stałym rozmiarze może działać jak pamięć świadoma zawartości. Ponieważ unika porównywania każdego tokena z każdym innym tokenem, Mamba skaluje się liniowo wraz z długością sekwencji i działa szybko w przypadku bardzo długich danych wejściowych, takich jak genomy, dźwięk lub tekst o długości książki.
Modelowanie niezwykle długich sekwencji DNA, w przypadku których transformatory o wartości miliona żetonów są zbyt drogie
Wspieranie asystentów językowych o długim kontekście, którzy podsumowują całe książki bez obcinania
Generowanie dźwięku w czasie rzeczywistym i modelowanie mowy, które efektywnie przetwarzają surowe przebiegi
Wdrożenia na urządzeniu lub na krawędzi, w których mały, cykliczny stan o stałym rozmiarze oszczędza pamięć w porównaniu z rosnącą pamięcią podręczną uwagi
Mamba i jej następca Mamba-2 idą w stronę architektur hybrydowych, które przeplatają kilka warstw uwagi wieloma warstwami SSM, wychwytując mocne strony obu. SSM można spodziewać się w asystentach długokontekstowych, modelach na urządzeniach, w których pamięć jest ograniczona, oraz w domenach nietekstowych, takich jak DNA i dźwięk. W ramach badań sprawdza się, czy czyste moduły SSM mogą dorównać transformatorom w przypadku zadań wymagających precyzyjnego przypomnienia i czy można je skalować do największych rozmiarów modeli.
Darmowy newsletter
Trzy zweryfikowane historie AI każdego poranka w dni robocze, napisane prostym angielskim. Bezpłatne na zawsze, bez reklam.
Jedno e‑mail każdego dnia roboczego. Wypisz się jednym kliknięciem. Nigdy nie sprzedajemy ani nie udostępniamy Twojego adresu.
Sprawdź się
Natychmiastowa informacja zwrotna na każde pytanie oraz udostępnialny certyfikat z weryfikowalnym ID po zaliczeniu kursu.
Wspieraj darmową edukację AI. AI Understanding jest organizacją non-profit 501(c)(3) — bez reklam, bez płatnej ściany, zawsze. Przekaż darowiznę
Mamba to model sekwencji zbudowany na modelach przestrzeni stanów (SSM), który przetwarza tekst w czasie liniowym, oferując szybką alternatywę dla uwagi kwadratowej Transformera. Jego kluczową sztuczką jest sprawienie, aby model selektywnie decydował, co zapamiętać, a co zapomnieć, na podstawie samych danych wejściowych.
Mamba unika porównywania żetonów wszystkich par, więc jego koszt rośnie liniowo wraz z długością sekwencji, a nie kwadratowo jak uwaga.
Selektywność oznacza, że parametry takie jak B, C i delta stają się funkcjami bieżącego wejścia, zapewniając pamięć świadomą zawartości.
SSM to modele w stylu powtarzalnym, które składają przeszłość w stan o ustalonym rozmiarze, podobnie jak RNN.
Widok splotu wymaga stałych (niezmiennych w czasie) parametrów; Parametry zależne od wejścia to zakłócają, więc Mamba zamiast tego używa skanowania równoległego.
Mamba wykorzystuje selektywne skanowanie uwzględniające sprzęt, które utrzymuje stany pośrednie w szybkiej pamięci SRAM w celu przywrócenia przepustowości.
Ucz się dalej
Wybrano więcej przewodników na ten temat
NastępnyNastępny poradnik
Modele przestrzeni stanów i Mamba
Podstawy