├── LibMTL ├── __init__.py ├── _record.py ├── architecture │ ├── CGC.py │ ├── Cross_stitch.py │ ├── DSelect_k.py │ ├── HPS.py │ ├── MMoE.py │ ├── MTAN.py │ ├── PLE.py │ ├── __init__.py │ └── abstract_arch.py ├── config.py ├── loss.py ├── metrics.py ├── model │ ├── __init__.py │ ├── resnet.py │ └── resnet_dilated.py ├── trainer.py ├── utils.py └── weighting │ ├── CAGrad.py │ ├── DWA.py │ ├── EW.py │ ├── GLS.py │ ├── GradDrop.py │ ├── GradNorm.py │ ├── GradVac.py │ ├── IMTL.py │ ├── MGDA.py │ ├── PCGrad.py │ ├── RLW.py │ ├── UW.py │ ├── __init__.py │ └── abstract_weighting.py ├── README.md ├── baseline ├── EDDA.py ├── GRecConv.py └── __init__.py ├── data ├── AliAd │ ├── d2_test.csv │ ├── d2_train.csv │ ├── d2_val.csv │ ├── d3_test.csv │ ├── d3_train.csv │ ├── d3_val.csv │ ├── d4_test.csv │ ├── d4_train.csv │ ├── d4_val.csv │ ├── d5_test.csv │ ├── d5_train.csv │ ├── d5_val.csv │ ├── d6_test.csv │ ├── d6_train.csv │ ├── d6_val.csv │ ├── d7_test.csv │ ├── d7_train.csv │ ├── d7_val.csv │ ├── d8_test.csv │ ├── d8_train.csv │ ├── d8_val.csv │ ├── d9_test.csv │ ├── d9_train.csv │ └── d9_val.csv ├── AliCCP │ ├── d0_test.csv │ ├── d0_train.csv │ ├── d0_val.csv │ ├── d1_test.csv │ ├── d1_train.csv │ ├── d1_val.csv │ ├── d2_test.csv │ ├── d2_train.csv │ └── d2_val.csv └── Amazon_5core │ ├── Arts_test.csv │ ├── Arts_train.csv │ ├── Arts_val.csv │ ├── Inst_test.csv │ ├── Inst_train.csv │ ├── Inst_val.csv │ ├── Luxury_test.csv │ ├── Luxury_train.csv │ ├── Luxury_val.csv │ ├── Music_test.csv │ ├── Music_train.csv │ ├── Music_val.csv │ ├── Pantry_test.csv │ ├── Pantry_train.csv │ ├── Pantry_val.csv │ ├── Video_test.csv │ ├── Video_train.csv │ └── Video_val.csv ├── requirements.txt ├── run_edda.py ├── task_dataloader.py └── util_functions.py /LibMTL/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/__init__.py -------------------------------------------------------------------------------- /LibMTL/_record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/_record.py -------------------------------------------------------------------------------- /LibMTL/architecture/CGC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/architecture/CGC.py -------------------------------------------------------------------------------- /LibMTL/architecture/Cross_stitch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/architecture/Cross_stitch.py -------------------------------------------------------------------------------- /LibMTL/architecture/DSelect_k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/architecture/DSelect_k.py -------------------------------------------------------------------------------- /LibMTL/architecture/HPS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/architecture/HPS.py -------------------------------------------------------------------------------- /LibMTL/architecture/MMoE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/architecture/MMoE.py -------------------------------------------------------------------------------- /LibMTL/architecture/MTAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/architecture/MTAN.py -------------------------------------------------------------------------------- /LibMTL/architecture/PLE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/architecture/PLE.py -------------------------------------------------------------------------------- /LibMTL/architecture/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/architecture/__init__.py -------------------------------------------------------------------------------- /LibMTL/architecture/abstract_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/architecture/abstract_arch.py -------------------------------------------------------------------------------- /LibMTL/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/config.py -------------------------------------------------------------------------------- /LibMTL/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/loss.py -------------------------------------------------------------------------------- /LibMTL/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/metrics.py -------------------------------------------------------------------------------- /LibMTL/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/model/__init__.py -------------------------------------------------------------------------------- /LibMTL/model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/model/resnet.py -------------------------------------------------------------------------------- /LibMTL/model/resnet_dilated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/model/resnet_dilated.py -------------------------------------------------------------------------------- /LibMTL/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/trainer.py -------------------------------------------------------------------------------- /LibMTL/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/utils.py -------------------------------------------------------------------------------- /LibMTL/weighting/CAGrad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/CAGrad.py -------------------------------------------------------------------------------- /LibMTL/weighting/DWA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/DWA.py -------------------------------------------------------------------------------- /LibMTL/weighting/EW.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/EW.py -------------------------------------------------------------------------------- /LibMTL/weighting/GLS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/GLS.py -------------------------------------------------------------------------------- /LibMTL/weighting/GradDrop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/GradDrop.py -------------------------------------------------------------------------------- /LibMTL/weighting/GradNorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/GradNorm.py -------------------------------------------------------------------------------- /LibMTL/weighting/GradVac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/GradVac.py -------------------------------------------------------------------------------- /LibMTL/weighting/IMTL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/IMTL.py -------------------------------------------------------------------------------- /LibMTL/weighting/MGDA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/MGDA.py -------------------------------------------------------------------------------- /LibMTL/weighting/PCGrad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/PCGrad.py -------------------------------------------------------------------------------- /LibMTL/weighting/RLW.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/RLW.py -------------------------------------------------------------------------------- /LibMTL/weighting/UW.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/UW.py -------------------------------------------------------------------------------- /LibMTL/weighting/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/__init__.py -------------------------------------------------------------------------------- /LibMTL/weighting/abstract_weighting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/LibMTL/weighting/abstract_weighting.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/README.md -------------------------------------------------------------------------------- /baseline/EDDA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/baseline/EDDA.py -------------------------------------------------------------------------------- /baseline/GRecConv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/baseline/GRecConv.py -------------------------------------------------------------------------------- /baseline/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/AliAd/d2_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d2_test.csv -------------------------------------------------------------------------------- /data/AliAd/d2_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d2_train.csv -------------------------------------------------------------------------------- /data/AliAd/d2_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d2_val.csv -------------------------------------------------------------------------------- /data/AliAd/d3_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d3_test.csv -------------------------------------------------------------------------------- /data/AliAd/d3_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d3_train.csv -------------------------------------------------------------------------------- /data/AliAd/d3_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d3_val.csv -------------------------------------------------------------------------------- /data/AliAd/d4_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d4_test.csv -------------------------------------------------------------------------------- /data/AliAd/d4_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d4_train.csv -------------------------------------------------------------------------------- /data/AliAd/d4_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d4_val.csv -------------------------------------------------------------------------------- /data/AliAd/d5_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d5_test.csv -------------------------------------------------------------------------------- /data/AliAd/d5_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d5_train.csv -------------------------------------------------------------------------------- /data/AliAd/d5_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d5_val.csv -------------------------------------------------------------------------------- /data/AliAd/d6_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d6_test.csv -------------------------------------------------------------------------------- /data/AliAd/d6_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d6_train.csv -------------------------------------------------------------------------------- /data/AliAd/d6_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d6_val.csv -------------------------------------------------------------------------------- /data/AliAd/d7_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d7_test.csv -------------------------------------------------------------------------------- /data/AliAd/d7_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d7_train.csv -------------------------------------------------------------------------------- /data/AliAd/d7_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d7_val.csv -------------------------------------------------------------------------------- /data/AliAd/d8_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d8_test.csv -------------------------------------------------------------------------------- /data/AliAd/d8_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d8_train.csv -------------------------------------------------------------------------------- /data/AliAd/d8_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d8_val.csv -------------------------------------------------------------------------------- /data/AliAd/d9_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d9_test.csv -------------------------------------------------------------------------------- /data/AliAd/d9_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d9_train.csv -------------------------------------------------------------------------------- /data/AliAd/d9_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliAd/d9_val.csv -------------------------------------------------------------------------------- /data/AliCCP/d0_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliCCP/d0_test.csv -------------------------------------------------------------------------------- /data/AliCCP/d0_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliCCP/d0_train.csv -------------------------------------------------------------------------------- /data/AliCCP/d0_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliCCP/d0_val.csv -------------------------------------------------------------------------------- /data/AliCCP/d1_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliCCP/d1_test.csv -------------------------------------------------------------------------------- /data/AliCCP/d1_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliCCP/d1_train.csv -------------------------------------------------------------------------------- /data/AliCCP/d1_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliCCP/d1_val.csv -------------------------------------------------------------------------------- /data/AliCCP/d2_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliCCP/d2_test.csv -------------------------------------------------------------------------------- /data/AliCCP/d2_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliCCP/d2_train.csv -------------------------------------------------------------------------------- /data/AliCCP/d2_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/AliCCP/d2_val.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Arts_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Arts_test.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Arts_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Arts_train.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Arts_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Arts_val.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Inst_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Inst_test.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Inst_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Inst_train.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Inst_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Inst_val.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Luxury_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Luxury_test.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Luxury_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Luxury_train.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Luxury_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Luxury_val.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Music_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Music_test.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Music_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Music_train.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Music_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Music_val.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Pantry_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Pantry_test.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Pantry_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Pantry_train.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Pantry_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Pantry_val.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Video_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Video_test.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Video_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Video_train.csv -------------------------------------------------------------------------------- /data/Amazon_5core/Video_val.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/data/Amazon_5core/Video_val.csv -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_edda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/run_edda.py -------------------------------------------------------------------------------- /task_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/task_dataloader.py -------------------------------------------------------------------------------- /util_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Stevenn9981/EDDA/HEAD/util_functions.py --------------------------------------------------------------------------------