更改

跳到导航 跳到搜索
删除258字节 、 2025年12月26日 (五) 11:54
第32行: 第32行:     
=== 模运算 ===
 
=== 模运算 ===
−
[[文件:二元运算.png|缩略图|发现顿悟现象的部分二元运算|替代=]]
   
最初关于模型顿悟的研究大多在transformer执行模运算任务上进行<ref name="Power_grokking" /><ref name=":0">{{cite journal | author1=Nanda N | author2=Chan L | author3=Lieberum T | author4=Smith J | author5=Steinhardt J | year=2023 | title=Progress measures for grokking via mechanistic interpretability | journal=International Conference on Learning Representations (ICLR) | volume= | issue= | page= | url=https://openreview.net/forum?id=9XFSbDPmdW }}</ref>。为什么使用模运算?是因为小型算法数据集在研究神经网络泛化能力方面具有重要作用。通过在这些数据集上进行实验,可以更显著地展现神经网络的泛化能力,因为在数据有限、规则清晰的情况下,模型需要更好地理解数据中的模式和规律才能实现泛化,有助于快速验证顿悟理论和假设,加深对神经网络泛化能力的理解。
 
最初关于模型顿悟的研究大多在transformer执行模运算任务上进行<ref name="Power_grokking" /><ref name=":0">{{cite journal | author1=Nanda N | author2=Chan L | author3=Lieberum T | author4=Smith J | author5=Steinhardt J | year=2023 | title=Progress measures for grokking via mechanistic interpretability | journal=International Conference on Learning Representations (ICLR) | volume= | issue= | page= | url=https://openreview.net/forum?id=9XFSbDPmdW }}</ref>。为什么使用模运算?是因为小型算法数据集在研究神经网络泛化能力方面具有重要作用。通过在这些数据集上进行实验,可以更显著地展现神经网络的泛化能力,因为在数据有限、规则清晰的情况下,模型需要更好地理解数据中的模式和规律才能实现泛化,有助于快速验证顿悟理论和假设,加深对神经网络泛化能力的理解。
   −
数据集通常由形如<math>a \circ b \bmod p = c</math>的方程组成,其中每个元素"<math>a</math>"、"<math>\circ</math>"、"<math>b</math>"、"<math>mod</math>"、"<math>p</math>"、"<math>=</math>"和"<math>c</math>"均为独立token。由于通常只有<math>a,b,c</math>三个变量,<math><c>=<a><op><b></math>,其中<math><op></math>是运算符。研究者们已经在多种模运算上验证了顿悟的存在,下一节将给出几个典型的例子。
+
数据集通常由形如<math>a \circ b \bmod p = c</math>的方程组成,其中每个元素"<math>a</math>"、"<math>\circ</math>"、"<math>b</math>"、"<math>mod</math>"、"<math>p</math>"、"<math>=</math>"和"<math>c</math>"均为独立token。由于通常只有<math>a,b,c</math>三个变量,<math><c>=<a><op><b></math>,其中<math><op></math>是运算符。研究者们已经在多种模运算上验证了顿悟的存在:
 +
 
 +
[[文件:二元运算.png|发现顿悟现象的部分二元运算|居中|无框|600x600像素|替代=]]
    
=== 其他算法任务 ===
 
=== 其他算法任务 ===
 
除了模运算,研究者还在以下数学任务中观察到了顿悟:
 
除了模运算,研究者还在以下数学任务中观察到了顿悟:
   −
'''群运算(Group Operations):'''这是模运算的推广。Chughtai 等人<ref>{{cite journal | author1=Chughtai B | author2=Chan L | author3=Nanda N | year=2023 | title=A toy model of universality: Reverse engineering how networks learn group operations | journal=International Conference on Machine Learning (ICML), PMLR | volume= | issue= | page= | url= }}</ref>训练MLP模型在阶数为<math>∣G∣=n</math>的有限群上执行群组合运算,发现模型表现出"顿悟"现象。模型输入为有序对 <math>(a,b)</math>(其中<math>a, b \in G</math>),训练目标是预测群元素 <math>c=ab</math>。值得注意的是,Nanda等人<ref name=":0" />提出的任务是该任务的特例——因为模113加法运算等价于113阶循环群的组合运算。
+
'''1. 群运算(Group Operations):'''这是模运算的推广。Chughtai 等人<ref>{{cite journal | author1=Chughtai B | author2=Chan L | author3=Nanda N | year=2023 | title=A toy model of universality: Reverse engineering how networks learn group operations | journal=International Conference on Machine Learning (ICML), PMLR | volume= | issue= | page= | url= }}</ref>训练MLP模型在阶数为<math>∣G∣=n</math>的[[有限群]]上执行群组合运算,发现模型表现出顿悟现象。该任务模型输入为有序对 <math>(a,b)</math>(其中<math>a, b \in G</math>),训练目标是预测群元素 <math>c=ab</math>。值得注意的是,模运算任务是该任务的特例——因为模113加法运算等价于113阶[[循环群]]的组合运算。
   −
'''n位k稀疏奇偶校验(Parity Check):'''这是一个典型[[离散搜索]]问题。Barak等人<ref>{{cite journal | author1=Barak B | author2=Edelman B | author3=Goel S | author4=Kakade S | author5=Malach E | author6=Zhang C | year=2022 | title=Hidden progress in deep learning: SGD learns parities near the computational limit | journal=Advances in Neural Information Processing Systems | volume=35 | issue= | page=21750–21764 | url= }}</ref>通过大量实验证明,使用标准初始化(均匀、高斯、伯努利)和标准训练(SGD)的各种神经网络架构(包括 2 层 MLP、特定设计的单神经元网络、Transformer、PolyNet),能够在训练步数接近计算极限<math> n^{O(k)}</math> 的情况下,成功学习 k-稀疏 n 位奇偶函数,即从<math> n </math>个比特中找出 <math>k </math>个关键比特并计算它们的异或作为标签。
+
'''2. n位k稀疏奇偶校验(Parity Check):'''这是一个典型[[离散搜索]]问题。Barak等人<ref>{{cite journal | author1=Barak B | author2=Edelman B | author3=Goel S | author4=Kakade S | author5=Malach E | author6=Zhang C | year=2022 | title=Hidden progress in deep learning: SGD learns parities near the computational limit | journal=Advances in Neural Information Processing Systems | volume=35 | issue= | page=21750–21764 | url= }}</ref>通过大量实验证明,使用标准初始化(均匀、高斯、伯努利)和标准训练(SGD)的各种神经网络架构(包括 2 层 MLP、特定设计的单神经元网络、Transformer、PolyNet),能够在训练步数接近计算极限<math> n^{O(k)}</math> 的情况下,成功学习 k-稀疏 n 位奇偶函数,即从<math> n </math>个比特中找出 <math>k </math>个关键比特,并计算它们的异或作为标签。
    
2023年,Nanda等人<ref name=":0" />发现有限数据是顿悟现象的必要条件,而无限数据会导致模型直接泛化,并在以下三种运算中证实这一现象:
 
2023年,Nanda等人<ref name=":0" />发现有限数据是顿悟现象的必要条件,而无限数据会导致模型直接泛化,并在以下三种运算中证实这一现象:
   −
'''5位数加法:'''该任务输入两个随机生成的5位数(例如 12345 + 67890),模型需预测其和的每一位数字(如 80235 的 8,0,2,3,5)。在1层完整Transformer上,训练数据限制为700个样本时出现顿悟;使用随机生成的无限数据时未出现顿悟。
+
'''3. 5位数加法:'''该任务输入两个随机生成的5位数(例如 12345 + 67890),模型需预测其和的每一位数字(如 80235 的 8,0,2,3,5)。在1层完整Transformer上,训练数据限制为700个样本时出现顿悟;使用随机生成的无限数据时未出现顿悟。
   −
'''重复子序列预测:'''该任务生成均匀分布的随机符号序列,随机选择子序列进行重复,训练模型预测被重复的符号。(如 7,2,8,3, 1,9,3,8,3, 1,9,9,2,5,其中 1,9 为重复子序列)。在2层仅注意力Transformer上,训练数据限制为512个样本时出现顿悟;使用随机生成的无限数据时未出现顿悟,形成归纳头后直接泛化。
+
'''4. 重复子序列预测:'''该任务生成均匀分布的随机符号序列,随机选择子序列进行重复,训练模型预测被重复的符号。(如 7,2,8,3, 1,9,3,8,3, 1,9,9,2,5,其中 1,9 为重复子序列)。在2层仅注意力Transformer上,训练数据限制为512个样本时出现顿悟;使用随机生成的无限数据时未出现顿悟,形成归纳头后直接泛化。
   −
'''跳跃三元组:'''该任务输入由0到19构成的符号序列(其中恰好一个符号≥10),模型需输出≥10的符号。该任务可通过学习10个跳跃三元组解决。如 [0,3,15,7,...,2],模型需输出该特殊token 15。在1层仅注意力Transformer上,训练数据限制为512个样本时出现顿悟;使用随机生成的无限数据时未出现顿悟,而是直接学习了所有跳跃三元组规则。
+
'''5. 跳跃三元组:'''该任务输入由0到19构成的符号序列(其中恰好一个符号≥10),模型需输出≥10的符号。该任务可通过学习10个跳跃三元组解决。如 [0,3,15,7,...,2],模型需输出该特殊token 15。在1层仅注意力Transformer上,训练数据限制为512个样本时出现顿悟;使用随机生成的无限数据时未出现顿悟,而是直接学习了所有跳跃三元组规则。
    
=== 非算法任务 ===
 
=== 非算法任务 ===
第56行: 第57行:  
顿悟不仅仅存在于完美的数学公式中。
 
顿悟不仅仅存在于完美的数学公式中。
   −
刘子鸣团队在2022年<ref name=":1">{{cite journal | author1=Liu Z | author2=Kitouni O | author3=Nolte N S | author4=Michaud E | author5=Tegmark M | author6=Williams M | year=2022 | title=Towards understanding grokking: An effective theory of representation learning | journal=Advances in Neural Information Processing Systems | volume=35 | issue= | page=34651–34663 | url=https://proceedings.neurips.cc/paper_files/paper/2022/file/dfc310e81992d2e4cedc09ac47eff13e-Paper-Conference.pdf }}</ref>首次证明,顿悟现象是机器学习中更为普遍的现象,不仅会出现在算法数据集上,也会出现在主流基准数据集中,例如MNIST数据集,并通过调整优化超参数实现了对顿悟的控制。进一步地,Ziming Liu等人<ref name=":2">{{cite journal | author1=Liu Z | author2=Michaud E J | author3=Tegmark M | year=2023 | title=OMNIGROK: Grokking Beyond Algorithmic Data | journal=International Conference on Learning Representations (ICLR) | volume= | issue= | page= | url= }}</ref>发现显式增加初始化权重范数会诱发顿悟现象,并在以下三种非算数任务上证明:
+
刘子鸣团队在2022年<ref name=":1">{{cite journal | author1=Liu Z | author2=Kitouni O | author3=Nolte N S | author4=Michaud E | author5=Tegmark M | author6=Williams M | year=2022 | title=Towards understanding grokking: An effective theory of representation learning | journal=Advances in Neural Information Processing Systems | volume=35 | issue= | page=34651–34663 | url=https://proceedings.neurips.cc/paper_files/paper/2022/file/dfc310e81992d2e4cedc09ac47eff13e-Paper-Conference.pdf }}</ref>首次证明,顿悟现象是机器学习中更为普遍的现象,不仅会出现在算法数据集上,也会出现在主流基准数据集中,例如MNIST数据集,并通过调整优化超参数实现了对顿悟的控制。进一步地,他们发现显式增加初始化权重范数会诱发顿悟现象,并在以下三种非算数任务上证明<ref name=":2">{{cite journal | author1=Liu Z | author2=Michaud E J | author3=Tegmark M | year=2023 | title=OMNIGROK: Grokking Beyond Algorithmic Data | journal=International Conference on Learning Representations (ICLR) | volume= | issue= | page= | url= }}</ref>:
   −
'''IMDb文本情感分析任务''':该数据集包含5万条需分类为正面或负面的电影评论。数据预处理阶段提取前1000个高频词,并将每条评论分词为词元索引数组。低频词被忽略,每条评论数组填充至长度500。采用LSTM模型进行分类,该模型包含两层结构,嵌入维度64,隐藏维度128。使用Adam优化器以0.001学习率最小化二元交叉熵损失,保留25%数据集用于测试。
+
'''IMDb文本情感分析任务''':该数据集包含5万条需分类为正面或负面的电影评论。数据预处理阶段提取前1000个高频词,并将每条评论分词为词元索引数组。低频词被忽略,每条评论数组填充至长度500。采用LSTM模型进行分类,该模型包含两层结构,嵌入维度64,隐藏维度128。
   −
'''分子QM9任务''':该数据是小分子及其性质的数据库。采用[[图卷积神经网络]](GCNN)预测各向同性极化率。该GCNN包含2个带ReLU激活的卷积层和1个线性层。使用Adam优化器以0.001学习率最小化均方误差损失。数据集按50/50比例划分为训练集/测试集。
+
'''分子QM9任务''':该数据是小分子及其性质的数据库。采用[[图卷积神经网络]](GCNN)预测各向同性极化率。该GCNN包含2个带ReLU激活的卷积层和1个线性层。数据集按50/50比例划分为训练集/测试集。
   −
'''MNIST任务''':在MNIST数据集上训练宽度200、深度3的ReLU[[多层感知机]](MLP),采用均方误差损失。使用AdamW优化器,学习率0.001,批量大小200。
+
'''MNIST任务''':在MNIST数据集上训练宽度200、深度3的ReLU[[多层感知机]](MLP),采用均方误差损失。
   −
除此之外,Abramov等人<ref>{{cite journal | author1=Abramov R | author2=Steinbauer F | author3=Kasneci G | year=2025 | title=Grokking in the Wild: Data Augmentation for Real-World Multi-Hop Reasoning with Transformers | journal=arXiv | volume= | issue= | page= | url=https://arxiv.org/abs/2504.20752 }}</ref>提出定向数据合成方法,在基于真实维基百科的多跳问答(Multi-hop QA)任务中释放泛化电路的形成潜力。实验表明当比例系数超过特定阈值时,顿悟随即出现。这意味着,不需要复杂的提示工程,只要给模型足够的时间去“顿悟”,它就能自动学会进行多步推理。
+
除此之外,Abramov等人<ref>{{cite journal | author1=Abramov R | author2=Steinbauer F | author3=Kasneci G | year=2025 | title=Grokking in the Wild: Data Augmentation for Real-World Multi-Hop Reasoning with Transformers | journal=arXiv | volume= | issue= | page= | url=https://arxiv.org/abs/2504.20752 }}</ref>提出定向数据合成方法,在基于真实维基百科的多跳问答(Multi-hop QA)任务中释放泛化电路的形成潜力。实验表明当比例系数超过特定阈值时,顿悟随即出现。这意味着,不需要复杂的提示工程,只要给模型足够的时间去顿悟,它就能自动学会进行多步推理。
    
=== 大语言模型 ===
 
=== 大语言模型 ===
81

个编辑

导航菜单