详细信息
Research on Wood Defects Feature Imbalance Optimization and Recognition ( SCI-EXPANDED收录 EI收录)
文献类型:期刊文献
英文题名:Research on Wood Defects Feature Imbalance Optimization and Recognition
第一作者:王霄
通信作者:Wang, X[1]
机构:[1]Chinese Acad Forestry, Res Inst Wood Ind, Beijing 100091, Peoples R China
年份:2025
卷号:13
起止页码:23841-23850
外文期刊名:IEEE ACCESS
收录:;EI(收录号:20250717866907);Scopus(收录号:2-s2.0-85217520801);WOS:【SCI-EXPANDED(收录号:WOS:001420279300031)】;
基金:This work was supported in part by the Special Fund of Basic Research Funding of Chinese Central Nonprofit Research Institutions under Grant CAFYBB2022ZC004-2.
语种:英文
外文关键词:Feature extraction; Entropy; Accuracy; Deep learning; Training; Inspection; Image recognition; Image color analysis; Production; Mathematical models; Computer vision; wood knot; classification model; deep learning; loss function
摘要:Deep learning is a promising method to achieve automatic wood defects detection which is indispensable for wood production; however, such a technique faces challenges caused by poor generalization ability and low recognition accuracy on light defects. In this study, the problems are attributed to imbalanced feature distribution of wood defects which are rich in diversity. To this end, two improvements on CNN classifier loss function are proposed to enhance the performance on wood defects recognition. Firstly, Cross Entropy loss is modified to allow the model to pay more attention to individual sample discrepancy during training. Secondly, a new loss, Var Loss is proposed to add to main loss function in order to decrease the variance of classification results, thus reducing the impact of sample diversity on model performance. Classical CNN classifiers are employed to distinguish three kinds of wood images: dead knot, live knot and ordinary grain. Results show that Modified Cross Entropy makes model more sensitive to hard samples regardless of the batchsize. Var Loss tends to decrease the fluctuation of prediction confidences, making model more robust in practical use. An overall accuracy increase of 6.6% is achieved, the accuracy on live knot has an increase of 17%, and the missing rate of defects is decreased by 15%. Besides, generalization ability test indicates that new methods allow the classifier to have comparable accuracies on five additional datasets. The proposed loss functions improve the model performance through optimizing the model training process, providing a new idea for deep learning application in wood defects detection.
参考文献:
正在载入数据...