https://github.com/huanglizi/wtc

This repository includes the official project of WTC Model, presented in our paper: Semi-WTC: A Practical Semi-supervised Framework for Attack Categorization through Weight-Task Consistency(TDSC major revision)
https://github.com/huanglizi/wtc

Last synced: about 1 month ago
JSON representation

This repository includes the official project of WTC Model, presented in our paper: Semi-WTC: A Practical Semi-supervised Framework for Attack Categorization through Weight-Task Consistency(TDSC major revision)

• Host: GitHub
• URL: https://github.com/huanglizi/wtc
• Owner: HUANGLIZI
• Created: 2021-09-04T06:50:36.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2022-09-05T20:13:06.000Z (over 2 years ago)
• Last Synced: 2023-05-02T15:24:43.826Z (over 1 year ago)
• Language: Python
• Homepage:
• Size: 2.49 MB
• Stars: 4
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Semi-WTC

This repository includes the official project of Semi-WTC Model, presented in our paper: Semi-WTC: A Practical Semi-supervised Framework for Attack Categorization through Weight-Task Consistency (Under Major Revision)

![image](https://github.com/HUANGLIZI/WTC/blob/main/img/SEMI-WTC.jpg)

*Paper Link*: https://arxiv.org/abs/2205.09669

*Email*: [email protected]

Please contact me if you need the dataset.

# Usage

dataset/ : the used dataset for the model

utils/ : all the supplementary tools used for our Semi-WTC

you can use the command "python AAR-NSLKDD.py" for ACTIVE ADAPTION RESAMPLING (AAR).

you can use the command "python main_demo.py" for demo training and testing.

At present, the demo file "main_demo.py" is under preparation.

# Environment

Please prepare an environment with python=3.8, and then use the command "pip install -r requirements.txt" for the dependencies.

# Experimental results for NSL-KDD demo

Acc & F1:

![image](https://github.com/HUANGLIZI/WTC/blob/main/img/results.jpg)

Confusion Matrix:

![image](https://github.com/HUANGLIZI/WTC/blob/main/img/CM.jpg)

# Citation

```bash

@article{li2022semi,

  title={Semi-WTC: A Practical Semi-supervised Framework for Attack Categorization through Weight-Task Consistency},

  author={Li, Zihan and Chen, Wentao and Wei, Zhiqing and Luo, Xingqi and Su, Bing},

  journal={arXiv preprint arXiv:2205.09669},

  year={2022}

}

```