Classification of Toona sinensis Young Leaves Using Machine Learning and UAV-Borne Hyperspectral Imagery

Abstract

Rapid and accurate distinction between young and old leaves of Toona sinensis in the wild is of great significance to the selection of T. sinensis varieties and the evaluation of relative yield. In this study, UAV hyperspectral imaging technology was used to obtain canopy hyperspectral data of biennial seedlings of different varieties of T. sinensis to distinguish young and old leaves. Five classification models were trained, namely Random Forest (RF), Artificial Neural Network (ANN), Decision Tree (DT), Partial Least Squares Discriminant Analysis (PLSDA), and Support Vector Machine (SVM). Raw spectra and six preprocessing methods were used to fit the best classification model. Satisfactory accuracy was obtained from all the five models using the raw spectra. The SVM model showed good performance on raw spectra and all preprocessing methods, and yielded higher accuracy, sensitivity, precision, and specificity than other models. In the end, the SVM model based on the raw spectra produced the most reliable and robust prediction results (99.62% accuracy and 99.23% sensitivity on the validation set only, and 100.00% for the rest). Three important spectral regions of 422.7~503.2, 549.2, and 646.2~687.2 nm were found to be highly correlated with the identification of young leaves of T. sinensis. In this study, a fast and effective method for identifying young leaves of T. sinensis was found, which provided a reference for the rapid identification of young leaves of T. sinensis in the wild.

Keywords: Toona sinensis young leaves; classification; machine learning; preprocessing; variable selection.