RNN与LSTM详解:AI是如何“记住”信息的?

RNN(循环神经网络)的基本原理

RNN是一种处理序列数据的神经网络,其核心在于通过隐藏状态(hidden state)传递历史信息。每个时间步的输入不仅包括当前数据,还包含上一时间步的隐藏状态,形成循环连接。数学表达为:

h_t = \\sigma(W_{xh}x_t + W_{hh}h_{t-1} + b_h)

其中,h_t是当前隐藏状态,x_t是输入,W为权重矩阵,b为偏置,\\sigma为激活函数(如tanh)。

RNN的局限性

传统RNN存在梯度消失或爆炸问题,难以捕获长距离依赖关系。例如,在文本生成任务中,早期的单词信息可能无法有效传递到后续时间步。


LSTM(长短期记忆网络)的改进

LSTM通过引入门控机制(输入门、遗忘门、输出门)和细胞状态(cell state)解决RNN的缺陷。其核心结构如下:

遗忘门 :决定哪些信息从细胞状态中丢弃

f_t = \\sigma(W_f \\cdot \[h_{t-1}, x_t\] + b_f)

输入门 :更新细胞状态

i_t = \\sigma(W_i \\cdot \[h_{t-1}, x_t\] + b_i) \\ \\tilde{C}*t = \\tanh(W_C \\cdot \[h*{t-1}, x_t\] + b_C)

细胞状态更新

C_t = f_t \\odot C_{t-1} + i_t \\odot \\tilde{C}_t

输出门 :控制当前隐藏状态输出

o_t = \\sigma(W_o \\cdot \[h_{t-1}, x_t\] + b_o) \\ h_t = o_t \\odot \\tanh(C_t)


实际应用差异

RNN适用场景 :短序列任务(如字符级文本生成),计算资源有限时。
LSTM适用场景:长序列任务(如机器翻译、语音识别),需捕获长期依赖关系。

代码示例(PyTorch实现LSTM单元)

python 复制代码
import torch.nn as nn
lstm = nn.LSTM(input_size=100, hidden_size=128, num_layers=2)
input_seq = torch.randn(10, 3, 100)  # (seq_len, batch, input_size)
output, (h_n, c_n) = lstm(input_seq)

关键结论

  • RNN通过循环连接传递信息,但受限于梯度问题。
  • LSTM的门控机制和细胞状态设计显式控制信息流,更适合长期记忆。
  • 现代变体(如GRU)在LSTM基础上进一步简化结构,平衡性能与效率。
相关推荐
ai生成式引擎优化技术15 分钟前
全球唯一四元结构底层架构问世:TSPR-WEB-LLM-HIC v2.0 终结大模型投毒与幻觉的终极技术范式
人工智能
听你说3224 分钟前
伊萨推出 ROBBI 360 协作机器人焊接工作站 简化自动化焊接部署流程
人工智能·机器人·自动化
weixin_4080996725 分钟前
【实战对比】在线 OCR 识别 vs OCR API 接口:从个人工具到系统集成该怎么选?
图像处理·人工智能·后端·ocr·api·图片文字识别·文字识别ocr
罗西的思考39 分钟前
【OpenClaw】通过Nanobot源码学习架构---(2)外层控制逻辑
人工智能·机器学习
火山引擎开发者社区1 小时前
明天武汉!用好“龙虾”的关键要素全在这儿
人工智能
梦想很大很大1 小时前
从 0 到 1 实现 AI Agent(02):设计可扩展的 Tool 调用系统
人工智能·llm·agent
木斯佳1 小时前
HarmonyOS 6实战:AI时代的“信任危机“,如何处理应用的请求拦截与安全防护
人工智能·安全·harmonyos
jkyy20141 小时前
食物识别与卡路里估算技术:以视觉技术重构膳食健康管理新范式
人工智能·语言模型·自动化·健康医疗
冬奇Lab1 小时前
一天一个开源项目(第61篇):knowledge_graph - 把任意文本转成知识图谱
人工智能·llm
wdf80881 小时前
算力随行:UltraLAB便携工作站如何将多卡深度学习带入户外与现场
人工智能·深度学习·大模型推理·无人机影像