Image Classification Using Convolutional Neural Network and Extreme Learning Machine Classifier Based on ReLU Function
Jung-Soo Han, Keun-Chang Kwak · The Journal of Korean Institute of Information Technology · 2017
본 논문은 컨벌루션 신경회로망(CNN, Convolutional Neural Network)과 ReLU(Rectified Linear Unit) 기반 ELMC(Extreme Learning Machine Classifier)를 이용하여 영상 분류하는 방법을 제안한다. 이를 위해 첫 번째로 컨벌루션 신경회로망으로 미리 학습된 모델(Pre-trained model)을 사용하여 특징을 추출하고 두 번째로 ReLU 기반 ELMC를 사용하여 영상을 분류한다. 본 논문에서 사용한 데이터베이스는 Caltech-101 데이터베이스와 OxfordFlowers-17 데이터베이스를 사용하였다. 두 개의 데이터베이스에 실험을 한 결과, 분류율 및 분류 속도에서 다른 머신 러닝(Machine Learning) 분류 알고리즘과 결합된 것과 비교하여 우수한 성능을 보여주고 있으며, 기존 활성화 함수를 사용한 결과보다 ReLU 함수를 사용하였을 때 더 우수한 성능을 보임을 확인하였다.