用强化学习自动选最优模型,提升小数据下动物检测效率
Policy-Driven Transfer Learning in Resource-Limited Animal Monitoring
- 用上置信度算法自动筛选最适合的预训练模型
- 检测率更高,计算时间减少显著
- 适合缺乏经验的研究者快速部署动物监测系统
野生动物保护与牲畜管理中的动物健康监测与种群管理日益依赖自动化检测与追踪系统。基于无人机与计算机视觉的方案虽能非侵入式地覆盖复杂地形,但标注数据有限仍是构建有效深度学习模型的主要障碍。迁移学习可缓解此问题,使大模型在数据稀缺场景中适配使用。然而,众多预训练网络架构使模型选择困难,尤其对新手而言。本文提出一种基于强化学习的迁移学习框架,采用上置信度(UCB)算法自动选取最合适的预训练模型用于动物检测任务。该方法系统评估并排序候选模型,简化了模型选择流程。实验表明,本框架在保持更高检测率的同时,显著降低了计算时间,优于传统方法。
原文摘要 · Abstract (English)
Animal health monitoring and population management are critical aspects of wildlife conservation and livestock management that increasingly rely on automated detection and tracking systems. While Unmanned Aerial Vehicle (UAV) based systems combined with computer vision offer promising solutions for non-invasive animal monitoring across challenging terrains, limited availability of labeled training data remains an obstacle in developing effective deep learning (DL) models for these applications. Transfer learning has emerged as a potential solution, allowing models trained on large datasets to be adapted for resource-limited scenarios such as those with limited data. However, the vast landscape of pre-trained neural network architectures makes it challenging to select optimal models, particularly for researchers new to the field. In this paper, we propose a reinforcement learning (RL)-based transfer learning framework that employs an upper confidence bound (UCB) algorithm to automatically select the most suitable pre-trained model for animal detection tasks. Our approach systematically evaluates and ranks candidate models based on their performance, streamlining the model selection process. Experimental results demonstrate that our framework achieves a higher detection rate while requiring significantly less computational time compared to traditional methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。