Final Year IEEE Project | AI-Powered Phishing Detection System
AI-powered phishing detection with 93 ML features, IDN protection, and enterprise security.
# Install Python dependencies
pip install -r requirements.txt# CLI mode
python 04_inference/service.py https://example.com
# API server
python 04_inference/api.py
# Run tests
pytest tests/phishing_detection_project/
βββ 01_data/ # Datasets & TLDs
βββ 02_models/ # ML models (joblib)
βββ 03_training/ # Training scripts + MLflow
βββ 04_inference/ # API + Service
βββ 05_utils/ # 93 feature extractors
βββ 06_notebooks/ # Jupyter notebooks
βββ 07_configs/ # Configuration files
βββ browser-extension/ # Chrome/Firefox extension
βββ tests/ # Test suite
β
βββ Dockerfile # Container deployment
βββ docker-compose.yml # Docker orchestration
βββ requirements.txt # Python dependencies
βββ README.md # This file
- 93 ML Features - Advanced feature engineering
- 4-Category Classification - Legitimate, Phishing, AI-Generated, Phishing Kit
- IDN Protection - Unicode homograph attack detection
- Browser Extension - Real-time link protection
- MLflow Integration - Model versioning & experiment tracking
- API Server - RESTful endpoints with JWT authentication
# Run test suite
pytest tests/ -v
# Run specific tests
python -m pytest tests/test_security.py
python -m pytest tests/test_comprehensive.py| Metric | Value |
|---|---|
| F1 Score | 99.82% |
| Precision | 99.81% |
| Recall | 99.83% |
| Features | 93 |
| Classification | 4 categories |
- JWT Authentication (24hr tokens)
- Rate Limiting (100 req/min)
- SSRF Protection
- TLS 1.3 Enforcement
- Input Validation
βββββββββββββββββββββββββββββββββββββββββββ
β User Interfaces β
β CLI β’ API β’ Browser Extension β
βββββββββββββββββββ¬ββββββββββββββββββββββββ
β
ββββββββββ΄βββββββββ
β Detection Core β
β 93 Features β
ββββββββββ¬βββββββββ
β
βββββββββββββββΌββββββββββββββ
β β β
βββββ΄ββββ ββββββ΄βββββ ββββββ΄βββββ
β Tier 1β β Tier 2 β β Tier 3 β
β ML β β Typosquatβ β MLLM β
βββββββββ βββββββββββ βββββββββββ
- Random Forest Classifier (200 estimators)
- XGBoost Classifier (50 estimators)
- Ensemble Method: Soft Voting
- Optional: Qwen2.5-3B for AI-generated phishing detection
See browser-extension/ for installation instructions.
cd browser-extension
npm install
npm run build- API Documentation - API endpoints
- Security Policy - Security details
- Deployment Guide - Docker deployment
- Browser Extension - Extension setup
- Author: Akarsh Bandi
- Email: akarshbandi82@gmail.com
- GitHub: BandiAkarsh
MIT License - See LICENSE file
π Final Year IEEE Project