Analyze weeds classification with visual explanation based on Convolutional Neural Networks
1
조회
0
다운로드
3
인용
0
참고문헌
키워드
초록
To understand how a Convolutional Neural Network (CNN) model captures the features of a pattern to determine which class it belongs to, in this paper, we use Gradient-weighted Class Activation Mapping (Grad-CAM) to visualize and analyze how well a CNN model behave on the CNU weeds dataset. We apply this technique to Resnet model and figure out which features this model captures to determine a specific class, what makes the model get a correct/wrong classification, and how those wrong label images can cause a negative effect to a CNN model during the training process. In the experiment, Grad-CAM highlights the important regions of weeds, depending on the patterns learned by Resnet, such as the lobe and limb on 미국가막사리, or the entire leaf surface on 단풍잎돼지풀. Besides, Grad-CAM points out a CNN model can localize the object even though it is trained only for the classification problem.
Abstract
To understand how a Convolutional Neural Network (CNN) model captures the features of a pattern to determine which class it belongs to, in this paper, we use Gradient-weighted Class Activation Mapping (Grad-CAM) to visualize and analyze how well a CNN model behave on the CNU weeds dataset. We apply this technique to Resnet model and figure out which features this model captures to determine a specific class, what makes the model get a correct/wrong classification, and how those wrong label images can cause a negative effect to a CNN model during the training process. In the experiment, Grad-CAM highlights the important regions of weeds, depending on the patterns learned by Resnet, such as the lobe and limb on 미국가막사리, or the entire leaf surface on 단풍잎돼지풀. Besides, Grad-CAM points out a CNN model can localize the object even though it is trained only for the classification problem.
참고문헌 (9)
- 박선, "오픈 소스 기반의 딥러닝을 이용한 적조생물 이미지 분류", 스마트미디어저널, 7(2), pp.34~39 (2018), (사)한국스마트미디어학회[DOI]
- 김서정, "딥러닝을 이용한 양파 밭의 잡초검출 연구", 스마트미디어저널, 7(3), pp.16~21 (2018), (사)한국스마트미디어학회[DOI]
- R. R. Selvaraju, "Grad-cam: Visual explanations from deep networks via gradient-based localization", Proc. of the IEEE International Conference on Computer Vision, pp.618~626 (2017)
- M. D. Zeiler, "Visualizing and understanding convolutional networks", European conference on computer vision, pp.818~833 (2014)
- B. Zhou, "Learning deep features for discriminative localization", Proc. of the IEEE conference on computer vision and pattern recognition, pp.2921~2929 (2016)
- K. Simonyan, "Very deep convolutional networks for large-scale image recognition", ~ (2014)
- K. He, "Deep residual learning for image recognition", Proc. of the IEEE conference on computer vision and pattern recognition, pp.770~778 (2016)
- K. He, "Identity mappings in deep residual networks", European conference on computer vision, pp.630~645 (2016), Springer
- "ImageNet"[URL]