arXiv:2412.05049cs.AIcs.CR2024-12中稿 · Applied Cryptograp…被引 5

OCEAN通过对比学习实现二进制代码作者身份识别,可在未知作者场景下精准定位代码来源。

OCEAN: Open-World Contrastive Authorship Identification

  • 基于对比学习构建函数级作者归属框架,首次支持开世界场景下的二进制代码比对。
  • 在未见过的数据集上达到0.86的AUROC,高编译优化下仍保持高精度。
  • 可检测未知作者的代码注入,适合软件供应链安全防护场景。

在软件供应链面临日益严峻网络攻击的背景下,准确识别二进制文件中的代码作者身份对提升网络安全至关重要。我们提出OCEAN,一种基于对比学习的函数级作者归属系统。OCEAN是首个在开世界和极端场景下探索二进制代码作者归属的框架,可比较两名未知作者的代码样本以判断是否出自同一人。为评估OCEAN,我们引入两个新数据集:CONAN用于提升真实场景下的性能表现,SNOOPY用于增强评估鲁棒性。使用CONAN训练模型并在完全未见过的SNOOPY数据集上测试,即使在高编译优化条件下,仍取得0.86的AUROC。相较此前常用的Google Code Jam数据集,CONAN使性能提升7%。此外,OCEAN在源代码分析场景中相比现有最优方法SCS-Gan提升10%。实验还表明,OCEAN可有效检测软件更新中来自未知作者的代码注入,凸显其在保障软件供应链安全中的价值。

原文摘要 · Abstract (English)

In an era where cyberattacks increasingly target the software supply chain, the ability to accurately attribute code authorship in binary files is critical to improving cybersecurity measures. We propose OCEAN, a contrastive learning-based system for function-level authorship attribution. OCEAN is the first framework to explore code authorship attribution on compiled binaries in an open-world and extreme scenario, where two code samples from unknown authors are compared to determine if they are developed by the same author. To evaluate OCEAN, we introduce new realistic datasets: CONAN, to improve the performance of authorship attribution systems in real-world use cases, and SNOOPY, to increase the robustness of the evaluation of such systems. We use CONAN to train our model and evaluate on SNOOPY, a fully unseen dataset, resulting in an AUROC score of 0.86 even when using high compiler optimizations. We further show that CONAN improves performance by 7% compared to the previously used Google Code Jam dataset. Additionally, OCEAN outperforms previous methods in their settings, achieving a 10% improvement over state-of-the-art SCS-Gan in scenarios analyzing source code. Furthermore, OCEAN can detect code injections from an unknown author in a software update, underscoring its value for securing software supply chains.

代码归属对比学习供应链安全

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。