-
Notifications
You must be signed in to change notification settings - Fork 38
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Pre-trained weights #66
Comments
What task are you referring to? |
Hi, we are interested in the Deep AUC Maximization (DAM) model from the paper "Large-scale Robust Deep AUC Maximization: A New Surrogate Loss and Empirical Studies on Medical Image Classification". The model achieves 0.93 AUC on CheXpert. |
Thank you for your interest! The result is actually an ensemble of several models trained using deep AUC maximization. You can follow the tutorial here https://github.com/Optimization-AI/LibAUC/blob/main/examples/05_Optimizing_AUROC_Loss_with_DenseNet121_on_CheXpert.ipynb to train a model. You can use different losses/optimizers available in our library to train different models and get an ensemble. The original models for achieving 0.93 AUC on CheXpert were not available anymore. |
Hi all,
Is it possible to release your best weights? Our team will use them for research rather than commercial purposes, and will appropriately cite your work in the paper.
The text was updated successfully, but these errors were encountered: