对比多种模型在非洲野生动物图像分类中的表现,找出高精度与低资源消耗的平衡点。
Evaluating Deep Learning Models for African Wildlife Image Classification: From DenseNet to Vision Transformers
- 采用冻结特征提取器的迁移学习,比较四种主流模型在四类动物上的表现。
- 视觉变换器ViT-H/14准确率达99%,但计算成本极高;DenseNet-201以67%准确率成为最佳轻量选择。
- 将最优CNN模型部署至Hugging Face实时应用,适合野外环保人员使用。
非洲野生动物面临严重威胁,过去五十年脊椎动物数量下降超65%。为应对这一挑战,深度学习图像分类成为生物多样性监测与保护的有力工具。本文针对非洲野生动物图像自动分类问题,开展深度学习模型的对比研究,重点采用冻结特征提取器的迁移学习策略。基于包含水牛、大象、犀牛和斑马四个物种的公开数据集,评估了DenseNet-201、ResNet-152、EfficientNet-B4及视觉变换器ViT-H/14的表现。结果显示,卷积网络中DenseNet-201表现最佳(准确率67%),而ViT-H/14达到最高整体准确率(99%),但计算开销显著增加,带来部署难题。实验揭示了准确率、资源需求与可部署性之间的权衡。最优卷积模型DenseNet-201被集成至Hugging Face Gradio Space,实现野外实时应用,验证了轻量化模型在保护场景中的可行性。本研究为基于非洲数据的AI研究提供了模型选型、数据准备与负责任部署的实践指导。
原文摘要 · Abstract (English)
Wildlife populations in Africa face severe threats, with vertebrate numbers declining by over 65% in the past five decades. In response, image classification using deep learning has emerged as a promising tool for biodiversity monitoring and conservation. This paper presents a comparative study of deep learning models for automatically classifying African wildlife images, focusing on transfer learning with frozen feature extractors. Using a public dataset of four species: buffalo, elephant, rhinoceros, and zebra; we evaluate the performance of DenseNet-201, ResNet-152, EfficientNet-B4, and Vision Transformer ViT-H/14. DenseNet-201 achieved the best performance among convolutional networks (67% accuracy), while ViT-H/14 achieved the highest overall accuracy (99%), but with significantly higher computational cost, raising deployment concerns. Our experiments highlight the trade-offs between accuracy, resource requirements, and deployability. The best-performing CNN (DenseNet-201) was integrated into a Hugging Face Gradio Space for real-time field use, demonstrating the feasibility of deploying lightweight models in conservation settings. This work contributes to African-grounded AI research by offering practical insights into model selection, dataset preparation, and responsible deployment of deep learning tools for wildlife conservation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。