DOI QR코드

DOI QR Code

MAPPO based Hyperparameter Optimization for CNN

MAPPO 기반 CNN 하이퍼 파라미터 최적화

  • Ma, Zhixin (Dept. of Computer Science, Hanyang University) ;
  • Joe, Inwhee (Dept. of Computer Science, Hanyang University)
  • 마지흔 (한양대학교 컴퓨터 소프트웨어학과) ;
  • 조인휘 (한양대학교 컴퓨터 소프트웨어학과)
  • Published : 2022.05.17

Abstract

대부분의 머신러닝 및 딥러닝 모델의 경우 하이퍼 파라미터 선택은 모델의 성능에 큰 영향을 미친다. 따라서 전문가들은 작업을 수행하기 위해 모델을 구축할 때 하이퍼 파라미터 튜닝을 수행하는 데 상당한 시간을 소비해야 한다. Hyperparameter Optimization(HPO)을 해결하기 위한 알고리즘은 많지만 대부분의 방법은 검색을 수행하기 위해 각 epoch에서 실제 실험 결과를 필요로 한다. 따라서 HPO 검색을 위한 시간과 계산 지원을 줄이기 위해 본 논문에서는 Multi-agent Proximal Policy Optimization(MAPPO) 강화 학습 알고리즘을 제안한다. 2개의 이미지 분류 데이터 세트에 대한 실험 결과는 우리의 모델이 속도와 정확성에서 다른 기존 방법보다 우수하다는 것을 보여준다.

Keywords