Chexpert icon indicating copy to clipboard operation
Chexpert copied to clipboard

Training the network gives high AUC but low ACC

Open erikbabu opened this issue 4 years ago • 4 comments

Thanks for open-sourcing this solution. I am using it as the backbone to evaluate a Federated Learning approach to medical imaging for my final year individual undergraduate project.

I trained the model (not using the pre-saved weights) on a 20% sample of the data, which follows the same distribution as the overall training set, and although the AUC scores are quite good, the ACC scores are quite low for some of the observations. I was wondering if you had some insight on what might be causing this. What Acc scores were you getting for your best model?

Atelectasis - AUCROC: 0.871, Acc: 0.375 Cardiomegaly - AUCROC: 0.834, Acc: 0.670 Consolidation - AUCROC: 0.908, Acc: 0.860 Edema - AUCROC: 0.888, Acc: 0.790 Pleural Effusion - AUCROC: 0.901, Acc: 0.335

Also, I am using the CheXpert downsampled dataset (11GB). The config file I am using is similar to the provided one except I have had to reduce the batch size to 8 due to GPU memory constraints.

Thanks!

erikbabu avatar Apr 30 '20 14:04 erikbabu

Hey the threshold used to calculate accuracy is 0.5 you can change the threshold to reach desired accuracy. You should focus on AUC and not accuracy as accuracy changes with threshold but AUC NOT

aakash-saboo avatar Apr 30 '20 15:04 aakash-saboo

@yww211 @deadpoppy I guess we could also consider open sourcing our calibration module that exactly deals with the accuracy/threshold issue lol

yil8 avatar May 01 '20 02:05 yil8

Wow! It must be the most powerful tool for this problem. @yil8

deadpoppy avatar May 06 '20 03:05 deadpoppy

@yww211 @deadpoppy I guess we could also consider open sourcing our calibration module that exactly deals with the accuracy/threshold issue lol

Hi man, i'm facing the same problem now, higher AUC with lower ACC. I've tried to calculate the cutoff of ROC as the optimal threshold , but that didn't work. Could you share your secret about the calibration module you mentioned? Thanks a lot, best wishes

Danie1Hayes avatar Dec 10 '20 07:12 Danie1Hayes