遇见 YOLO26: 下一代视觉 AI。
Ultralytics
返回 Ultralytics 词汇表

ReLU (Rectified Linear Unit)

探索线性整流函数 (ReLU) 激活函数。了解它如何提高神经网络效率、防止梯度消失并为 AI 模型提供动力。

线性整流函数(Rectified Linear Unit),通常称为 ReLU,是深度学习领域中最基本且应用最广泛的 激活函数 之一。作为 神经网络 (NN) 中的数学守门人,ReLU 通过应用一个简单的非线性变换来决定神经元的输出:它允许正输入值直接通过,同时将所有负输入值转换为零。这种简单而强大的机制为模型引入了必要的非线性,使其能够学习数据中复杂的模式和结构——这是基础线性模型无法实现的。由于其计算效率高且在缓解 梯度消失 问题等训练难题方面表现有效,ReLU 已成为许多现代架构(包括 卷积神经网络 (CNN))中隐藏层的默认选择。

Link to this sectionReLU 的工作原理#

机器学习 (ML) 中使用的其他数学运算相比,ReLU 的核心逻辑非常简单。从概念上讲,它充当了一个在网络中引入稀疏性的过滤器。通过强制将负输入变为零,ReLU 确保在任何给定时刻只有一部分神经元处于激活状态。这种稀疏性模仿了生物神经元在人脑中的放电方式,并使网络的处理效率更高。

使用 ReLU 的好处包括:

  • 计算效率: 与涉及复杂指数计算的函数(如 Sigmoid 或 Tanh 函数)不同,ReLU 只需要简单的阈值运算。当在 GPU 等高性能硬件上训练大型模型时,这种速度至关重要。
  • 改善梯度流动:反向传播 过程中,ReLU 有助于为正输入保持良好的梯度流动。这解决了梯度消失问题,即误差信号变得太小而无法有效更新深度网络中的 模型权重
  • 稀疏激活: 通过将负值输出为绝对零,ReLU 创建了数据的稀疏表示,这可以在某些情况下简化模型并减少 过拟合 的可能性。

Link to this section实际应用#

ReLU 是无数 AI 应用的引擎,特别是那些需要快速处理图像和视频等高维数据的应用。

Link to this section自动驾驶感知#

自动驾驶 领域,安全性取决于实时检测和分类物体的能力。感知系统依赖深层骨干网络来识别行人、交通灯和其他车辆。ReLU 在这些网络中被广泛用于快速提取特征,有助于实现低 推理延迟。这种速度使车辆的 AI 能够立即做出关键的驾驶决策。

Link to this section医学图像分析#

医疗保健 AI 使用深度学习来辅助放射科医生识别异常。例如,在 医学图像分析 中,模型分析 MRI 扫描以检测肿瘤。ReLU 提供的非线性使这些网络能够以高精度区分健康组织和病变。这种能力对于像 脑肿瘤检测 这样的数据集至关重要,在这些数据集中,早期且准确的诊断可改善患者的预后。

Link to this section使用 PyTorch 实现 ReLU#

以下示例演示了如何使用 torch 库(这是 深度学习 (DL) 的标准工具)应用 ReLU 激活。注意输入张量中的负值是如何被“整流”为零,而正值保持线性的。

import torch
import torch.nn as nn

# Initialize the ReLU function
relu = nn.ReLU()

# Input data with a mix of positive and negative values
data = torch.tensor([-5.0, 0.0, 5.0, -1.2])

# Apply activation: Negatives become 0, Positives stay linear
output = relu(data)

print(f"Input:  {data}")
print(f"Output: {output}")
# Output: tensor([0., 0., 5., 0.])

Link to this section与相关激活函数的比较#

虽然 ReLU 是许多任务的标准,但存在特定的变体和替代方案来解决其局限性或针对特定场景优化性能。

  • ReLU 与 Leaky ReLU 标准 ReLU 可能会遭受“死亡 ReLU”问题的困扰,即神经元卡在输出零的状态,完全停止学习。Leaky ReLU 通过允许负输入有一个小的非零梯度(例如乘以 0.01)来解决这个问题,确保神经元在训练期间保持“活跃”。
  • ReLU 与 Sigmoid Sigmoid 将输出压缩到 0 到 1 之间的范围内。虽然它在预测最终输出层的概率时很有用,但如今它在隐藏层中很少使用,因为它会导致梯度消失,从而减慢 模型训练
  • ReLU vs. SiLU (Sigmoid Linear Unit): SiLU is a smoother, probabilistic approximation of ReLU. It is often used in state-of-the-art architectures like YOLO26 because its smoothness can lead to better accuracy in deep layers, though it is slightly more computationally expensive than ReLU.

Link to this section延伸阅读与资源#

Understanding activation functions is a key step in mastering neural network design. For those looking to dive deeper, the PyTorch documentation on ReLU offers technical specifications for implementation. Additionally, the original AlexNet paper provides historical context on how ReLU revolutionized computer vision. To experiment with training your own models using advanced activations, explore the Ultralytics Platform, which simplifies the workflow for annotating, training, and deploying vision models.

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 的未来!

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