AI面試第五彈(圖像分割常用Loss)


https://github.com/JunMa11/SegLoss

前文:推薦一個(gè)很好的github倉(cāng)庫(kù)腕铸,里面幾乎有所有圖像分割loss的實(shí)現(xiàn)造寝,且目前仍在更新:
https://github.com/JunMa11/SegLoss

一荐吉、Distributation-based loss?

1钠龙、Binary Cross-Etropy(二進(jìn)制交叉熵?fù)p失函數(shù))

交叉熵定義為對(duì)給定隨機(jī)變量或事件集的兩個(gè)概率分布之間的差異的度量惑淳。它被廣泛用于分類(lèi)任務(wù)沛膳,并且由于分割是像素級(jí)分類(lèi),因此小郭很好汛聚。在多分類(lèi)任務(wù)中,經(jīng)常采用softmax激活函數(shù)+交叉熵?fù)p失函數(shù)短荐,因此交叉熵描述了兩個(gè)概率分布的差異倚舀,然而神經(jīng)網(wǎng)絡(luò)輸出的向量,并不是概率分布的形式忍宋。所以需要將softmax激活函數(shù)將一個(gè)向量進(jìn)行“歸一化”成概率分布的習(xí)慣是痕貌,再采用交叉熵?fù)p失函數(shù)計(jì)算loss。


2糠排、WCE Loss

帶權(quán)重的交叉熵loss — Weighted cross-entropy (WCE)

R為標(biāo)準(zhǔn)的分割圖舵稠,其中rn?為label 分割圖中的某一個(gè)像素的GT。P為預(yù)測(cè)的概率圖,pnp_npn?為像素的預(yù)測(cè)概率值哺徊,背景像素圖的概率值就為1-P室琢。

只有兩個(gè)類(lèi)別的帶權(quán)重的交叉熵為:


w為權(quán)重。

這種損失函數(shù)的缺點(diǎn)是需要認(rèn)為的調(diào)整困難樣本的權(quán)重落追,增加調(diào)參難度盈滴。

3、Focal loss

focal loss的提出是在目標(biāo)檢測(cè)領(lǐng)域轿钠,為了解決正負(fù)樣本比例嚴(yán)重失衡的問(wèn)題巢钓。是由log loss改進(jìn)而來(lái)的,為了于log loss進(jìn)行對(duì)比疗垛,公式如下:


4症汹、Distance penalized CE loss

5、Log loss

(不同于交叉熵贷腕,具體區(qū)別可見(jiàn)夕小瑤文章)對(duì)于二分類(lèi)而言背镇,對(duì)數(shù)損失函數(shù)如下公式所示:

log loss

其中,yi為輸入實(shí)例xi的真實(shí)類(lèi)別,?pi為預(yù)測(cè)輸入實(shí)例?xi屬于類(lèi)別 1 的概率. 對(duì)所有樣本的對(duì)數(shù)損失表示對(duì)每個(gè)樣本的對(duì)數(shù)損失的平均值, 對(duì)于完美的分類(lèi)器, 對(duì)數(shù)損失為 0花履。

此loss function每一次梯度的回傳對(duì)每一個(gè)類(lèi)別具有相同的關(guān)注度芽世!所以極易受到類(lèi)別不平衡的影響,在圖像分割領(lǐng)域尤其如此诡壁。

例如目標(biāo)在整幅圖像當(dāng)中占比也就僅僅千分之一济瓢,那么在一副圖像中,正樣本(像素點(diǎn))與父樣本的比例約為1~1000妹卿,如果訓(xùn)練圖像中還包含大量的背景圖旺矾,即圖像當(dāng)中不包含任何的疾病像素,那么不平衡的比例將擴(kuò)大到>10000夺克,那么訓(xùn)練的后果將會(huì)是箕宙,網(wǎng)絡(luò)傾向于什么也不預(yù)測(cè)!生成的mask幾乎沒(méi)有病灶像素區(qū)域铺纽!

此處的案例可以參考airbus-ship-detection

二柬帕、基于區(qū)域的loss

1、Dice loss

dice loss 的提出是在U-net中狡门,其中的一段原因描述是在感興趣的解剖結(jié)構(gòu)僅占據(jù)掃描的非常小的區(qū)域陷寝,從而使學(xué)習(xí)過(guò)程陷入損失函數(shù)的局部最小值。所以要加大前景區(qū)域的權(quán)重其馏。

Dice 可以理解為是兩個(gè)輪廓區(qū)域的相似程度凤跑,用A、B表示兩個(gè)輪廓區(qū)域所包含的點(diǎn)集叛复,定義為:

也可表示為

實(shí)際實(shí)現(xiàn)時(shí)可照以下公式實(shí)現(xiàn):

2仔引、IOU loss




3扔仓、Tversky loss

公式如下

α=β=0.5時(shí),Tversky Loss是DICE loss(分子分母同乘2) 咖耘。?α=β=1時(shí)翘簇,Tversky Loss是IOU loss。

同時(shí)我們也應(yīng)注意到 鲤看,如果設(shè)A是預(yù)測(cè)區(qū)域缘揪,B是真實(shí)區(qū)域,|A-B|代表著假陽(yáng)义桂,|B-A|代表著假陰找筝。在這種情況下,我們改變?chǔ)梁挺碌娜≈稻湍軌蚋淖兾覀儍?yōu)化時(shí)是希望假陽(yáng)少一點(diǎn)還是假陰少一點(diǎn)慷吊。

4袖裕、Generalized Dice loss

在計(jì)算DICE loss時(shí),針對(duì)每一類(lèi)都有一個(gè)dice loss溉瓶,這樣就會(huì)考慮各個(gè)類(lèi)急鳄。這個(gè)loss可以解決訓(xùn)練數(shù)據(jù)集不均衡的問(wèn)題。

5堰酿、sensitivity-specificity loss

6疾宏、Penalty loss

7、Lovász-Softmax loss

三触创、基于邊界的Loss

1坎藐、Boundary loss

2、Hausdorff distance(HD) loss

(1)HD介紹:

在談?wù)摼嚯x時(shí)哼绑,我們通常指的是最短的距離:例如岩馍,如果說(shuō)一個(gè)點(diǎn)X在多邊形P的距離D處,我們通常假設(shè)D是從X到P的最近點(diǎn)的距離抖韩。

相同的邏輯也適用于多邊形:如果兩個(gè)多邊形A和B彼此相距一定距離蛀恩,我們通常將距離理解為A的任意點(diǎn)與B的任意點(diǎn)之間的最短距離。通常茂浮,這稱(chēng)為minimin?function双谆。該距離A和B之間的D由下式給出

(2)HD loss

由CNN的輸出預(yù)測(cè)HD,并將預(yù)測(cè)出來(lái)的HD當(dāng)作loss來(lái)優(yōu)化

五席揽、各種Loss相組合

1佃乘、Exponential Logarithmic loss

2、其他各種組合...



?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末驹尼,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子庞呕,更是在濱河造成了極大的恐慌新翎,老刑警劉巖程帕,帶你破解...
    沈念sama閱讀 222,464評(píng)論 6 517
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異地啰,居然都是意外死亡愁拭,警方通過(guò)查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 95,033評(píng)論 3 399
  • 文/潘曉璐 我一進(jìn)店門(mén)亏吝,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)岭埠,“玉大人,你說(shuō)我怎么就攤上這事蔚鸥∠郏” “怎么了?”我有些...
    開(kāi)封第一講書(shū)人閱讀 169,078評(píng)論 0 362
  • 文/不壞的土叔 我叫張陵止喷,是天一觀的道長(zhǎng)馆类。 經(jīng)常有香客問(wèn)我,道長(zhǎng)弹谁,這世上最難降的妖魔是什么乾巧? 我笑而不...
    開(kāi)封第一講書(shū)人閱讀 59,979評(píng)論 1 299
  • 正文 為了忘掉前任,我火速辦了婚禮预愤,結(jié)果婚禮上沟于,老公的妹妹穿的比我還像新娘。我一直安慰自己植康,他們只是感情好旷太,可當(dāng)我...
    茶點(diǎn)故事閱讀 69,001評(píng)論 6 398
  • 文/花漫 我一把揭開(kāi)白布。 她就那樣靜靜地躺著向图,像睡著了一般泳秀。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上榄攀,一...
    開(kāi)封第一講書(shū)人閱讀 52,584評(píng)論 1 312
  • 那天嗜傅,我揣著相機(jī)與錄音,去河邊找鬼檩赢。 笑死吕嘀,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的贞瞒。 我是一名探鬼主播偶房,決...
    沈念sama閱讀 41,085評(píng)論 3 422
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼军浆!你這毒婦竟也來(lái)了棕洋?” 一聲冷哼從身側(cè)響起,我...
    開(kāi)封第一講書(shū)人閱讀 40,023評(píng)論 0 277
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤乒融,失蹤者是張志新(化名)和其女友劉穎掰盘,沒(méi)想到半個(gè)月后摄悯,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,555評(píng)論 1 319
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡愧捕,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,626評(píng)論 3 342
  • 正文 我和宋清朗相戀三年奢驯,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片次绘。...
    茶點(diǎn)故事閱讀 40,769評(píng)論 1 353
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡瘪阁,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出邮偎,到底是詐尸還是另有隱情管跺,我是刑警寧澤,帶...
    沈念sama閱讀 36,439評(píng)論 5 351
  • 正文 年R本政府宣布钢猛,位于F島的核電站伙菜,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏命迈。R本人自食惡果不足惜贩绕,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 42,115評(píng)論 3 335
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望壶愤。 院中可真熱鬧淑倾,春花似錦、人聲如沸征椒。這莊子的主人今日做“春日...
    開(kāi)封第一講書(shū)人閱讀 32,601評(píng)論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)勃救。三九已至碍讨,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間蒙秒,已是汗流浹背勃黍。 一陣腳步聲響...
    開(kāi)封第一講書(shū)人閱讀 33,702評(píng)論 1 274
  • 我被黑心中介騙來(lái)泰國(guó)打工, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留晕讲,地道東北人覆获。 一個(gè)月前我還...
    沈念sama閱讀 49,191評(píng)論 3 378
  • 正文 我出身青樓,卻偏偏與公主長(zhǎng)得像瓢省,于是被迫代替她去往敵國(guó)和親弄息。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,781評(píng)論 2 361