Skip to content

Discussion: Consider learning rate 0.01 as a default setting for the combination AUCM x PESG #100

Description

@kdg1993

What

Change the default learning rate setting of combination AUCM x PESG as 0.01.
In detail, the learning rate of PESG

Why

  • Fundamentally, it is important because of the high performance of AUCM x PESG
  • Empirically, the learning rate affects the performance a lot
  • Despite that learning rate 0.1 is good, it seems too high compare to the default learning rate of Adam
  • Learning rate 0.01 was better than 0.1 for both 224x224 & 512x512 images (CheXpert)
  • The validation loss curves of learning rate 0.01 show less fluctuation
  • Experiment results were quite clear that 0.01~0.1 could be a sweet spot

Experiment result : https://wandb.ai/snuh_interns/kdg_aucm_pesg_lr_test_w_img_size/table?workspace=user-snuh_interns

image
Fig. 1 Experiments and best validation scores (descending order)
image
Fig. 2 Validation loss & best validation score curves of two different learning rate based on 512 images
image
Fig. 3 Validation loss & best validation score curves of two different learning rate based on 224 images


How

Metadata

Metadata

Labels

DiscussionExtra attention is needed

Projects

Status
🤔 Discussion

Relationships

None yet

Development

No branches or pull requests

Issue actions