A novel method combining deep learning with the Kennard–Stone algorithm for training dataset selection for image‐based rice seed variety identification
Chen Jin, Xinyue Zhou, Mengyu He, Li Cheng, Zeyi Cai, Lei Zhou, Hengnian Qi, Chu Zhang
Abstract
BACKGROUND: Different varieties of rice vary in planting time, stress resistance, and other characteristics. With advances in rice-breeding technology, the number of rice varieties has increased significantly, making variety identification crucial for both trading and planting. RESULTS: This study collected RGB images of 20 hybrid rice seed varieties. An enhanced deep super-resolution network (EDSR) was employed to enhance image resolution, and a variety classification model utilizing the high-resolution dataset demonstrated superior performance to that of the model using the low-resolution dataset. A novel training sample selection methodology was introduced integrating deep learning with the Kennard-Stone (KS) algorithm. Convolutional neural networks (CNN) and autoencoders served as supervised and unsupervised feature extractors, respectively. The extracted feature vectors were subsequently processed by the KS algorithm to select training samples. The proposed methodologies exhibited superior performance over the random selection approach in rice variety classification, with an approximately 10.08% improvement in overall classification accuracy. Furthermore, the impact of noise on the proposed methodology was investigated by introducing noise to the images, and the proposed methodologies maintained superior performance relative to the random selection approach on the noisy image dataset. CONCLUSION: The experimental results indicate that both supervised and unsupervised learning models performed effectively as feature extractors, and the deep learning framework significantly influenced the selection of training set samples. This study presents a novel approach for training sample selection in classification tasks and suggests the potential for extending the proposed method to image datasets and other types of datasets. Further exploration of this potential is warranted. © 2024 Society of Chemical Industry.