vix.ing · top · new · best · stats · spec

Deep AUC Maximization for Medical Image Classification: Challenges and Opportunities

2021/11/01 by Yang, Tianbao · 1 citation
#Artificial Intelligence (cs.AI) #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #FOS: Electrical engineering #FOS: Mathematics #Image and Video Processing (eess.IV) #Machine Learning (cs.LG) #Optimization and Control (math.OC) #electronic engineering #information engineering

paper · doi:10.48550/arxiv.2111.02400

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 Deep \underline\bf AUC \underline\bf Maximization 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.

Cited by

Related