术语表

微调

了解如何通过微调预训练模型(如YOLO )来优化人工智能,以更少的数据和资源完成特定领域的任务。

使用Ultralytics HUB 对YOLO 模型进行简单培训

了解更多

微调是将预先训练好的机器学习模型,通过在新数据上进一步训练,使其适应特定任务或数据集的过程。与从头开始训练模型相比,这种方法充分利用了模型已经获得的基础知识,减少了所需的计算资源和时间。在标注数据稀缺的情况下,或者在针对特定领域应用优化模型时,微调尤其有效。

微调的重要性

预训练模型(如迁移学习中使用的模型)是在大型通用数据集(如 ImageNet 或 COCO)上进行训练的。虽然这些模型擅长一般任务,但通过微调,它们可以专注于特定领域的细微差别,从而实现专业化。例如,针对一般物体检测训练的模型可以进行微调,以检测放射摄影中的医学异常或野生动物监测中的特定动物物种。

微调是一种具有成本效益和可扩展性的技术,可最大限度地减少对大型数据集和大量计算资源的需求。因此,它在医疗保健制造农业等行业的应用中尤为重要。

微调工作原理

微调通常包括以下步骤:

  1. 选择预训练模型:选择在与目标任务类似的任务上预先训练过的模型。例如,Ultralytics YOLO 模型是在 COCO 等数据集上预先训练的,用于物体检测任务。
  2. 冻结层:最初,预训练模型的某些层会被冻结,以保留广义特征。只有顶层会在新数据集上重新训练。
  3. 自定义模型:必要时修改架构。这可能包括添加特定任务层,如用于图像分类的分类头或用于姿态估计的回归层。
  4. 使用新数据训练:使用较小的学习率在新数据集上训练模型,确保原始特征不会被过度覆盖。
  5. 评估性能:使用mAPF1 分数等指标评估微调模型,确保模型在特定任务中表现良好。

有关培训和微调YOLO 模型的详细指南,请参阅Ultralytics YOLO 培训文档

微调的实际应用

1.医学影像

微调技术广泛应用于医学成像中的肿瘤检测或器官分割等任务。例如,在一般物体检测任务中预先训练好的模型,可以利用医疗数据集进行微调,以识别核磁共振扫描中的脑肿瘤。这样,医疗专业人员就能获得更高的诊断准确性和效率。

了解更多有关人工智能如何改变医疗保健行业的信息,请参阅《医疗保健中的人工智能》。

2.智慧农业

在农业领域,微调技术可以针对害虫检测或作物监测等特定挑战开发模型。例如,可以利用本地农业害虫数据集对一般物体检测模型进行微调,从而提高识别和减轻作物威胁的精度。

更多有关人工智能在农业中的作用的信息,请参阅《农业中的人工智能》。

3.零售自动化

在零售环境中,库存管理和顾客行为分析等任务都需要微调模型。可以对预先训练好的模型进行微调,以准确检测特定的产品类型或分析商店的人流模式。

要深入了解人工智能对零售效率的影响,请浏览《制造业中的人工智能》。

微调与相关技术

微调与迁移学习有相似之处,但在范围和应用上有所不同。迁移学习是将预先训练好的模型用作特征提取器,而不改变其权重,而微调则是针对新任务调整模型的权重。微调也不同于超参数调整,后者优化模型设置,但不涉及重新训练模型层。

此外,微调有别于零点学习少量学习。这些技术旨在以最少或没有新数据的情况下泛化到未见过的任务中,而微调则依赖于特定的数据集来提高目标任务的性能。

使用Ultralytics HUB 进行微调的好处

Ultralytics HUB 为训练和部署模型提供了一个直观的平台,从而简化了微调过程。借助基于云的培训和预训练Ultralytics YOLO 模型等功能,用户无需丰富的专业技术知识,即可轻松地针对各种应用对模型进行微调。了解有关Ultralytics HUB 及其功能的更多信息。

微调是现代机器学习的基石,它使开发人员和研究人员能够针对特定任务高效地定制模型。通过利用预训练模型和Ultralytics HUB 等工具,微调成为了在现实世界应用中优化人工智能性能的一种便捷而强大的方法。

阅读全部