Skip to content

Repository files navigation

Attack-Phase Aware Dynamic Vulnerability Prioritization Framework

This repository contains the full implementation of an attack-phase aware, machine-learning–adaptive vulnerability prioritization framework that unifies:

  • CVSS (severity)
  • EPSS (exploit likelihood)
  • MITRE ATT&CK (adversarial techniques)
  • Unified Cyber Kill Chain (UCKC) (attack phase context)
  • Adaptive ML Weighting (dynamic α, β, γ, δ coefficients)
  • Explainability (SHAP)

🔥 Key Features

✔ Full CVSS v3.1 parsing from NVD
✔ Missing CVSS metric prediction using ML
✔ EPSS dataset integration
✔ ATT&CK → UCKC phase mapping
✔ Static and adaptive PSSS scoring
✔ Weight evolution simulation
✔ SHAP explainability
✔ Ready-to-run Python script


📁 Repository Structure

src/               → main Python script  
data/              → datasets (NVD, EPSS, ATT&CK)  
results/           → plots generated by experiment  
notebooks/         → demo Jupyter notebook  
docs/              → diagrams & documentation  

🧠 How It Works

1️⃣ Data Pipeline

  • Parse NVD CVEs
  • Extract/Infer CVSS metrics
  • Merge EPSS
  • Map ATT&CK techniques
  • Convert ATT&CK → Unified Cyber Kill Chain (UCKC) phases

2️⃣ Scoring Models

✔ Static PSSS

PSSS = 0.4·CVSS + 0.4·EPSS + 0.2·ATT&CK

✔ Adaptive PSSS

Machine learning model learns the weights dynamically:

PSSS = α·CVSS + β·EPSS + γ·ATT&CK + δ·UCKC

3️⃣ Explainability

  • SHAP values
  • Feature importance charts
  • Attack-phase influence

🚀 Run the Project

Install Dependencies

pip install -r requirements.txt

Run Script

python src/adaptive_vulnerability_prioritization_with_uckc_integration.py

📊 Generated Outputs

Results stored in /results:

  • Static PSSS distribution
  • Adaptive PSSS distribution
  • Weight evolution curves
  • SHAP feature importance
  • Top 20 prioritized CVEs

📘 Research Paper

The methodology, results, discussion, and diagrams are stored under:

docs/PAPER_SECTIONS.md

📄 License

MIT License.


🏷️ GitHub Project Tags

#cvss #epss #mitre-attack #uckc #cybersecurity #vulnerability-management #machine-learning #shap #adaptive-weighting #risk-scoring

License: MIT Python Version Status

About

A machine-learning–driven, attack-phase–aware vulnerability prioritization framework integrating CVSS, EPSS, MITRE ATT&CK, and the Unified Cyber Kill Chain (UCKC) to compute dynamic risk scores with adaptive weighting and SHAP explainability.

Resources

Stars

1 star

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages