三亩地 三亩地SAN MU DI · CODE DIARY
ARTICLE DETAIL

日记详情

真实记录编程学习的某一天,欢迎挑你感兴趣的翻一翻。

Dromedary大模型深度解析:NeurIPS 2023焦点成果如何实现最小人工监督下的自对齐?

Dromedary大模型深度解析:NeurIPS 2023焦点成果如何实现最小人工监督下的自对齐?

Dromedary大模型深度解析:NeurIPS 2023焦点成果如何实现最小人工监督下的自对齐?

【免费下载链接】DromedaryDromedary: towards helpful, ethical and reliable LLMs.项目地址: https://gitcode.com/gh_mirrors/dr/Dromedary

Dromedary是一款致力于打造有帮助、符合伦理且可靠的大型语言模型(LLM),其核心创新在于最小人工监督下的自对齐技术。作为NeurIPS 2023的焦点成果,该模型通过独特的"自我对齐" pipeline,仅使用少于300行的人工标注,就能实现与人类价值观的高度一致,为AI伦理对齐提供了突破性解决方案。

什么是自对齐(Self-Alignment)技术?

自对齐是Dromedary模型的核心创新点,指的是AI系统通过自我指导而非大规模人工标注来实现与人类价值观对齐的过程。传统LLM对齐需要成千上万条人工标注数据,而Dromedary通过以下四个关键步骤,大幅降低了对人工监督的依赖:

  1. 主题引导的自我指令生成:使用195个种子提示和7条规则,自动生成36万条合成指令
  2. 原则驱动的自我对齐:基于16条AI助手原则和5个上下文学习示例,生成26万条对齐响应
  3. 原则雕刻(Principle Engraving):修剪原则和示例后微调原始模型
  4. 详细克隆(Verbose Cloning):优化模型以产生深入且详细的响应

图:Dromedary的四阶段自对齐 pipeline,仅需少于300行人工标注即可实现伦理对齐

16条核心原则:Dromedary的"道德指南针"

Dromedary的自对齐能力源于其精心设计的16条核心原则,这些原则作为模型的"道德指南针",指导其在各种情境下的行为。关键原则包括:

  • 伦理优先:主动拒绝非法、不道德或有害话题,优先考虑用户安全
  • 信息准确:提供准确、相关和最新的信息,确保内容兼具教育性和吸引力
  • 诚实谦逊:当信息不在内部知识库中时,承认自己的知识局限
  • 多语言支持:能够使用用户所用的语言进行对话(如用中文回复中文查询)
  • 分步推理:在提供答案前,先呈现逐步的推理过程

这些原则被编码在prompts/watson_self_align_prompt.txt文件中,作为模型自我对齐的基础框架。通过这些原则,Dromedary能够在保持高可用性的同时,避免生成有害或不适当的内容。

技术实现:从理论到实践的桥梁

Dromedary的技术实现主要集中在以下几个核心模块:

模型架构与训练

Dromedary基于LLaMA模型架构,并引入了多查询注意力(MQA)技术以提高推理效率。项目提供了完整的训练流程,分为三个主要步骤:

  1. 提示清洗:training/step1_prompt_cleaning/目录包含聚合和清洗提示数据的工具
  2. 原则驱动自对齐:training/step2_principle_driven_self_alignment/实现了核心的自对齐响应生成
  3. 原则雕刻:training/step3_principle_engraving/提供了模型微调的脚本和配置

推理与部署

项目的inference/目录提供了完整的推理脚本和示例,包括:

  • 流式聊天机器人演示:inference/run_stream_chatbot_demo.py
  • 多分片部署脚本:inference/scripts/demo_dromedary_stream_8shards.sh支持大规模部署

实际应用:Dromedary如何改变AI交互体验?

Dromedary的自对齐技术带来了更安全、更可靠的AI交互体验。以下是几个典型应用场景:

安全对话系统

Dromedary能够识别并拒绝处理有害请求,同时保持友好和帮助性。例如,当被问及未来天气时,模型会诚实地说明其知识截止日期(2022年9月),并建议用户查阅实时来源,而不是编造信息。

教育辅助工具

凭借其"信息准确"和"分步推理"原则,Dromedary成为理想的学习伙伴。它能提供详细的解释和示例,如在编程问题中,不仅给出代码,还会解释每一步的逻辑。

多语言助手

支持17种语言的能力使Dromedary能够服务全球用户。无论是中文、英文还是其他语言,模型都能保持一致的响应质量和伦理标准。

快速开始:体验Dromedary的强大功能

要开始使用Dromedary,只需按照以下简单步骤操作:

  1. 克隆仓库:

    git clone https://gitcode.com/gh_mirrors/dr/Dromedary
  2. 安装依赖:

    cd Dromedary/llama_dromedary pip install -e .
  3. 运行示例对话:

    torchrun --nproc_per_node 1 example_chat_completion.py \ --ckpt_dir llama-2-7b-chat/ \ --tokenizer_path tokenizer.model \ --max_seq_len 512 --max_batch_size 6

结语:AI伦理对齐的新范式

Dromedary通过最小人工监督实现自对齐的创新方法,为解决AI伦理挑战提供了新范式。其少于300行人工标注就能实现高度伦理对齐的成果,不仅大幅降低了模型训练成本,也为构建更安全、更可靠的AI系统开辟了新路径。

随着技术的不断发展,Dromedary有望在教育、医疗、客服等多个领域发挥重要作用,真正实现AI的"有帮助、符合伦理且可靠"的承诺。对于AI研究者和开发者而言,Dromedary的开源代码和文档提供了宝贵的学习资源,推动整个行业向更负责任的AI发展方向前进。

想了解更多细节?可以查阅项目的llama_dromedary/README.md和training/README.md获取完整技术文档。

【免费下载链接】DromedaryDromedary: towards helpful, ethical and reliable LLMs.项目地址: https://gitcode.com/gh_mirrors/dr/Dromedary

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

← 返回列表