Исследователи представили HiFi-LLP — высокоточный и экономичный метод прогнозирования задержки нейронных сетей при их развертывании на периферийных устройствах. Решение позволяет значительно ускорить поиск оптимальных архитектур (HW-NAS), заменяя ресурсоемкие прямые измерения на аппаратном обеспечении быстрой оценкой с высокой степенью достоверности, что критически важно для эффективной оптимизации моделей под конкретное «железо».
Традиционные методы оптимизации нейросетей, такие как поиск архитектур с учетом аппаратных ограничений, требуют постоянного тестирования моделей на целевых устройствах. Этот процесс создает «бутылочное горлышко» из-за длительности замеров задержки, особенно при масштабировании поиска. HiFi-LLP решает эту проблему, предлагая прокси-модель, которая не только предсказывает время выполнения, но и оценивает уверенность в своем прогнозе.
Использование таких предсказателей позволяет разработчикам отсеивать неэффективные архитектуры на ранних этапах проектирования. Внедрение метода в пайплайны автоматизированного поиска архитектур сокращает время поиска и снижает требования к вычислительным мощностям, необходимым для подготовки моделей к работе на edge-девайсах.
Ключевые факты
- Метод HiFi-LLP предназначен для ускорения HW-NAS (Hardware-aware Neural Architecture Search) за счет замены реальных замеров на предсказания.
- Система обеспечивает высокую точность оценки задержки при минимальных вычислительных затратах по сравнению с прямым профилированием.
- Внедрение предсказателей с оценкой уверенности позволяет повысить надежность выбора архитектур для различных типов аппаратных ускорителей.
- Технология ориентирована на оптимизацию глубоких нейронных сетей для работы на устройствах с ограниченными ресурсами (edge devices).