├── README.md ├── code ├── .gitignore ├── Data_split.py ├── LICENSE ├── appnpconv.py ├── config │ ├── __init__.py │ └── default.py ├── configs │ ├── bsce.yaml │ ├── cbce.yaml │ ├── cbf.yaml │ ├── cdt.yaml │ ├── ce.yaml │ ├── csce.yaml │ ├── focal.yaml │ ├── ibloss.yaml │ ├── kld.yaml │ ├── lasce.yaml │ ├── ldam.yaml │ ├── lsce.yaml │ └── seql.yaml ├── data_loader │ ├── __init__.py │ ├── batch_graph.py │ └── dataset.py ├── layers.py ├── loss │ ├── __init__.py │ ├── loss_base.py │ ├── loss_base_f.py │ ├── loss_impl │ │ ├── __init__.py │ │ ├── balanced_softmax_cross_entropy_loss.py │ │ ├── class_balanced_loss.py │ │ ├── class_dependent_temperatures_loss.py │ │ ├── cost_sensitive_cross_entropy_loss.py │ │ ├── cross_entropy_label_aware_smooth_loss.py │ │ ├── cross_entropy_label_smooth_loss.py │ │ ├── equalization_loss.py │ │ ├── focal_loss.py │ │ ├── influence_balanced_loss.py │ │ ├── kld_loss.py │ │ └── ldam_loss.py │ └── utils.py ├── loss_base.py ├── main_sta.py ├── mlp_readout.py ├── modules │ ├── __init__.py │ └── model.py ├── trainer_sta.py └── utils.py ├── data └── readme.md ├── figure ├── Architecture1_cropped.pdf ├── NVD_number.pdf ├── case_cropped.pdf └── cweid.pdf └── preprocessing ├── joern.md ├── ori_ourdevign+token.py ├── process.py ├── utils.py └── word2vec_multi.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/README.md -------------------------------------------------------------------------------- /code/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/.gitignore -------------------------------------------------------------------------------- /code/Data_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/Data_split.py -------------------------------------------------------------------------------- /code/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/LICENSE -------------------------------------------------------------------------------- /code/appnpconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/appnpconv.py -------------------------------------------------------------------------------- /code/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/config/__init__.py -------------------------------------------------------------------------------- /code/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/config/default.py -------------------------------------------------------------------------------- /code/configs/bsce.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/configs/bsce.yaml -------------------------------------------------------------------------------- /code/configs/cbce.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/configs/cbce.yaml -------------------------------------------------------------------------------- /code/configs/cbf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/configs/cbf.yaml -------------------------------------------------------------------------------- /code/configs/cdt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/configs/cdt.yaml -------------------------------------------------------------------------------- /code/configs/ce.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/configs/ce.yaml -------------------------------------------------------------------------------- /code/configs/csce.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/configs/csce.yaml -------------------------------------------------------------------------------- /code/configs/focal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/configs/focal.yaml -------------------------------------------------------------------------------- /code/configs/ibloss.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/configs/ibloss.yaml -------------------------------------------------------------------------------- /code/configs/kld.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/configs/kld.yaml -------------------------------------------------------------------------------- /code/configs/lasce.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/configs/lasce.yaml -------------------------------------------------------------------------------- /code/configs/ldam.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/configs/ldam.yaml -------------------------------------------------------------------------------- /code/configs/lsce.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/configs/lsce.yaml -------------------------------------------------------------------------------- /code/configs/seql.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/configs/seql.yaml -------------------------------------------------------------------------------- /code/data_loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/data_loader/__init__.py -------------------------------------------------------------------------------- /code/data_loader/batch_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/data_loader/batch_graph.py -------------------------------------------------------------------------------- /code/data_loader/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/data_loader/dataset.py -------------------------------------------------------------------------------- /code/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/layers.py -------------------------------------------------------------------------------- /code/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/__init__.py -------------------------------------------------------------------------------- /code/loss/loss_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/loss_base.py -------------------------------------------------------------------------------- /code/loss/loss_base_f.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/loss_base_f.py -------------------------------------------------------------------------------- /code/loss/loss_impl/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | -------------------------------------------------------------------------------- /code/loss/loss_impl/balanced_softmax_cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/loss_impl/balanced_softmax_cross_entropy_loss.py -------------------------------------------------------------------------------- /code/loss/loss_impl/class_balanced_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/loss_impl/class_balanced_loss.py -------------------------------------------------------------------------------- /code/loss/loss_impl/class_dependent_temperatures_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/loss_impl/class_dependent_temperatures_loss.py -------------------------------------------------------------------------------- /code/loss/loss_impl/cost_sensitive_cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/loss_impl/cost_sensitive_cross_entropy_loss.py -------------------------------------------------------------------------------- /code/loss/loss_impl/cross_entropy_label_aware_smooth_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/loss_impl/cross_entropy_label_aware_smooth_loss.py -------------------------------------------------------------------------------- /code/loss/loss_impl/cross_entropy_label_smooth_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/loss_impl/cross_entropy_label_smooth_loss.py -------------------------------------------------------------------------------- /code/loss/loss_impl/equalization_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/loss_impl/equalization_loss.py -------------------------------------------------------------------------------- /code/loss/loss_impl/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/loss_impl/focal_loss.py -------------------------------------------------------------------------------- /code/loss/loss_impl/influence_balanced_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/loss_impl/influence_balanced_loss.py -------------------------------------------------------------------------------- /code/loss/loss_impl/kld_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/loss_impl/kld_loss.py -------------------------------------------------------------------------------- /code/loss/loss_impl/ldam_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/loss_impl/ldam_loss.py -------------------------------------------------------------------------------- /code/loss/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss/utils.py -------------------------------------------------------------------------------- /code/loss_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/loss_base.py -------------------------------------------------------------------------------- /code/main_sta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/main_sta.py -------------------------------------------------------------------------------- /code/mlp_readout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/mlp_readout.py -------------------------------------------------------------------------------- /code/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/modules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/modules/model.py -------------------------------------------------------------------------------- /code/trainer_sta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/trainer_sta.py -------------------------------------------------------------------------------- /code/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/code/utils.py -------------------------------------------------------------------------------- /data/readme.md: -------------------------------------------------------------------------------- 1 | Data's url: 2 | 3 | https://drive.google.com/file/d/1j8WUSNte6Tda2uXR8Ue3Hk9jRa-33LGL/view?usp=sharing -------------------------------------------------------------------------------- /figure/Architecture1_cropped.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/figure/Architecture1_cropped.pdf -------------------------------------------------------------------------------- /figure/NVD_number.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/figure/NVD_number.pdf -------------------------------------------------------------------------------- /figure/case_cropped.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/figure/case_cropped.pdf -------------------------------------------------------------------------------- /figure/cweid.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/figure/cweid.pdf -------------------------------------------------------------------------------- /preprocessing/joern.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/preprocessing/joern.md -------------------------------------------------------------------------------- /preprocessing/ori_ourdevign+token.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/preprocessing/ori_ourdevign+token.py -------------------------------------------------------------------------------- /preprocessing/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/preprocessing/process.py -------------------------------------------------------------------------------- /preprocessing/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/preprocessing/utils.py -------------------------------------------------------------------------------- /preprocessing/word2vec_multi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xin-Cheng-Wen/LIVABLE/HEAD/preprocessing/word2vec_multi.py --------------------------------------------------------------------------------