跟李沐学AI:长短期记忆网络LSTM

输入们、遗忘门和输出门

LSTM引入输入门、忘记门和输出门

输入门计算公式为:。

遗忘门计算公式为:。

输出门计算公式为:。

它们由三个具有sigmoid激活函数的全连接层处理, 以计算输入门、遗忘门和输出门的值。 因此,这三个门的值都在(0,1)的范围内。

候选记忆元

类似RNN中的,计算公式为:。

记忆元

LSTM中,输入门和遗忘门类似GRU中控制输入或遗忘的机制。输入们用于控制采用多少来自的新数据,遗忘门用于控制保留多少过去的记忆元的内容。随后按元素乘法,得出。

如果遗忘门始终为1且输入门始终为0, 则过去的记忆元Ct−1 将随时间被保存并传递到当前时间步。 引入这种设计是为了缓解梯度消失问题, 并更好地捕获序列中的长距离依赖关系。

隐状态

LSTM中,输出门用于计算隐状态:。只要输出门接近1,我们就能够有效地将所有记忆信息传递给预测部分, 而对于输出门接近0,我们只保留记忆元内的所有信息,而不需要更新隐状态。

相关推荐
七七安7 分钟前
标准卷积和转置卷积
人工智能
nhdh13 分钟前
Higress:AI时代云原生网关新选择
人工智能·云原生
一木 之林16 分钟前
DeepSeek Agent 开发
java·前端·人工智能
王中阳Go16 分钟前
简历写「QPS 提升 3 倍」,面试官问「怎么压测的」,我卡在并发数怎么定
人工智能·后端·面试
方方洛18 分钟前
ai-agent教程-03-大模型接口与工具调用
人工智能·llm·agent
miofly19 分钟前
语言判别增强多语言语音模型的语言学习能力
人工智能
EatFan20 分钟前
当 CubeMX 遇上 AI Agent:用 MCP 让 AI 直接生成 STM32 HAL 工程
人工智能·stm32·嵌入式硬件·cubemx·stm32cubemx·hal·mcp
这张生成的图像能检测吗21 分钟前
(论文速读)Object-Driven Multi-Layer Scene Decomposition:从单张图像恢复遮挡后的多层场景
人工智能·算法·计算机视觉·rgb-d·场景分解
YOLO数据集集合23 分钟前
河道环境智能检测系统(YOLO + DeepSeek) | 河道巡检 YOLO DeepSeek 大语言模型 智慧水利 9144期
人工智能·yolo·目标检测·语言模型·河道污染·城市治理
猛犸象限24 分钟前
[Grok Bot 仓颉实践] 让 AI 助手替你干活之前,先想清楚谁来拍板——我用 Grok Bot 做游戏的做法
人工智能·grok