遇见 YOLO26: 下一代视觉 AI。
Ultralytics
指南

什么是计算机视觉中的过拟合,以及如何预防?

学习什么是计算机视觉中的过拟合,以及如何使用数据增强、正则化和预训练模型来防止它。

ABAbdelrahman Elgendy
6 min read
计算机视觉模型中的过拟合以及如何预防

计算机视觉模型旨在识别模式、检测对象和分析图像。然而,它们的性能取决于它们对未见数据的泛化能力。泛化是指模型在处理新图像(而不仅仅是训练所用的图像)时表现良好的能力。在训练这些模型时,一个常见的问题是过拟合,即模型从训练数据中学到了太多内容,包括不必要的噪声,而不是识别出有意义的模式。

当这种情况发生时,模型在训练数据上表现良好,但在处理新图像时却很吃力。例如,一个仅在高清、光照充足的图像上训练的对象检测模型,在现实环境的模糊或阴影图像中可能会失效。过拟合限制了模型的适应性,从而限制了它在自动驾驶、医学成像和安全系统等现实应用中的使用。

在本文中,我们将探讨什么是过拟合、它为何发生以及如何预防它。我们还将了解 Ultralytics YOLO11 等计算机视觉模型如何帮助减少过拟合并提高泛化能力。

Link to this section什么是过拟合?#

当模型记住了训练数据而不是学习可广泛应用于新输入的模式时,就会发生过拟合。模型过于关注训练数据,因此在处理它以前没见过的新图像或情况时会感到困难。

在计算机视觉中,过拟合会影响不同的任务。仅在明亮、清晰图像上训练的分类模型可能在低光照条件下表现不佳。从完美图像中学习的对象检测模型可能会在拥挤或混乱的场景中失败。同样,实例分割模型在受控环境下可能运行良好,但在处理阴影或重叠物体时可能会遇到问题。

这在现实世界的 AI 应用中成为一个问题,因为模型必须能够在受控训练条件之外进行泛化。例如,自动驾驶汽车必须能够在不同的光照条件、天气和环境下检测行人。过拟合训练集的模型在这些不可预测的场景中将无法可靠地运行。

Link to this section过拟合何时发生以及为什么会发生?#

过拟合通常是由于数据集不平衡、模型复杂度过高和过度训练引起的。以下是主要原因:

  • 有限的训练数据:小型数据集使模型倾向于记住模式而不是泛化模式。一个仅在 50 张鸟类图像上训练的模型在检测该数据集之外的鸟类品种时可能会遇到困难。
  • 参数过多的复杂模型:具有过多层和神经元的深度网络倾向于记住细节,而不是专注于基本特征。
  • 缺乏数据增强:如果没有裁剪、翻转或旋转等变换,模型可能只会从其精确的训练图像中学习。
  • 过度训练:如果模型遍历训练数据的次数过多,即所谓的轮次,它会记住细节而不是学习通用模式,从而使其适应性变差。
  • 不一致或嘈杂的标签:错误标记的数据会导致模型学习到错误的模式。这在手动标记的数据集中很常见。

在模型复杂度、数据集质量和训练技术之间保持良好的平衡可以确保更好的泛化效果。

Link to this section过拟合与欠拟合#

过拟合和欠拟合是深度学习中完全截然相反的两个问题。

计算机视觉模型中欠拟合、最佳学习和过拟合的对比

图 1. 计算机视觉模型中欠拟合、最佳学习和过拟合的比较。

过拟合发生在模型过于复杂时,导致它过度关注训练数据。它不是学习通用模式,而是记住了小细节,甚至是像背景噪声这样无关紧要的内容。这导致模型在训练数据上表现良好,但在处理新图像时却很吃力,这意味着它并没有真正学会如何识别适用于不同情况的模式。

欠拟合发生在模型过于简单时,导致它错过了数据中的重要模式。这可能发生在模型层数太少、训练时间不够或数据有限时。结果是,它无法识别重要模式并做出不准确的预测。这会导致在训练数据和测试数据上的表现都很差,因为模型没有学到足够的内容来正确理解任务。

训练有素的模型能在复杂性和泛化之间找到平衡。它应该足够复杂以学习相关模式,但又不至于复杂到在记住数据的同时却无法识别潜在的关系。

Link to this section如何识别过拟合#

以下是一些表明模型出现过拟合的迹象:

  • 如果训练准确率显著高于验证准确率,模型很可能已经过拟合。
  • 训练损失和验证损失之间差距的扩大是另一个强有力的指标。
  • 模型对错误的答案过于自信,这表明它记住了细节而不是理解了模式。

为了确保模型具有良好的泛化能力,它需要在反映现实世界条件的多样化数据集上进行测试。

Link to this section如何预防计算机视觉中的过拟合#

过拟合并非不可避免,是可以预防的。通过正确的技术,计算机视觉模型可以学习通用模式,而不是记住训练数据,从而使其在现实应用中更加可靠。

以下是预防计算机视觉中过拟合的五个关键策略。

Link to this section通过增强和合成数据提高数据多样性#

帮助模型在新数据上表现良好的最佳方式是利用数据增强和合成数据来扩展数据集。合成数据是计算机生成的,而不是从现实世界的图像中收集的。它有助于在真实数据不足时填补空白。

结合真实世界数据与合成数据以减少过拟合并提高检测准确率

图 2. 结合现实世界和合成数据可以减少过拟合并提高对象检测的准确性。

数据增强通过翻转、旋转、裁剪或调整亮度来稍微改变现有图像,因此模型不会仅仅记住细节,而是学会识别不同情况下的对象。

当难以获得真实图像时,合成数据非常有用。例如,自动驾驶汽车模型可以在计算机生成的道路场景上进行训练,以学习如何在不同天气和光照条件下检测对象。这使得模型更加灵活可靠,而无需数千张真实世界的图像。

Link to this section优化模型复杂度和架构#

深度神经网络(一种具有多层处理数据而非单层的机器学习模型类型)并不总是越深越好。当模型具有太多的层或参数时,它会记住训练数据而不是识别更广泛的模式。减少不必要的复杂度有助于防止过拟合。

为了实现这一点,一种方法是剪枝,它会移除冗余的神经元和连接,使模型更精简、更高效。

另一种方法是通过减少层数或神经元数量来简化架构。YOLO11 等预训练模型旨在通过较少的参数在任务中实现良好的泛化,这使得它们比从头开始训练深度模型更具抗过拟合能力。

在模型深度和效率之间找到正确的平衡,有助于它在不简单记住训练数据的情况下学习有用的模式。

Link to this section应用正则化技术#

正则化技术可以防止模型过度依赖训练数据中的特定特征。以下是一些常用的技术:

  • Dropout 在训练期间随机关闭模型的一部分,这样它就能学习识别不同的模式,而不是过于依赖少数几个特征。
  • 权重衰减 (L2 正则化) 抑制极端权重值,从而控制模型的复杂度。
  • 批归一化 (Batch normalization) 通过确保模型对数据集中的变化不太敏感来帮助稳定训练。

这些技术有助于保持模型的灵活性和适应性,在保持准确性的同时降低过拟合的风险。

Link to this section利用验证和提前停止来监控训练#

为了预防过拟合,追踪模型的学习方式并确保其在新数据上具有良好的泛化能力非常重要。以下是几种有助于实现此目的的技术:

  • 提前停止 (Early stopping):在模型不再提升时自动结束训练,这样它就不会继续学习不必要的细节。
  • 交叉验证 (Cross-validation):将数据分为多个部分并在每个部分上训练模型。这有助于它学习模式,而不是记住特定的图像。

这些技术有助于模型保持平衡,从而在不至于过于关注单一训练数据的情况下学习到足够的准确性。

Link to this section使用预训练模型并改善数据集标注#

使用 YOLO11 等预训练模型而不是从头开始训练,可以减少过拟合。YOLO11 在大规模数据集上进行了训练,使其能够在不同条件下实现良好的泛化。

预训练计算机视觉模型以增强准确率并防止过拟合

图 3. 预训练计算机视觉模型可提高准确性并预防过拟合。

对预训练模型进行微调有助于它在学习新任务的同时保留已有的知识,这样它就不会仅仅记住训练数据。

此外,确保高质量的数据集标注至关重要。错误标注或不平衡的数据可能会误导模型学习到错误的模式。清理数据集、修复错误标注的图像并平衡类别可以提高准确性并降低过拟合风险。另一种有效的方法是对抗训练,通过让模型接触略有改动或更具挑战性的示例来测试其极限。

Link to this section关键要点#

过拟合是计算机视觉中的常见问题。模型在训练数据上可能表现良好,但在处理真实世界图像时却会遇到困难。为了避免这种情况,数据增强、正则化以及使用 YOLO11 等预训练模型等技术有助于提高准确性和适应性。

通过应用这些方法,AI 模型可以保持可靠并在不同环境中表现良好。随着深度学习的进步,确保模型能够正确泛化将是实现现实世界 AI 成功的关键。

加入我们日益壮大的社区!浏览我们的 GitHub 仓库以了解有关 AI 的更多信息。准备好开始你自己的计算机视觉项目了吗?查看我们的许可选项。通过访问我们的解决方案页面,发现自动驾驶中的视觉 AI医疗保健中的 AI

Explore solutions

Real-time AI that works with your team

机器人技术中的 AI

使用 Ultralytics YOLO 模型为智能机器赋能。机器人技术中的视觉 AI 可推动自主导航、感知、物体跟踪和实时控制。
了解更多
Real-time AI that works with your team

物流中的 AI

使用 Ultralytics YOLO 模型简化物流。视觉 AI 可实现包裹检查、分拣、车辆跟踪和实时仓库安全监控。
了解更多
Real-time AI that works with your team

零售业 AI

使用 Ultralytics YOLO 模型重塑零售业。视觉 AI 推动库存跟踪、货架监控、队列管理和更智能的客户洞察。
了解更多
Real-time AI that works with your team

医疗保健中的 AI

利用 Ultralytics YOLO 模型构建医疗保健解决方案。医疗保健中的视觉 AI 可助力更快速的医学影像分析、更智能的诊断和患者监测。
了解更多
Real-time AI that works with your team

制造业中的 AI

使用 Ultralytics YOLO 模型优化制造业。视觉 AI 推动质量控制、缺陷检测、PPE 合规性和装配线自动化。
了解更多
Real-time AI that works with your operation

汽车中的 AI

将计算机视觉应用于汽车行业,并配合 Ultralytics YOLO 模型。汽车视觉 AI 可提升道路安全、辅助驾驶和车辆自动化,打造更智能的道路。
了解更多
Real-time AI tailored to your operation

农业中的 AI

借助 Ultralytics YOLO 模型,将视觉 AI 引入智慧农业。赋能作物监测、牲畜追踪和精准农业,实现更高、更智能的产量。
了解更多
Real-time AI that works with your team

机器人技术中的 AI

使用 Ultralytics YOLO 模型为智能机器赋能。机器人技术中的视觉 AI 可推动自主导航、感知、物体跟踪和实时控制。
了解更多
Real-time AI that works with your team

物流中的 AI

使用 Ultralytics YOLO 模型简化物流。视觉 AI 可实现包裹检查、分拣、车辆跟踪和实时仓库安全监控。
了解更多
Real-time AI that works with your team

零售业 AI

使用 Ultralytics YOLO 模型重塑零售业。视觉 AI 推动库存跟踪、货架监控、队列管理和更智能的客户洞察。
了解更多
Real-time AI that works with your team

医疗保健中的 AI

利用 Ultralytics YOLO 模型构建医疗保健解决方案。医疗保健中的视觉 AI 可助力更快速的医学影像分析、更智能的诊断和患者监测。
了解更多
Real-time AI that works with your team

制造业中的 AI

使用 Ultralytics YOLO 模型优化制造业。视觉 AI 推动质量控制、缺陷检测、PPE 合规性和装配线自动化。
了解更多
Real-time AI that works with your operation

汽车中的 AI

将计算机视觉应用于汽车行业,并配合 Ultralytics YOLO 模型。汽车视觉 AI 可提升道路安全、辅助驾驶和车辆自动化,打造更智能的道路。
了解更多
Real-time AI tailored to your operation

农业中的 AI

借助 Ultralytics YOLO 模型,将视觉 AI 引入智慧农业。赋能作物监测、牲畜追踪和精准农业,实现更高、更智能的产量。
了解更多
Real-time AI that works with your team

机器人技术中的 AI

使用 Ultralytics YOLO 模型为智能机器赋能。机器人技术中的视觉 AI 可推动自主导航、感知、物体跟踪和实时控制。
了解更多
Real-time AI that works with your team

物流中的 AI

使用 Ultralytics YOLO 模型简化物流。视觉 AI 可实现包裹检查、分拣、车辆跟踪和实时仓库安全监控。
了解更多
Real-time AI that works with your team

零售业 AI

使用 Ultralytics YOLO 模型重塑零售业。视觉 AI 推动库存跟踪、货架监控、队列管理和更智能的客户洞察。
了解更多
Real-time AI that works with your team

医疗保健中的 AI

利用 Ultralytics YOLO 模型构建医疗保健解决方案。医疗保健中的视觉 AI 可助力更快速的医学影像分析、更智能的诊断和患者监测。
了解更多
Real-time AI that works with your team

制造业中的 AI

使用 Ultralytics YOLO 模型优化制造业。视觉 AI 推动质量控制、缺陷检测、PPE 合规性和装配线自动化。
了解更多
Real-time AI that works with your operation

汽车中的 AI

将计算机视觉应用于汽车行业,并配合 Ultralytics YOLO 模型。汽车视觉 AI 可提升道路安全、辅助驾驶和车辆自动化,打造更智能的道路。
了解更多
Real-time AI tailored to your operation

农业中的 AI

借助 Ultralytics YOLO 模型,将视觉 AI 引入智慧农业。赋能作物监测、牲畜追踪和精准农业,实现更高、更智能的产量。
了解更多

让我们一起构建 AI 的未来!

开启你的机器学习未来之旅