crash-information-extraction/llama3.1-8B-Crash-Location-Casualty
0
Llama3.1-8B-Crash-Location-Casualty
Fine-tuned Llama 3.1 for extracting crash location and casualties from Persian social media.
๐ Paper: Extracting traffic crash information from social media: an LLM-based approach โ Transportation Letters (2026)
๐ฏ What it does
Extracts two key types of information from Persian social media texts:
- Location (streets, highways, areas)
- Casualty Count (fatalities & injuries)
โ๏ธ Fine-tuning
- Method: QLoRA
- Base Model:
unsloth/meta-llama-3.1-8b-bnb-4bit - Data: Proprietary Persian social media crash dataset (Damavand County, Iran)
๐ Performance
Social media captured 64.1% of crash hotspots compared to official records.
๐ Quick Start
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("crash-information-extraction/llama3.1-8B-Crash-Location-Casualty")
tokenizer = AutoTokenizer.from_pretrained("crash-information-extraction/llama3.1-8B-Crash-Location-Casualty")