bert 大白话
keras 版本的bert源码
tensorflow版本的bert源码
大家记住768这个数字
我只问一个问题,那就是bert 词向量化输出的是什么?
在这个问题之前,大家可以想想,word2vec 输出的是什么?
大家可以看看这篇文章大白话讲解word2vec到底在做些什么
我们要获取的dense vector其实就是Hidden Layer的输出单元。有的地方定为Input Layer和Hidden Layer之间的权重
我看了bert的tensorflow 源码,大家可以看看bert tensorflow 源码的36行和877行,都出现了 hidden_size=768
def __init__(self,vocab_size,#大家看看下面这个768hidden_size=768,num_hidden_layers=12,num_attention_heads=12,intermediate_size=3072,hidden_act="gelu",hidden_dropout_prob=0.1,attention_probs_dropout_prob=0.1,max_position_embeddings=512,type_vocab_size=16,initializer_range=0.02): # Down-project back to `hidden_size` then add the residual.with tf.variable_scope("output"):layer_output = tf.layers.dense(intermediate_output,hidden_size,kernel_initializer=create_initializer(initializer_range))layer_output = dropout(layer_output, hidden_dropout_prob)layer_output = layer_norm(layer_output + attention_output)prev_output = layer_outputall_layer_outputs.append(layer_output)我又看了bert keras 版本,在源码的47行和121行都出现了 embed_dim=768
def get_model(token_num,pos_num=512,seq_len=512,embed_dim=768,transformer_num=12,head_num=12,feed_forward_dim=3072,dropout_rate=0.1,attention_activation=None,feed_forward_activation='gelu',training=True,trainable=None,output_layer_num=1): mlm_dense_layer = keras.layers.Dense(units=embed_dim,activation=feed_forward_activation,name='MLM-Dense',)(transformed)mlm_norm_layer = LayerNormalization(name='MLM-Norm')(mlm_dense_layer)mlm_pred_layer = EmbeddingSimilarity(name='MLM-Sim')([mlm_norm_layer, embed_weights])masked_layer = Masked(name='MLM')([mlm_pred_layer, inputs[-1]])extract_layer = Extract(index=0, name='Extract')(transformed)nsp_dense_layer = keras.layers.Dense(units=embed_dim,activation='tanh',name='NSP-Dense',)(extract_layer)nsp_pred_layer = keras.layers.Dense(units=2,activation='softmax',name='NSP',)(nsp_dense_layer)结论:
word2vec本质上就是一个三层的神经网络,word2vec输出的是隐藏层到输出层权重矩阵的某一行,
bert是一个8层的神经网络,分别是
1.Dense=>
2.LayerNormalization=>
3.EmbeddingSimilarity=>
5.Masked=>
6Extract=>
7Dense=>(这一层的神经元个数是768)
8Dense=>
bert的深度更加深,所以训练的时间要长,并且bert 词向量的长度是固定的,要想改变,需要改变网络参数,重新训练
总结
- 上一篇: Adaboost、GBDT与XGBoos
- 下一篇: inceptionv4 keras 实