Sign In

Lightweight Software Kernels and Hardware Extensions for Efficient Sparse Deep Neural Networks on Microcontrollers

Created by
  • Haebom
Category
Empty

저자

Francesco Daghero, Daniele Jahier Pagliari, Francesco Conti, Luca Benini, Massimo Poncino, Alessio Burrello

개요

본 논문은 마이크로컨트롤러(MCU)와 같은 에지 디바이스에서 가지치기된 심층 신경망(DNN)의 가속화 문제를 해결하기 위한 세 가지 기여를 제안합니다. 첫째, 초저전력 멀티코어 RISC-V MCU를 타겟으로 N:M 가지치기 계층을 위한 최적화된 소프트웨어 커널을 설계하여 1:8 및 1:16 스파스에서 각각 기존 대비 최대 2.1배 및 3.4배의 속도 향상을 달성했습니다. 둘째, 커널에 필요한 간접 로드 및 0이 아닌 인덱스 압축 해제 연산을 가속화하기 위해 경량 명령어 집합 아키텍처(ISA) 확장을 구현하여 면적 오버헤드 5%로 최대 1.9배의 추가 속도 향상을 얻었습니다. 마지막으로, 스파스 커널을 활용하여 오픈소스 DNN 컴파일러를 확장하여 ResNet18과 Vision Transformer(ViT)에서 각각 3.21배 및 1.81배의 속도 향상을 보였으며, 기존 대비 정확도 저하가 1.5% 미만이었습니다.

시사점, 한계점

시사점:
초저전력 MCU에서 가지치기된 DNN의 효율적인 실행을 위한 최적화된 소프트웨어 및 하드웨어 솔루션을 제시합니다.
RISC-V 기반 MCU에서의 DNN 가속화에 대한 실질적인 성능 향상을 보여줍니다.
오픈소스 DNN 컴파일러 확장을 통해 실제 애플리케이션에의 적용 가능성을 높였습니다.
한계점:
제안된 ISA 확장은 특정 아키텍처에 종속적일 수 있습니다.
평가는 특정 DNN 모델(ResNet18, ViT)에 국한되어 일반화 가능성에 대한 추가 연구가 필요합니다.
더욱 다양한 스파스 패턴에 대한 성능 평가가 필요합니다.
에너지 소비량에 대한 정량적인 분석이 부족합니다.
👍