数学基础
- 线性代数
- 概率论与随机过程
- 数值计算
- 蒙特卡洛方法与 MCMC 采样
- 机器学习方法概论
统计学习
深度学习
- 深度学习简介
- 深度前馈网络
- 反向传播算法
- 正则化
- 深度学习中的最优化问题
- 卷积神经网络
- CNN:图像分类
- 循环神经网络 RNN
- Transformer
- 一、Transformer [2017]
- 二、Universal Transformer [2018]
- 三、Transformer-XL [2019]
- 四、GPT1 [2018]
- 五、GPT2 [2019]
- 六、GPT3 [2020]
- 七、OPT [2022]
- 八、BERT [2018]
- 九、XLNet [2019]
- 十、RoBERTa [2019]
- 十一、ERNIE 1.0 [2019]
- 十二、ERNIE 2.0 [2019]
- 十三、ERNIE 3.0 [2021]
- 十四、ERNIE-Huawei [2019]
- 十五、MT-DNN [2019]
- 十六、BART [2019]
- 十七、mBART [2020]
- 十八、SpanBERT [2019]
- 十九、ALBERT [2019]
- 二十、UniLM [2019]
- 二十一、MASS [2019]
- 二十二、MacBERT [2019]
- 二十三、Fine-Tuning Language Models from Human Preferences [2019]
- 二十四 Learning to summarize from human feedback [2020]
- 二十五、InstructGPT [2022]
- 二十六、T5 [2020]
- 二十七、mT5 [2020]
- 二十八、ExT5 [2021]
- 二十九、Muppet [2021]
- 三十、Self-Attention with Relative Position Representations [2018]
- 三十一、USE [2018]
- 三十二、Sentence-BERT [2019]
- 三十三、SimCSE [2021]
- 三十四、BERT-Flow [2020]
- 三十五、BERT-Whitening [2021]
- 三十六、Comparing the Geometry of BERT, ELMo, and GPT-2 Embeddings [2019]
- 三十七、CERT [2020]
- 三十八、DeCLUTR [2020]
- 三十九、CLEAR [2020]
- 四十、ConSERT [2021]
- 四十一、Sentence-T5 [2021]
- 四十二、ULMFiT [2018]
- 四十三、Scaling Laws for Neural Language Models [2020]
- 四十四、Chinchilla [2022]
- 四十七、GLM-130B [2022]
- 四十八、GPT-NeoX-20B [2022]
- 四十九、Bloom [2022]
- 五十、PaLM [2022] (粗读)
- 五十一、PaLM2 [2023](粗读)
- 五十二、Self-Instruct [2022]
- 句子向量
- 词向量
- 传统CTR 预估模型
- CTR 预估模型
- 一、DSSM [2013]
- 二、FNN [2016]
- 三、PNN [2016]
- 四、DeepCrossing [2016]
- 五、Wide 和 Deep [2016]
- 六、DCN [2017]
- 七、DeepFM [2017]
- 八、NFM [2017]
- 九、AFM [2017]
- 十、xDeepFM [2018]
- 十一、ESMM [2018]
- 十二、DIN [2017]
- 十三、DIEN [2019]
- 十四、DSIN [2019]
- 十五、DICM [2017]
- 十六、DeepMCP [2019]
- 十七、MIMN [2019]
- 十八、DMR [2020]
- 十九、MiNet [2020]
- 二十、DSTN [2019]
- 二十一、BST [2019]
- 二十二、SIM [2020]
- 二十三、ESM2 [2019]
- 二十四、MV-DNN [2015]
- 二十五、CAN [2020]
- 二十六、AutoInt [2018]
- 二十七、Fi-GNN [2019]
- 二十八、FwFM [2018]
- 二十九、FM2 [2021]
- 三十、FiBiNET [2019]
- 三十一、AutoFIS [2020]
- 三十三、AFN [2020]
- 三十四、FGCNN [2019]
- 三十五、AutoCross [2019]
- 三十六、InterHAt [2020]
- 三十七、xDeepInt [2023]
- 三十九、AutoDis [2021]
- 四十、MDE [2020]
- 四十一、NIS [2020]
- 四十二、AutoEmb [2020]
- 四十三、AutoDim [2021]
- 四十四、PEP [2021]
- 四十五、DeepLight [2021]
- 图的表达
- 一、DeepWalk [2014]
- 二、LINE [2015]
- 三、GraRep [2015]
- 四、TADW [2015]
- 五、DNGR [2016]
- 六、Node2Vec [2016]
- 七、WALKLETS [2016]
- 八、SDNE [2016]
- 九、CANE [2017]
- 十、EOE [2017]
- 十一、metapath2vec [2017]
- 十二、GraphGAN [2018]
- 十三、struc2vec [2017]
- 十四、GraphWave [2018]
- 十五、NetMF [2017]
- 十六、NetSMF [2019]
- 十七、PTE [2015]
- 十八、HNE [2015]
- 十九、AANE [2017]
- 二十、LANE [2017]
- 二十一、MVE [2017]
- 二十二、PMNE [2017]
- 二十三、ANRL [2018]
- 二十四、DANE [2018]
- 二十五、HERec [2018]
- 二十六、GATNE [2019]
- 二十七、MNE [2018]
- 二十八、MVN2VEC [2018]
- 二十九、SNE [2018]
- 三十、ProNE [2019]
- Graph Embedding 综述
- 图神经网络
- 一、GNN [2009]
- 二、Spectral Networks 和 Deep Locally Connected Networks [2013]
- 三、Fast Localized Spectral Filtering On Graph [2016]
- 四、GCN [2016]
- 五、神经图指纹 [2015]
- 六、GGS-NN [2016]
- 七、PATCHY-SAN [2016]
- 八、GraphSAGE [2017]
- 九、GAT [2017]
- 十、R-GCN [2017]
- 十一、 AGCN [2018]
- 十二、FastGCN [2018]
- 十三、PinSage [2018]
- 十四、GCMC [2017]
- 十五、JK-Net [2018]
- 十六、PPNP [2018]
- 十七、VRGCN [2017]
- 十八、ClusterGCN [2019]
- 十九、LDS-GNN [2019]
- 二十、DIAL-GNN [2019]
- 二十一、HAN [2019]
- 二十二、HetGNN [2019]
- 二十三、HGT [2020]
- 二十四、GPT-GNN [2020]
- 二十五、Geom-GCN [2020]
- 二十六、Graph Network [2018]
- 二十七、GIN [2019]
- 二十八、MPNN [2017]
- 二十九、UniMP [2020]
- 三十、Correct and Smooth [2020]
- 三十一、LGCN [2018]
- 三十二、DGCNN [2018]
- 三十三、AS-GCN
- 三十四、DGI [2018]
- 三十五、DIFFPOLL [2018]
- 三十六、DCNN [2016]
- 三十七、IN [2016]
- 图神经网络 2
- 图神经网络 3
- 推荐算法(传统方法)
- 一、Tapestry [1992]
- 二、GroupLens [1994]
- 三、ItemBased CF [2001]
- 四、Amazon I-2-I CF [2003]
- 五、Slope One Rating-Based CF [2005]
- 六、Bipartite Network Projection [2007]
- 七、Implicit Feedback CF [2008]
- 八、PMF [2008]
- 九、SVD++ [2008]
- 十、MMMF 扩展 [2008]
- 十一、OCCF [2008]
- 十二、BPR [2009]
- 十三、MF for RS [2009]
- 十四、 Netflix BellKor Solution [2009]
- 推荐算法(神经网络方法 1)
- 一、MIND [2019](用于召回)
- 二、DNN For YouTube [2016]
- 三、Recommending What Video to Watch Next [2019]
- 四、ESAM [2020]
- 五、Facebook Embedding Based Retrieval [2020](用于检索)
- 六、Airbnb Search Ranking [2018]
- 七、MOBIUS [2019](用于召回)
- 八、TDM [2018](用于检索)
- 九、DR [2020](用于检索)
- 十、JTM [2019](用于检索)
- 十一、Pinterest Recommender System [2017]
- 十二、DLRM [2019]
- 十三、Applying Deep Learning To Airbnb Search [2018]
- 十四、Improving Deep Learning For Airbnb Search [2020]
- 十五、HOP-Rec [2018]
- 十六、NCF [2017]
- 十七、NGCF [2019]
- 十八、LightGCN [2020]
- 十九、Sampling-Bias-Corrected Neural Modeling [2019](检索)
- 二十、EGES [2018](Matching 阶段)
- 二十一、SDM [2019](Matching 阶段)
- 二十二、COLD [2020 ] (Pre-Ranking 模型)
- 二十三、ComiRec [2020](https://www.wenjiangs.com/doc/0b4e1736-ac78)
- 二十四、EdgeRec [2020]
- 二十五、DPSR [2020](检索)
- 二十六、PDN [2021](mathcing)
- 二十七、时空周期兴趣学习网络ST-PIL [2021]
- 推荐算法之序列推荐
- 一、FPMC [2010]
- 二、GRU4Rec [2015]
- 三、HRM [2015]
- 四、DREAM [2016]
- 五、Improved GRU4Rec [2016]
- 六、NARM [2017]
- 七、HRNN [2017]
- 八、RRN [2017]
- 九、Caser [2018]
- 十、p-RNN [2016]
- 十一、GRU4Rec Top-k Gains [2018]
- 十二、SASRec [2018]
- 十三、RUM [2018]
- 十四、SHAN [2018]
- 十五、Phased LSTM [2016]
- 十六、Time-LSTM [2017]
- 十七、STAMP [2018]
- 十八、Latent Cross [2018]
- 十九、CSRM [2019]
- 二十、SR-GNN [2019]
- 二十一、GC-SAN [2019]
- 二十二、BERT4Rec [2019]
- 二十三、MCPRN [2019]
- 二十四、RepeatNet [2019]
- 二十五、LINet(2019)
- 二十六、NextItNet [2019]
- 二十七、GCE-GNN [2020]
- 二十八、LESSR [2020]
- 二十九、HyperRec [2020]
- 三十、DHCN [2021]
- 三十一、TiSASRec [2020]
- 推荐算法(综述)
- 多任务学习
- 系统架构
- 实践方法论
- 深度强化学习 1
- 自动代码生成
工具
- CRF
- lightgbm
- xgboost
- scikit-learn
- spark
- numpy
- matplotlib
- pandas
- huggingface_transformer
- 一、Tokenizer
- 二、Datasets
- 三、Model
- 四、Trainer
- 五、Evaluator
- 六、Pipeline
- 七、Accelerate
- 八、Autoclass
- 九、应用
- 十、Gradio
Scala
- 环境搭建
- 基础知识
- 函数
- 类
- 样例类和模式匹配
- 测试和注解
- 集合 collection(一)
- 集合collection(二)
- 集成 Java
- 并发
七、共轭梯度法
7.1 最速下降法
最速下降法是梯度下降法的一个特例:在每次梯度下降时,学习率的选择是使得当前函数值下降最快的那个学习率。
$ \epsilon^*=\min_{\epsilon} f(\vec\theta_k-\epsilon \mathbf{\vec g}_k) $梯度下降法仅仅指明:负梯度的方向是使得代价函数值下降的方向。它并没有说明沿着负梯度的方向推进多少步长。
最速下降法通过显式对学习率建模,求解沿着负梯度的方向需要推进多少步长。
在最速下降法中,假设上一次搜索方向是 $ \mathbf{\vec d}_{t-1} $ ,当前搜索方向是 $ \mathbf{\vec d}_{t} $ 。可以证明有:
$ \mathbf{\vec d}_{t} \cdot \mathbf{\vec d}_{t-1}=0 $即:前后两次搜索的方向是正交的。
证明过程:
已知 $ \vec\theta_k $ 的梯度 $ \mathbf{\vec g}_k $ ,根据 $ 0=\frac{\partial f}{\partial \epsilon} $ ,则有:
$ 0=\frac{\partial f}{\partial \epsilon}=-\nabla f(\vec\theta_k-\epsilon\mathbf{\vec g}_k)\cdot \mathbf{\vec g}_k \\ \rightarrow \nabla f(\vec\theta_{k+1})\cdot \mathbf{\vec g}_k=0\\ \rightarrow \mathbf{\vec g}_{k+1}\cdot \mathbf{\vec g}_k=0 $在最速下降法中,前进过程是锯齿形的。在某种意义上,这意味着消除了之前的线性搜索方向上取得的进展:
如果前一次搜索已经找出了某个方向上的最小值,最速下降法不会保持这一方向。
7.2 共轭梯度法
- 共轭:给定一个正定矩阵 $ \mathbf Q $ ,如果非零向量 $ \mathbf{\vec x},\mathbf{\vec y} $ 满足: $ \mathbf{\vec x}^T\mathbf Q \mathbf{\vec y}=0 $ ,则称向量 $ \mathbf{\vec x},\mathbf{\vec y} $ 关于 $ \mathbf Q $ 共轭。
- 如果 $ \mathbf Q=\mathbf I $ 为单位矩阵,则有 $ \mathbf{\vec x}^T\mathbf{\vec y}=0 $ 。即:向量共轭是向量点积为0的推广。
7.2.1 共轭梯度原理
对于线性空间 $ \mathbb R^n $ ,如果能找到 $ n $ 个向量 $ \mathbf{\vec d}_1,\mathbf{\vec d_2},\cdots,\mathbf{\vec d}_n $ ,且它们满足: $ \mathbf{\vec d}_i^T\mathbf Q\mathbf{\vec d}_j=0,i\ne j $ ,则可以证明这一组向量是线性无关的。
此时这一组向量可以作为线性空间的一组基向量,空间中任意向量 $ \mathbf{\vec x} $ 都可以用这组向量表示: $ \mathbf{\vec x}=\sum_{i=1}^na_i\mathbf{\vec d}_i $ 。
对于二次函数极值问题(其中 $ \mathbf Q $ 为正定矩阵): $ \min_{\mathbf{\vec x}\in \mathbb R^n}\frac 12\mathbf{\vec x}^T\mathbf Q\mathbf{\vec x}-\mathbf{\vec b}^T\mathbf{\vec x} $ ,将 $ \mathbf{\vec x}=\sum_{i=1}^na_i\mathbf{\vec d}_i $ 代入上式,令 $ \mathbf{\vec a}=(a_1,a_2,\cdots,a_n)^T $ :
$ \min_{\mathbf{\vec a}\in \mathbb R^n}\frac 12\left(\sum_{i=1}^na_i\mathbf{\vec d}_i\right)^T\mathbf Q\left(\sum_{i=1}^na_i\mathbf{\vec d}_i\right)-\mathbf{\vec b}^T\left(\sum_{i=1}^na_i\mathbf{\vec d}_i\right)\\ =\min_{\mathbf{\vec a}\in \mathbb R^n}\frac 12 \left( \sum_{i=1}^n\sum_{j=1}^na_ia_j\mathbf{\vec d}_i^T\mathbf Q\mathbf{\vec d}_j\right)-\left(\sum_{i=1}^na_i\mathbf{\vec b}^T\mathbf{\vec d}_i\right) $根据 $ \mathbf{\vec d}_i\mathbf Q\mathbf{\vec d}_j=0 $ ,有:
$ \min_{\mathbf{\vec a}\in \mathbb R^n}\frac 12\left(\sum_{i=1}^na_i^2\mathbf{\vec d}_i^T\mathbf Q\mathbf{\vec d}_i\right) -\left(\sum_{i=1}^na_i\mathbf{\vec b}^T\mathbf{\vec d}_i\right) =\min_{\mathbf{\vec a}\in \mathbb R^n}\sum_{i=1}^n\left(\frac 12 a_i^2 \mathbf{\vec d}_i^T\mathbf Q\mathbf{\vec d}_i-a_i\mathbf{\vec b}^T\mathbf{\vec d}_i\right) $现在变量 $ a_1,a_2,\cdots,a_n $ 已经被分开,可以分别优化。
于是求解第 $ i $ 项的最小值:
$ \min_{a_i}\left(\frac 12 a_i^2 \mathbf{\vec d}_i^T\mathbf Q\mathbf{\vec d}_i-a_i\mathbf{\vec b}^T\mathbf{\vec d}_i\right) $直接求导即可得到:
$ a_i^*=\frac{\mathbf{\vec b}^T\mathbf{\vec d_i}}{\mathbf{\vec d}_i^T\mathbf Q\mathbf{\vec d}_i} $最优解为:
$ \mathbf{\vec x}^*=\sum_{i=1}^na_i^*\mathbf{\vec d}_i=\sum_{i=1}^n\frac{\mathbf{\vec b}^T\mathbf{\vec d_i}}{\mathbf{\vec d}_i^T\mathbf Q\mathbf{\vec d}_i} \mathbf{\vec d}_i $上述最优解可以这样理解:
- 目标函数沿着方向 $ \mathbf{\vec d}_i $ (即: $ \mathbf{\vec x}=a_i \mathbf{\vec d}_i $ )求解极小值,得到 $ a_i^* $ 。
- 总和所有的 $ n $ 个方向求得的极小值,得到 $ \mathbf{\vec x}^*=\sum_{i=1}^na_i^*\mathbf{\vec d}_i $ 。
现在的问题是: $ n $ 个关于 $ \mathbf Q $ 共轭的向量组 $ \{\mathbf{\vec d}_1,\mathbf{\vec d}_2,\cdots,\mathbf{\vec d}_n\} $ 未知。
7.2.2 共轭梯度搜索
共轭梯度法通过迭代来搜索关于 $ \mathbf Q $ 共轭的向量组 $ \{\mathbf{\vec d}_1,\mathbf{\vec d}_2,\cdots,\mathbf{\vec d}_n\} $ 。
令梯度 $ \mathbf{\vec g}=\nabla_{\mathbf{\vec x}}f(\mathbf{\vec x}) $ ,由于 $ f(\mathbf{\vec x})=\frac 12\mathbf{\vec x}^T\mathbf Q\mathbf{\vec x}-\mathbf{\vec b}^T\mathbf{\vec x} $ ,则有:
$ \mathbf{\vec g}=\frac {\mathbf Q^T+\mathbf Q}{2}\mathbf{\vec x}-\mathbf{\vec b}\\ $假设 $ \mathbf Q $ 为对称的正定矩阵,则有: $ \mathbf Q=\mathbf Q^T $ 。因此有: $ \mathbf{\vec g}=\mathbf Q\mathbf{\vec x}-\mathbf{\vec b} $ 。
任取一个初始点 $ \mathbf{\vec x}_1 $ 。
如果 $ \nabla_{\mathbf{\vec x}}f(\mathbf{\vec x}_1)=\vec 0 $ ,则停止计算。因为此时 $ \mathbf{\vec x}_1 $ 就是极值点。
否则,令 $ \mathbf{\vec d}_1=\mathbf {\vec g}_1=-\nabla_{\mathbf{\vec x}}f(\mathbf{\vec x}_1) $ ,即: $ \mathbf{\vec d}_1 $ 沿着点 $ \mathbf{\vec x}_1 $ 的梯度方向。
沿着 $ \mathbf{\vec d}_1 $ 方向搜索: $ \mathbf{\vec x}_2=\mathbf{\vec x}_1+\lambda \mathbf{\vec d}_1 $ ,使得沿着这个方向,目标函数下降最多。
$ \min_{\lambda}\left(\frac 12(\mathbf{\vec x}_1+\lambda \mathbf{\vec d}_1)^T\mathbf Q(\mathbf{\vec x}_1+\lambda \mathbf{\vec d}_1)-\mathbf{\vec b}^T(\mathbf{\vec x}_1+\lambda \mathbf{\vec d}_1)\right)-\left(\frac 12\mathbf{\vec x}_1^T\mathbf Q\mathbf{\vec x}_1-\mathbf{\vec b}^T\mathbf{\vec x}_1\right) $解得:
$ \lambda=-\frac{\mathbf{\vec x}_1^T\mathbf Q\mathbf{\vec d}_1-\mathbf{\vec b}^T\mathbf{\vec d}_1}{\mathbf{\vec d}_1^T\mathbf Q\mathbf{\vec d}_1} $由于 $ \mathbf{\vec g}^T=\mathbf{\vec x}^T\mathbf Q-\mathbf{\vec b}^T $ ,因此有:
$ \lambda=- \frac{ \mathbf{\vec g}_1 ^T\mathbf{\vec d}_1}{\mathbf{\vec d}_1^T\mathbf Q\mathbf{\vec d}_1} $
选择第二个迭代点:
$ \mathbf{\vec x}_2=\mathbf{\vec x}_1+\lambda \mathbf{\vec d}_1=\mathbf{\vec x}_1- \frac{ \mathbf{\vec g}_1 ^T\mathbf{\vec d}_1}{\mathbf{\vec d}_1^T\mathbf Q\mathbf{\vec d}_1}\mathbf{\vec d}_1 $注意:这里有 $ \mathbf{\vec g}_1=\mathbf{\vec d}_1 $ ,但是不保证后面的 $ \mathbf{\vec g}_i=\mathbf{\vec d}_i $ 。
构建下一个搜索方向,使得 $ \mathbf{\vec d}_2 $ 与 $ \mathbf{\vec d}_1 $ 关于 $ \mathbf Q $ 共轭。
令: $ \mathbf {\vec d}_2=-\mathbf{\vec g}_2+\beta_1\mathbf{\vec d}_1 $ ,代入 $ \mathbf{\vec d}_1^T\mathbf Q\mathbf{\vec d}_2=0 $ ,有:
$ \beta_1=\frac{\mathbf{\vec d}_1^T\mathbf Q\mathbf{\vec g}_2}{\mathbf{\vec d_1^T\mathbf Q\mathbf{\vec d}_1}},\quad \mathbf {\vec d}_2=-\mathbf{\vec g}_2+\frac{\mathbf{\vec d}_1^T\mathbf Q\mathbf{\vec g}_2}{\mathbf{\vec d_1^T\mathbf Q\mathbf{\vec d}_1}}\mathbf{\vec d}_1 $同理:得到 $ \mathbf{\vec x}_{k+1} $ 时,计算梯度 $ \mathbf{\vec g}_{k+1} $ ,同时用 $ -\mathbf{\vec g}_{k+1} $ 和 $ \mathbf{\vec d}_k $ 来构建下一个搜索方向:
$ \mathbf {\vec d}_{k+1}=-\mathbf{\vec g}_{k+1}+\beta_k\mathbf{\vec d}_k,\quad \mathbf{\vec d}_k^T\mathbf Q\mathbf{\vec d}_{k+1}=0 $得到:
$ \beta_k=\frac{\mathbf{\vec d}_k^T\mathbf Q\mathbf{\vec g}_{k+1}}{\mathbf{\vec d_k^T\mathbf Q\mathbf{\vec d}_k}},\quad \mathbf {\vec d}_{k+1}=-\mathbf{\vec g}_{k+1}+\frac{\mathbf{\vec d}_k^T\mathbf Q\mathbf{\vec g}_{k+1}}{\mathbf{\vec d_k^T\mathbf Q\mathbf{\vec d}_k}}\mathbf{\vec d}_k $.
7.2.3 共轭梯度算法
共轭梯度法的搜索方向 $ \mathbf{\vec d}_t $ 会保持前一次线性搜索方向上 $ \mathbf{\vec d}_{t-1} $ 取得的进展:
$ \mathbf{\vec d}_t=-\nabla_{\vec\theta} J(\vec\theta)+\beta_t\mathbf{\vec d}_{t-1} $其中 $ \beta_t $ 的大小控制了:要保留多大比例的上一次搜索方向。
在实际应用中,目标函数往往是高于二次的函数。此时为非线性共轭梯度, $ \mathbf Q $ 就是海森矩阵 $ \mathbf H $ 。
直接计算 $ \beta_t $ 也需要计算海森矩阵的逆矩阵,比较复杂。有两个计算 $ \beta_t $ 的流行方法(不需要计算海森矩阵的逆矩阵):
这里不再保证 $ \mathbf{\vec d}_t $ 和 $ \mathbf{\vec d}_{t-1} $ 关于 $ \mathbf Q $ 是共轭的。
$ \beta_t=\frac{\mathbf{\vec g}_t^{T}\mathbf{\vec g}_t}{\mathbf{\vec g}_{t-1}^T\mathbf{\vec g}_{t-1}} $Fletcher-Reeves
:
$ \beta_t=\frac{(\mathbf{\vec g}_{t}-\mathbf{\vec g}_{t-1})^{T}\mathbf{\vec g}_{t}}{\mathbf{\vec g}_{t-1}^T\mathbf{\vec g}_{t-1}} $Polak_Ribiere
:
目标函数在高于二次的函数时,往往可能存在局部极值。此时共轭梯度法不能在 $ n $ 维空间内依靠 $ n $ 步搜索到达极值点。
此时需要重启共轭梯度法,继续迭代,以完成搜索极值点的工作。
共轭梯度法:
输入:
- 初始参数 $ \vec\theta_0 $
- 包含 $ m $ 个样本的训练集
算法步骤:
初始化:
$ \vec{\mathbf d}_0=\mathbf{\vec 0}\\ \mathbf{\vec g}_0=\mathbf{\vec 0}\\ t=1 $迭代,直到到达停止条件。迭代步骤为:
计算梯度:
$ \mathbf{\vec g}_t\leftarrow \frac 1m\nabla_{\vec\theta}\sum_{i=1}^{m}L(f(\mathbf{\vec x}_i;\vec\theta),y_i) $计算 $ \beta_t $
$ \beta_t=\frac{(\mathbf{\vec g}_t-\mathbf{\vec g}_{t-1})^{T}\mathbf{\vec g}_t}{\mathbf{\vec g}_{t-1}^{T}\mathbf{\vec g}_{t-1}} $如果是非线性共轭梯度:则可以以一定的频率重置 $ \beta_t $ 为零(如每当 $ t $ 为 5 的倍数时)。
计算搜索方向: $ \vec {\mathbf d}_t=-\mathbf{\vec g}_t+\beta_t\vec{\mathbf d}_{t-1} $
执行线性搜索: $ \epsilon^{*}=\arg\min_{\epsilon}\frac 1m\sum_{i=1}^{m}L(f(\mathbf{\vec x}_i;\vec\theta_t+\epsilon\vec{\mathbf d}_t),y_i) $
在前面推导过程中,并没有出现这一步。这是因为:在非线性共轭梯度下,以及修改的 $ \beta_t $ 的情况下,不再满足 $ \mathbf{\vec d}_t $ 和 $ \mathbf{\vec d}_{t-1} $ 关于 $ \mathbf Q $ 是共轭的。
对于真正的二次函数:存在 $ \epsilon^{*} $ 的解析解,无需显式搜索。
- 应用更新: $ \vec\theta_{t+1}=\vec\theta_t+ \epsilon^{*}\vec{\mathbf d}_t $
- $ t\leftarrow t+1 $
7.2.4 共轭梯度算法性质
神经网络的目标函数比二次函数复杂得多,但是共轭梯度法在这种情况下仍然适用。此时非线性共轭梯度算法会偶尔采取一些重置操作(重置 $ \beta_t $ 为零)。
尽管共轭梯度算法是批方法(需要所有的样本来计算梯度),目前也开发出了
mini-batch
版本。理论上,对于二次函数的目标函数,需要迭代 $ n $ 步, $ n $ 为参数的个数。每一步中都需要计算梯度。
在神经网络中,这样有两个问题:
- 当参数的数量 $ n $ 巨大时,迭代的量非常大,而且难以并行(后面的迭代依赖前面的迭代),计算量非常庞大。
- 当样本的数量巨大时,每次迭代中的梯度计算的计算量非常庞大。
如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论