常見的損失函數

在深度學習中，損失函數扮演着至關重要的角色。通過對最小化損失函數，使模型達到收斂狀態，減少模型預測值的誤差。因此，不同的損失函數，對模型的影響是重大的。接下來，總結一下，在工作中經常用到的損失函數：

• 圖像分類：交叉熵
• 目標檢測：Focal loss，L1/L2損失函數，IOU Loss，GIOU ，DIOU，CIOU
• 圖像識別：Triplet Loss，Center Loss，Sphereface，Cosface，Arcface

---

圖像分類

交叉熵

在圖像分類中，經常使用softmax+交叉熵作為損失函數，具體的推導可以參考我以前的博客。

 

CrossEntropy=−∑i=1np(xi)ln(q(xi))CrossEntropy=−∑i=1np(xi)ln(q(xi))

 

其中，p(x)p(x)表示真實概率分布，q(x)q(x)表示預測概率分布。交叉熵損失函數通過縮小兩個概率分布的差異，來使預測概率分布盡可能達到真實概率分布。

后來，谷歌在交叉熵的基礎上，提出了label smoothing（標簽平滑），具體介紹，可以參考這篇博客。

在實際中，需要用預測概率去擬合真實概率，而擬合one-hot的真實概率函數會帶來兩個問題：

1. 無法保證模型的泛化能力，容易造成過擬合；
2. 全概率和0概率鼓勵所屬類別和其他類別之間的差距盡可能加大，而由梯度有界可知，這種情況很難adapt，會造成模型過於相信預測的類別。

因此，為了減少這種過於自信，同時減緩人為標注的誤差帶來的影響，需要對p(x)p(x)進行變化：

 

p′(x)=(1−ϵ)δ(k,y)+ϵu(k)p′(x)=(1−ϵ)δ(k,y)+ϵu(k)

 

其中，δ(k,y)δ(k,y)為Dirac函數，u(k)u(k)為均勻分布。簡單而言，降低標簽y的置信度，提高其余類別的置信度。從而，交叉熵變成了：

 

H(p′,q)=−∑i=1np′(xi)ln(q(xi))=(1−ϵ)H(p,q)+ϵH(p,u)H(p′,q)=−∑i=1np′(xi)ln(q(xi))=(1−ϵ)H(p,q)+ϵH(p,u)

 

 

---

目標檢測

最近，看到一篇很好的博文，是介紹目標檢測中的損失函數的，可以參考一下：https://mp.weixin.qq.com/s/ZbryNlV3EnODofKs2d01RA

在目標檢測中，損失函數一般由兩部分組成，classification loss和bounding box regression loss。calssification loss的目的是使類別分類盡可能正確；bounding box regression loss的目的是使預測框盡可能與GT框匹對上。

Focal loss

該Focal loss損失函數出自於論文《Focal Loss for Dense Object Detection》，主要是解決正負樣本之間的不平衡問題。通過降低easy example中的損失值，間接提高了hard example中損失值的權重。Focal loss是基於交叉熵進行改進的：

 

Focalloss=−αt(1−pt)γlog(pt)Focalloss=−αt(1−pt)γlog(pt)

 

可以看到，在交叉熵前增加了(1−pt)γ(1−pt)γ，當圖片被錯分時，ptpt會很小，(1−pt)(1−pt)接近於1，所以損失受到的影響不大；而增加參數γγ是為了平滑降低esay example的權重。當γ=0γ=0時，Focal loss退化成交叉熵。對於不同的γγ，其影響如下圖所示。

 

 L1，L2，smooth L1損失函數

利用L1,L2或者smooth L1損失函數，來對4個坐標值進行回歸。smooth L1損失函數是在Fast R-CNN中提出的。三個損失函數，如下所示：

 

L1=|x|L1=|x|

 

 

L2=x2L2=x2

 

 

smoothL1={0.5x2if|x|<1|x|−0.5otherwisesmoothL1={0.5x2if|x|<1|x|−0.5otherwise

 

從損失函數對x的導數可知：L1L1損失函數對x的導數為常數，在訓練后期，x很小時，如果learning rate 不變，損失函數會在穩定值附近波動，很難收斂到更高的精度。L2L2損失函數對x的導數在x值很大時，其導數也非常大，在訓練初期不穩定。smooth L1完美的避開了L1L1和L2L2損失的缺點。

 在一般的目標檢測中，通常是計算4個坐標值與GT框之間的差異，然后將這4個loss進行相加，構成regression loss。

但使用上述的3個損失函數，會存在以下的不足：

• 上面的三種Loss用於計算目標檢測的Bounding Box Loss時，獨立的求出4個點的Loss，然后進行相加得到最終的Bounding Box Loss，這種做法的假設是4個點是相互獨立的，實際是有一定相關性的；
• 實際評價框檢測的指標是使用IOU，這兩者是不等價的，多個檢測框可能有相同大小的Loss，但IOU可能差異很大，為了解決這個問題就引入了IOU LOSS

 IOU Loss

該IOU Loss是曠視在2016年提出的《UnitBox: An Advanced Object Detection Network》。該論文的主要觀點之一是：

• 使用基於歐式距離的L-n損失函數，其前提是假設4個坐標變量都是獨立的，但實際上，這些坐標變量是具有一定的關聯性。
• 評價指標使用了IOU，而回歸坐標框又使用4個坐標變量，這兩者是不等價的。
• 具有相同的歐式距離的框，其IOU值卻不是唯一的。

所以，提出了IOU loss，直接使用IOU作為損失函數：

 

LossIOU=−ln(IOU)LossIOU=−ln(IOU)

 

同時，也會有人使用的是：

 

LossIOU=1−IOULossIOU=1−IOU

 

GIOU

該GIOU Loss損失函數是斯坦福於2019年提出的《Generalized Intersection over Union: A Metric and A Loss for Bounding Box Regression》。在上面的IOU Loss中，無法對兩個不重疊的框進行優化，而且IOU Loss無法反映出兩個框到底距離有多遠。為了解決這個問題，作者提了GIOU來作為損失函數：

 

GIOU=IOU−C−(A⋃B)CGIOU=IOU−C−(A⋃B)C

 

其中，CC表示兩個框的最小外接矩陣面積。即先求出兩個框的IOU，然后求出外接矩陣C的面積，減去A與B的面積。最終得到GIOU的值。

 

GIOU具有以下的性質：

• GIOU可以作為一種衡量距離的方式，LossGIOU=1−GIOULossGIOU=1−GIOU
• GIOU具有尺度不變性
• GIOU是IOU的下限，GIOU(A,B)≤IOU(A,B)GIOU(A,B)≤IOU(A,B)。當矩形框A、B類似時，limA→BGIOU(A,B)=IOU(A,B)limA→BGIOU(A,B)=IOU(A,B)
• 當矩形框A、B重疊時，GIOU(A,B)=IOU(A,B)GIOU(A,B)=IOU(A,B)
• 當矩形框A、B不相交時，GIOU(A,B)=−1GIOU(A,B)=−1

總的來說，GIOU包含了IOU所有的優點，同時克服了IOU的不足。

DIOU和CIOU

DIOU和CIOU是天津大學於2019年提出的《Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression》。為了解決GIOU收斂慢和提高回歸精度，提出DIOU來加速收斂。同時考慮到框回歸的3個幾何因素（重疊區域，中心點距離，寬高比），基於DIOU，再次提出CIOU，進一步提高收斂速度和回歸精度。另外，可以將DIOU結合NMS組成DIOU-NMS，來對預測框進行后處理。

當出現下圖的情況（GT框完全包含預測框）時，IOU與GIOU的值相同，此時GIOU會退化成IOU，無法區分其相對位置關系。同時由於嚴重依賴於IOU項，GIOU會致使收斂慢。

 基於上述問題，作者提出兩個問題

 

1. 直接最小化預測框與目標框之間的歸一化距離是否可行，以達到更快的收斂速度。
2. 如何使回歸在與目標框有重疊甚至包含時更准確、更快

好的目標框回歸損失應該考慮三個重要的幾何因素：重疊面積，中心點距離，長寬比。基於問題一，作者提出了DIoU Loss，相對於GIoU Loss收斂速度更快，該Loss考慮了重疊面積和中心點距離，但沒有考慮到長寬比；針對問題二，作者提出了CIoU Loss，其收斂的精度更高，以上三個因素都考慮到了。

 

首先，定義一下基於IOU Loss的損失函數：

 

Loss=1−IOU+R(B,Bgt)Loss=1−IOU+R(B,Bgt)

 

其中，R(B,Bgt)R(B,Bgt)表示預測框與GT框的懲罰項。在IOU Loss中，R(B,Bgt)=0R(B,Bgt)=0；在GIOU中，R(B,Bgt)=C−A⋃BCR(B,Bgt)=C−A⋃BC。

而在DIOU中，該懲罰項R(B,BGT)=ρ2(b,bgt)c2R(B,BGT)=ρ2(b,bgt)c2，其中bb和bgtbgt表示預測框與GT框的中心點，ρ()ρ()表示歐式距離，cc表示預測框BB與GT框BgtBgt的最小外接矩陣的對角線距離，如下圖所示。

因此，LossDIOULossDIOU定義為：

 

LossGIOU=1−IOU+ρ2(b,bgt)c2LossGIOU=1−IOU+ρ2(b,bgt)c2

 

所以，LossDIOULossDIOU具有如下性質：

1. DIOU依然具有尺度不變性；
2. DIOU直接最小化兩個框的距離，因此收斂會更快；
3. 對於目標框包裹預測框的這種情況，DIoU Loss可以收斂的很快，而GIoU Loss此時退化為IoU Loss收斂速度較慢

DIOU同時考慮了重疊面積和中心點之間的距離，但是沒有考慮到寬高比。進一步提出了CIOU，同時考慮了這3個因素，在DIOU的懲罰項中加入了αυαυ：

 

R(B,Bgt)=RCIOU=ρ2(b,bgt)c2+αυR(B,Bgt)=RCIOU=ρ2(b,bgt)c2+αυ

 

其中，αα表示trade-off參數，υυ表示寬高比一致性參數。

 

υ=4π2(arctanwgthgt−arctanwh)2υ=4π2(arctanwgthgt−arctanwh)2

 

 

α=υ(1−IOU)+υα=υ(1−IOU)+υ

 

這里的υυ為什么會有4π24π2呢？這里arctanarctan的范圍是[0,π2)[0,π2)。

所以，CIOU的損失函數為：

 

LossCIOU=1−IOU+ρ2(b,bgt)c2+αυLossCIOU=1−IOU+ρ2(b,bgt)c2+αυ

 

而在實際操作中，w2+h2w2+h2是很小的數字，在后向傳播時，容易造成梯度爆炸。通常用1來代替w2+h2w2+h2。

另外，提醒一點的是，GIOU、CIOU、DIOU都是衡量方式，在使用時可以代替IOU。但是這里需要考慮的一個問題是，預測框與GT框的匹配規則問題。並不是說anchor一定會去匹配一個不重疊的GT框。類似於SSD中所說，anchor會選擇一個重疊最大的GT框進行預測，而這個重疊最大可以使用IOU、GIOU、CIOU、DIOU來進行衡量。

 

---

圖像識別

圖像識別問題，包含了行人重識別，人臉識別等問題。此類損失都是通用的，因此放在一起匯總。同樣，也看到一篇很好的博客介紹了大量人臉識別的損失函數：https://mp.weixin.qq.com/s/wJ-JNsUv60vXtGIV-mDrTA

Triplet Loss

該Triplet Loss損失函數提出於2015年的《FaceNet: A Unified Embedding for Face Recognition and Clustering》。該損失函數的主要想法是，拉近同一id之間的距離，擴大不同id之間的距離。如下圖所示，圖中的anchor與positive屬於同一id，即yanchor=ypositiveyanchor=ypositive；而anchor與negative屬於不同的id，即yanchor≠ynegativeyanchor≠ynegative。通過不斷學習后，使得anchor與positive的歐式距離變小，anchor與negative的歐式距離變大。其中，這里的anchor、Positive、negative是圖片的d維嵌入向量（embedding）。

 使用數學公式進行表達，triplet loss想達到的效果是：

 

d(xai,xpi)+α≤d(xai,xni)d(xia,xip)+α≤d(xia,xin)

 

其中，d()d()表示兩個向量之間的歐氏距離，αα表示兩個向量之間的margin，防止d(xai,xpi)=d(xai,xni)=0d(xia,xip)=d(xia,xin)=0。因此，可以最小化triplet loss損失函數來達到此目的：

 

tripletloss=[d(xai,xpi)−d(xai,xni)+α]+tripletloss=[d(xia,xip)−d(xia,xin)+α]+

 

在實際中，通常使用在線訓練方式，選擇P的不同的id，每個id包含K張圖片，形成了batchsize=PKbatchsize=PK的mini-batch。從而在這mini-batch種選擇hard/easy example構成loss，具體可以參考這篇博客。

Center Loss

該Center Loss損失函數提出於《A Discriminative Feature Learning Approach for Deep Face Recognition》。為了提高特征的區分能力，作者提出了center loss損失函數，不僅能縮小類內差異，而且能擴大類間差異。

作者首先在MNIST數據集上進行試驗，將隱藏層的最后輸出維度改為2，使用softmax+交叉熵作為損失函數，將其結果可視化出來，如下圖所示。可以看出，交叉熵可以使每一類分開，數據分布呈射線形，但卻不夠區分性，即類內差異大。

 

 因此，作者想要在保持數據的可分性前提下，進一步縮小類內之間的差異。為了達到這個目的，提出了center loss損失函數：

 

LC=12∑i=1m∥∥xi−cyi∥∥22LC=12∑i=1m‖xi−cyi‖22

 

其中，cyicyi表示第yiyi類的中心。因此，通常將center loss和交叉熵進行結合，構成組合損失函數：

 

L=LS+λLC=−∑i=1mlogeWTyixi+byi∑nj=1eWTjxj+byi+λ2∑i=1m∥∥xi−cyi∥∥22L=LS+λLC=−∑i=1mlogeWyiTxi+byi∑j=1neWjTxj+byi+λ2∑i=1m‖xi−cyi‖22

 

其中，λλ表示center loss的懲罰力度。同樣在MNIST中，其結果如下圖所示。可以看到隨着λλ的增加，約束性更強，每一類會更聚集在類內中心處。

 

在使用Center Loss損失函數時，需要引入兩個超參：αα和λλ。其中，λλ表示center loss的懲罰力度；而αα控制類內中心點cyicyi的學習率。類內中心點cyicyi應該隨着特征的不同，會產生變化。一般會在每個mini-batch中更新類內中心點cyicyi：

 

ct+1j=ctj−αΔctjcjt+1=cjt−αΔcjt

 

Sphereface

該Sphereface提出於《SphereFace: Deep Hypersphere Embedding for Face Recognition》，其也稱A-Softmax損失函數。作者認為，triplet loss需要精心構建三元組，不夠靈活；center loss損失函數只是強調了類內的聚合度，對類間的可分性不夠重視。因此，作者提出了疑問：基於歐式距離的損失函數是否適合模型學習到具有區分性的特征呢？

首先，重新看一下softmax loss損失函數（即softmax+交叉熵）：

 

Lossi=−log⎛⎝eWTyixi+byi∑jeWTjxi+bj⎞⎠=−log(e||Wyi||||xi||cos(θyi,i)+byi∑je||Wj||||xi||cos(θj,i)+bj)Lossi=−log(eWyiTxi+byi∑jeWjTxi+bj)=−log(e||Wyi||||xi||cos(θyi,i)+byi∑je||Wj||||xi||cos(θj,i)+bj)

 

其中，θj,i(0≤θj,i≤π)θj,i(0≤θj,i≤π)表示向量WjWj和xixi的夾角。可以看到，損失函數與||Wj||||Wj||，θj,iθj,i和bjbj有關，令||Wj||=1||Wj||=1和bj=0bj=0，則可以得到modified-softmax損失函數，更加關注角度信息：

 

Lmodified−softmax=−log(e||xi||cos(θyi,i)∑je||xi||cos(θj,i))Lmodified−softmax=−log(e||xi||cos(θyi,i)∑je||xi||cos(θj,i))

 

雖然使用modified-softmax損失函數可以學習到特征具有角度區分性，但這個區分力度仍然不夠大。因此，在θj,iθj,i上乘以一個大於1的整數，來提高區分度：

 

Lang=−log(e||xi||cos(mθyi,i)e||xi||cos(mθj,i)+∑j≠yie||xi||cos(θj,i))Lang=−log(e||xi||cos(mθyi,i)e||xi||cos(mθj,i)+∑j≠yie||xi||cos(θj,i))

 

這樣，能擴大類間距離，縮小類內距離。

下圖是論文的實驗結果，從超球面的角度進行解釋，不同的m值的結果。其中，不同顏色的點表示不同的類別。可以看出，使用A-Softmax損失函數，會將學習到的向量特征映射到超球面上，m=1m=1表示退化成modified-softmax損失函數，可以看出，每個類別雖然有明顯的分布，但區分性不夠明顯。隨着mm的增大，區分性會越來越大，但也越來越難訓練。

 

最后，給出該損失函數的實現方式，請參考這篇博客。

Cosface

該Cosface損失函數是由騰訊在2018年《CosFace: Large Margin Cosine Loss for Deep Face Recognition》中提出的。Cosface損失函數，也稱Large Margin Cosine Loss(LMCL)。從名字可以看出，通過對cos的間隔最大化，來實現擴大類間距離，縮小類內距離。

從softmax出發（與Sphereface類似），作者發現，為了實現有效的特征學習，||Wj=1||||Wj=1||是十分有必要的，即對權重進行歸一化。同時，在測試階段，測試用的人臉對的得分通常是根據兩個特征向量之間的余弦相似度進行計算的。這表明，||x||||x||對得分計算影響不大，因此，在訓練階段將||x||=s||x||=s固定下來（在本論文中，s=64s=64）：

 

Lns=1N∑i−logescos(θyi,i)∑jescos(θj,i)Lns=1N∑i−logescos(θyi,i)∑jescos(θj,i)

 

其中ns表示歸一化版本的softmax loss，θj,iθj,i表示WjWj和xx之間的角度。為了加大區分性，類似Sphereface一樣，引入常數m：

 

Llmc=1N∑i−loges(cos(θyi,i)−m)es(cos(θyi,i)−m)+∑j≠yiescos(θj,i)Llmc=1N∑i−loges(cos(θyi,i)−m)es(cos(θyi,i)−m)+∑j≠yiescos(θj,i)

 

其中，W=W∗||W∗||,x=x∗||x∗||,cos(θj,i)=WTjxiW=W∗||W∗||,x=x∗||x∗||,cos(θj,i)=WjTxi。

下圖是作者的解釋圖。第一個表示正常的sotfmax loss，可以看出兩個類別的分類邊界具有重疊性，即區分性不強；第二個表示歸一化版本的softmax loss，此時邊界已經很明顯，相互沒有重疊，但是區分性不足；第三個表示A-softmax，此時橫縱坐標變成了θθ，從這個角度解釋，使用兩條線作為區分邊界，作者也提出，該損失函數的缺點是不連續；第四個表示Cosface，在cos(θ)cos(θ)下，使用兩條線作為區分邊界，特征之間沒有交集，隨着m值增大，區分性也會越來越明顯，但訓練難度會加大。

 

 Arcface

該Arcface損失函數提出於《ArcFace: Additive Angular Margin Loss for Deep Face Recognition》。類似於Sphereface和Cosface，Arcface同樣需要令||W||=1,||x||=s||W||=1,||x||=s，同時也引入常數m，但與前面兩者不同的是，這里的m是對θθ進行修改：

 

Larcface=1N∑i−loges(cos(θyi,i+m))es(cos(θyi,i+m))+∑j≠yiescos(θj,i)Larcface=1N∑i−loges(cos(θyi,i+m))es(cos(θyi,i+m))+∑j≠yiescos(θj,i)

 

下圖是Arcface的計算流程圖，首先對xx與WW進行標准化，然后進行相乘得到cos(θj,i)cos(θj,i)，通過arccos(cos(θj,i))arccos(cos(θj,i))來得到角度θj,iθj,i，加上常數mm來加大間距得到θj,i+mθj,i+m，之后計算cos(θj,i+m)cos(θj,i+m)並乘上常數ss，最后進行常規的softmax loss就行。

 

通過對Sphereface、Cosface和Arcface進行整合，得到了統一的形式：

 

L=1N∑i−loges(cos(m1θyi,i+m2)−m3)es(cos(m1θyi,i+m2)−m3)+∑j≠yiescos(θj,i)L=1N∑i−loges(cos(m1θyi,i+m2)−m3)es(cos(m1θyi,i+m2)−m3)+∑j≠yiescos(θj,i)

 

此時，就可以對該損失函數進行魔改了，作者實驗得到，對於部分數據集，m1=1,m2=0.3,m3=0.2m1=1,m2=0.3,m3=0.2和m1=0.9,m2=0.4,m3=0.15m1=0.9,m2=0.4,m3=0.15的效果較好。