Исследователи представили HiFi-LLP — высокоточный и экономичный метод прогнозирования задержки нейронных сетей при их развертывании на периферийных устройствах. Решение позволяет значительно ускорить поиск оптимальных архитектур (HW-NAS), заменяя ресурсоемкие прямые измерения на аппаратном обеспечении быстрой оценкой с высокой степенью достоверности, что критически важно для эффективной оптимизации моделей под конкретное «железо».

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

Использование таких предсказателей позволяет разработчикам отсеивать неэффективные архитектуры на ранних этапах проектирования. Внедрение метода в пайплайны автоматизированного поиска архитектур сокращает время поиска и снижает требования к вычислительным мощностям, необходимым для подготовки моделей к работе на edge-девайсах.

Ключевые факты

  • Метод HiFi-LLP предназначен для ускорения HW-NAS (Hardware-aware Neural Architecture Search) за счет замены реальных замеров на предсказания.
  • Система обеспечивает высокую точность оценки задержки при минимальных вычислительных затратах по сравнению с прямым профилированием.
  • Внедрение предсказателей с оценкой уверенности позволяет повысить надежность выбора архитектур для различных типов аппаратных ускорителей.
  • Технология ориентирована на оптимизацию глубоких нейронных сетей для работы на устройствах с ограниченными ресурсами (edge devices).