云起无垠：SecGPT-全球首个网络安全开源大模型

  2023年是人工智能的奇迹年,OpenAI的ChatGPT横空出世,在自然语言的人机对话领域实现了突破性的智能表现。人工智能技术的革命性应用已在各行业引发了热潮,医疗、金融、法律等领域都涌现出了垂类开源大模型,为行业应用带来了无限可能性。与别的行业类似,网络安全从业者们也一直在积极探索安全垂直大模型的实际应用。这一趋势在网络安全领域引发了深刻的变革,为保护互联网生态系统和应对一直增长的威胁提供了新的工具和方法。

  近日,云起无垠公司发布了名为「SecGPT」的网络安全大模型开源项目。该项目是全球首个开源的网络安全大模型,截止目前已经在GitHub上积累了超过400个star,受到了广泛的业界关注。云起无垠希望贡献自己的力量,推动网络安全领域迈向智能化时代,实现“让软件更安全,让安全更智能”的企业使命。

  SecGPT的愿景是将人工智能技术引入网络安全领域,以提高网络防御的效率和效果。其使命是推动网络安全智能化,为社会提供更安全的数字生活环境。

  SecGPT可当作基座安全模型,用于探索各种网络安全任务。以下是对SecGPT在网络安全任务中可能应用的进一步探讨:

  1.漏洞分析:SecGPT可以与安全研究人员或开发团队进行多轮对话,共同分析和审查应用程序或系统中的潜在漏洞。它能够在一定程度上帮助识别和理解漏洞的性质、潜在影响以及可能的修复方法。

  2.溯源分析:在网络入侵事件调查中,SecGPT可以与调查员合作,协助分析网络流量、日志和事件记录,以追踪攻击者的活动路径,从而支持安全溯源分析。

  3.流量分析:SecGPT可以分析网络流量数据,识别异常流量模式,帮助检测潜在的网络攻击或入侵行为,并提供应对这些行为的建议。

  4.攻击研判:在面对未知攻击时,SecGPT可以与安全团队一起进行多轮对话,共同分析攻击的特征、模式和可能的来源,以便更好地理解和应对威胁。

  5.命令解释:SecGPT可以解释和分析安全相关的命令、脚本或代码,帮助安全专家理解其功能、潜在风险和执行结果。

  6.网安知识问答:SecGPT能够适用于回答网络安全领域的问题,为安全从业者提供支持和解释,包括解释安全概念、提供最佳实践建议和分析最新的威胁趋势。

  1.自写训练代码,节省显存:SecGPT提供了自己的训练代码,包括预训练、有监督学习和DPO强化对齐等功能。这使得即使使用小型显卡,也能够训练大规模的网络安全模型,为更多的开发者和研究者提供了参与网络安全研究的机会。

  2.高质量网络安全训练集:SecGPT提供了精选的高质量网络安全类无监督训练集,使模型能够学习网络安全问题。这有助于提高模型对于网络安全领域的理解。

  3.DPO强化学习:通过DPO强化学习方法,SecGPT引入了安全专家的见解和答案,与模型原始输出一起作为数据来进行训练。这种方法有助于模型更好地模拟安全专家的思维方法,提高了模型输出的实用性和准确性,使其更符合安全领域的需求。

  4. 无道德限制:在网络安全领域,无道德限制的GPT模型具有优势。比如模型可以深入分析和解释恶意软件的代码、行为和影响,而不受道德限制的束缚。当然,必须要格外注意的是,无道德限制并不代表可以不受限制地使用这一些模型。在执行敏感网络安全任务时,仍然需要谨慎考虑法律、道德和隐私问题,并遵循适用的法律和法规和道德准则。

  目前,SecGPT不仅开源了基座模型,还开源了所有的训练代码以及数据(包括预训练,有监督训练,DPO训练等大模型全流程训练代码)和网络安全训练数据。我们鼓励任何人都使用这一些代码和数据去训练自己的网络安全大模型!该模型的训练过程大致如下图所示:

  (1)选用Baichuan-13B模型作为基座模型进行训练(无道德限制,较好中文支持,显存资源占用小)

  ①收集安全书籍,安全知识库,安全论文,安全社区文章,漏洞库等等安全内容。

  ①利用ChatGPT+人工构造各类有监督安全能力数据集,让模型能了解各类安全指令。

  ②思维链:基于思维链方式构造有监督数据集让模型可以依据问题逐步推理到最终答案,展现推理过程。

  ③知乎回答:加入了部分高质量知乎数据集,在一些开放性问题上模型能通过讲故事举例子等方式回答答案和观点,更易读懂。

  ④为防止灾难性遗忘,在有监督数据中,通用能力数据:安全能力数据,约为5:1。

  为了改善模型的输出质量,有以下两种策略可供选择。首先,能够最终靠使用更明确的提示词来引导模型,使其更专注于客户的真实需求,从而生成更有明确的目的性的回答。其次,能够使用RLHF强化学习方法,模型将生成多个答案,用户都能够手动选择最优的答案,在后面进行强化学习,从而逐步的提升模型对齐的能力。SecGPT模型具备内置的RLHF选择器,每次会输出三个不同的答案。用户都能够根据需求选择最佳答案,同时这也为后续的RLHF微调提供了有价值的数据,有助于模型逐步提升输出的质量。这一些方法共同作用,有助于确保模型输出使用户得到满足的预期,并在网络安全等领域中发挥更大的作用。

  关于SecGPT网络安全大模型,这只是我们分享的一部分内容。云起无垠致力于将SecGPT开源大模型分享给更多的安全从业者,我们鼓励大家热情参加共同的研讨和交流,以一直在改进和增强其功能。通过开源的方式,我们也期望加强网络安全从业者对这一大模型的理解和应用。

  如果您有任何关于该模型的问题或建议,请随时在GitHub(搜:SecGPT)上提交issue或PR。我们热切欢迎各位共同参与建设,让SecGPT网络安全大模型能够惠及网络安全领域的每一位需求者!共同的努力将有利于提高网络安全的水平,确保数字生活的安全性。

  【广告】免责声明：本内容为广告，相关素材由广告主提供，广告主对本广告内容的真实性负责。本网发布目的是传递更加多信息，并不代表本网赞同其观点和对其真实性负责，广告内容仅供读者参考。

  新闻信息服务许可证音像制品出版许可证广播电视节目制作经营许可证网络视听许可证网络文化经营许可证