Learning Non-parametric Surrogate Losses with Correlated Gradients

Seungdong Yoa, Jinyoung Park, Hyunwoo J. Kim
2021 IEEE Access  
Training models by minimizing surrogate loss functions with gradient-based algorithms is a standard approach in various vision tasks. This strategy often leads to suboptimal solutions due to the gap between the target evaluation metrics and surrogate loss functions. In this paper, we propose a framework to learn a surrogate loss function that approximates the evaluation metric with correlated gradients. We observe that the correlated gradients significantly benefit the gradient-based algorithms
more » ... to improve the quality of solutions. We verify the effectiveness of our method in various tasks such as multi-class classification, ordinal regression, and pose estimation with three evaluation metrics and five datasets. Our extensive experiments showed that our method outperforms conventional loss functions and surrogate loss learning methods.
doi:10.1109/access.2021.3120092 fatcat:q6eoekgo2rhl7gg37ism7i3smu