學(xué)習(xí)筆記|語(yǔ)義分割常用損失函數(shù)

參考:https://mp.weixin.qq.com/s/ra2qpFSbSuuJPDj39A5MWA

源于GiantPandaCV?庆揪,作者BBuf

文章僅為學(xué)習(xí)筆記无切,感謝文章各位貢獻(xiàn)者,侵刪


交叉熵loss

L=-\sum\nolimits_{c=1}^M y_{c} \lg Pc

M表示類(lèi)別數(shù)的圆,y_{c} 為one-hot向量(即元素只取0溪食、1囊卜,若與樣本類(lèi)別相同則取1,否則取0)

缺點(diǎn):若前景像素遠(yuǎn)小于背景像素错沃,損失函數(shù)中y=0部分會(huì)占據(jù)主導(dǎo)栅组,模型嚴(yán)重偏向背景

代碼實(shí)現(xiàn):

#二值交叉熵,input需要經(jīng)過(guò)sigmoid處理

nn.BCELoss(F.sigmoid(input), target)

#多分類(lèi)交叉熵

nn.CrossEntropyLoss(input, target)


帶權(quán)交叉熵loss

L=-\sum\nolimits_{c=1}^M \omega y_{c} \lg p_{c}

添加權(quán)重參數(shù)w_{c} 枢析,計(jì)算公式為w_{c}=\frac{N-Nc}{N}  玉掸,N表示總的像素?cái)?shù),Nc表示gt類(lèi)別為c的像素個(gè)數(shù)醒叁,對(duì)于樣本不均衡可以獲得更好的效果


Focal Loss

retinanet引入focal loss解決難易樣本不均衡的問(wèn)題司浪。一階段目標(biāo)檢測(cè)器會(huì)產(chǎn)生10k數(shù)量級(jí)的候選框,但只有少數(shù)是正樣本把沼。計(jì)算分類(lèi)時(shí)常用的交叉熵?fù)p失公式為:

為了解決政府樣本不均衡的問(wèn)題啊易,利用加權(quán)的形式:

由于候選目標(biāo)包含容易區(qū)分的和不易區(qū)分樣本,由于易分樣本占大多數(shù)(易分樣本指置信度高的樣本)饮睬,盡管損失很低租谈,但是仍然主導(dǎo)了損失韓式。易分樣本對(duì)模型效果提升很少捆愁,應(yīng)該主要關(guān)注難分樣本——focal loss:

focal loss pytorch代碼:

參考網(wǎng)址


dice loss

dice系數(shù):

上述函數(shù)用來(lái)度量集合的相似度割去。分子2是由于分母重復(fù)計(jì)算并集的原因窟却,s取值[0,1]。針對(duì)分割任務(wù)劫拗,X间校、Y指gt和預(yù)測(cè)的圖像。還可以改寫(xiě)為:

dice loss 定義公式:

注意一般情況下使用dice loss會(huì)對(duì)反向傳播不利页慷,使得訓(xùn)練不穩(wěn)定憔足。一般用于樣本極度不均衡的情況

原因說(shuō)明:參考網(wǎng)址


IOU loss

定義:(1-jaccard系數(shù))

與dice loss一樣屬于metric learning的衡量方式,一樣存在訓(xùn)練過(guò)程不穩(wěn)定的問(wèn)題酒繁,在分割任務(wù)中不怎么使用


Tversky loss

dice loss實(shí)質(zhì)上屬于tversky loss的特殊形式滓彰。

Tversky系數(shù):是dice系數(shù)和jaccard系數(shù)的廣義系數(shù)

alpha和β均為0.5時(shí)為dice系數(shù),為1時(shí)為jaccard系數(shù)州袒。A-B為假陽(yáng)性揭绑,B-A為假陰性


Generalized Dice loss

由于dice loss對(duì)小目標(biāo)預(yù)測(cè)十分不利,一旦小目標(biāo)有部分像素預(yù)測(cè)錯(cuò)誤就可能引起dice系數(shù)巨大波動(dòng)(分母)郎哭,導(dǎo)致梯度變化大訓(xùn)練不穩(wěn)定他匪。又因?yàn)閐ice loss針對(duì)的是某一個(gè)特定類(lèi)別的分割的損失,多個(gè)類(lèi)別要用到多個(gè)dice loss夸研。GDL就是對(duì)多個(gè)dice loss 進(jìn)行加權(quán)整合邦蜜。


BCE + Dice loss

適用于數(shù)據(jù)均衡的情況下,若數(shù)據(jù)不均衡亥至,容易退化成dice loss


Focal loss + Dice loss

https://arxiv.org/pdf/1808.05238.pdf


Exponential Logarithmic loss

https://arxiv.org/abs/1809.00076

整合focal loss和dice loss悼沈。表達(dá)式:

其中


Lovasz-Softmax Loss

https://arxiv.org/pdf/1705.08790.pdf

iou loss的擴(kuò)展

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市姐扮,隨后出現(xiàn)的幾起案子絮供,更是在濱河造成了極大的恐慌,老刑警劉巖茶敏,帶你破解...
    沈念sama閱讀 217,509評(píng)論 6 504
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件壤靶,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡惊搏,警方通過(guò)查閱死者的電腦和手機(jī)贮乳,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,806評(píng)論 3 394
  • 文/潘曉璐 我一進(jìn)店門(mén),熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)胀屿,“玉大人塘揣,你說(shuō)我怎么就攤上這事包雀∷拚福” “怎么了?”我有些...
    開(kāi)封第一講書(shū)人閱讀 163,875評(píng)論 0 354
  • 文/不壞的土叔 我叫張陵才写,是天一觀的道長(zhǎng)葡兑。 經(jīng)常有香客問(wèn)我奖蔓,道長(zhǎng),這世上最難降的妖魔是什么讹堤? 我笑而不...
    開(kāi)封第一講書(shū)人閱讀 58,441評(píng)論 1 293
  • 正文 為了忘掉前任吆鹤,我火速辦了婚禮,結(jié)果婚禮上洲守,老公的妹妹穿的比我還像新娘疑务。我一直安慰自己,他們只是感情好梗醇,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,488評(píng)論 6 392
  • 文/花漫 我一把揭開(kāi)白布知允。 她就那樣靜靜地躺著,像睡著了一般叙谨。 火紅的嫁衣襯著肌膚如雪温鸽。 梳的紋絲不亂的頭發(fā)上,一...
    開(kāi)封第一講書(shū)人閱讀 51,365評(píng)論 1 302
  • 那天手负,我揣著相機(jī)與錄音涤垫,去河邊找鬼。 笑死竟终,一個(gè)胖子當(dāng)著我的面吹牛蝠猬,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播衡楞,決...
    沈念sama閱讀 40,190評(píng)論 3 418
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼吱雏,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來(lái)了瘾境?” 一聲冷哼從身側(cè)響起歧杏,我...
    開(kāi)封第一講書(shū)人閱讀 39,062評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎迷守,沒(méi)想到半個(gè)月后犬绒,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,500評(píng)論 1 314
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡兑凿,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,706評(píng)論 3 335
  • 正文 我和宋清朗相戀三年凯力,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片礼华。...
    茶點(diǎn)故事閱讀 39,834評(píng)論 1 347
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡咐鹤,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出圣絮,到底是詐尸還是另有隱情祈惶,我是刑警寧澤,帶...
    沈念sama閱讀 35,559評(píng)論 5 345
  • 正文 年R本政府宣布,位于F島的核電站捧请,受9級(jí)特大地震影響凡涩,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜疹蛉,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,167評(píng)論 3 328
  • 文/蒙蒙 一活箕、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧可款,春花似錦育韩、人聲如沸。這莊子的主人今日做“春日...
    開(kāi)封第一講書(shū)人閱讀 31,779評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至翠拣,卻和暖如春版仔,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背误墓。 一陣腳步聲響...
    開(kāi)封第一講書(shū)人閱讀 32,912評(píng)論 1 269
  • 我被黑心中介騙來(lái)泰國(guó)打工蛮粮, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人谜慌。 一個(gè)月前我還...
    沈念sama閱讀 47,958評(píng)論 2 370
  • 正文 我出身青樓然想,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親欣范。 傳聞我的和親對(duì)象是個(gè)殘疾皇子变泄,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,779評(píng)論 2 354

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