目標檢測網(wǎng)絡(luò)之retinanet解讀(一)

[Paper:Focal Loss for Dense Object Detection]
[https://arxiv.org/abs/1708.02002]
關(guān)于目標檢測主要分為兩大類別:one stage压固,two stage
one stage 代表:yolo系列,ssd(特點:檢測精度低浙踢,但檢測速度快)
two stage 代表:R-CNN系列,SPPNet(特點:檢測精度高,但檢測速度慢)

在論文中作者去探討了造成one stage精度低的原因哺窄,發(fā)現(xiàn)在訓(xùn)練密集目標檢測器的過程中出現(xiàn)了嚴重的foreground-background類別不平衡笆檀。

首先,什么是“類別不平衡”呢露戒?

檢測算法在早期會生成很多bbox描验,而在一幅正常的圖像中需要檢測的object不會很多白嘁,這就意味著多數(shù)的bbox是屬于background,使得foreground-background類別不平衡膘流。

存在類別平衡問題為什么會導(dǎo)致檢測精度低呢絮缅?

因為bbox數(shù)量很多,而屬于background的bbox太多了呼股,假設(shè)分類器將所有的bbox全部歸為background耕魄,那么精度也會很高,而這樣的分類器是一個失敗的分類器彭谁,所以導(dǎo)致目標檢測的精度很低吸奴。

那么,為什么two stage機制的目標檢測器可以達到很高的精度呢?

因為two stage的第一個階段生成一個候選目標位置組成的稀疏樣本集则奥,即RPN簡單的對anchor進行二分類(只是區(qū)分是foreground和background考润,并不會區(qū)別細類),這樣读处,屬于background類別的bbox會大量減少糊治,雖然其數(shù)量依然遠大于foreground的bbox(例如3:1),但已不像最初生成anchor差別那么大了罚舱,這一階段最終結(jié)果是從“類別極不平衡”到“類別較不平衡”轉(zhuǎn)變井辜,也就是說two stage并不能完全解決類別不平衡問題,第二個階段使用一個卷積神經(jīng)網(wǎng)絡(luò)將各候選位置歸置foreground類別或者background類別管闷,即在初步篩選后的bbox上進行難度小得多的分類(細分類)粥脚,這樣分類器便得到了較好的訓(xùn)練,從而精度提高了包个。但因為是經(jīng)過了兩個階段的處理阿逃,操作復(fù)雜,使得檢測速度變慢了赃蛛。

為什么one stage系列無法避免這個問題呢?

因為one stage系列的檢測器直接在“類別極不平衡”的bbox中進行難度極大的細分類搀菩,直接輸出bbox和標簽呕臂,而原有的交叉熵損失(CE)作為分類任務(wù)的損失函數(shù),無法抗衡“類別極不平衡”(會在另一篇文章中詳細講解交叉熵損失)肪跋,容易導(dǎo)致分類器訓(xùn)練失敗歧蒋。因此,one stage檢測雖然速度快州既,但檢測精度低谜洽。

既然one stage中的交叉熵損失函數(shù)無法抗衡“類別極不平衡”,所以retinanet作者何愷明及其團隊提出Focal Loss替換交叉熵損失來提高檢測精度吴叶。

Focal Loss的形式是怎樣的阐虚,有什么含義?

首先介紹一下二分類的交叉熵損失:

wenyilab

公式(1)中蚌卤,y∈{±1} 指定了ground-true class实束,p∈[0,1] 是模型對于標簽y = 1的類的估計概率。為了方便逊彭,我們定義Pt:
wenyilab

公式(2)可以轉(zhuǎn)換為:


wenyilab

解決類別不平衡的常見方法是分別為class1和class-1加入加權(quán)因子α∈[0; 1]咸灿、1-α。所以α-balanced 的CE損耗可寫為:

wenyilab

鑒于以上的改進過程的思想侮叮,可以將交叉熵損失增加一個調(diào)節(jié)因子(1 - pt)γ避矢,其中γ≥0。這樣便出現(xiàn)了Focal Loss,其定義為:


wenyilab

在實踐過程中使用的Focal Loss形式為:


wenyilab
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末审胸,一起剝皮案震驚了整個濱河市亥宿,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌歹嘹,老刑警劉巖箩绍,帶你破解...
    沈念sama閱讀 219,039評論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異尺上,居然都是意外死亡材蛛,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,426評論 3 395
  • 文/潘曉璐 我一進店門怎抛,熙熙樓的掌柜王于貴愁眉苦臉地迎上來卑吭,“玉大人,你說我怎么就攤上這事马绝《股停” “怎么了?”我有些...
    開封第一講書人閱讀 165,417評論 0 356
  • 文/不壞的土叔 我叫張陵富稻,是天一觀的道長掷邦。 經(jīng)常有香客問我,道長椭赋,這世上最難降的妖魔是什么抚岗? 我笑而不...
    開封第一講書人閱讀 58,868評論 1 295
  • 正文 為了忘掉前任,我火速辦了婚禮哪怔,結(jié)果婚禮上宣蔚,老公的妹妹穿的比我還像新娘。我一直安慰自己认境,他們只是感情好胚委,可當我...
    茶點故事閱讀 67,892評論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著叉信,像睡著了一般亩冬。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上茉盏,一...
    開封第一講書人閱讀 51,692評論 1 305
  • 那天鉴未,我揣著相機與錄音,去河邊找鬼鸠姨。 笑死铜秆,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的讶迁。 我是一名探鬼主播连茧,決...
    沈念sama閱讀 40,416評論 3 419
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了啸驯?” 一聲冷哼從身側(cè)響起客扎,我...
    開封第一講書人閱讀 39,326評論 0 276
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎罚斗,沒想到半個月后徙鱼,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,782評論 1 316
  • 正文 獨居荒郊野嶺守林人離奇死亡针姿,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,957評論 3 337
  • 正文 我和宋清朗相戀三年袱吆,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片距淫。...
    茶點故事閱讀 40,102評論 1 350
  • 序言:一個原本活蹦亂跳的男人離奇死亡绞绒,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出榕暇,到底是詐尸還是另有隱情蓬衡,我是刑警寧澤,帶...
    沈念sama閱讀 35,790評論 5 346
  • 正文 年R本政府宣布彤枢,位于F島的核電站狰晚,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏缴啡。R本人自食惡果不足惜家肯,卻給世界環(huán)境...
    茶點故事閱讀 41,442評論 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望盟猖。 院中可真熱鬧,春花似錦换棚、人聲如沸式镐。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,996評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽娘汞。三九已至,卻和暖如春夕玩,著一層夾襖步出監(jiān)牢的瞬間你弦,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,113評論 1 272
  • 我被黑心中介騙來泰國打工燎孟, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留禽作,地道東北人。 一個月前我還...
    沈念sama閱讀 48,332評論 3 373
  • 正文 我出身青樓揩页,卻偏偏與公主長得像旷偿,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 45,044評論 2 355

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