Intelligent Fish Recognition Method Based on Variable-Step Size Learning Rate Optimization Strategy

基于变步长学习率优化策略的智能鱼类识别方法

阅读:1

Abstract

Fish capture usually requires classification of fish species, and the cost of manual classification is relatively high. Recently, deep learning has been widely applied in the fishery field. Transfer learning was conducted on ResNet18, ShuffleNet, EfficientNet, MobileNetV3, and YOLOv8. Through analysis of the influence of the law of learning rate on accuracy during the network learning process, a variable-step learning rate optimization strategy was proposed. Experimental results indicate that the optimal learning rates for fish classification utilizing this strategy were determined to be 0.01, 0.015, 0.001, 0.001, and 0.006 for ResNet18, ShuffleNet, EfficientNet, MobileNetV3, and YOLOv8, respectively. The recognition accuracy rates on the sample set reach 96.33%, 96.74%, 97.50%, 86.73%, 88.49%, respectively, and the average recognition accuracy rate between the sample set and other multi-species interfering fish reaches 93.13%, 93.44%, 96.13%, 95.21%, and 92.16%, respectively. This enables high-precision and rapid sorting of the target fish and other multi-species interfering fish. Compared with global optimization, the number of optimizations can be reduced by more than 97.1%; and compared with the same number of optimizations, the accuracy can be improved by more than 34.21%, which improves the efficiency and accuracy of network training and provides a theoretical reference for the setting of learning rate during model training in the field of deep learning.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。