数学基础
- 线性代数
- 概率论与随机过程
- 数值计算
- 蒙特卡洛方法与 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
- 并发
四、TADW [2015]
网络在我们的日常生活中无处不在,例如
Facebook
用户之间的友谊或学术论文之间的引用。近年来,研究人员广泛研究了网络中许多重要的机器学习应用,例如顶点分类vertex classification
、标签推荐tag recommendation
、异常检测anomaly detection
、链接预测link prediction
。数据稀疏data sparsity
是这些任务面临的常见问题。为了解决数据稀疏性问题,network representation learning: NRL
在统一的低维空间中编码和表达每个顶点。NRL
有助于我们更好地理解顶点之间的语义相关性semantic relatedness
,并进一步缓解稀疏性带来的不便。NRL
中的大多数工作从网络结构中学习representation
。例如,social dimensions
是计算网络的拉普拉斯Laplacian
矩阵或modularity
矩阵的特征向量eigenvectors
。最近,NLP
中的word representation
模型SkipGram
被引入,用于从社交网络中的随机游走序列中学习顶点的representation
,称作DeepWalk
。social dimensions
和DeepWalk
方法都将网络结构作为输入来学习network representation
,但是没有考虑任何其它信息。在现实世界中,网络中的顶点通常具有丰富的信息,例如文本内容和其它元数据
meta data
。例如,维基百科文章相互连接并形成网络,每篇文章作为一个顶点都包含大量的文本信息,这些文本信息可能对NRL
也很重要。因此,论文《Network Representation Learning with Rich Text Information》
提出了同时从网络结构和文本信息中学习network representation
的想法,即text-associated DeepWalk: TADW
。- 一种直接的方法是:分别独立地从文本特征和网络特征中学习
representation
,然后将两个独立的representation
拼接起来。然而,该方法没有考虑网络结构和文本信息之间复杂的交互interaction
,因此通常会导致效果不佳。 - 在现有的
NRL
框架中加入文本信息也不容易。例如,DeepWalk
在网络中的随机游走过程中,无法轻松地处理附加信息。
幸运的是,给定网络 $ G=(V,E) $ ,论文证明
DeepWalk
实际上是分解矩阵 $ \mathbf M\in \mathbb R^{|V|\times |V|} $ ,其中每一项 $ M_{i,j} $ 是顶点 $ v_i $ 以固定steps
随机游走到顶点 $ v_j $ 的平均概率的对数。下图(a)
展示了MF
风格的DeepWalk
:将矩阵 $ \mathbf M $ 分解为两个低维矩阵 $ \mathbf W\in \mathbb R^{k\times |V| }, \mathbf H\in \mathbb R^{ k\times |V| } $ ,其中 $ k\ll |V| $ 。DeepWalk
将矩阵 $ \mathbf W $ 作为顶点representation
。我们将在后面进一步详细介绍。DeepWalk
的矩阵分解视角启发了作者将文本信息引入到NRL
的MF
中。如上图(b)
展示了论文方法的主要思想:将矩阵 $ \mathbf M $ 分解为三个矩阵的乘积: $ \mathbf W\in \mathbb R^{k\times |V|}, \mathbf H\in \mathbb R^{k\times f_t}, \mathbf T\in \mathbb R^{f_t\times |V|} $ ,其中 $ \mathbf T $ 为文本特征矩阵。然后论文拼接 $ \mathbf W $ 和 $ \mathbf H\mathbf T $ 作为每个顶点的 $ 2k $ 维的representation
。论文在三个数据集上针对多个
baseline
测试了TADW
算法。当训练集比率training ratio
在10% ~ 50%
的范围内时,TADW
的representation
的分类准确率比其它baseline
高达2% ~ 10%
。当训练集比率小于10%
时,论文还使用半监督分类器Transductive SVM: TSVM
测试这些方法。TADW
在1%
的训练集比率下,相比其它baseline
方法提升了5% ~ 20%
,尤其是在网络信息包含噪音的情况下。论文主要贡献:
- 论文证明了
DeepWalk
算法实际上分解了一个矩阵 $ \mathbf M $ ,并找出了 $ \mathbf M $ 的封闭形式closed form
。 - 论文将文本特征引入
NRL
,并相比其它baseline
获得了5% ~ 20%
的优势,尤其是在训练集比率较小的情况下。
- 一种直接的方法是:分别独立地从文本特征和网络特征中学习
相关工作:
representation learning
广泛应用于计算机视觉、自然语言处理、knowledge representation learning
。一些研究集中在NRL
,但它们都无法泛化来处理顶点的其它特征。据作者所知,很少有工作致力于考虑NRL
中的文本信息。有一些主题模型
topic model
,如NetPLSA
,在主题建模时同时考虑了网络和文本信息,然后可以用主题分布topic distribution
来表示每个顶点。在本文中,作者以NetPLSA
作为baseline
方法。
4.1 模型
4.1.1 DeepWalk 作为矩阵分解
公式化
NRL
:给定网络 $ G=(V,E) $ ,我们希望为每个顶点 $ v $ 构建一个低维representation
向量 $ \mathbf{\vec w}_v\in \mathbb R^k $ ,其中 $ k \ll |V| $ 。作为一种稠密的real-valued representation
, $ \mathbf{\vec w}_v $ 可以缓解邻接矩阵等network representation
的稀疏性。我们可以将 $ \mathbf{\vec w}_v $ 视为顶点 $ v $ 的特征,并将这些特征应用到许多机器学习任务中,如顶点分类。这些特征可以方便地提供给许多分类器,例如逻辑回归、SVM
。注意,representation
不是特定于任务task-specific
的,可以在不同任务之间共享。我们首先介绍
DeepWalk
,然后给出DeepWalk
和矩阵分解之间的等价证明。
a. DeepWalk
DeepWalk
首次将应用广泛的分布式word representation
方法SkipGram
引入到社交网络的研究中,从而根据网络结构来学习vertex representation
。
$ \frac{1}{|\mathcal S|}\sum_{i=1}^{|\mathcal S|}\sum_{-t\le j\le t,j\ne 0} \log p(v_{i+j}\mid v_i) $DeepWalk
首先生成短的随机游走short random walk
(短的随机游走也被用作相似性度量)。给定一条由随机游走生成的顶点序列 $ \mathcal S=\{v_1,v_2,\cdots,v_{|\mathcal S|}\} $ ,我们将顶点 $ v \in \{v_{i-t},\cdots,v_{i+t}\},v\ne v_i $ 视为中心顶点 $ v_i $ 的上下文,其中 $ t $ 为窗口大小window size
。遵循SkipGram
的思想,DeepWalk
旨在最大化随机游走顶点序列 $ \mathcal S $ 中所有vertex-context pair
的平均对数概率average log probability
:其中 $ p(v_j\mid v_i) $ 以
$ p(v_j\mid v_i) = \frac{\exp\left(\mathbf{\vec c}_{v_j}^\top \mathbf{\vec w}_{v_i}\right)}{\sum_{v\in V}\exp\left(\mathbf{\vec c}_v^\top \mathbf{\vec w}_{v_i}\right)} $softmax
函数的方式来定义:其中 :
- $ \mathbf{\vec w}_{v_i} $ 为顶点 $ v_i $ 作为
center vertex
时的represetation
向量。 - $ \mathbf{\vec c}_{v_j} $ 为顶点 $ v_j $ 作为
context vertex
时的representation
向量。
即,每个顶点 $ v $ 有两个
representation
向量:当 $ v $ 作为center vertex
时的 $ \mathbf{\vec w}_v $ 、当 $ v $ 作为context vertex
时的 $ \mathbf{\vec c}_v $ 。之后,
DeepWalk
使用SkipGram
和Hierarchical Softmax
从随机游走生成的序列中学习顶点的representation
。注意,Hierarchical Softmax
是用于加速的softmax
的变体。- $ \mathbf{\vec w}_{v_i} $ 为顶点 $ v_i $ 作为
b. 等价性证明
假设一个
vertex-context
集合 $ \mathcal D $ 是从随机游走序列中生成,其中每个元素为一个vertex-context pair
$ (v,c) $ 。定义vertex
集合为 $ V $ ,定义context
集合为 $ V_C $ ,并且大多数情况下 $ V=V_C $ 。DeepWalk
将一个vertex
$ v $ 嵌入到一个 $ k $ 维向量 $ \mathbf{\vec w}_v\in \mathbb R^k $ 。另外,一个context
$ v\in V_C $ 被嵌入到一个 $ k $ 维向量 $ \mathbf{\vec c}_v\in \mathbb R^k $ 。令 $ \mathbf W\in \mathbb R^{k\times |V| } $ 为所有vertex embedding
组成的矩阵,其中第 $ i $ 行表示顶点 $ v_i $ 的vertex representation
。令 $ \mathbf H\in \mathbb R^{k\times|V| } $ 为 所有context embedding
组成的矩阵,其中第 $ i $ 行表示顶点 $ v_i $ 的context representation
。我们的目标是找出矩阵 $ \mathbf M $ 的封闭形式closed form
,其中 $ \mathbf M = \mathbf W^\top\mathbf H $ 。现在我们考虑一个
vertex-context pair
$ (v,c) $ 。定义 $ N(v,c) $ 为 $ (v,c) $ 出现在 $ \mathcal D $ 中的次数, $ N(v)=\sum_{c\in V_C}N(v,c) $ 为vertex
$ v $ 出现在 $ \mathcal D $ 中的次数, $ N(c) = \sum_{v\in V}N(v,c) $ 为context
$ c $ 出现在 $ \mathcal D $ 中的次数。
$ M_{i,j} = \log \frac{N(v_i,c_j)\times |\mathcal D|}{N(v_i)\times N(c_j)} - \log \lambda $《Neural word embedding as implicit matrix factorization》
已经证明:当维度 $ k $ 足够大时,带负采样的SkipGram
(SkipGram with Negative Sampling: SGNS
) 相当于隐式地分解word-context
矩阵 $ \mathbf M $ 。其中 $ \mathbf M $ 的元素为:其中 $ \lambda $ 为每个
word-context pair
的负样本数量。$ M_{i,j} $ 可以解释为
word-context pair
$ (v_i,c_j) $ 的Pointwise Mutual Information: PMI
的 $ \log \lambda $ 偏移。类似地,我们可以证明带
$ M_{i,j} = \log \frac{N(v_i,c_j)}{N(v_i)} $softmax
的SkipGram
相当于分解矩阵 $ \mathbf M $ ,其中:我们现在讨论
DeepWalk
中的 $ M_{i,j} $ 代表什么。显然,vertex-context pair
的采样方法会影响矩阵 $ \mathbf M $ 。假设网络是连通connected
和无向undirected
的,窗口大小为 $ t $ 。我们将基于DeepWalk
算法的理想ideal
的采样方法来讨论 $ N(v)/|\mathcal D| $ , $ N(c)/|\mathcal D| $ ,以及 $ N(v,c)/N(v) $ :- 首先我们生成一条足够长的随机游走序列 $ RW $ 。令 $ RW_i $ 表示随机游走序列 $ RW $ 上位置 $ i $ 的顶点。
- 然后我们将
vertex-context pair
$ (RW_i,RW_j) $ 添加到 $ \mathcal D $ 中,当且仅当 $ 0\lt |i-j|\le t $ 。
$ N(v_i)/|\mathcal D| $ 就是顶点 $ v_i $ 在随机游走中出现的频率,也就是 $ v_i $ 的
PageRank
值。 $ 2tN(v_i,v_j)/N(v_i) $ 为在 $ v_i $ 的左侧或右侧 $ t $ 个邻居中观察到 $ v_j $ 的期望次数。现在我们尝试基于这种理解来计算 $ N(v_i,v_j)/N(v_i) $ 。将
$ A_{i,j} = \begin{cases} 1/d_i,& (v_i,v_j)\in E\\ 0,&\text{else} \end{cases} $PageRank
中的转移矩阵表示为 $ \mathbf A $ 。令 $ d_i $ 为顶点 $ i $ 的degree
,则有:我们使用 $ \mathbf{\vec e}_i $ 来表示一个 $ V $ 维的行向量,其中除了第 $ i $ 项为
$ \frac{N(v_i,v_j)}{N(v_i)} = \frac{\left[\mathbf{\vec e}_i(\mathbf A + \mathbf A^2+\cdots+\mathbf A^t)\right]_j}{t} $1
之外其它项均为零。假设我们从顶点 $ i $ 开始随机游走并使用 $ \mathbf{\vec e}_i $ 表示初始状态。那么 $ \mathbf{\vec e}_i \mathbf A $ 是所有顶点的分布,它的第 $ j $ 项是顶点 $ i $ 随机游走到顶点 $ j $ 的概率。因此, $ \mathbf{\vec e}_i \mathbf A^t $ 的第 $ j $ 项是顶点 $ i $ 以恰好 $ t $ 步随机游走到顶点 $ j $ 的概率,其中 $ \mathbf A^t $ 是矩阵 $ \mathbf A $ 的 $ t $ 次幂。因此, $ \left[\mathbf{\vec e}_i(\mathbf A + \mathbf A^2+\cdots+\mathbf A^t)\right] $ 的第 $ j $ 项是 $ v_j $ 出现在 $ v_i $ 的右侧 $ t $ 个邻居中的期望次数。因此我们有:上述证明也适用于有向图。因此,我们可以看到 $ M_{i,j} = \log N(v_i,v_j)/N(v_i) $ 为顶点 $ v_i $ 在 $ t $ 步中随机游走到顶点 $ j $ 的平均概率的对数。
通过证明
DeepWalk
等价于矩阵分解,我们提出融合丰富的文本信息到基于DeepWalk
派生的矩阵分解中。
4.1.2 TADW
- 这里我们首先简要介绍低秩矩阵分解,然后我们提出从网络和文本信息中学习
representation
的方法。
a. 低秩矩阵分解
矩阵是表示关系型数据
relational data
的常用方法。矩阵分析的一个有趣主题是通过矩阵的一部分项来找出矩阵的内在结构。一个假设是矩阵 $ \mathbf M\in \mathbb R^{b\times d} $ 可以用低秩 $ k $ 的另一个矩阵来近似,其中 $ k \ll \{b,d\} $ 。然后我们可以在这个假设下用这样一个低秩近似来补全complete
矩阵 $ \mathbf M $ 的缺失值。然而,求解秩约束优化rank constraint optimization
始终是NP
难的。因此,研究人员求助于寻找矩阵 $ \mathbf W\in \mathbb R^{k\times b} $ 和 $ \mathbf H\in \mathbb R^{k\times d} $ ,从而最小化损失函数 $ \mathcal L(\mathbf M,\mathbf W^\top \mathbf H) $ ,并且带有迹范数的约束trace norm constraint
(这个约束可以通过在损失函数中添加一个惩罚项从而进一步地移除)。在本文中,我们使用平方损失函数。低秩分解:形式上,令矩阵 $ \mathbf M $ 的观察集为 $ \mathbf\Omega $ 。我们希望找到矩阵 $ \mathbf W\in \mathbb R^{k\times b} $ 和 $ \mathbf H\in \mathbb R^{k\times d} $ ,从而最小化损失函数:
$ \min_{\mathbf W,\mathbf H} \sum_{(i,j)\in \mathbf\Omega} \left(M_{i,j} - \left(\mathbf W^\top\mathbf H\right)_{i,j}\right)^2 + \frac{\lambda}{2}\left(\|\mathbf W\|_F^2 + \|\mathbf H\|_F^2\right) $其中 $ \|\cdot\|_F $ 表示矩阵的
Frobenius
范数, $ \lambda $ 为正则化系数。归纳矩阵补全:低秩矩阵分解仅基于 $ \mathbf M $ 的低秩假设来补全矩阵 $ \mathbf M $ 。如果矩阵 $ \mathbf M $ 中的
item
具有附加特征additional feature
,那么我们可以应用归纳矩阵补全inductive matrix completion
来利用它们。归纳矩阵补全通过融合两个特征矩阵feature matrix
到目标函数中从而利用更多行单元row unit
和列单元column unit
的信息。假设我们有特征矩阵 $ \mathbf X\in \mathbb R^{ f_x\times b} $ 和 $ \mathbf Y\in \mathbb R^{f_y\times d} $ ,其中 $ f_x,f_y $ 分别为两个矩阵列向量的维度。我们的目标是求解矩阵 $ \mathbf W\in \mathbb R^{k\times f_x} $ 和 $ \mathbf H\in \mathbb R^{k\times f_y } $ ,从而最小化:
$ \min_{\mathbf W,\mathbf H} \sum_{(i,j)\in \mathbf\Omega} \left(M_{i,j} - \left(\mathbf X^\top \mathbf W^\top \mathbf H \mathbf Y \right)_{i,j}\right)^2 + \frac{\lambda}{2}\left(\|\mathbf W\|_F^2 + \|\mathbf H\|_F^2\right) $注意,归纳矩阵补全最初是为了补全具有基因特征和疾病特征的 “基因-疾病”矩阵,其目标与我们的工作有很大不同。受归纳矩阵补全思想的启发,我们将文本信息引入
NRL
。
b. TADW
给定一个网络 $ G=(V,E) $ 以及它对应的文本特征矩阵 $ \mathbf T\in \mathbb R^{ f_t\times |V|} $ ,我们提出了文本相关的
DeepWalk
(text-associated DeepWalk: TADW
)从网络结构 $ G $ 和文本特征 $ \mathbf T $ 中学习每个顶点 $ v\in V $ 的representaion
。回想一下,
DeepWalk
分解矩阵 $ \mathbf M $ ,其中 $ M_{i,j} = \log \left[\mathbf{\vec e}_i(\mathbf A + \mathbf A^2+\cdots+\mathbf A^t)\right]_j/t $ 。 当 $ t $ 较大时,准确地计算 $ \mathbf M $ 的计算复杂度为 $ O(|V|^3) $ 。实际上,DeepWalk
使用基于随机游走的采样方法来避免显式地、精确地计算矩阵 $ \mathbf M $ 。当DeepWalk
采样更多的随机游走序列时,性能会更好,但是计算效率会更低。在
TADW
中,我们找到了速度和准确性之间的tradeoff
:分解矩阵 $ \mathbf M = (\mathbf A + \mathbf A^2)/2 $ 。在这里,为了计算效率,我们分解矩阵 $ \mathbf M $ 而不是 $ \log \mathbf M $ 。原因是 $ \log \mathbf M $ 比 $ \mathbf M $ 具有更多的非零项,并且具有平方损失的矩阵分解的复杂度与矩阵 $ \mathbf M $ 中非零元素的数量成正比。由于大多数现实世界的网络都是稀疏的,即 $ O(|E|) = O(|V|) $ ,因此计算矩阵 $ \mathbf M $ 需要 $ O(|V|^2) $ 的时间。如果网络是稠密的,我们甚至可以直接分解矩阵 $ \mathbf A $ 。即使是 $ O(|V|^2) $ 的计算复杂度,对于百万甚至亿级顶点的网络而言,这个计算复杂度仍然无法接受。
我们的任务是求解矩阵 $ \mathbf W\in \mathbb R^{k\times |V| } $ 和 $ \mathbf H\in \mathbb R^{k\times f_t } $ ,从而最小化:
$ \min_{\mathbf W,\mathbf H} \sum_{(i,j)\in \mathbf\Omega} \left(M_{i,j} - \left( \mathbf W^\top \mathbf H \mathbf T\right)_{i,j}\right)^2 + \frac{\lambda}{2}\left(\|\mathbf W\|_F^2 + \|\mathbf H\|_F^2\right) $为了优化 $ \mathbf W $ 和 $ \mathbf H $ ,我们交替最小化 $ \mathbf W $ 和 $ \mathbf H $ ,因为它是 $ \mathbf W $ 或 $ \mathbf H $ 的凸函数。虽然
TADW
可能收敛到局部最小值而不是全局最小值,但我们的方法在实践中效果很好,如我们的实验所示。也可以直接应用随机梯度下降法来优化。
不同于低秩矩阵分解
low-rank matrix factorization
和归纳矩阵补全inductive matrix completion
(它们聚焦于补全矩阵 $ \mathbf M $ ),TADW
的目标是结合文本特征从而获得更好的network representation
。此外,归纳矩阵补全直接从原始数据中获得矩阵 $ \mathbf M $ ,而我们从MF
风格的DeepWalk
的推导中人为地构建矩阵 $ \mathbf M $ 。由于从
TADW
获得的 $ \mathbf W $ 和 $ \mathbf T\mathbf H $ 都可以视为顶点的低维representation
,因此我们可以通过拼接它们为network representation
构建一个统一的、 $ 2k $ 维的矩阵。在实验中,我们将证明统一的representation
显著优于将network representation
和文本特征(即矩阵 $ \mathbf T $ ) 的简单组合。复杂度分析:在
TADW
中,计算 $ \mathbf M $ 的过程需要 $ O(|V|^2) $ 的时间。我们使用《Large-scale multi-label learning with missing labels》
引入的快速过程来求解TADW
的优化问题。最小化 $ \mathbf W $ 和 $ \mathbf H $ 的每次迭代的复杂度为 $ O(\text{nnz}(\mathbf M)k + |V|f_t k + |V|k^2) $ ,其中
nnz(.)
为矩阵非零元素的个数。作为比较,传统矩阵分解的复杂度(即低秩矩阵分解的优化问题) 为 $ O(\text{nnz}(\mathbf M)k + |V|k^2) $ 。在我们的实验中,优化在10
次迭代中收敛。未来工作:针对大规模网络数据场景的
TADW
在线学习和分布式学习。另外,还可以研究矩阵分解的其它技术,例如matrix co-factorization
,从而包含来自其它来源的丰富信息。
4.2 实验
我们使用顶点分类问题来评估
NRL
的质量。正式地,我们得到顶点的低维representation
作为顶点特征,然后我们的任务是基于顶点特征用标记顶点集合 $ \mathbb L $ 来预测未标记顶点集合 $ \mathbb U $ 的label
。机器学习中的许多分类器可以处理这个任务。我们分别选择
SVM
和transductive SVM
从而进行监督的、半监督的学习和测试。注意,由于representation learning
过程忽略了训练集中的顶点label
,因此representation learning
是无监督的。我们使用三个公开可用的数据集,并使用
representation learning
的五种baseline
方法来评估TADW
。我们从文档之间的链接或引用,以及这些文档的term frequency-inverse document frequency: TF-IDF
矩阵(即矩阵 $ \mathbf T $ )中学习representation
。数据集:
Cora
数据集:包含来自七个类别的2708
篇机器学习论文。论文之间的链接代表引用关系,共有5429
个链接。每篇文档映射为一个1433
维的binary
向量,每一位为0/1
表示对应的单词是否存在。Citeseer
数据集:包含来自六个类别的3312
篇公开发表出版物。文档之间的链接代表引用关系,共4732
个链接。每篇文档映射为一个3703
维的binary
向量,每一位为0/1
表示对应的单词是否存在。Wiki
数据集:包含来自十九个类别的2405
篇维基百科文章。文章之间的链接代表超链接,共17981
个链接。我们移除了没有任何超链接的文档。每篇文章都用内容单词的TFIDF
向量来表示,向量维度为4973
维。
Cora
和Citeseer
数据集从标题、摘要中生成短文本作为文档,并经过停用词处理、低频词处理(过滤文档词频低于10
个的单词),并将每个单词转化为one-hot
向量。Cora、Citeseer
数据集平均每篇文档包含18~32
个单词,数据集被认为是有向图。Wiki
数据集是长文本,平均每篇文档包含640
个单词,数据集被认为是无向图。baseline
方法及其配置:TADW
:通过SVD
分解TFIDF
矩阵到200
维的文本特征矩阵 $ \mathbf T\in \mathbb R^{200\times |V|} $ ,这是为了降低矩阵 $ \mathbf H $ 的规模。我们也将文本特征矩阵 $ \mathbf T $ 视为一个content-only
的baseline
。对于
Cora,Citeseer
数据集 $ k = 80,\lambda = 0.2 $ ;对于Wiki
数据集 $ k = 100,200,\lambda = 0.2 $ 。注意:最终每个顶点的
representation
向量的维度为 $ 2k $ 。DeepWalk
:DeepWalk
是一种network-only representation learning
方法。我们选择参数为:窗口尺寸 $ t=10 $ ,每个顶点的游走序列数量 $ \gamma = 80 $ 。对于Cora,Citeseer
数据集维度 $ k=100 $ ,对于Wiki
数据集维度 $ k=200 $ ,这些是在50 ~200
之间选择的性能最佳的维度。我们还通过求解“低秩分解”方程并将 $ \mathbf W $ 和 $ \mathbf H $ 拼接起来作为顶点
representation
从而评估MF-style DeepWalk
的性能。结果与DeepWalk
相比差不多,因此我们仅报告了原始DeepWalk
的性能。pLSA
:我们使用PLSA
通过将每个顶点视为一个文档来从TF-IDF
矩阵训练主题模型。因此,PLSA
是content-only baseline
方法。PLSA
通过EM
算法估计文档的主题分布和主题的word
分布。我们使用文档的主题分布作为顶点representation
。Text Features
:使用文本特征矩阵 $ \mathbf T\in \mathbb R^{200\times |V| } $ 作为每篇文档的200
维representation
,这也是一种content-only baseline
方法。Naive Combination
:直接拼接DeepWalk
的embedding
向量和文本特征向量。对于Cora,Citeseer
数据集这将得到一个300
维向量;对于Wiki
数据集这将得到一个400
维向量。NetPLSA
:NetPLSA
将文档之间的链接视为网络正则化来学习文档的主题分布,存在链接的文档应该共享相似的主题分布。我们使用网络增强的文档主题分布作为network representation
。NetPLSA
可以视为一种兼顾网络和文本信息的NRL
方法。我们将Cora,Citeseer
数据集主题数量设置为160
,将Wiki
数据集主题数量设置为200
。
评估方式:对于监督分类器,我们使用由
Liblinear
实现的linear SVM
。对于半监督分类器,我们使用SVM-Light
实现的transductive SVM
。我们对TVSM
使用线性核。我们为每个类别训练一个one-vs-rest
分类器,并选择linear SVM
和transductive SVM
中得分最高的类别。我们将顶点的
representation
作为特征来训练分类器,并以不同的训练比率training ratio
来评估分类准确性。训练比率从linear SVM
的10% ~ 50%
,以及从TSVM
的1% ~ 10%
不等。对于每个训练比率,我们随机选择文档作为训练集,剩余文档作为测试集。我们重复试验10
次并报告平均准确率。实验结果:下表给出了
Cora
数据集、Citeseer
数据集、Wiki
数据集的分类准确率。这里-
表示TSVM
不能在12
小时内收敛,因为representation
的质量较低(对于TADW
,TADM
总是可以在5
分钟内收敛)。我们没有在Wiki
数据集上展示半监督学习的结果,因为监督SVM
已经在这个数据集上以较小的训练比率获得了有竞争力的、甚至更好的性能。因此,我们仅报告Wiki
数据集的有监督SVM
的结果。Wiki
数据集的类别要比其它两个数据集多得多,这需要更多的数据来进行足够的训练,因此我们将最小训练比率设为3%
。从这些表中我们可以看到:
TADW
在所有三个数据集上始终优于所有其它baseline
。此外,TADW
可以在Cora
数据集和Citeseer
数据集上减少50%
的训练数据从而击败其它baseline
。这些实验证明TADW
是有效且鲁棒的。TADW
对于半监督学习有更显著的提升。TADW
的表现优于最佳的baseline
(即naive combination
),在Cora
数据集上提升4%
、在Citeseer
数据集上提升10% ~ 20%
。这是因为在Citeseer
上的network representation
质量很差,而TADW
从噪音的数据中学习比naive combination
更鲁棒。TADW
在训练比率较小时有更好的表现。大多数baseline
的准确率随着训练比率的降低而迅速下降,因为它们的vertex representation
对于training
和testing
而言非常noisy
和inconsistent
。相反,由于TADW
从网络和文本信息中联合学习representation
,因此representation
具有更少的噪音并且更加一致。
这些观察结果证明了
TADW
生成的高质量representation
。此外,TADW
不是task-specific
的,representation
可以方便地用于不同的任务,例如链接预测、相似性计算、顶点分类等等。TADW
的分类准确性也与最近的几种collective
分类算法相媲美,尽管我们在学习representation
时没有对任务执行特别的优化。参数敏感性:
TADW
有两个超参数:维度 $ k $ 和正则化系数 $ \lambda $ 。我们将训练比率固定为10%
,并使用不同的 $ k $ 和 $ \lambda $ 测试分类准确率。对于Cora
数据集和Citeseer
数据集,我们让 $ k $ 在40 ~ 120
之间变化,而 $ \lambda $ 在0.1 ~ 1
之间变化。对于Wiki
数据集,我们让 $ k $ 在100 ~ 200
之间变化,而 $ \lambda $ 在0.1 ~ 1
之间变化。下图显示了不同 $ k $ 和 $ \lambda $ 下分类准确率的变化。可以看到:
- 对于
Cora, Citeseer, Wiki
上的固定 $ k $ ,不同 $ \lambda $ 对应的准确率分别在1.5%, 1%, 2%
的波动范围内。 - 当
Cora
和Citeseer
上的 $ k\ge 80 $ 、Wiki
上的 $ k\ge 140 $ 时,准确率具有竞争性competitive
。因此,当 $ k $ 和 $ \lambda $ 在合理范围内变化时,TADW
可以保持稳定。
- 对于
案例研究:为了更好地理解
NRL
文本信息的有效性,我们在Cora
数据集中提供了一个案例。document
标题为Irrelevant Features and the Subset Selection Problem
(不相关特征和子集选择问题)。我们将这篇论文简称为IFSSP
。IFSSP
的类别标签为Theory
。如下表所示,使用DeepWalk
和TADW
生成的representation
,我们找到了5
篇最相似的论文,并基于余弦相似度来排序。我们发现,所有这些论文都被
IFSSP
引用。然而,DeepWalk
发现的5
篇论文中有3
篇具有不同的类别标签,而TADW
发现的前4
篇论文具有相同的类别标签Theory
。这表明:与单纯的基于网络的DeepWalk
相比,TADW
可以借助文本信息学到更好的network representation
。DeepWalk
发现的第5
篇论文也展示了仅考虑网络信息的另一个限制。《MLC Tutorial A Machine Learning library of C classes》
(简称MLC
)是一个描述通用toolbox
的论文,可以被不同主题的许多论文引用。一旦其中一些论文也引用了IFSSP
,DeepWalk
将倾向于给IFSSP
一个与MLC
类似的representation
,即使它们具有完全不同的主题。
如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论