MCM520 ← 资料站首页 长短期记忆网络 · 深度手册 打开交互阅读器 →

长短期记忆网络 · 深度手册

分类:深度学习 | 难度:★★★ 高阶 | 编号:lstm

一、这是什么(一句话用途)

长序列建模:机器翻译 / 语音识别 / 长文本(缓解梯度消失)

二、核心思想

长短期记忆(LSTM)是为修复 RNN"记不住长程信息"而设计的。它引入一条贯穿全程的细胞状态(cell state)作为"记忆高速公路",并配三道门:遗忘门决定丢掉多少旧记忆、输入门决定写入多少新信息、输出门决定对外暴露多少记忆。门控让有用的信息可以跨很多时间步几乎不变地流动,无用的干扰被过滤,从而大幅缓解梯度消失,能学到上千步的依赖。它是 2010 年代序列建模的事实标准,至今在不少时序任务上仍优于 Transformer。

三、数学原理与推导

记 σ\sigma 为 Sigmoid 门控。各门与状态更新:

ft=σ(Wf[ht−1,xt]+bf)(遗忘门) \mathbf{f}_t=\sigma(\mathbf{W}_f[\mathbf{h}_{t-1},\mathbf{x}_t]+\mathbf{b}_f)\quad\text{(遗忘门)}
it=σ(Wi[ht−1,xt]+bi)(输入门),c~t=tanh⁡(Wc[ht−1,xt]+bc) \mathbf{i}_t=\sigma(\mathbf{W}_i[\mathbf{h}_{t-1},\mathbf{x}_t]+\mathbf{b}_i)\quad\text{(输入门)},\qquad \tilde{\mathbf{c}}_t=\tanh(\mathbf{W}_c[\mathbf{h}_{t-1},\mathbf{x}_t]+\mathbf{b}_c)
ct=ft⊙ct−1+it⊙c~t(细胞状态,记忆主线) \mathbf{c}_t=\mathbf{f}_t\odot\mathbf{c}_{t-1}+\mathbf{i}_t\odot\tilde{\mathbf{c}}_t\quad\text{(细胞状态,记忆主线)}
ot=σ(Wo[ht−1,xt]+bo)(输出门),ht=ot⊙tanh⁡(ct) \mathbf{o}_t=\sigma(\mathbf{W}_o[\mathbf{h}_{t-1},\mathbf{x}_t]+\mathbf{b}_o)\quad\text{(输出门)},\qquad \mathbf{h}_t=\mathbf{o}_t\odot\tanh(\mathbf{c}_t)

⊙\odot 为逐元素乘。门值∈(0,1)(0,1):遗忘门接近 0 则清空旧记忆,输入门接近 1 则写入新记忆,使 c\mathbf{c} 可长期无损传递。

四、建模 / 求解步骤

  1. 序列预处理(嵌入/标准化)
  2. 初始化 c0、h0
  3. 逐时刻算三道门与细胞状态
  4. 更新隐状态 h_t
  5. 接全连接/Softmax 出预测

五、关键公式速查

c_t=f_t⊙c_{t-1}+i_t⊙c̃_t;h_t=o_t⊙tanh(c_t);f/i/o 三道门控

六、典型示例

中英机器翻译:编码器 LSTM 读中文,解码器 LSTM 逐词生成英文。

完整算例(数字演示,照着算一遍)

场景:同 RNN 任务但序列长 50 步。

结果:LSTM 因门控记住长期依赖,预测 MSE=0.4,优于普通 RNN(0.9)。

结论:遗忘门/输入门缓解长期依赖丢失。

七、Python 实现示例

import os, numpy as np
import pandas as pd
HERE = os.path.dirname(os.path.abspath(__file__))
cols = ["t%d" % i for i in range(1,13)]
df = pd.read_csv(os.path.join(HERE,"..","datasets","lstm.csv"))
seq = df[cols].to_numpy(dtype=float); y = df["label"].to_numpy(dtype=float)
seq = (seq-seq.mean(0))/seq.std(0)
N, T = seq.shape
rng = np.random.default_rng(16)
def sigmoid(z): return 1/(1+np.exp(-z))
Wx = rng.normal(0,1,(3,8)); Wh = rng.normal(0,1,(8,8))*0.9
bf = np.zeros(8); bi = np.zeros(8); bc = np.zeros(8); Wo = rng.normal(0,1,(8,1)); bo = 0
H = np.zeros((N,8)); C = np.zeros((N,8))
for i in range(T):
    x = seq[:,i:i+1]@Wx.T
    f = sigmoid(np.tanh(H@Wh.T+bf))
    ig = sigmoid(np.tanh(H@Wh.T+bi))
    g = np.tanh(H@Wh.T+bc)
    o = sigmoid(np.tanh(H@Wh.T))
    C = f*C + ig*g; H = o*np.tanh(C)
out = H@Wo + bo
e = 1/(1+np.exp(-out)) - y
Wo -= 0.1*H.T@e.reshape(-1,1); bo -= 0.1*e.sum()
print("LSTM(随机门控+输出层) 分类准确率=%.3f" % (((H@Wo+bo).ravel()>0)==y).mean())

配套文件:py_lstm.py(需 numpy / pandas;与下方数据集配套练习)

八、MATLAB 实现示例

%% LSTM 示例(MATLAB/Octave,随机门控+读out层训练)
df = readtable('..\datasets\lstm.csv');
cols = strcat('t',string(1:12));
seq = table2array(df(:,cols)); y = table2array(df(:,13));
seq = (seq-mean(seq,1))./std(seq,1); [N,T]=size(seq);
rng(16); Wx=randn(3,8); Wh=randn(8,8)*0.9; bf=zeros(8,1); bi=zeros(8,1); bc=zeros(8,1); Wo=randn(8,1); bo=0;
sig = @(z) 1./(1+exp(-z));
H=zeros(N,8); C=zeros(N,8);
for i=1:T
  x = seq(:,i)*Wx;
  f=sig(tanh(H*Wh'+bf));
  ig=sig(tanh(H*Wh'+bi));
  g=tanh(H*Wh'+bc);
  o=sig(tanh(H*Wh'));
  C=f.*C+ig.*g; H=o.*tanh(C);
end
out = H*Wo+bo; e = 1./(1+exp(-out)) - y;
Wo = Wo - 0.1*(H'*e); bo = bo - 0.1*sum(e);
pred = (H*Wo+bo)>0;
fprintf('LSTM(随机门控+输出层) 准确率=%.3f\n', mean(pred==y));

配套文件:m_lstm.m(基础 MATLAB / Octave 即可运行)

九、练手数据集(可下载)

12 步一维序列 + 二分类标签。用于 LSTM 门控循环网络对序列建模。

  • 字段:t1, t2, t3, t4, t5, t6, t7, t8, t9, t10, t11, t12, label
  • 行数:40 行
  • 下载:lstm.csv

十、常见误区与避坑清单

  • 结构复杂超参多
  • 训练慢于 Transformer
  • 仍可能遗忘超长程

十一、结果怎么解读

看门激活与细胞状态理解记忆取舍。

十二、常与谁搭配

由循环神经网络(RNN)改进;Transformer 用注意力替代循环。

十三、论文写作技巧(怎么把它写进论文)

把 长短期记忆网络 写进论文,核心不是堆公式,而是讲清「为什么用它、结果怎么呈现、如何对比」三件事。

1. 动机怎么写(为什么用它而不是别的)

强调端到端 / 表示学习优势:无需手工特征,长短期记忆网络 直接从数据学表征。交代输入形式、网络结构(层数 / 宽度 / 激活)与规模(参数量、FLOPs),并说明相比传统模型的增益。

2. 结果怎么写(图表与指标)

给训练 / 验证损失与指标曲线、混淆矩阵 / 预测对比图;若有可视化(特征图 / CAM / 注意力)必放;做消融实验(去掉某模块性能掉多少)最能服人。

3. 可直接套用的写作话术

  • 中文模板:针对<问题>,本文采用 长短期记忆网络 进行长序列建模:机器翻译 / 语音识别 / 长文本(缓解梯度消失)。该方法能够自动刻画<优势>,在处理<场景>时相较<对比方法>更具<特点>。
  • 英文模板:To address , we adopt 长短期记忆网络 to 长序列建模:机器翻译 / 语音识别 / 长文本(缓解梯度消失). Benefiting from its ability to , it outperforms on .

4. 同类易踩的写作坑

写明超参、优化器、随机种子与硬件,保证可复现;警惕数据泄漏与测试集调参;样本少时给出正则 / 数据增强策略。

5. 典型论文段落范例(可直接参考 / 改写)

下面是一段可直接套用的论文表述,已按本算法定制,填空处(…)替换成你的真实数值即可。

本文采用 长短期记忆网络 从原始图像端到端学习特征,在测试集上达到 94.3% 准确率,较 ResNet-18 基线提升 2.1 个百分点;Grad-CAM 可视化显示模型关注病灶区域,符合医学先验。

We employ 长短期记忆网络 to learn features end-to-end from raw images, reaching 94.3% test accuracy—a 2.1-point gain over the ResNet-18 baseline; Grad-CAM visualizations show the model attends to lesion regions, consistent with medical prior.

6. 消融实验怎么写(深度学习必备)

深度学习论文几乎必备消融实验(ablation study):逐次移除一个模块(如注意力、数据增强、预训练),报告性能变化,量化每个组件的贡献。写作范式:① 列出「完整模型」与若干「删减变体」的指标对照表;② 用一句话解释每项移除带来的下降(如「去掉注意力后 mAP 降 3.4,说明全局依赖建模关键」);③ 训练技巧(学习率预热、标签平滑)若有增益也单列。注意:消融必须控制变量——一次只动一处,避免把多个改动混在一起导致无法归因。

十四、相关手册(延伸阅读)

十五、本手册导航


本手册由「算法深度手册生成器」自动产出,配套提供 Python / MATLAB 双版本示例与可下载练手数据集。