滑動(dòng)平均模型

在學(xué)習(xí)神經(jīng)網(wǎng)絡(luò)中看到了一個(gè)平均滑動(dòng)模型赢笨,該方法可以使模型在測(cè)試數(shù)據(jù)上表現(xiàn)的更加健壯赚哗,而TensorFlow中提供的實(shí)現(xiàn)方法為tf.train.ExponentialMovingAverage展父,起初并不理解該方法為啥能使模型在測(cè)試數(shù)據(jù)上更健壯举农,多放查找資料之后幽告,記錄在此。

思想

在初始化ExponentialMovingAverage時(shí)谬返,需要提供一個(gè)衰減率(decay)來(lái)空值模型跟新的書讀。ExponentialMovingAverage會(huì)對(duì)TensorFlow中每一個(gè)變量會(huì)維護(hù)一個(gè)影子變量(shadow_variable)日杈,影子變量的初始值為變量的初始值遣铝,每次迭代時(shí),變量進(jìn)行更新之后莉擒,影子變量的值也會(huì)同步更新:

shadow\_variable = decay * shadow\_variable + (1-decay)*variable

從上式中可以看到酿炸,decay決定模型更新的速度,decay越大涨冀,模型越穩(wěn)定填硕。在實(shí)際應(yīng)用中,decay一般是接近1的數(shù)(0.99鹿鳖,0.999等)扁眯。

當(dāng)decay設(shè)置較大時(shí),模型訓(xùn)練比較慢翅帜,為了使模型在前期能夠更新更快姻檀,ExponentialMovingAverage還提供了num_updates參數(shù)來(lái)動(dòng)態(tài)設(shè)置decay大小。而此時(shí)的衰減率為:

decay = min\left\{decay, \frac{1+num\_updates}{10+num\_updates} \right\}

在使用梯度下降算法進(jìn)行模型訓(xùn)練時(shí)涝滴,每次更新參數(shù)權(quán)重時(shí)绣版,該權(quán)重的影子變量也會(huì)隨著模型的訓(xùn)練而更新周荐,最終穩(wěn)定在一個(gè)接近真實(shí)權(quán)重值的附近。在測(cè)試集上使用影子變量替換原來(lái)的變量進(jìn)行預(yù)測(cè)時(shí)僵娃,可以得到一個(gè)更好的結(jié)果概作。

即,滑動(dòng)平均的使用步驟為:

  1. 訓(xùn)練階段:為每個(gè)可訓(xùn)練的權(quán)重維護(hù)影子變量默怨,并隨著迭代的進(jìn)行更新讯榕;
  2. 預(yù)測(cè)階段:使用影子變量替代真實(shí)變量值,進(jìn)行預(yù)測(cè)匙睹。

滑動(dòng)平均為什么在測(cè)試過(guò)程中被使用

訓(xùn)練中一直使用原來(lái)不帶滑動(dòng)的參數(shù)愚屁,可以得到新的參數(shù),如此就可以更新該參數(shù)的影子變量shadow_variable痕檬■保基于上面的式子可以看到,shadow_variable的更新比較平滑梦谜,對(duì)于隨機(jī)梯度下降算法而言丘跌,更平滑的更新效果較好。

代碼示例

import tensorflow as tf

v1 = tf.Variable(0, dtype=tf.float32)
step = tf.Variable(0, trainable=False)

ema = tf.train.ExponentialMovingAverage(0.99, step)

# 定義一個(gè)平滑平均偏亮的操作唁桩,每次執(zhí)行時(shí)闭树,會(huì)更新列表中的變量
maintain_averages_op = ema.apply([v1])

with tf.Session() as sess:
    init_op = tf.global_variables_initializer()
    sess.run(init_op)

    # 通過(guò)ema.average(v1)獲取滑動(dòng)平均之后變量的取值
    print(sess.run([v1, ema.average(v1)]))

    # 更新變量v1的值到5
    sess.run(tf.assign(v1, 5))

    # 更新v1的滑動(dòng)平均值。decay = min{0.99, (1+step)/(10+step)} = 0.1
    # v1 的滑動(dòng)平均更新為 0.1 * 0 + 0.9 * 5 = 4.5
    sess.run(maintain_averages_op)
    print(sess.run([v1, ema.average(v1)]))

    # 更新step為10000
    sess.run(tf.assign(step, 10000))
    # 更新v1的值為10
    sess.run(tf.assign(v1, 10))
    # 更新v1的滑動(dòng)平均值荒澡,decay = min{0.99, (1+step)/(10+step)} = 0.99
    # v1的滑動(dòng)平均更新為 0.99 * 4.5 + 0.01 * 10 = 4.555
    sess.run(maintain_averages_op)
    print(sess.run([v1, ema.average(v1)]))

    # 再次更新滑動(dòng)平均值 0.99 * 4.555 + 0.001 * 10 = 4.60945
    sess.run(maintain_averages_op)
    print(sess.run([v1, ema.average(v1)]))
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末报辱,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子单山,更是在濱河造成了極大的恐慌碍现,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,546評(píng)論 6 507
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件米奸,死亡現(xiàn)場(chǎng)離奇詭異昼接,居然都是意外死亡,警方通過(guò)查閱死者的電腦和手機(jī)躏升,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,224評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門辩棒,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái),“玉大人膨疏,你說(shuō)我怎么就攤上這事一睁。” “怎么了佃却?”我有些...
    開(kāi)封第一講書人閱讀 164,911評(píng)論 0 354
  • 文/不壞的土叔 我叫張陵者吁,是天一觀的道長(zhǎng)。 經(jīng)常有香客問(wèn)我饲帅,道長(zhǎng)复凳,這世上最難降的妖魔是什么瘤泪? 我笑而不...
    開(kāi)封第一講書人閱讀 58,737評(píng)論 1 294
  • 正文 為了忘掉前任,我火速辦了婚禮育八,結(jié)果婚禮上对途,老公的妹妹穿的比我還像新娘。我一直安慰自己髓棋,他們只是感情好实檀,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,753評(píng)論 6 392
  • 文/花漫 我一把揭開(kāi)白布。 她就那樣靜靜地躺著按声,像睡著了一般膳犹。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上签则,一...
    開(kāi)封第一講書人閱讀 51,598評(píng)論 1 305
  • 那天须床,我揣著相機(jī)與錄音,去河邊找鬼渐裂。 笑死豺旬,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的芯义。 我是一名探鬼主播哈垢,決...
    沈念sama閱讀 40,338評(píng)論 3 418
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼扛拨!你這毒婦竟也來(lái)了?” 一聲冷哼從身側(cè)響起举塔,我...
    開(kāi)封第一講書人閱讀 39,249評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤绑警,失蹤者是張志新(化名)和其女友劉穎,沒(méi)想到半個(gè)月后央渣,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體计盒,經(jīng)...
    沈念sama閱讀 45,696評(píng)論 1 314
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,888評(píng)論 3 336
  • 正文 我和宋清朗相戀三年芽丹,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了北启。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 40,013評(píng)論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡拔第,死狀恐怖咕村,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情蚊俺,我是刑警寧澤懈涛,帶...
    沈念sama閱讀 35,731評(píng)論 5 346
  • 正文 年R本政府宣布,位于F島的核電站泳猬,受9級(jí)特大地震影響批钠,放射性物質(zhì)發(fā)生泄漏宇植。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,348評(píng)論 3 330
  • 文/蒙蒙 一埋心、第九天 我趴在偏房一處隱蔽的房頂上張望指郁。 院中可真熱鬧,春花似錦拷呆、人聲如沸坡氯。這莊子的主人今日做“春日...
    開(kāi)封第一講書人閱讀 31,929評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)箫柳。三九已至,卻和暖如春啥供,著一層夾襖步出監(jiān)牢的瞬間悯恍,已是汗流浹背。 一陣腳步聲響...
    開(kāi)封第一講書人閱讀 33,048評(píng)論 1 270
  • 我被黑心中介騙來(lái)泰國(guó)打工伙狐, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留涮毫,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 48,203評(píng)論 3 370
  • 正文 我出身青樓贷屎,卻偏偏與公主長(zhǎng)得像罢防,于是被迫代替她去往敵國(guó)和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子唉侄,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,960評(píng)論 2 355

推薦閱讀更多精彩內(nèi)容