TLDR：視頻分割一直是重標(biāo)注的一個(gè) task，這篇 CVPR 2023 文章研究了完全不需要標(biāo)注的視頻物體分割。僅使用 ResNet，RCF模型在 DAVIS16/STv2/FBMS59 上提升了 7/9/5%。文章里還提出了不需要標(biāo)注的調(diào)參方法。代碼已公開(kāi)可用。

論文鏈接:

作者機(jī)構(gòu)：

分割效果視頻：

項(xiàng)目主頁(yè):

代碼鏈接:

視頻物體分割真的可以不需要人類監(jiān)督嗎？

視頻分割一直是重標(biāo)注的一個(gè) task，可是要標(biāo)出每一幀上的物體是非常耗時(shí)費(fèi)力的。然而人類可以輕松地分割移動(dòng)的物體，而不需要知道它們是什么類別。為什么呢？

Gestalt 定律嘗試解釋人類是怎么分割一個(gè)場(chǎng)景的，其中有一條定律叫做 Common Fate，即移動(dòng)速度相同的物體屬于同一類別。比如一個(gè)箱子從左邊被拖到右邊，箱子上的點(diǎn)是均勻運(yùn)動(dòng)的，人就會(huì)把這個(gè)部分給分割出來(lái)理解。然而人并不需要理解這是個(gè)箱子來(lái)做這個(gè)事情，而且就算是嬰兒之前沒(méi)有見(jiàn)過(guò)箱子也能知道這是一個(gè)物體。

運(yùn)用Common Fate來(lái)分割視頻

這個(gè)定律啟發(fā)了基于運(yùn)動(dòng)的無(wú)監(jiān)督分割。然而，Common Fate 并不是物體性質(zhì)的可靠指標(biāo)：關(guān)節(jié)可動(dòng)（articulated）/可變形物體（deformable objects）的一些 part 可能不以相同速度移動(dòng)，而物體的陰影/反射（shadows/reflections）始終隨物體移動(dòng)，但并非其組成部分。

舉個(gè)例子，下面這個(gè)人的腿和身子的運(yùn)動(dòng)是不同的（Optical Flow 可視化出來(lái)顏色不同）。這很常見(jiàn)，畢竟人有關(guān)節(jié)嘛（articulated），要是這個(gè)處理不了的話，很多視頻都不能分割了。然而很多 baseline 是處理不了這點(diǎn)的（例如 AMD+ 和 OCLR），他們把人分割成了幾個(gè)部分。

還有就是影子和反射，比如上面這只天鵝，它的倒影跟它的運(yùn)動(dòng)是一致的（Optical Flow 可視化顏色一樣），所以之前的方法認(rèn)為天鵝跟倒影是一個(gè)物體。很多視頻里是有這類現(xiàn)象的（畢竟大太陽(yáng)下物體都有個(gè)影子嘛），如果這個(gè)處理不了的話，很多視頻也不能分割了。

那怎么解決？放松。Relax.

長(zhǎng)話短說(shuō)，那我們的方法是怎么解決這個(gè)問(wèn)題的呢？無(wú)監(jiān)督學(xué)習(xí)的一個(gè)特性是利用神經(jīng)網(wǎng)絡(luò)自己內(nèi)部的泛化和擬合能力進(jìn)行學(xué)習(xí)。既然 Common Fate 有自己的問(wèn)題，那么我們沒(méi)有必要強(qiáng)制神經(jīng)網(wǎng)絡(luò)去擬合 Common Fate。于是我們提出了 Relaxed Common Fate，通過(guò)一個(gè)比較弱的學(xué)習(xí)方式讓神經(jīng)網(wǎng)絡(luò)真正學(xué)到物體的特性而不是 noise。

具體來(lái)說(shuō)，我們的方法認(rèn)為物體運(yùn)動(dòng)由兩部分組成：物體總體的 piecewise-constant motion （也就是 Common Fate）和物體內(nèi)部的 segment motion。比如你看下圖這個(gè)舞者，他全身的運(yùn)動(dòng)就可以被理解成 piecewise-constant motion 來(lái)建模，手部腿部這些運(yùn)動(dòng)就可以作為 residual motion 進(jìn)行擬合，最后合并成一個(gè)完整的 flow，跟 RAFT 生成的 flow 進(jìn)行比較來(lái)算 loss。我們用的 RAFT 是用合成數(shù)據(jù)（FlyingChairs 和 FlyingThings）進(jìn)行訓(xùn)練的，不需要人工標(biāo)注。

Relaxed Common Fate

這里的 Residual Flow 會(huì)盡量初始化得小一些，來(lái)鼓勵(lì)先學(xué) piecewise-constant 的部分（有點(diǎn)類似 ControlNet），再慢慢學(xué)習(xí) residual 部分。

引入Appearance信息來(lái)幫助無(wú)監(jiān)督視頻分割

光是 Relaxed Common Fate 就能在 DAVIS 上相對(duì) baseline 提 5%了，但這還不夠。前面說(shuō) Relaxed Common Fate 的只用了 motion 而沒(méi)有使用 appearance 信息。

讓我們?cè)俅位氐缴厦孢@個(gè)例子。這個(gè)舞者的手和身子是一個(gè)顏色，然而 AMD+ 直接把舞者的手忽略了。下面這只天鵝和倒影明明在 appearance 上差別這么大，卻在 motion 上沒(méi)什么差別。如果整合 appearance 和 motion，是不是能提升分割質(zhì)量呢？

因此我們引入了 Appearance 來(lái)進(jìn)行進(jìn)一步的監(jiān)督。在學(xué)習(xí)完 motion 信息之后，我們直接把取得的 Mask 進(jìn)行兩步優(yōu)化：一個(gè)是 low-level 的 CRF refinement，強(qiáng)調(diào)顏色等細(xì)節(jié)一致的地方應(yīng)該屬于同一個(gè) mask（或背景），一個(gè)是 semantic constraint，強(qiáng)調(diào) Unsupervised Feature 一直的地方應(yīng)該屬于同一個(gè) mask。

把優(yōu)化完的 mask 再和原 mask 進(jìn)行比較，計(jì)算 L2 Loss，再更新神經(jīng)網(wǎng)絡(luò)。這樣訓(xùn)練的模型的無(wú)監(jiān)督分割能力可以進(jìn)一步提升。具體細(xì)節(jié)歡迎閱讀原文。

無(wú)監(jiān)督調(diào)參

很多無(wú)監(jiān)督方法都需要使用有標(biāo)注的數(shù)據(jù)集來(lái)調(diào)參，而我們的方法提出可以利用前面說(shuō)的 motion 和 appearance 的一致性來(lái)進(jìn)行調(diào)參。簡(jiǎn)單地說(shuō)，motion 學(xué)習(xí)出的 mask 在 appearance 上不一致代表這個(gè)參數(shù)可能不是最優(yōu)的。具體方法是在 Unsupervised Feature 上計(jì)算 Normalized Cuts （但是不用算出最優(yōu)值），Normalized Cuts 越小越代表分割效果好。原文里面對(duì)此有詳細(xì)描述。

方法效果

無(wú)論是否有 Post-processing，我們的方法在三個(gè)視頻分割數(shù)據(jù)集上都有很大提升，在 STv2 上更是提升了 12%。

Ablation 可以看出 Residual pathway （Relaxed Common Fate）的貢獻(xiàn)是最大的，其他部分總計(jì)貢獻(xiàn)了 11.9% 的增長(zhǎng)。

Visualizations

總結(jié)

這篇 CVPR 2023 文章研究了完全不需要標(biāo)注的視頻物體分割。通過(guò) Relaxed Common Fate 來(lái)利用 motion 信息，再通過(guò)改進(jìn)和利用 appearance 信息來(lái)進(jìn)一步優(yōu)化，RCF 模型在 DAVIS16/STv2/FBMS59 上提升了 7/9/5%。文章里還提出了不需要標(biāo)注的調(diào)參方法。代碼和模型已公開(kāi)可用。