Skip to content

Latest commit

 

History

History
31 lines (15 loc) · 813 Bytes

README.md

File metadata and controls

31 lines (15 loc) · 813 Bytes

Code for the paper:

Yinjun Wu, Adam Stein, Jacob Gardner, Mayur Naik, "Learning to Select Pivotal Samples for Meta Re-weighting " (AAAI 2023 oral)

plot

Install the pre-requiste:

pip install -r requirements.txt

Minimal code block for using our algorithm for selecting meta samples for meta reweighting algorithm:


#select valid samples with rbc
valid_ids, new_valid_representations = get_representative_valid_ids_rbc(train_dataset, criterion, optimizer, trainloader, args, net, valid_count)



#select valid samples with gbc
valid_ids, new_valid_representations = get_representative_valid_ids_gbc(train_dataset, criterion, optimizer, trainloader, args, net, valid_count)


See the jupyter notebook main_example.ipynb for details