Compression and Acceleration of Face Detector using L1 Loss and Channel Pruning

L1 목적 함수와 채널 프루닝을 이용한 얼굴 검출기 경량화

  • Lee, Seok Hee (Seoul National University Institute of New Media and Communications) ;
  • Jang, Young Kyun (Seoul National University Institute of New Media and Communications) ;
  • Cho, Nam Ik (Seoul National University Institute of New Media and Communications)
  • 이석희 (서울대학교 뉴미디어통신공동연구소) ;
  • 장영균 (서울대학교 뉴미디어통신공동연구소) ;
  • 조남익 (서울대학교 뉴미디어통신공동연구소)
  • Published : 2020.07.13

Abstract

본 논문에서는 합성곱 기반의 얼굴 검출기 Dual Shot Face Detector (DSFD)에 대하여, 특징점 맵의 희소화와 채널 프루닝 목적 함수를 사용하여 네트웍 경량화를 수행하였다. 특징점 맵을 희소화하기 위해 L1 목적 함수를 사용했고, 특징점 맵의 채널 프루닝을 하기 위해 채널 최대값이 가장 낮은 채널들의 합을 최소화 시키는 목적함수를 적용했다. 기존의 신경망은 특징점 맵 희소화 비율이 45%였고 두 목적 함수를 적용했을 때 69.67% 로 희소화 비율이 높아진 것을 확인했다. 얼굴 검출 성능을 다양한 조명, 크기, 환경, 각도, 표정의 얼굴들을 포함하는 영상들로 이뤄진 Wider Face 데이터 셋으로 실험한 결과, average precision은 하락 했고 easy validation set에서 0.9257, hard validation set에서 0.8363 였다.

Keywords