news 2026/7/10 10:20:12

Transformer 大语言模型(LLM)基石 - Transformer架构详解 - 残差连接(Residual Connection)详解以及算法实现

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
Transformer 大语言模型(LLM)基石 - Transformer架构详解 - 残差连接(Residual Connection)详解以及算法实现

锋哥原创的Transformer 大语言模型(LLM)基石视频教程:

https://www.bilibili.com/video/BV1X92pBqEhV

课程介绍

本课程主要讲解Transformer简介,Transformer架构介绍,Transformer架构详解,包括输入层,位置编码,多头注意力机制,前馈神经网络,编码器层,解码器层,输出层,以及Transformer Pytorch2内置实现,Transformer基于PyTorch2手写实现等知识。

Transformer 大语言模型(LLM)基石 - Transformer架构详解 - 残差连接(Residual Connection)详解以及算法实现

在深层网络中,随着网络深度的增加,梯度在反向传播时容易消失或爆炸,导致网络难以训练。为了解决这个问题,Transformer采用了残差连接,即将每一层的输入与该层的输出进行相加,而不是直接使用该层的输出。

形式:

这种连接方式保证了在网络初期,若子层变换的结果接近于零,则输出与输入相同,从而缓解了深度神经网络中梯度消失的问题。

优点:

  1. 恒等映射:使得网络能容易地学习到恒等映射,即如果某一层的变换没有效果,模型可以跳过该层。

  2. 梯度流畅:在反向传播时,梯度可以通过残差连接直接传递,避免了梯度消失的问题。

代码实现:

# 实现残差连接 class ResidualConnection(nn.Module): def __init__(self, d_model, dropout=0.1): # d_model 词嵌入维度 512 dropout 随机丢失率 super().__init__() self.norm = LayerNorm(features=d_model) self.dropout = nn.Dropout(dropout) def forward(self, x, sublayer): """ 前向传播 参数: x: 输入张量 [batch_size, seq_len, d_model] sublayer: 残差连接的子层 多头自注意力机制,或者来自前馈神经网络 返回: 残差连接后的张量 """ return x + self.dropout(self.norm(sublayer(x))) if __name__ == '__main__': vocab_size = 2000 # 词表大小 embedding_dim = 512 # 词嵌入维度的大小 embeddings = Embeddings(vocab_size, embedding_dim) embed_result = embeddings( torch.tensor([[1999, 2, 99, 4, 5], [66, 2, 3, 22, 5], [66, 2, 3, 4, 5]])) print("embed_result.shape:", embed_result.shape) print("embed_result", embed_result) positional_encoding = PositionalEncoding(embedding_dim) result = positional_encoding(embed_result) print("result:", result) print("result.shape:", result.shape) # 测试自注意力机制 # query = key = value = result # mask = create_sequence_mask(5) # dropout = nn.Dropout(0.1) # attention_output, attention_weights = self_attention(query, key, value, mask, dropout) # print("attention_output.shape:", attention_output.shape) # [3, 5, 512] # print("attention_weights.shape:", attention_weights.shape) # [3, 5, 5] mha = MultiHeadAttention(d_model=512, num_heads=8) # print(mha) mask = create_sequence_mask(5) # 定义匿名函数 sublayer = lambda x: mha(x, x, x, mask) # 测试残差连接 rc = ResidualConnection(d_model=512) result_rc = rc(result, sublayer) print("result_rc.shape:", result_rc.shape)

运行输出:

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/7/10 8:35:10

蛋白质模型选择困境终结指南:从入门到精通的5大实战策略

你是否曾因蛋白质序列分析任务而陷入模型选择的困境?看着从8M到15B的庞大参数规模,既担心小模型精度不足,又害怕大模型资源消耗过高?本文将通过问题导向的分析方法,为你提供一套完整的蛋白质语言模型选型解决方案。 【…

作者头像 李华
网站建设 2026/7/10 10:22:15

CIDR合并工具:高效管理IP地址段的智能解决方案

CIDR合并工具:高效管理IP地址段的智能解决方案 【免费下载链接】cidr-merger A simple command line tool to merge ip/ip cidr/ip range, supports IPv4/IPv6 项目地址: https://gitcode.com/gh_mirrors/ci/cidr-merger 还在为处理大量分散的IP地址段而烦恼…

作者头像 李华
网站建设 2026/7/7 13:22:21

Node.js打包终极指南:快速解决pkg工具90%常见问题

还在为Node.js应用的分发和部署而烦恼吗?😩 每次打包都遇到各种诡异错误,让人抓狂?别担心,今天我将带你全面掌握pkg工具的使用技巧,让你轻松实现"一次打包,处处运行"的梦想&#xff0…

作者头像 李华
网站建设 2026/7/9 1:23:15

如何快速搭建Android媒体画廊:Gallery完整使用指南

如何快速搭建Android媒体画廊:Gallery完整使用指南 【免费下载链接】Gallery Light-weight Media Gallery app for Android made with Jetpack Compose 项目地址: https://gitcode.com/gh_mirrors/galler/Gallery Gallery是一款基于Jetpack Compose构建的轻量…

作者头像 李华
网站建设 2026/7/8 13:18:12

Kornia几何计算机视觉库:可微分视觉处理的完整指南

Kornia几何计算机视觉库:可微分视觉处理的完整指南 【免费下载链接】kornia Geometric Computer Vision Library for AI 项目地址: https://gitcode.com/gh_mirrors/ko/kornia 1. 项目概述与核心优势 Kornia是一个基于PyTorch构建的可微分计算机视觉库&…

作者头像 李华
网站建设 2026/7/10 7:51:06

代码检索效率革命:OASIS-1.3B如何用5M数据超越OpenAI同类模型

代码检索效率革命:OASIS-1.3B如何用5M数据超越OpenAI同类模型 【免费下载链接】OASIS-code-1.3B 项目地址: https://ai.gitcode.com/hf_mirrors/Kwaipilot/OASIS-code-1.3B 导语 仅用500万训练数据就超越OpenAI同类模型,快手Kwaipilot团队开源的…

作者头像 李华