arXiv:2411.07528cs.CRcs.AI2024-11被引 2

用安全日志预训练小模型,提升安全任务表现

SecEncoder: Logs are All You Need in Security

  • 用安全日志而非自然语言预训练小型语言模型
  • 在日志分析、威胁情报等任务上优于BERT等通用模型
  • 适合安全领域研究者与工程师使用

大模型和小模型通常基于大量公开文本(如维基百科、书籍语料)进行预训练,虽具备强大泛化能力,但在特定领域表现不佳。本文提出SecEncoder,一种基于安全日志预训练的小型语言模型,旨在解决通用模型在安全领域的局限性。实验表明,尽管仅在日志数据上预训练,SecEncoder在日志分析、事件优先级排序、威胁情报文档检索等任务上均优于BERT-large、DeBERTa-v3-large及OpenAI的textembedding-ada-002等通用模型。这一结果表明,基于日志的领域特定预训练能显著提升模型在安全任务中的性能,为安全专用语言模型的研究与应用提供新路径。

原文摘要 · Abstract (English)

Large and Small Language Models (LMs) are typically pretrained using extensive volumes of text, which are sourced from publicly accessible platforms such as Wikipedia, Book Corpus, or through web scraping. These models, due to their exposure to a wide range of language data, exhibit impressive generalization capabilities and can perform a multitude of tasks simultaneously. However, they often fall short when it comes to domain-specific tasks due to their broad training data. This paper introduces SecEncoder, a specialized small language model that is pretrained using security logs. SecEncoder is designed to address the domain-specific limitations of general LMs by focusing on the unique language and patterns found in security logs. Experimental results indicate that SecEncoder outperforms other LMs, such as BERTlarge, DeBERTa-v3-large and OpenAI's Embedding (textembedding-ada-002) models, which are pretrained mainly on natural language, across various tasks. Furthermore, although SecEncoder is primarily pretrained on log data, it outperforms models pretrained on natural language for a range of tasks beyond log analysis, such as incident prioritization and threat intelligence document retrieval. This suggests that domain specific pretraining with logs can significantly enhance the performance of LMs in security. These findings pave the way for future research into security-specific LMs and their potential applications.

安全日志小模型预训练语言模型

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