Learning Non-Parametric Surrogate Losses with Correlated Gradients

Seungdong Yoa, Jinyoung Park, Hyunwoo J. Kim

Research output: Contribution to journalArticlepeer-review

Abstract

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 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.

Original languageEnglish
Pages (from-to)141199-141209
Number of pages11
JournalIEEE Access
Volume9
DOIs
Publication statusPublished - 2021

Bibliographical note

Funding Information:
This work was supported in part by the Institute for Information & Communications Technology Planning & Evaluation (IITP) grant funded by the Korea Government Ministry of Science and ICT (MSIT) (Regional Strategic Industry Convergence Security Core Talent Training Business) under Grant 2019-0-01343; in part by the ICT Creative Consilience Program under Grant IITP-2021-2020-0-01819; in part by the Korea Institute of Planning and Evaluation for Technology in Food, Agriculture and Forestry (IPET) and Korea Smart Farm Research and Development Foundation (KosFarm) through Smart Farm Innovation Technology Development Program, funded by the Ministry of Agriculture, Food and Rural Affairs (MAFRA) and MSIT, Rural Development Administration (RDA) under Grant 421025-04; and in part by Samsung Electronics Company Ltd.

Publisher Copyright:
© 2013 IEEE.

Keywords

  • Learning loss
  • computer vision
  • deep learning
  • machine learning

ASJC Scopus subject areas

  • General Computer Science
  • General Materials Science
  • General Engineering

Fingerprint

Dive into the research topics of 'Learning Non-Parametric Surrogate Losses with Correlated Gradients'. Together they form a unique fingerprint.

Cite this