Image Channel as an Input Method for Deep Learning Ensemble

Javokhir Musaev, Ngoc Thanh Nguyên, Dosam Hwang · 2021 International Conference on INnovations in Intelligent SysTems and Applications (INISTA) · 2021

The application of computer vision (CV) in many fields is becoming an important factor in development, and the demand for high accuracy CV products is increasing. Therefore, a new method that ensembles three trained models with the same architecture but different inputs is proposed. Each of the different channels of the images were used as input. The first channel of the image was used in the dataset for the first model, the second channel was used for the second model, and the third channel was used for the third model. The classification probabilities were achieved after training three different models with three channels. The model with the highest accuracy was selected as the main model, and it was added to the classification probabilities of certain classes of the main model. Applying this method to the classification task resulted in 75.45% accuracy, while using the entire image for the model resulted in 70.9% accuracy. The advantage of using this ensemble method is that it can be used simultaneously with other ensemble methods and achieves better results. Dividing an image into three different channels than simply using the entire image helps the model learn the image better.

Read the paper · More papers on PaperTik