Skip to content

Pre-trained weights #66

Open
Open

Description

@LiGuo12

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.

Activity

  1. optmai commented on Sep 24, 2024

    @optmai
    Collaborator

    What task are you referring to?

  2. LiGuo12 commented on Sep 24, 2024

    @LiGuo12
    Author

    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.

  3. optmai commented on Sep 25, 2024

    @optmai
    Collaborator

    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.

  4. reopened this on Mar 3, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions