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

什么是 EfficientNet?快速概览

了解 EfficientNet 架构及其复合缩放魔法!探索 EfficientNet B0-B7 以获得顶级的图像分类和分割效率。

ABAbirami Vina
6 min read
EfficientNet 架构概述

2019 年,Google AI 的研究人员推出了 EfficientNet,这是一款尖端的计算机视觉模型,旨在识别图像中的物体和模式。它最初是为图像分类而设计的,即将图像分配到几个预定义类别中的一类。然而,如今 EfficientNet 也被用作目标检测、分割和迁移学习等更复杂任务的主干网络。

在 EfficientNet 出现之前,此类机器学习和视觉 AI 模型试图通过增加层数或扩大这些层的大小来提高准确性。层是神经网络模型(一种受人脑启发的深度学习模型)中处理数据以学习模式并提高准确性的步骤。

这些改变造成了一种权衡,使得传统的 AI 模型变得更大、更慢,而所增加的准确性与所需的巨大计算能力增长相比,往往微不足道。

EfficientNet 采取了不同的方法。它以平衡的方式同时增加了深度(层数)、宽度(每层的单元数)和图像分辨率(输入图像的细节级别)。这种称为复合缩放的方法能够可靠地利用所有可用的处理能力。最终结果是一个更小、更快的模型,其表现优于 ResNet 或 DenseNet 等旧模型。

如今,像 Ultralytics YOLO11 这样更新的计算机视觉模型提供了更高的准确性、速度和效率。即便如此,EfficientNet 仍然是一个重要的里程碑,影响了许多高级架构的设计。

在本文中,我们将用五分钟的时间剖析 EfficientNet,涵盖它的工作原理、独特之处以及它在计算机视觉中仍然重要的原因。让我们开始吧!

Link to this section什么是 EfficientNet?#

在设计 EfficientNet 之前,大多数图像识别模型通过调整层数或增加输入图像的大小来获取更多细节以提高准确性。虽然这些策略改善了结果,但也使模型变得更重、要求更高。这意味着它们需要更多的内存和更好的硬件。

EfficientNet 没有更改单个层,而是使用一种称为复合缩放的方法同时缩放深度、宽度和图像分辨率。这种方法使模型能够高效增长,而不会过分侧重于任何单一维度。

EfficientNet 架构通过一系列块处理图像,每个块都由更小的模块构成。每个块中的模块数量取决于模型的大小。

EfficientNet 的构建模块

图 1. EfficientNet 的构建块。(来源)

较小的版本使用较少的模块,而较大的版本则更频繁地重复模块。这种灵活的设计使 EfficientNet 能够在从移动设备到大规模系统的各种应用中提供高准确性和效率。

Link to this section复合缩放的工作原理#

复合缩放方法扩展了模型的深度、宽度和图像分辨率,但保持了它们的平衡。这使得高效利用计算能力成为可能。该系列从一个称为 EfficientNet-B0 的较小基准模型开始,它构成了所有其他版本的基础。

从 B0 开始,模型扩展为更大的变体,命名为 EfficientNet-B1 到 EfficientNet-B7。随着每一步的推进,网络增加了额外的层,增加了通道数量(用于处理的单元),并处理更高分辨率的输入图像。每一步的增长量由一个称为复合系数的参数决定,该参数确保深度、宽度和分辨率按固定比例增加,而不是独立增加。

复合缩放增加了模型的宽度、深度和图像分辨率

图 2. 复合缩放增加了模型的宽度、深度和图像分辨率。(来源)

Link to this sectionEfficientNet 架构#

接下来,让我们看看 EfficientNet 的架构。

它构建在 MobileNetV2 之上,这是一款针对移动和嵌入式设备优化的轻量级计算机视觉模型。其核心是移动倒置瓶颈卷积 (MBConv) 块,这是一个特殊的层,它像标准卷积一样处理图像数据,但计算量更少。这个块使模型既快速又更具内存效率。

每个 MBConv 块内部都有一个挤压与激励 (SE) 模块。此模块调整网络中不同通道的强度。它增强基本通道的强度并降低其他通道的强度。该模块帮助网络专注于图像中最重要的特征,同时忽略其余部分。EfficientNet 模型还使用了 Swish 激活函数(一种帮助网络学习模式的数学函数),这比旧方法能更好地发现图像中的模式。

此外,它还使用了 DropConnect,即在训练期间随机关闭网络内部的一些连接。这种随机正则化方法(一种防止模型记忆训练数据而不是泛化的随机化技术)通过强制网络学习更稳健的特征表示(数据中更强、更通用的模式)来减少过拟合,从而更好地迁移到未见数据。

EfficientNet-B0 的架构

图 3. EfficientNet-B0 的架构 (来源)

Link to this sectionEfficientNet 模型变体简述#

既然我们已经更好地了解了 EfficientNet 模型的工作原理,让我们讨论一下不同的模型变体。

EfficientNet 模型从 B0 到 B7 进行缩放,以 B0 作为平衡速度和准确性的基准。每个版本都增加了深度、宽度和图像分辨率,从而提高了准确性。然而,它们也需要更多的计算能力,从 B1 和 B2 到高性能的 B6 和 B7。

虽然 EfficientNet-B3 和 EfficientNet-B4 模型在较大图像方面取得了平衡,但 B5 通常用于需要高精度的复杂数据集。除了这些模型之外,最新的模型 EfficientNet V2 可以提高训练速度,更好地处理小型数据集,并针对现代硬件进行了优化。

Link to this sectionEfficientNet 的应用#

EfficientNet 可以在使用比许多其他模型更少的内存和处理能力的同时产生准确的结果。这使得它在从科学研究到人们日常使用的产品等许多领域都很有用。

Link to this section医学图像分析#

医学图像,例如肺部 CT 扫描,通常包含对准确诊断至关重要的细微细节。AI 模型可以协助分析这些图像,以揭示人类难以检测到的模式。EfficientNet 在这方面的一个应用是 MONAI (Medical Open Network for AI) EfficientNet,它是专门为医学图像分析设计的。

在 EfficientNet 架构的基础上,研究人员还开发了 Lung-EffNet,这是一种对肺部 CT 扫描进行分类以检测肿瘤的模型。它可以将肿瘤分类为良性、恶性或正常,在实验环境中报告的准确率超过 99%。

使用 Lung-EffNet 进行肿瘤图像分类

图 4. 使用 Lung-EffNet 进行肿瘤图像分类。(来源)

Link to this section实时目标检测#

目标检测是查找图像中的对象并确定其位置的过程。它是安防系统、自动驾驶汽车和无人机等应用的关键部分。

EfficientNet 在该领域变得重要,因为它提供了一种非常有效的图像特征提取方式。它缩放深度、宽度和分辨率的方法展示了模型如何在不变得过重或缓慢的情况下保持准确。这就是为什么许多检测系统(如 EfficientDet)使用 EfficientNet 作为其主干网络的原因。

更新的模型,如 Ultralytics YOLO11,也有着将速度与准确性相结合的共同目标。这种向高效模型发展的趋势深受 EfficientNet 等架构理念的影响。

Link to this sectionEfficientNet 的优缺点#

以下是在计算机视觉项目中使用 EfficientNet 的一些好处:

  • 高准确性且参数更少: EfficientNet 可以提供与 ResNet 或 DenseNet 等旧模型相似或更好的准确性。然而,它使用的参数更少,因此训练更快,部署也更容易。
  • 可扩展的模型系列: 从 B0 到 B7,你可以选择与你的硬件和精度要求相匹配的版本,而无需更改基准网络。
  • 适合迁移学习: EfficientNet 可以为迁移学习提供可靠的模型性能,这是一个为自定义任务重新训练预训练模型的过程。它可以作为各种计算机视觉任务的主干。它在微调后也显示出了强大的结果。例如,它在 CIFAR-100(一种广泛使用的图像分类数据集)上实现了最先进的准确率,且参数比以前的模型少得多。

虽然使用 EfficientNet 有很多好处,但以下是需要牢记的局限性:

  • 需要更多内存: EfficientNet-B6 和 EfficientNet-B7 等版本需要大量的 GPU 内存。
  • 缩放针对 ImageNet 进行了调整: 缩放设置是为 ImageNet 数据集设计的,因此在非常不同的数据集上,如果不进行微调,性能可能会下降。对于小型数据集尤其如此,因为 EfficientNet 的架构和缩放是为像 ImageNet 这样庞大且多样化的数据集而设计的,后者提供了足够的数据来证明其深度和宽度的合理性。
  • 在某些硬件上较慢: EfficientNet 使用称为 MBConv 的层,这些层旨在提高现代硬件上的效率。在较旧的 GPU 或 CPU 上,这些层的运行速度可能会变慢。

Link to this section关键要点#

EfficientNet 通过保持深度、宽度和图像分辨率的平衡,改变了计算机视觉模型的增长方式。它仍然是一个重要的模型,也影响了更新的架构。特别是,它在计算机视觉史上占有重要地位。

加入我们的社区GitHub 存储库以探索更多关于 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 的未来!

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