English

Deep AUC Maximization for Medical Image Classification: Challenges and Opportunities

Machine Learning 2021-11-05 v1 Artificial Intelligence Computer Vision and Pattern Recognition Image and Video Processing Optimization and Control

Abstract

In this extended abstract, we will present and discuss opportunities and challenges brought about by a new deep learning method by AUC maximization (aka \underline{\bf D}eep \underline{\bf A}UC \underline{\bf M}aximization or {\bf DAM}) for medical image classification. Since AUC (aka area under ROC curve) is a standard performance measure for medical image classification, hence directly optimizing AUC could achieve a better performance for learning a deep neural network than minimizing a traditional loss function (e.g., cross-entropy loss). Recently, there emerges a trend of using deep AUC maximization for large-scale medical image classification. In this paper, we will discuss these recent results by highlighting (i) the advancements brought by stochastic non-convex optimization algorithms for DAM; (ii) the promising results on various medical image classification problems. Then, we will discuss challenges and opportunities of DAM for medical image classification from three perspectives, feature learning, large-scale optimization, and learning trustworthy AI models.

Keywords

Cite

@article{arxiv.2111.02400,
  title  = {Deep AUC Maximization for Medical Image Classification: Challenges and Opportunities},
  author = {Tianbao Yang},
  journal= {arXiv preprint arXiv:2111.02400},
  year   = {2021}
}

Comments

Medical Imaging meets NeurIPS 2021 workshop

R2 v1 2026-06-24T07:24:54.494Z