L2-Regularization 实现的话,需要把所有的参数放在一个集合内,最后计算loss时,再减去加权值。
相比自己乱搞,代码一团糟,Tensorflow 提供了更优美的实现方法。
一、tf.GraphKeys : 多个包含Variables(Tensor)集合
(1)GLOBAL_VARIABLES:使用tf.get_variable()时,默认会将vairable放入这个集合。
我们熟悉的tf.global_variables_initializer()就是初始化这个集合内的Variables。
import tensorflow as tf
sess=tf.Session()
a=tf.get_variable("a",[3,3,32,64],initializer=tf.random_normal_initializer())
b=tf.get_variable("b",[64],initializer=tf.random_normal_initializer())
#collections=None等价于 collection=[tf.GraphKeys.GLOBAL_VARIABLES]
print("I am a:", a)
print("I am b:", b)
print("I am gv:", tf.GraphKeys.GLOBAL_VARIABLES)
gv= tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES)
#tf.get_collection(collection_name)返回某个collection的列表
print("I am gv:", gv)
for var in gv:
print("Iam var:",var)
print(var is a)
print(var.get_shape())
print("----------------")
输出如下图所示:
Tips: tf.GraphKeys.GLOBAL_VARIABLES == "variable"。即其保存的是一个字符串。所以,我们也可以自己定义集合,然后用 tf.get_collection(“string_name”)获取集合变量,详细看下面内容。
(2)自定义集合
想个集合的名字,然后在tf.get_variable时,把集合名字传给 collection 就好了。
import tensorflow as tf
sess=tf.Session()
a=tf.get_variable("a",shape=[10],collections=["mycollection"]) #不把GLOBAL_VARIABLES加进去,那么就不在那个集合里了。
b=tf.get_variable("b",shape=[10],collections=["mycollection"]) #不把GLOBAL_VARIABLES加进去,那么就不在那个集合里了。
keys=tf.get_collection("mycollection")
print("keys:", keys)
for key in keys:
print(key.name)##也就是变量的名字
输出结果如下图所示:
二、L2正则化
先看看tf.contrib.layers.l2_regularizer(weight_decay)都执行了什么:
import tensorflow as tf
sess=tf.Session()
weight_decay=0.1
tmp=tf.constant([0,1,2,3],dtype=tf.float32)
"""
l2_reg=tf.contrib.layers.l2_regularizer(weight_decay)
a=tf.get_variable("I_am_a",regularizer=l2_reg,initializer=tmp)
"""
#**上面代码的等价代码**************************
a=tf.get_variable("I_am_a",initializer=tmp)
a2=tf.reduce_sum(a*a)*weight_decay/2;
a3=tf.get_variable(a.name.split(":")[0]+"/Regularizer/l2_regularizer",initializer=a2)
tf.add_to_collection(tf.GraphKeys.REGULARIZATION_LOSSES,a3)
#****************************************************
####把上面这四行屏蔽了,把上面的上面2行代码解除屏蔽,得到的结果完全一样
sess.run(tf.global_variables_initializer())
keys = tf.get_collection(tf.GraphKeys.REGULARIZATION_LOSSES)
for key in keys:
print("%s : %s" %(key.name,sess.run(key)))
输出结果:
我们很容易可以模拟出tf.contrib.layers.l2_regularizer都做了什么,不过会让代码变丑。
以下比较完整实现L2 正则化。
import tensorflow as tf
sess=tf.Session()
weight_decay=0.1 #(1)定义weight_decay
l2_reg=tf.contrib.layers.l2_regularizer(weight_decay) #(2)定义l2_regularizer()
tmp=tf.constant([0,1,2,3],dtype=tf.float32)
a=tf.get_variable("I_am_a",regularizer=l2_reg,initializer=tmp) #(3)创建variable,l2_regularizer复制给regularizer参数。
#目测REXXX_LOSSES集合
#regularizer定义会将a加入REGULARIZATION_LOSSES集合
print("Global Set:")
keys = tf.get_collection("variables")
for key in keys:
print(key.name)
print("Regular Set:")
keys = tf.get_collection(tf.GraphKeys.REGULARIZATION_LOSSES)
for key in keys:
print(key.name)
print("--------------------")
sess.run(tf.global_variables_initializer())
print(sess.run(a))
reg_set=tf.get_collection(tf.GraphKeys.REGULARIZATION_LOSSES) #(4)则REGULARIAZTION_LOSSES集合会包含所有被weight_decay后的参数和,将其相加
l2_loss=tf.add_n(reg_set)
print("loss=%s" %(sess.run(l2_loss)))
"""
此处输出0.7,即:
weight_decay*sigmal(w*2)/2=0.1*(0*0+1*1+2*2+3*3)/2=0.7
其实代码自己写也很方便,用API看着比较正规。
在网络模型中,直接将l2_loss加入loss就好了。(loss变大,执行train自然会decay)
"""