A Method for Enhancing the Accuracy of Pet Breeds Identification Model in Complex Environments
Source abstract
Abstract
Most existing studies on pet breeds classification focus on images with simple backgrounds, leading to the unsatisfactory performance of models in practical applications. This paper investigates training pet breeds classification models using complex images and constructs a dataset for identifying breeds of pet cats and dogs. We use this dataset to fine-tune three SOTA models: ResNet34, DenseNet121, and Swin Transformer. Specifically, in terms of top-1 accuracy, the performance of DenseNet is improved from 89.10% to 89.19%, while that of the Swin Transformer is increased by 1.26%, marking the most significant enhancement. The results show that training with our dataset significantly enhances the models’ classification capabilities in complex environments. Additionally, we offer a lightweight pet breeds identification model based on PBI-EdgeNeXt (Pet Breeds Identification EdgeNeXt). We utilizes the PolyLoss function and Sophia optimizer for model training. Furthermore, we compare our model with five commonly used lightweight models and find that the proposed model achieves the highest top-1 accuracy of 87.12%. These results demonstrate that the model achieves high accuracy, reaching the SOTA level.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Food Supply Chain Traceability · Gaussian Processes and Bayesian Inference · Identification and Quantification in Food
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
School of Electronic and Information Engineering, Guangxi Normal University, Guilin 541004, China
References (36)
Show source-provided references
1. Forensic DNA phenoty**: Canis familiaris breed classification and skeletal phenotype prediction using functionally significant skeletal SNPs and indels10.1111/age.13165 ↗
2. Imagenet classification with deep convolutional neural networks
3. Long, J., Shelhamer, E., and Darrell, T. (2015, January 7–12). Fully convolutional networks for semantic segmentation. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Boston, MA, USA.10.1109/CVPR.2015.7298965 ↗
4. Ronneberger, O., Fischer, P., and Brox, T. (2015, January 5–9). U-net: Convolutional networks for biomedical image segmentation. Proceedings of the Medical Image Computing and Computer-Assisted Intervention–MICCAI 2015: 18th International Conference, Munich, Germany. proceedings, part III 18.
5. Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., Dehghani, M., Minderer, M., Heigold, G., and Gelly, S. (2020). An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale. arXiv.
6. An end-to-end steel strip surface defects recognition system based on convolutional neural networks10.1002/srin.201600068 ↗
7. Deng, J., Dong, W., Socher, R., Li, L.J., Li, K., and Fei-Fei, L. (2009, January 20–25). Imagenet: A large-scale hierarchical image database. Proceedings of the 2009 IEEE Conference on Computer Vision and Pattern Recognition, Miami, FL, USA.10.1109/CVPR.2009.5206848 ↗
8. Krizhevsky, A., and Hinton, G. (2009). Learning multiple layers of features from tiny images. Handb. Syst. Autoimmune Dis., 7.
9. Fe-Fei, L. (2003, January 13–16). A Bayesian approach to unsupervised one-shot learning of object categories. Proceedings of the Ninth IEEE International Conference on Computer Vision, Nice, France.
10. Griffin, G., Holub, A., and Perona, P. (2007). Caltech-256 Object Category Dataset, California Institute of Technology.
11. Hybrid Deep Learning Approach for Enhanced Animal Breed Classification and Prediction10.18280/ts.400526 ↗
12. Plant leaf classification using GIST texture features10.1049/iet-cvi.2018.5028 ↗
13. Mabrouk, A.B., Najjar, A., and Zagrouba, E. (2014, January 5–8). Image flower recognition based on a new method for color feature extraction. Proceedings of the 2014 International Conference on Computer Vision Theory and Applications (VISAPP), Lisbon, Portugal.
14. Cuthill.How background complexity impairs target detection10.1016/j.anbehav.2024.02.003 ↗
15. Parkhi, O.M., Vedaldi, A., Zisserman, A., and Jawahar, C.V. (2012, January 16–21). Cats and dogs. Proceedings of the 2012 IEEE Conference on Computer Vision and Pattern Recognition, Providence, RI, USA.10.1109/CVPR.2012.6248092 ↗
16. Liu, Z., Lin, Y., Cao, Y., Hu, H., Wei, Y., Zhang, Z., Lin, S., and Guo, B. (2021, January 11–17). Swin transformer: Hierarchical vision transformer using shifted windows. Proceedings of the IEEE/CVF International Conference on Computer Vision, Virtual Conference.10.1109/ICCV48922.2021.00986 ↗
17. Huang, G., Liu, Z., Van Der Maaten, L., and Weinberger, K.Q. (2017, January 21–26). Densely connected convolutional networks. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, HI, USA.10.1109/CVPR.2017.243 ↗
18. He, K., Zhang, X., Ren, S., and Sun, J. (2016, January 27–30). Deep residual learning for image recognition. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA.10.1109/CVPR.2016.90 ↗
19. Remote Sensing Cross-Modal Text-Image Retrieval Based on Global and Local Information
20. A Joint Saliency Temporal–Spatial–Spectral Information Network for Hyperspectral Image Change Detection10.1109/TGRS.2024.3428551 ↗
21. Convolutional LSTM-Based Hierarchical Feature Fusion for Multispectral Pan-Sharpening
22. Sabour, S., Frosst, N., and Hinton, G.E. (2017). Dynamic routing between capsules. Adv. Neural Inf. Process. Syst., 30.
23. Gabor capsule network with preprocessing blocks for the recognition of complex images10.1007/s00138-021-01221-6 ↗
24. Chen, Y., Fan, H., Xu, B., Yan, Z., Kalantidis, Y., Rohrbach, M., Yan, S., and Feng, J. (November, January 27). Drop an octave: Reducing spatial redundancy in convolutional neural networks with octave convolution. Proceedings of the IEEE/CVF International Conference on Computer Vision, Seoul, Republic of Korea.
25. Maaz, M., Shaker, A., Cholakkal, H., Khan, S., Zamir, S.W., Anwer, R.M., and Shahbaz Khan, F. (2022, January 23–27). Edgenext: Efficiently amalgamated cnn-transformer architecture for mobile vision applications. Proceedings of the European Conference on Computer Vision, Tel Aviv, Israel.10.1007/978-3-031-25082-8_1 ↗
26. Islam, M.A., Jia, S., and Bruce, N.D. (2020). How much position information do convolutional neural networks encode?. arXiv.
27. Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A.N., Kaiser, Ł., and Polosukhin, I. (2017). Attention is all you need. Adv. Neural Inf. Process. Syst., 30.
28. A tutorial on the cross-entropy method10.1007/s10479-005-5724-z ↗
29. Lin, T.Y., Goyal, P., Girshick, R., He, K., and Dollár, P. (2017, January 22–29). Focal loss for dense object detection. Proceedings of the IEEE International Conference on Computer Vision, Venice, Italy.10.1109/ICCV.2017.324 ↗
30. Cui, Y., Jia, M., Lin, T.Y., Song, Y., and Belongie, S. (2019, January 15–20). Class-Balanced Loss Based on Effective Number of Samples. Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) (2019), Long Beach, CA, USA.10.1109/CVPR.2019.00949 ↗
31. Leng, Z., Tan, M., Liu, C., Cubuk, E.D., Shi, X., Cheng, S., and Anguelov, D. (2022). PolyLoss: A Polynomial Expansion Perspective of Classification Loss Functions. arXiv.
32. Kingma, D.P., and Ba, J. (2014). Adam: A Method for Stochastic Optimization. arXiv.
33. Keskar, N.S., and Socher, R. (2017). Improving generalization performance by switching from adam to sgd. arXiv.
34. Liu, H., Li, Z., Hall, D., Liang, P., and Ma, T. (2023). Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training. arXiv.
35. Aarizou, A., and Merah, M. (2022, January 8–9). Transfer Learning for Plant Disease Detection on Complex Images. Proceedings of the 2022 7th International Conference on Image and Signal Processing and Their Applications (ISPA), Mostaganem, Algeria.10.1109/ISPA54004.2022.9786306 ↗
36. Cahyo, D.D.N., Sunyoto, A., and Ariatmanto, D. (2023, January 10–11). Transfer Learning and Fine-tuning Effect Analysis on Classification of Cat Breeds using a Convolutional Neural Network. Proceedings of the 2023 6th International Conference on Information and Communications Technology (ICOIACT), Yogyakarta, Indonesia.10.1109/ICOIACT59844.2023.10455771 ↗