Thanks to visit codestin.com
Credit goes to Github.com

Skip to content

Predicting DDoS attacks using Machine learning models (Random Forest, XGBoost, TabNet) with feature selection and evaluation on network traffic data.

Notifications You must be signed in to change notification settings

NurFakhri/DDos-Attack-Predict

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

4 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

πŸ” DDoS Attack Prediction using Machine Learning

This project aims to detect potential Distributed Denial of Service (DDoS) attacks using machine learning techniques on historical network traffic data. It explores multiple models and compares performance across different feature subsets.


πŸ“‚ Project Structure

  • DDoS_Attack_Prediction.ipynb: Main notebook containing EDA, preprocessing, training, and evaluation
  • Data/: All dataset (unprocessed, preprocessed)

πŸš€ Models Used

We implemented and compared the following classification models:

  1. Machine Learning
  • Random Forest
  • XGBoost
  1. Deep Learning
  • TabNet

Each model was trained on:

  • Full feature set
  • Top 6 highly correlated features
  • 6 least correlated features

Data was split into 80% train, 10% validation, 10% test.


πŸ“ˆ Results Summary

Model Full Features High Correlation Only Low Correlation Only
Random Forest 100% 99% 75%
XGBoost 99% 98% 74%
TabNet 97% 92% 72%

Strongly correlated features proved nearly as effective as using the full dataset. Low-correlated features performed poorly, highlighting the importance of feature selection.


πŸ’‘ Key Features Used

  • bytecount
  • pktcount
  • byteperflow
  • pktperflow
  • pktrate
  • tot_dur

🧠 Future Work

  • Real-time implementation using live traffic streams
  • Automated alert system for suspicious behavior
  • Explainability with SHAP/TabNet attention to interpret model decisions
  • Scalability testing on larger datasets or multiple network sources

πŸ“¬ Contact

For questions or collaboration, feel free to reach out:

  • Name: Muhammad Hadi Nur Fakhri
  • LinkedIn: linkedin.com/in/nur-fakhri/

πŸ“„ License

This project is open-source

About

Predicting DDoS attacks using Machine learning models (Random Forest, XGBoost, TabNet) with feature selection and evaluation on network traffic data.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published