Учёные из Пекина ускорили нейросеть в 149 раз на слабых чипах FPGA

Представьте ситуацию: у вас есть суперкар, который жрет бензин ведрами. Вы гоните по трассе, пока топливо не заканчивается. И только тогда вы вспоминаете, что можно было ехать экономичнее. Примерно так сейчас выглядит вся мировая индустрия искусственного интеллекта. Гонка за «больше, быстрее, мощнее» привела к тому, что мы забыли про оптимизацию. И парадокс в том, что именно санкции США против Китая заставили последних вспомнить об этом. Результат оказался не просто утешительным призом — это потенциальный прорыв.

Речь о проекте ученых из Пекинского университета. Они собрали экспериментальную систему, которая на первый взгляд выглядит как шаг назад. Вместо того чтобы использовать один мощнейший GPU (графический процессор), они взяли пять слабеньких ПЛИС (FPGA) — программируемых логических матриц. И соединили их... светом.

Свет вместо проводов: как это работает?

Суть технологии в том, чтобы заменить медные провода и традиционную память на оптические каналы. В системе использовались кремниевые фотонные передатчики со скоростью 400 Гбит/с и оптический коммутатор. Данные передаются по оптоволокну на разных длинах волн — это как многополосное шоссе, где каждый автомобиль едет по своей полосе, не мешая другим.

Но главная фишка не в скорости передачи, а в архитектуре. Это конвейер, а не конвейер с перекурами. Обычно GPU обрабатывает слой нейросети, сохраняет результат в память, потом загружает его обратно и обрабатывает следующий слой. Этот процесс называется «стена памяти» — именно он съедает львиную долю времени и энергии. Китайские инженеры сделали проще: каждая ПЛИС постоянно обрабатывает свой слой и сразу передает результат следующему чипу. Никаких сохранений, никаких загрузок. Просто поток.

Цифры говорят сами за себя. Система с производительностью всего 1,97 Тфлопс (это 11,6% от мощности контрольного GPU) выполнила задачу шумоподавления в 149 раз быстрее, чем GPU с его 16,96 Тфлопс. Тысяча изображений была обработана за 105 микросекунд против 15,6 миллисекунд у GPU. Эффективность использования вычислительных блоков ПЛИС достигла 94,7%. Для справки: у обычных GPU этот показатель редко превышает 30-40%.

Почему это важно прямо сейчас, а не в будущем?

Личное наблюдение автора. Недавно я общался с инженером из дата-центра, который жаловался, что счета за электричество выросли настолько, что руководство всерьез рассматривает отключение части серверов по ночам. И это не единичный случай. Мы привыкли, что ИИ жрет ресурсы — это норма. Но когда речь идет о масштабах целой страны, такой подход становится роскошью.

Китайские ученые показали, что можно обойтись малой кровью. Их система — это не про то, как сделать ИИ мощнее. Это про то, как сделать его эффективнее. И здесь есть несколько ключевых моментов, которые стоит запомнить:

• Оптическая связь снимает «стену памяти». Данные не нужно гонять туда-сюда между процессором и памятью. Они текут непрерывно.
• Конвейерная обработка убивает простои. Каждый чип работает постоянно, а не ждет, пока освободится шина данных.
• Энергопотребление падает радикально. Оптические каналы потребляют меньше энергии, чем электрические, и не греются так сильно.

Да, эксперимент проводился на простой пятислойной нейросети с маленькими изображениями 32x32 пикселя. Это не GPT-4. Но сам принцип доказан. Представьте, что эту архитектуру масштабируют на современные генеративные модели. Ускорение в 149 раз — это не шутка. Это разница между тем, чтобы ждать ответа от нейросети минуту или полсекунды.

Что это значит для нас?

Авторы работы сами говорят: «Конкретные задачи могут быть реализованы при ограниченных вычислительных ресурсах, когда совместно разрабатываются алгоритмы, микроархитектуры процессоров и межсоединения». Переводя на человеческий: мы привыкли думать, что для крутого ИИ нужны суперкомпьютеры. А они говорят — нет. Нужна умная архитектура.

Это меняет правила игры. Если технологию удастся масштабировать, мы получим не просто более быстрые вычисления. Мы получим возможность запускать сложные нейросети на слабом железе. Это значит, что ИИ сможет работать на вашем смартфоне, на дешевых серверах в развивающихся странах, на борту дрона. И это не требует создания новых процессоров — достаточно грамотно соединить старые.

Резюме от автора. Санкции США против Китая в сфере высоких технологий — это не просто политика. Это катализатор инноваций. Когда у вас нет доступа к самым мощным GPU, вы начинаете думать головой, а не кошельком. И результат оказывается не просто дешевле — он оказывается умнее. Гонка за сырой мощью зашла в тупик. Будущее — за оптимизацией. И оно, как выясняется, светлое. В прямом смысле.

По материалам: 3DNews