MARGIN: Margin-Aware Regularized Geometry for Imbalanced Vulnerability Detection
Software vulnerability detection is critical for ensuring software security and reliability. Despite recent advances in deep learning, real-world vulnerability datasets suffer from two severe challenges: frequency imbalance and difficulty imbalance. We reinterpret these challenges from an embedding geometry perspective, observing that such imbalances induce geometric distortions in hyperspherical representation space. To address this issue, we propose MARGIN, a metric-based framework that learns discriminative vulnerability representations through adaptive margin metric learning and hyperspherical prototype modeling. MARGIN dynamically adjusts geometric regularization according to the dis...
| Authors | Yuteng Zhang, Huifang Ma, Jiahui Wei et al. |
|---|---|
| Primary category | cs.SE |
| Published | 11 May 2026 |
| Last updated at source | 8 July 2026 |
Connected in the knowledge graph
- codemetic/MARGINDataset · cited by
Models and datasets referencing this paper
- codemetic/MARGINDataset
Silytics shows public metadata from arXiv and links to the original. Information may be incomplete or out of date. Last refreshed 17 October 2026. Licenses and terms of the original apply to the work itself. Not financial, legal or professional advice.