CenterLoss在Mnist數據集上的實現

centerloss,顧名思義,中心損失函數,它的原理主要是在softmax loss的基礎上,通過對訓練集的每個類別在特徵空間分別維護一個類中心,在訓練過程,增加樣本經過網絡映射後在特徵空間與類中心的距離約束,從而兼顧了類內聚合與類間分離,centerloss只是一個輔助損失函數,softmaxloss纔是主打,但softmaxloss只能簡單的將類分開,還得加上centerloss這一個強力輔
相關文章
相關標籤/搜索