├── .gitignore ├── README.md ├── baselines ├── model │ └── bert_classify.py ├── preprocess │ ├── data_augment.py │ ├── dataloader.py │ ├── preprocess.py │ └── read_data.py ├── script │ ├── train_asym.sh │ └── train_idn.sh ├── train_base.py ├── train_cl.py ├── train_coteaching.py ├── train_elr.py ├── train_sce.py └── utils │ ├── augmentation │ ├── __init__.py │ ├── saliency.py │ ├── ssmix.py │ └── unk.py │ ├── clean_data.py │ ├── common.py │ ├── entropy.py │ ├── loss.py │ ├── metric.py │ └── my_learning_with_noise.py ├── data ├── corrupt.py └── trec │ ├── test.csv │ ├── train.csv │ └── train_corrupted.csv ├── datasets.py ├── demo_config ├── trec-bert-asym_train.json ├── trec-bert-idn_train.json └── trec-bert_eval.json ├── evaluation.py ├── figure └── model.png ├── model.py ├── requirements.txt ├── train.py └── trainer.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.pt 2 | bert-base-uncased 3 | save_model 4 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/README.md -------------------------------------------------------------------------------- /baselines/model/bert_classify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/model/bert_classify.py -------------------------------------------------------------------------------- /baselines/preprocess/data_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/preprocess/data_augment.py -------------------------------------------------------------------------------- /baselines/preprocess/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/preprocess/dataloader.py -------------------------------------------------------------------------------- /baselines/preprocess/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/preprocess/preprocess.py -------------------------------------------------------------------------------- /baselines/preprocess/read_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/preprocess/read_data.py -------------------------------------------------------------------------------- /baselines/script/train_asym.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/script/train_asym.sh -------------------------------------------------------------------------------- /baselines/script/train_idn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/script/train_idn.sh -------------------------------------------------------------------------------- /baselines/train_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/train_base.py -------------------------------------------------------------------------------- /baselines/train_cl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/train_cl.py -------------------------------------------------------------------------------- /baselines/train_coteaching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/train_coteaching.py -------------------------------------------------------------------------------- /baselines/train_elr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/train_elr.py -------------------------------------------------------------------------------- /baselines/train_sce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/train_sce.py -------------------------------------------------------------------------------- /baselines/utils/augmentation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/utils/augmentation/__init__.py -------------------------------------------------------------------------------- /baselines/utils/augmentation/saliency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/utils/augmentation/saliency.py -------------------------------------------------------------------------------- /baselines/utils/augmentation/ssmix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/utils/augmentation/ssmix.py -------------------------------------------------------------------------------- /baselines/utils/augmentation/unk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/utils/augmentation/unk.py -------------------------------------------------------------------------------- /baselines/utils/clean_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/utils/clean_data.py -------------------------------------------------------------------------------- /baselines/utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/utils/common.py -------------------------------------------------------------------------------- /baselines/utils/entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/utils/entropy.py -------------------------------------------------------------------------------- /baselines/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/utils/loss.py -------------------------------------------------------------------------------- /baselines/utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/utils/metric.py -------------------------------------------------------------------------------- /baselines/utils/my_learning_with_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/baselines/utils/my_learning_with_noise.py -------------------------------------------------------------------------------- /data/corrupt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/data/corrupt.py -------------------------------------------------------------------------------- /data/trec/test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/data/trec/test.csv -------------------------------------------------------------------------------- /data/trec/train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/data/trec/train.csv -------------------------------------------------------------------------------- /data/trec/train_corrupted.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/data/trec/train_corrupted.csv -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/datasets.py -------------------------------------------------------------------------------- /demo_config/trec-bert-asym_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/demo_config/trec-bert-asym_train.json -------------------------------------------------------------------------------- /demo_config/trec-bert-idn_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/demo_config/trec-bert-idn_train.json -------------------------------------------------------------------------------- /demo_config/trec-bert_eval.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/demo_config/trec-bert_eval.json -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/evaluation.py -------------------------------------------------------------------------------- /figure/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/figure/model.png -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/model.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | transformers==4.18.0 2 | torch==1.11.0 3 | numpy 4 | scikit-learn 5 | pandas 6 | -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/train.py -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/noise-learning/SelfMix/HEAD/trainer.py --------------------------------------------------------------------------------