DA(語義分割3)Bidirectional Learning for Domain Adaptation of Semantic Segmentation

Bidirectional Learning for Domain Adaptation of Semantic Segmentation

來源:CVPR 2019

作者:Yunsheng Li,Lu Yuan雷逆,Nuno Vasconcelos

機(jī)構(gòu):加州大學(xué)圣地亞哥分校(UC San Diego)盾剩,微軟

數(shù)據(jù)集:GTA5和SYNTHIA是原域筷登,Cityscapes是目標(biāo)域呻疹。

網(wǎng)絡(luò):translation model(F) is CycleGAN适贸, DeepLab V2? with the backbone ResNet101? and FCN-8s with VGG16 as our segmentation model.



代碼:https://github.com/liyunsheng13/BDL

本篇論文是基于Cycle-consistent adversarialdomain adaptation(CyCADA)提出的子網(wǎng)絡(luò)框架戈钢,一個(gè)子網(wǎng)絡(luò)是F馆蠕,一個(gè)是M期升。

貢獻(xiàn):

1.雙向?qū)W習(xí)(之前是單向的)

2.訓(xùn)練M時(shí)惊奇,提出用SSL(自監(jiān)督訓(xùn)練)

3.訓(xùn)練F時(shí),提出了一個(gè)peceptual loss

作者通過消融實(shí)驗(yàn)結(jié)果發(fā)現(xiàn)播赁,單獨(dú)使用雙向?qū)W習(xí)時(shí)不用SSL颂郎,效果并沒有使用雙向?qū)W習(xí)并使用SSL效果好。

1.動(dòng)機(jī)

目前行拢,語義分割的域適配分成兩步祖秒,第一步是用圖片到圖片的翻譯模型(比如cycleGAN)把圖片從原域轉(zhuǎn)到目標(biāo)域,第二步是在分割模型中加入一個(gè)判別器舟奠,來進(jìn)一步減小域差異。這兩步是順序進(jìn)行的房维。

該方法的不足:

第二步嚴(yán)重依賴第一步沼瘫,第一步效果好,則第二步效果也好咙俩。第一步效果不好耿戚,第二步就無能為力。

解決方法:

仍然用圖片到圖片的翻譯模型以及分割模型阿趁,但是改變訓(xùn)練的方法膜蛔,變成“translation-to-segmentation”以及“segmentation-to-translation”。同時(shí)脖阵,作者說皂股,如何讓其中一個(gè)模塊給另一個(gè)模塊提供正向反饋是關(guān)鍵。




順序?qū)W習(xí)和雙向?qū)W習(xí)的差別

1.1本文方法的詳細(xì)介紹:

本文方法由正向和反向兩個(gè)方向組成:

正向命黔,也就是translation-to-segmentation,作者提出用自監(jiān)督訓(xùn)練(self-supervised training,SSL)來訓(xùn)練分割適配模型呜呐,也就是更新網(wǎng)絡(luò)M。

反向悍募,也就是segmentation-to-translation蘑辑,作者提出一個(gè)新的perceptual loss來讓分割適配模型來更新translation模型,也就是更新網(wǎng)絡(luò)F(這個(gè)更新在之前的順序?qū)W習(xí)中是沒有的)坠宴。

自監(jiān)督學(xué)習(xí):自監(jiān)督學(xué)習(xí)是監(jiān)督學(xué)習(xí)的一個(gè)特例洋魂,它與眾不同,值得單獨(dú)分為一類喜鼓。自監(jiān)督學(xué)習(xí)是沒有人工標(biāo)注標(biāo)簽的監(jiān)督學(xué)習(xí)副砍,可以將它看作沒有人類參與的監(jiān)督學(xué)習(xí)。標(biāo)簽仍然存在(因?yàn)榭傄惺裁礀|西來監(jiān)督學(xué)習(xí)過程)颠通,但它們是從輸入數(shù)據(jù)中生成的址晕,通常使用啟發(fā)式算法生成的。

自監(jiān)督學(xué)習(xí) 參考博客https://blog.csdn.net/gdengden/article/details/84196883

2.訓(xùn)練過程

F和M是兩個(gè)子網(wǎng)絡(luò)顿锰,在參考論文Cycle-consistent adversarialdomain adaptation中提出的谨垃。F用于把一幅圖片從原域翻譯到目標(biāo)域启搂。而M則用翻譯的結(jié)果F(S)來訓(xùn)練網(wǎng)絡(luò),F(xiàn)(S)的標(biāo)簽和原域的標(biāo)簽相同刘陶。

訓(xùn)練算法

S'=F(S)胳赌,S’是原域數(shù)據(jù)經(jīng)過網(wǎng)絡(luò)F后的輸出。

該算法由兩個(gè)循環(huán)匙隔,外層循環(huán)主要用于通過前向和反向來學(xué)習(xí)翻譯模型F和分割模型M疑苫。

內(nèi)層循環(huán)則用于實(shí)現(xiàn)SSL過程。

等式2

訓(xùn)練F的損失函數(shù)分成三對(duì)兒纷责,一對(duì)兒是對(duì)抗損失捍掺,用來使得S'和T的分布相近,一對(duì)兒是重構(gòu)損失再膳,用來衡量經(jīng)過F翻譯后再翻譯回來的原域數(shù)據(jù)和未經(jīng)過翻譯的原域數(shù)據(jù)之間的差異挺勿,還有一對(duì)兒是本文提出的認(rèn)知損失,用于保持S和S'之間的語義一致性喂柒,即S和S'有相同的標(biāo)簽不瓶。


等式1


這是最初訓(xùn)練M的損失函數(shù),包括兩項(xiàng)灾杰,一項(xiàng)是對(duì)抗損失蚊丐,用來使得S‘和T輸入M后的特征表示相近,該項(xiàng)損失用于讓目標(biāo)域中沒有和原域?qū)R的數(shù)據(jù)進(jìn)行對(duì)齊,另一項(xiàng)就是常規(guī)的分割損失艳吠。

等式3

這是后來訓(xùn)練M的損失函數(shù)麦备,除了最初的那兩項(xiàng),還加了一項(xiàng)用帶偽標(biāo)簽的目標(biāo)域數(shù)據(jù)的分割損失讲竿。目標(biāo)域數(shù)據(jù)的偽標(biāo)簽是分割網(wǎng)絡(luò)給出的泥兰。


對(duì)M的自監(jiān)督學(xué)習(xí)過程

本文的貢獻(xiàn)之一就是為網(wǎng)絡(luò)M提出一種自監(jiān)督學(xué)習(xí)算法,圖二中(a)是初始時(shí)用等式1對(duì)分割網(wǎng)絡(luò)M進(jìn)行訓(xùn)練后的結(jié)果题禀,可以看出鞋诗,訓(xùn)練后,一些目標(biāo)域的數(shù)據(jù)和原域的數(shù)據(jù)進(jìn)行了對(duì)齊迈嘹,然后選擇這些對(duì)齊的目標(biāo)域數(shù)據(jù)作為集合Tssl的元素削彬,這樣就用等式3去訓(xùn)練M,然后再往集合Tssl中增加元素秀仲,這樣融痛,目標(biāo)域中不帶標(biāo)簽的樣本就越來越少了,直到目標(biāo)域樣本都被標(biāo)記了標(biāo)簽神僵。


3.網(wǎng)絡(luò)結(jié)構(gòu)和損失函數(shù)


網(wǎng)絡(luò)結(jié)構(gòu)

注意:網(wǎng)絡(luò)中是有判別器的雁刷,只不過沒有畫出來。

3.1訓(xùn)練F的損失函數(shù)(等式2)

等式2里的GAN損失和重構(gòu)損失


等式2里的認(rèn)知損失

3.2訓(xùn)練M的損失函數(shù)(等式3)


等式3里的對(duì)抗損失



等式3里原域數(shù)據(jù)的分割損失


等式3里目標(biāo)域數(shù)據(jù)的分割損失

4.實(shí)驗(yàn)結(jié)果

We compare the results? between our method and the state-of-the-art method with? two different backbone networks: ResNet101 and VGG16?? respectively. We perform the comparison on two tasks: “GTA5 to Cityscapes” and “SYNTHIA to Cityscapes”



原文總結(jié)部分

Traditional methods for domain adaptation involve minimizing? some measure of distance between the source and the target distributions. Two commonly used measures are? the first and second order moment [2], and learning the distance metrics using Adversarial approaches保礼。


單詞

vise verse:相反亦然

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末沛励,一起剝皮案震驚了整個(gè)濱河市责语,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌目派,老刑警劉巖坤候,帶你破解...
    沈念sama閱讀 222,183評(píng)論 6 516
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異企蹭,居然都是意外死亡白筹,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,850評(píng)論 3 399
  • 文/潘曉璐 我一進(jìn)店門谅摄,熙熙樓的掌柜王于貴愁眉苦臉地迎上來徒河,“玉大人,你說我怎么就攤上這事送漠⌒榍啵” “怎么了?”我有些...
    開封第一講書人閱讀 168,766評(píng)論 0 361
  • 文/不壞的土叔 我叫張陵螺男,是天一觀的道長(zhǎng)。 經(jīng)常有香客問我纵穿,道長(zhǎng)下隧,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 59,854評(píng)論 1 299
  • 正文 為了忘掉前任谓媒,我火速辦了婚禮淆院,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘句惯。我一直安慰自己土辩,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 68,871評(píng)論 6 398
  • 文/花漫 我一把揭開白布抢野。 她就那樣靜靜地躺著拷淘,像睡著了一般。 火紅的嫁衣襯著肌膚如雪指孤。 梳的紋絲不亂的頭發(fā)上启涯,一...
    開封第一講書人閱讀 52,457評(píng)論 1 311
  • 那天,我揣著相機(jī)與錄音恃轩,去河邊找鬼结洼。 笑死,一個(gè)胖子當(dāng)著我的面吹牛叉跛,可吹牛的內(nèi)容都是我干的松忍。 我是一名探鬼主播,決...
    沈念sama閱讀 40,999評(píng)論 3 422
  • 文/蒼蘭香墨 我猛地睜開眼筷厘,長(zhǎng)吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼鸣峭!你這毒婦竟也來了宏所?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,914評(píng)論 0 277
  • 序言:老撾萬榮一對(duì)情侶失蹤叽掘,失蹤者是張志新(化名)和其女友劉穎楣铁,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體更扁,經(jīng)...
    沈念sama閱讀 46,465評(píng)論 1 319
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡盖腕,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,543評(píng)論 3 342
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了浓镜。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片溃列。...
    茶點(diǎn)故事閱讀 40,675評(píng)論 1 353
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖膛薛,靈堂內(nèi)的尸體忽然破棺而出听隐,到底是詐尸還是另有隱情,我是刑警寧澤哄啄,帶...
    沈念sama閱讀 36,354評(píng)論 5 351
  • 正文 年R本政府宣布雅任,位于F島的核電站,受9級(jí)特大地震影響咨跌,放射性物質(zhì)發(fā)生泄漏沪么。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 42,029評(píng)論 3 335
  • 文/蒙蒙 一锌半、第九天 我趴在偏房一處隱蔽的房頂上張望禽车。 院中可真熱鬧,春花似錦刊殉、人聲如沸殉摔。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,514評(píng)論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽逸月。三九已至,卻和暖如春亚亲,著一層夾襖步出監(jiān)牢的瞬間彻采,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,616評(píng)論 1 274
  • 我被黑心中介騙來泰國打工捌归, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留肛响,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 49,091評(píng)論 3 378
  • 正文 我出身青樓惜索,卻偏偏與公主長(zhǎng)得像特笋,于是被迫代替她去往敵國和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,685評(píng)論 2 360

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