基於度量學習的行人重識別方法中損失函數總結! |
文章目錄
- 本文主要是參考的文章:基於深度學習的行人重識別研究進展!
一、對比損失(Contrasive loss)
- 對比損失(
Contrasive loss
)用於訓練孿生網絡(Siamese network
)。孿生網絡的輸入爲一對(兩張)圖片 和 ,這兩張圖片可以爲同一行人,也可以爲不同行人。 每一對訓練圖片都有一個標籤 , 其中 表示兩張圖片屬於同一個行人(正樣本對),反之 表示它們屬於不同行人(負樣本對)。之後, 對比損失函數寫作:
- 其中: 是根據實際需求設置的訓練閾值參數, 代表兩個樣本對應的特徵的歐氏距離。
二、三元組損失(Triplet loss)
- 三元組損失 是一種被廣泛應用的度量學習損失,之後的大量度量學習方法也是基於三元組損失演變而來。顧名思義,三元組損失需要三張輸入圖片。和對比損失不同,一個輸入的三元組(Triplet) 包括一對正樣本對和一對負樣本對。三張圖片分別命名爲固定圖片 ,正樣本圖片 和負樣本圖片 。 圖片 和圖片 爲一對正樣本對,圖片 和圖片 爲一對負樣本對。 則三元組損失表示爲:
- 其中: 是根據實際需求設置的訓練閾值參數, 代表兩個樣本對應的特徵的歐氏距離。
triplet loss
的目標是:①. 兩個具有同樣標籤的樣本,它們在新的編碼空間裏距離很近。②. 兩個具有不同標籤的樣本,它們在新的編碼空間裏距離很遠。進一步,我們希望兩個positive examples
和一個negative example
中,negative example
與positive example
的距離,大於positive examples
之間的距離,或者大於某一個閾值:
理解:
- 優化的目的就是使
loss
在訓練迭代中下降的越小越好,也就是要使得Anchor
與Positive
越接近越好,Anchor
與Negative
越遠越好。基於上面這些,分析一下margin
值的取值。 - 首先:當
margin
值越小時,loss
也就較容易的趨近於 0,於是Anchor
與Positive
都不需要拉的太近,Anchor
與Negative
不需要拉的太遠,就能使得loss
很快的趨近於0。這樣訓練得到的結果,不能夠很好的區分相似的圖像。 - 其次:當
margin
越大時,就需要使得網絡參數要拼命地拉近Anchor、Positive
之間的距離,拉遠Anchor、Negative
之間的距離。如果margin
值設置的太大,很可能最後loss
保持一個較大的值,難以趨近於0 。 - 因此,設置一個合理的
margin
值很關鍵,這是衡量相似度的重要指標。簡而言之,margin
值設置的越小,loss
很容易趨近於0 ,但很難區分相似的圖像。margin
值設置的越大,loss
值較難趨近於0,甚至導致網絡不收斂,但可以較有把握的區分較爲相似的圖像。
三、改進三元組損失(Improved triplet loss)
- 文獻 認爲公式 只考慮正負樣本對之間的相對距離,而並沒有考慮正樣本對之間的絕對距離(存在下圖的情況,可能最後優化結束,正樣本對的絕對距離仍然很大!),爲此提出改進三元組損失(Improved triplet loss):
- 其中: 是根據實際需求設置的訓練閾值參數, 代表兩個樣本對應的特徵的歐氏距離。公式 添加 項,保證網絡不僅能夠在特徵空間把正負樣本推開,也能保證正樣本對之間的距離很近。
四、難樣本採樣三元組損失(Triplet loss with batch hard mining, TriHard loss)
- 難樣採樣三元組損失(
TriHard
損失)是三元組損失的改進版。傳統的三元組隨機從訓練數據中抽樣三張圖片,這樣的做法雖然比較簡單,但是抽樣出來的大部分都是簡單易區分的樣本對。如果大量訓練的樣本對都是簡單的樣本對,那麼這是不利於網絡學習到更好的表徵。大量論文發現用更難的樣本去訓練網絡能夠提高網絡的泛化能力,而採樣難樣本對的方法很多。有學者提出了一種基於訓練批量(Batch
)的在線難樣本採樣方法——TriHard Loss。 - 損失的核心思想是:對於每一個訓練
batch
,隨機挑選P
個ID的行人,每個行人隨機挑選K
張不同的圖片,即一個batch
含有P×K
張圖片。之後對於batch中的每一張圖片a
,我們可以挑選一個最難的正樣本和一個最難的負樣本和a
組成一個三元組。 - 首先我們定義 和爲相同ID的圖片集爲 ,剩下不同ID的圖片圖片集爲 ,則
TriHard
損失表示爲:
- 其中: 是人爲設定的閾值參數,TriHard損失會計算 和
batch
中的每一張圖片在特徵空間的歐式距離,然後選出與 距離最遠(最不像)的正樣本 和距離最近(最像)的負樣本 來計算三元組損失。 通常TriHard損失效果比傳統的三元組損失要好。
五、四元組損失(Quadruplet loss)
- 四元組損失 是三元組損失的另一個改進版本。顧名思義,四元組(Quadruplet) 需要四張輸入圖片,和三元組不同的是多了一張負樣本圖片。即四張圖片爲固定圖片 , 正樣本圖片 , 負樣本圖片 和負樣本圖片 。其中 和 是兩張不同行人ID的圖片。則四元組損失表示爲:
- 其中: 代表兩個樣本對應的特徵的歐氏距離, 是手動設置的正常數,通常設置 小於 ,前一項稱爲強推動, 後一項稱爲弱推動。其中前一項和三元組損失一樣,只考慮正負樣本間的相對距離,共享了固定圖片 。因此在推開負樣本對 和 的同時,也會直接影響 的特徵,造成正樣本對 和 的距離不好控制。改進三元組損失通過直接約束 和 之間的距離來解決這個問題。而四元組通過引入第二項弱推動實現,添加的第二項中負樣本對和正樣本對不共享ID, 所以考慮的是正負樣本間的絕對距離,在推開負樣本對的同時不會太過直接影響 的特徵。因此,四元組損失通常能讓模型學習到更好的表徵。
- 以上度量學習方法樣本示例如上圖所示,這些方法在計算度量損失時,樣本對都是從訓練集中隨機挑選。隨機挑選樣本對的方法可能經常挑選出一些容易識別的樣本對組成訓練批量(Batch),使得網絡泛化能力受限。爲此, 部分學者提出了難樣本採樣(Hard sample mining) 的方法,來挑選出難樣本對訓練網絡。常用的思路是挑選出一個訓練Batch 中特徵向量距離比較大(非常不像) 的正樣本對和特徵向量距離比較小(非常像) 的負樣本對來訓練網絡。難樣本採樣技術可以明顯改進度量學習方法的性能,加快網絡的收斂,並且可以很方便地在原有度量學習方法上進行擴展,是目前廣泛採用的一種技術。
參考文獻
-
這裏主要摘錄了浙大羅浩博士(大神)的文章:基於深度學習的行人重識別研究進展!
-
Person re-identification by multi-channel parts-based CNN with improved triplet loss function_2016
-
Siamese convolutional neural network architecture for human re-identification_2016
-
A Siamese long short-term memory architecture for human re-identification_2016
-
Person re-identification with cascaded pairwise convolutions_2018
-
Person re-identification by multi-channel parts-based CNN with improved triplet loss function_2016
-
In defense of the triplet loss for person re-identification_2017
-
End-to-end comparative attention networks for person re-identification_2017
-
Features for multi-target multi-camera tracking and re-identification_2018
-
Beyond triplet loss: a deep quadruplet network for person re-identification_2017