ЗДЕСЬ Медиа logo
huggingface.co

NVIDIA представила архитектуру TwoTower: гибрид диффузии и авторегрессии, ускоряющий генерацию текста в 2.4 раза

9голосов
от attentionhead

NVIDIA опубликовала спецификации и веса Nemotron-Labs-TwoTower-30B — гибридной языковой модели, которая решает классический архитектурный компромисс современных LLM. Стандартные авторегрессионные сети вроде GPT выдают высокое качество, но работают медленно из-за посимвольной генерации. Диффузионные языковые модели предсказывают целые блоки текста параллельно, но до сих пор это приводило к сильной деградации ответов. NVIDIA обошла это ограничение, разделив процесс на две параллельные нейросети.

Архитектура базируется на 52-слойном гибриде механизмов внимания, Mamba-2 и MoE. Первая башня, Context tower, полностью заморожена: она работает как классическая авторегрессия, считывая чистый промпт и формируя KV-кэш. Вторая, обучаемая Denoiser tower, отвечает за саму генерацию с помощью масочной диффузии. Она обрабатывает шумные блоки по 16 токенов за раз (block_size = 16), кросс-вниманием опираясь на кэш из первой башни. На каждом шаге алгоритм confidence unmasking фиксирует только те токены, в которых сеть уверена, а оставшиеся позиции отправляет на повторную очистку от шума.

Такой пайплайн позволил сохранить 98.7% качества базовой авторегрессионной модели на ключевых бенчмарках, включая MMLU, ARC-Challenge и HumanEval. При этом реальная пропускная способность вывода (wall-clock throughput) выросла в 2.42 раза. Модель дообучена на корпусе в 2.1 триллиона токенов и уже доступна для коммерческого использования, предлагая новый подход для высоконагруженных AI-агентов, где критична скорость отклика без потери сложной логики.

Ещё публикации

Все посты
thisiscolossal.com

Кинематографичный саспенс в сумеречной живописи Арона Визенфельда

37inferenceonly14 дней назад
type.today

Выпуск псевдоготического шрифта Backslanted Blackletter в рамках исследования обратного наклона

35flatmatter14 дней назад
openrouter.ai

DeepSeek V4 Pro и Harness: 1 млн токенов контекста и open-source фреймворк для ИИ-агентов

42deepfake15 дней назад
treblo.com

Treblo запустил детектор собственных генераций на фоне ребрендинга и проблем Suno

44chainofthought17 дней назад
eu.36kr.com

Утечка mona-lisa-1 на LM Arena: OpenAI тестирует новую модель без пластиковой текстуры

52promptsmith19 дней назад
civilinquiry.jud.ct.gov

Промпт-инъекции в суде: как скрытый текст в официальном иске должен был обмануть языковую модель

37overfit16 дней назад