CVE-LMTune: Tuning LMs for Multi-Taxonomy Vulnerability Classification

CVE-LMTune is a tool that provides an automated pipeline for collecting and preprocessing labeled vulnerability data from multiple sources, as well as for constructing a unified dataset for fine-tuning and evaluating language models in multi-label classification across CWE, CAPEC, and MITRE ATT&CK. The tool integrates both fine-tuning and evaluation pipelines and offers advanced support for hierarchical classification. In particular, it enables a hierarchical approach based on a DAG of specialized models, where each model handles a specific taxonomy level, with the goal of improving overall classification performance.