核心概念
提出一種新的邊際去偏網路(MDN)來學習去偏表示,通過引入邊際懲罰的概念來減少目標類別和偏差屬性之間的虛假相關,從而提高在無偏測試標準上的性能。
要約
本文提出了一種新的邊際去偏網路(MDN)來解決視覺識別中的偏差問題。具體來說:
-
設計了一種邊際softmax損失(MSL),通過為偏差相反樣本(少數樣本)和偏差一致樣本(多數樣本)分別設置不同的邊際,來減少虛假相關的負面影響,並提高在無偏測試集上的泛化能力。
-
提出了一種元學習框架,通過在元驗證集上優化元等化損失(MEL)來自適應地學習最優邊際參數,使模型在兼顧準確性的同時實現公平性。
-
在BiasedMNIST、Corrupted CIFAR-10、CelebA和UTK-Face數據集上進行了大量實驗,結果表明MDN在不同偏差情況下都能取得顯著的去偏效果,優於現有方法。
統計
在BiasedMNIST數據集上,當目標類別和偏差屬性的相關性為0.999時,MDN的無偏準確率達到74.3%,而傳統方法僅為11.8%。
在Corrupted CIFAR-10數據集上,當偏差相反樣本佔5%時,MDN的無偏準確率達到57.8%,而現有最佳方法僅為55.1%。
在CelebA數據集上,當以"性別"作為偏差屬性時,MDN的無偏準確率、最差組準確率和等化奇異度分別為78.01%、71.35%和6.97,優於其他方法。
在UTK-Face數據集上,當以"年齡"作為偏差屬性時,MDN的無偏準確率、最差組準確率和等化奇異度分別為81.98%、72.60%和13.60,顯著優於現有方法。
引用
"我們提出了一種新的邊際去偏網路(MDN)來學習去偏表示,通過引入邊際懲罰的概念來減少目標類別和偏差屬性之間的虛假相關,從而提高在無偏測試標準上的性能。"
"我們設計了一種邊際softmax損失(MSL),通過為偏差相反樣本(少數樣本)和偏差一致樣本(多數樣本)分別設置不同的邊際,來減少虛假相關的負面影響,並提高在無偏測試集上的泛化能力。"
"我們提出了一種元學習框架,通過在元驗證集上優化元等化損失(MEL)來自適應地學習最優邊際參數,使模型在兼顧準確性的同時實現公平性。"