Proxy for AUC When Training Neural Network?

What metrics is a good proxy for training deep learning model with an objective to maximize AUC? Is cross_entropy the best choice?

We can use roc_auc_score itself from sklearn.metrics

Don’t use accuracy…(that’s all I know…)

Have a look at this link

I know about sklearn but we cannot use auc as a training metric right? It’s not differentiable and how would we backprop it.

Didn’t see that you will use deep learning

1 Like

I don’t know if it’s appropriate in all situations, but you can try the Dice loss. This loss function is 1 - Dice coefficient, and the Dice coefficient is also known as the F1-score, which combines precision and recall into one number. It’s not exactly the same as the AUC but both try to maximize precision and recall.