СледваСледващо ръководство
Държавни космически модели и Мамба
Основи
Език AI РЪКОВОДСТВО
Mamba е модел на последователност, изграден върху модели на пространството на състоянията (SSM), който обработва текст в линейно време, предлагайки бърза алтернатива на квадратичното внимание на Transformer.
Чуйте това ръководствоAI-разказан · ~2 min
Ключовият му трик е да накара модела избирателно да реши какво да запомни и забрави въз основа на самия вход.
Mamba, представена от Албърт Гу и Три Дао в края на 2023 г., е изградена върху модели на структурирано пространство на състоянието. Класическият SSM компресира цялата история на последователност в скрито състояние с фиксиран размер и я актуализира стъпка по стъпка, като сложна повтаряща се мрежа. Пробивът е в селективността: Mamba прави параметрите на SSM (колко да се запази, колко да се пусне) в зависимост от текущия токен, така че моделът да може да се съсредоточи върху съответните думи и да игнорира пълнителя. Това позволява на едно състояние с фиксиран размер да действа като памет, осъзнаваща съдържанието. Тъй като избягва сравняването на всеки токен с всеки друг токен, Mamba се мащабира линейно с дължината на последователността и остава бърз при много дълги входове като геноми, аудио или текст с дължина на книга.
Моделиране на изключително дълги ДНК последователности, където трансформаторите с милиони жетони са твърде скъпи
Поддържане на езикови асистенти с дълъг контекст, които обобщават цели книги без съкращаване
Аудио генериране в реално време и моделиране на реч, които обработват ефикасно необработени вълнови форми
Внедрявания на устройство или периферия, при които малко повтарящо се състояние с фиксиран размер спестява памет в сравнение с нарастващ кеш на вниманието
Mamba и нейният наследник Mamba-2 налагат хибридни архитектури, които преплитат няколко слоя за внимание с много SSM слоеве, улавяйки силните страни и на двата. Очаквайте SSM в асистенти с дълъг контекст, модели на устройства, където паметта е ограничена, и нетекстови домейни като ДНК и аудио. Изследванията проучват дали чистите SSMs могат да се сравнят с Transformers при задачи, изискващи прецизно извикване, и дали те се мащабират до най-големите размери на модела.
Безплатен бюлетин
Три проверени AI истории всяка делнична сутрин, написани на достъпен език. Безплатно завинаги, без реклами.
Един имейл всеки делничен ден. Отписване с едно кликване. Никога не продаваме и не споделяме вашия имейл адрес.
Тествайте себе си
Незабавна обратна връзка за всеки отговор и сертификат за споделяне с подлежащ на проверка идентификационен номер, след като преминете курс.
Подкрепете безплатното AI образование. AI Understanding е 501(c)(3) с нестопанска цел — без реклами, без платен достъп, никога. Направете дарение
Mamba е модел на последователност, изграден върху модели на пространството на състоянията (SSM), който обработва текст в линейно време, предлагайки бърза алтернатива на квадратичното внимание на Transformer. Ключовият му трик е да накара модела избирателно да реши какво да запомни и забрави въз основа на самия вход.
Mamba избягва сравнението на всички двойки токени, така че цената му расте линейно с дължината на последователността, а не квадратично като вниманието.
Селективността означава, че параметри като B, C и делта стават функции на текущия вход, давайки памет за съдържание.
SSM са модели в повтарящ се стил, които сгъват миналото в състояние с фиксиран размер, подобно на RNN.
Конволюционният изглед изисква фиксирани (инвариантни във времето) параметри; зависимите от входа параметри нарушават това, така че вместо това Mamba използва паралелно сканиране.
Mamba използва хардуерно селективно сканиране, което поддържа междинните състояния в бърза SRAM за възстановяване на пропускателната способност.
Продължавай да учиш
Още ръководства, избрани за тази тема
СледваСледващо ръководство
Държавни космически модели и Мамба
Основи