Mimicking Very Efficient Network for Object Detection 筆記

1. 介紹

如今大多數(shù)的檢測模型的backbone都需要在ImageNet預(yù)訓(xùn)練话瞧,才能達到一個較好的結(jié)果嫩与。但是如果是新設(shè)計的網(wǎng)絡(luò)結(jié)構(gòu),就需要先在ImageNet預(yù)訓(xùn)練移稳,這將會非常耗時蕴纳。因此,作者提出是否能夠跳過這個步驟个粱,而不損失效果。
作者提出翻翩,假如我們擁有一個表現(xiàn)很好的檢測模型都许,我們可以用它來監(jiān)督其他檢測網(wǎng)絡(luò)。那么問題就是如何用一個好的檢測模型嫂冻,監(jiān)督訓(xùn)練一個更加高效的網(wǎng)絡(luò)胶征,同時保持它的準(zhǔn)確率。
大多數(shù)的檢測算法桨仿,都會有個基于CNN的特征提取器睛低,然后使用不同的方法來對特征進行解碼,得到用于檢測的數(shù)據(jù)服傍。因此钱雷,大的模型和更加高效的模型的區(qū)別主要在于這個CNN特征提取器,也就是backbone吹零。因此,作者認為,mimicking應(yīng)該加在CNN輸出的feature map上辉巡。
但是feature map通常維度很高太雨,直接進行模仿很難收斂∶S迹考慮到特征提取是基于region或者proposal的操刀,因此,需要從特征中提取對應(yīng)區(qū)域來優(yōu)化婴洼。
作者提出的模仿的方式骨坑,可以應(yīng)對圖片縮放,可以使用一種2階段的流程來進一步提高performance窃蹋。

3. Mimicking for Object Detection

3.1 Logits Mimic Learning

Mimicking的主要思想是通過大的模型的soft targets或者logits來訓(xùn)練一個小的神經(jīng)網(wǎng)絡(luò)卡啰。通常soft targets包含了大的模型學(xué)習(xí)得到的信息,是的小的模型能夠近似地擬合這種大模型的知識警没。loss函數(shù)可以是L2:


(1)

3.2 Feature Map Mimic Learning

為了能夠在檢測任務(wù)中應(yīng)用mimicking匈辱,作者采用的是feature map模仿。由于卷積出來的特征的維度都非常高杀迹,但是與此同時亡脸,圖片上的物體卻可能相對較少,因此直接模仿整個圖片的feature map就不是特別合適,因此作者提出浅碾,僅僅模仿proposals的feature map大州。具體loss如下:

loss

從公式可以看出,作者使用了模仿帶來的L2 loss和原本的檢測的loss垂谢,并用一個參數(shù)來控制相對大小厦画。
為了減少特征數(shù)目帶來的影響,作者又提出改用如下loss:
5

其中
m_i
為proposal提取的特征的維度

3.3 Network Architecture and Implementation Details

框架如下圖


Figure 1

訓(xùn)練的時候滥朱,大的網(wǎng)絡(luò)是以及訓(xùn)練好的Faster R-CNN或者R-FCN根暑,小的網(wǎng)絡(luò)隨機初始化,每次根據(jù)小的網(wǎng)絡(luò)的RPN徙邻,從大的網(wǎng)絡(luò)提取相應(yīng)的feature map排嫌,作為小的網(wǎng)絡(luò)的監(jiān)督信息。最后再使用訓(xùn)練數(shù)據(jù)進行finetune缰犁。

3.4 Two-stage Mimic

finetune可能導(dǎo)致退化淳地,結(jié)果變差,因此作者提出在2階段也假如mimic loss帅容。通過二階段的mimic颇象,就使得特征,proposal信息丰嘉,以及分類信息都傳遞給了小網(wǎng)絡(luò)夯到。

3.5 Mimic over Scales

當(dāng)輸入尺寸減少時,通常會導(dǎo)致結(jié)果很差饮亏。作者通過在feature map后增加一層deconv耍贾,再按照前文提出的mimic方法進行訓(xùn)練。這種方式可以增加performance路幸。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末荐开,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子简肴,更是在濱河造成了極大的恐慌晃听,老刑警劉巖,帶你破解...
    沈念sama閱讀 217,542評論 6 504
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件砰识,死亡現(xiàn)場離奇詭異能扒,居然都是意外死亡,警方通過查閱死者的電腦和手機辫狼,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,822評論 3 394
  • 文/潘曉璐 我一進店門初斑,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人膨处,你說我怎么就攤上這事见秤∩笆” “怎么了?”我有些...
    開封第一講書人閱讀 163,912評論 0 354
  • 文/不壞的土叔 我叫張陵鹃答,是天一觀的道長乎澄。 經(jīng)常有香客問我,道長测摔,這世上最難降的妖魔是什么置济? 我笑而不...
    開封第一講書人閱讀 58,449評論 1 293
  • 正文 為了忘掉前任,我火速辦了婚禮避咆,結(jié)果婚禮上舟肉,老公的妹妹穿的比我還像新娘。我一直安慰自己查库,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 67,500評論 6 392
  • 文/花漫 我一把揭開白布黄琼。 她就那樣靜靜地躺著樊销,像睡著了一般。 火紅的嫁衣襯著肌膚如雪脏款。 梳的紋絲不亂的頭發(fā)上围苫,一...
    開封第一講書人閱讀 51,370評論 1 302
  • 那天,我揣著相機與錄音撤师,去河邊找鬼剂府。 笑死,一個胖子當(dāng)著我的面吹牛剃盾,可吹牛的內(nèi)容都是我干的腺占。 我是一名探鬼主播,決...
    沈念sama閱讀 40,193評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼痒谴,長吁一口氣:“原來是場噩夢啊……” “哼衰伯!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起积蔚,我...
    開封第一講書人閱讀 39,074評論 0 276
  • 序言:老撾萬榮一對情侶失蹤意鲸,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后尽爆,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體怎顾,經(jīng)...
    沈念sama閱讀 45,505評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,722評論 3 335
  • 正文 我和宋清朗相戀三年漱贱,在試婚紗的時候發(fā)現(xiàn)自己被綠了槐雾。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 39,841評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡饱亿,死狀恐怖蚜退,靈堂內(nèi)的尸體忽然破棺而出闰靴,到底是詐尸還是另有隱情,我是刑警寧澤钻注,帶...
    沈念sama閱讀 35,569評論 5 345
  • 正文 年R本政府宣布蚂且,位于F島的核電站,受9級特大地震影響幅恋,放射性物質(zhì)發(fā)生泄漏杏死。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,168評論 3 328
  • 文/蒙蒙 一捆交、第九天 我趴在偏房一處隱蔽的房頂上張望淑翼。 院中可真熱鬧,春花似錦品追、人聲如沸玄括。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,783評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽遭京。三九已至,卻和暖如春泞莉,著一層夾襖步出監(jiān)牢的瞬間哪雕,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,918評論 1 269
  • 我被黑心中介騙來泰國打工鲫趁, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留斯嚎,地道東北人。 一個月前我還...
    沈念sama閱讀 47,962評論 2 370
  • 正文 我出身青樓挨厚,卻偏偏與公主長得像堡僻,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子幽崩,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 44,781評論 2 354

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