alihur-ecu/cve-cwe-crossencoder

Fine-tuned cross-encoder for CVE→CWE ranking (cve2graph Phase 26).

  • Base model: cross-encoder/ms-marco-MiniLM-L-6-v2
  • Training: 3 epochs on in-domain CVE/CWE pairs (hard negatives from Phase 24 fusion ranker)
  • Test Hits@10: 0.7165
  • Test MRR: 0.6026
  • Fusion: rerank top-200 CWE candidates with cross-encoder weight λ=0.5

Usage

from sentence_transformers import CrossEncoder

model = CrossEncoder("alihur-ecu/cve-cwe-crossencoder")
scores = model.predict([("CVE description text", "CWE description text")])
Downloads last month
16
Safetensors
Model size
22.7M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for alihur-ecu/cve-cwe-crossencoder