Proceedings of the Korean Society of Computer Information Conference (한국컴퓨터정보학회:학술대회논문집)
- 2023.07a
- /
- Pages.517-518
- /
- 2023
A Study on Hyper Parameters of Graph Neural Network
그래프 신경망 하이퍼 파라미터 연구
- Youn-A Min (Dept. of Applied SW Engineering, Hanyang Cyber University) ;
- Jin-Young Jun (Graduate School of Mechanical & IT Convergence Engineering, Hanyang Cyber University)
- Published : 2023.07.12
Abstract
본 논문에서는 인공지능 신경망의 하이퍼 파라미터들이 그래프 신경망 모델의 성능에 미치는 영향을 알아보기 위하여 대규모 그래프 데이터를 기반으로 이진 분류 문제를 예측하는 그래프 합성곱 신경망 모델(Graph Convolution Network Model)을 구현하고 모델의 다양한 하이퍼 파라미터 중 손실함수와 활성화 함수를 여러 가지 조합으로 적용하며 모델 학습과 예측 실험을 시행하였다. 실험 결과, 활성화 함수보다는 손실함수의 선택이 모델의 예측 성능에 좀 더 큰 영향을 미치는 것을 확인하였다.