DOI QR코드

DOI QR Code

Performance Analysis of Various Activation Functions in Super Resolution Model

초해상화 모델의 활성함수 변경에 따른 성능 분석

  • Yoo, YoungJun (Dept. of Computer Science, Kookmin University) ;
  • Kim, DaeHee (Dept. of Computer Science, Kookmin University) ;
  • Lee, JaeKoo (Dept. of Computer Science, Kookmin University)
  • 유영준 (국민대학교 컴퓨터공학과) ;
  • 김대희 (국민대학교 컴퓨터공학과) ;
  • 이재구 (국민대학교 컴퓨터공학과)
  • Published : 2020.05.29

Abstract

ReLU(Rectified Linear Unit) 함수는 제안된 이후로 대부분의 깊은 인공신경망 모델들에서 표준 활성함수로써 지배적으로 사용되었다. 이후에 ReLU 를 대체하기 위해 Leaky ReLU, Swish, Mish 활성함수가 제시되었는데, 이들은 영상 분류 과업에서 기존 ReLU 함수 보다 향상된 성능을 보였다. 따라서 초해상화(Super Resolution) 과업에서도 ReLU 를 다른 활성함수들로 대체하여 성능 향상을 얻을 수 있는지 실험해볼 필요성을 느꼈다. 본 연구에서는 초해상화 과업에서 안정적인 성능을 보이는 EDSR(Enhanced Deep Super-Resolution Network) 모델의 활성함수들을 변경하면서 성능을 비교하였다. 결과적으로 EDSR 의 활성함수를 변경하면서 진행한 실험에서 해상도를 2 배로 변환하는 경우, 기존 활성함수인 ReLU 가 실험에 사용된 다른 활성함수들 보다 비슷하거나 높은 성능을 보였다. 하지만 해상도를 4 배로 변환하는 경우에서는 Leaky ReLU 와 Swish 함수가 기존 ReLU 함수대비 다소 향상된 성능을 보임을 확인하였다. 구체적으로 Leaky ReLU 를 사용했을 때 기존 ReLU 보다 영상의 품질을 정량적으로 평가할 수 있는 PSNR 과 SSIM 평가지표가 평균 0.06%, 0.05%, Swish 를 사용했을 때는 평균 0.06%, 0.03%의 성능 향상을 확인할 수 있었다. 4 배의 해상도를 높이는 초해상화의 경우, Leaky ReLU 와 Swish 가 ReLU 대비 향상된 성능을 보였기 때문에 향후 연구에서는 다른 초해상화 모델에서도 성능 향상을 위해 활성함수를 Leaky ReLU 나 Swish 로 대체하는 비교실험을 수행하는 것도 필요하다고 판단된다.

Keywords

Acknowledgement

이 성과는 2020년도 정부(미래창조과학부)의 재원으로 한국연구재단의 지원과 과학기술정보통신부 및 정보통신기획 평가원의 SW 중심대학지원사업으로 수행된 연구임 (No. NRF2018R1C1B5086441)