摘要
arXiv:2502.00133v1 宣告类型: cross
摘要:深度学习方法在目标检测任务中表现出强大的性能;然而,在有限的训练数据下学习特定领域的应用仍然是一个重大挑战。迁移学习技术通过利用相关数据集预训练的知识,解决了这一问题,使新任务的学习更加快速和高效。找到合适的预训练数据集在确定迁移学习的成功以及整体模型性能方面起着关键作用。在本文中,我们研究了在七个不同的数据集上预训练YOLOv8n模型的影响,评估了这些数据集在转移到息肉检测任务中的有效性。我们比较了包含多样物体的大而通用的数据集与具有息肉类似特征的专一性数据集的效果。此外,我们评估了数据集大小对迁移学习效果的影响。在息肉数据集上的实验显示,预训练于相关数据集上的模型始终优于从头开始训练的模型,突显了在共享特定领域特征的数据集上预训练的好处。