An Automated CNN Recommendation System for Image Classification Tasks
This addresses the difficulty for ordinary users and experts in choosing CNN models for image classification tasks, though it appears incremental.
The authors tackled the problem of selecting optimal CNN models for image classification by proposing an automated recommendation system that evaluates task complexity and model ability without training, resulting in fast and accurate recommendations.
Nowadays the CNN is widely used in practical applications for image classification task. However the design of the CNN model is very professional work and which is very difficult for ordinary users. Besides, even for experts of CNN, to select an optimal model for specific task may still need a lot of time (to train many different models). In order to solve this problem, we proposed an automated CNN recommendation system for image classification task. Our system is able to evaluate the complexity of the classification task and the classification ability of the CNN model precisely. By using the evaluation results, the system can recommend the optimal CNN model and which can match the task perfectly. The recommendation process of the system is very fast since we don't need any model training. The experiment results proved that the evaluation methods are very accurate and reliable.