本文介绍了 RUGGED (Retrieval Under Graph-Guided Explainable disease Distinction),它将大型语言模型 (LLM) 推理与检索增强生成 (RAG) 集成在一起。它从专家策划的生物医学知识库和同行评审的生物医学出版物中汲取证据,从最新信息中综合新知识,确定可解释和可作的预测,并为假设驱动的调查确定有希望的方向。
Method Article
本文介绍了 RUGGED (Retrieval Under Graph-Guided Explainable disease Distinction),它将大型语言模型 (LLM) 推理与检索增强生成 (RAG) 集成在一起。它从专家策划的生物医学知识库和同行评审的生物医学出版物中汲取证据,从最新信息中综合新知识,确定可解释和可作的预测,并为假设驱动的调查确定有希望的方向。
生物医学知识的规模(涵盖科学文献和精选知识库)对研究人员在有效处理、评估和解释结果方面构成了重大挑战。大型语言模型 (LLM) 已成为驾驭这种复杂知识领域的强大工具,但可能会产生幻觉反应。检索增强生成 (RAG) 对于识别相关信息以提高准确性和可靠性至关重要。该协议引入了 RUGGED (Retrieval Under Graph-Guided Explainable disease Distinction),这是一个全面的工作流程,旨在支持知识整合、减少偏差以及探索和验证新的研究方向。通过文本挖掘关联分析和可解释的图形预测模型,对出版物和知识库中的生物医学信息进行综合和分析,以揭示潜在的药物-疾病关系。这些发现以及源文本语料库和知识库被整合到一个框架中,该框架采用 RAG 增强的 LLM,使用户能够探索假设并研究潜在机制。一个临床用例展示了 RUGGED 在评估和推荐致心律失常性心肌病 (ACM) 和扩张型心肌病 (DCM) 疗法、分析分子相互作用处方药和潜在新应用方面的能力。该平台减少了 LLM 幻觉,突出了可作的见解,并简化了新疗法的研究。
生物医学企业中的假设探索过程对于揭示发病机制背后的新型分子-药物-疾病相互依存关系和释放治疗潜力至关重要 1,2。这个过程从现有的生物医学知识中汲取证据,根据嵌入在同行评审文献中的逻辑线索(例如,来自 PubMed 的 >36 份报告)综合新发现,并整合植根于生物医学知识库中的高置信度精选证据。最近的进展通过在文学语料库 3,4,5 上应用文本挖掘以及采用基于图的分析 6,7,8,9 来综合相关信息并发现新的调查途径,从而减少了费力的手动工作。尽管做出了这些努力,但由于数据分散,当前方法通常不支持深入的上下文理解。此外,他们缺乏得出基于证据的推论和交互式探索新假设的能力。
大型语言模型 (LLM) 的最新发展为这些挑战提供了新的视角,通过跨多个学科的大量信息训练展示了高水平的上下文理解 10,11,12。在生物医学领域,LLM 在提取患者信息13 和一般临床问答 14,15 方面显示出有前途的作用,而在特定领域的问答16 和初级临床保健17 中的实用程序中的应用仍有待探索。这些模型表现出从复杂数据集中推理和得出推理的能力,使其可能适合进行假设探索和知识综合。此外,一些模型具有类似聊天的交互功能,以吸引用户并支持对主题的动态探索,超越了基于查询的搜索引擎和知识库的传统界限18,19。
除了这些可能性之外,LLM 还面临着重大挑战,例如可能出现信息幻觉、对可能不准确的解释表现出不必要的信心、缺乏可解释性以及容易受到有偏见或不适当的内容的影响 20,21,22,23,24.直接应用于指导临床决策,LLM 衍生的反应和预测具有很高的风险;任何错误都可能导致昂贵的实验室实验或对患者的健康轨迹产生负面影响25,26。因此,可靠和值得信赖的 LLM 回应至关重要,因为他们的建议必须牢牢植根于证据。在这些情况下,可解释性不是一种奢侈品,而是理解这些模型为什么会做出预测的必要条件。
为此,检索增强生成 (RAG) 是一个旨在最大限度地减少 LLM 幻觉的系统,将 LLM 反应建立在证据的基础上,以提高其准确性和可靠性27,28。这种方法通常涉及检索相关的文本段落,例如将 LLM(例如 ChatGPT)与 PubMed 集成,从而允许识别用户查询的相关引文29,30。不限于文本,知识图谱 (KG) 上的检索有望应用于 LLM 完成事实核查31、32、33、透明推理34、35、36、编码知识37、改进问答38 和完成知识图谱 39 等任务.通过对来自经过验证的来源的事实信息进行编码,KG 提高了 LLM 回答的准确性、透明度和可靠性。这些图表中的链接预测技术利用深度学习来识别分子、药物和疾病之间以前隐藏的关系 5,40,41。可解释 AI 预测的最新进展进一步提高了这些链接预测任务的透明度和可解释性,为解释生物医学假设作为可行的调查途径提供了潜在支持42,43,44。这些进步确保了 LLM 生成的反应是平衡的并从证据中得出的,从而显着提高了它们在生物医学企业中的适用性。
该协议将 RUGGED (Retrieval Under Graph-Guided Explainable disease Distinction) 作为一种可访问且高效的工作流程,用于探索和验证临床治疗见解(图 1)。该工作流程协议利用生物医学文献和知识库的大量资源来提取和验证相关信息,从而实现查询定制的检索过程(图 2)。采用可解释的人工智能预测模型从现有的生物医学知识中发现可解释和可作的见解,从而提高预测模型的透明度和实用性。完整的工作流程通过支持 RAG 的 LLM 简化了对知识图谱和模型预测的探索,为研究人员、临床医生和临床专业人员提供了直观和明智的交互。
本节为协议奠定了基础,下一节将介绍实施此方法的步骤。接下来,展示了一个转化临床用例来演示这种方法,应用于药物的分子相互作用评估以及心血管医学的治疗策略。最后,讨论了该协议的含义和讨论。
该协议是在 Python 3.10 中开发的,并在 Windows 中作为 Docker 容器实现。提供的命令基于 Docker 容器中的 Unix 环境。该软件可在 https://github.com/pinglab-utils/RUGGED 下载。表 1 显示了协议中所有步骤的计算时间估计值。
1. 安装软件
2. 获取生物医学知识和提取信息
注意:这些步骤概述了两个知识提取管道,作为构成 RUGGED 检索增强生成 (RAG) 系统的底层信息:(1) CaseOLAP LIFT 生物医学文本挖掘管道5 和 (2) Know2BIO 知识图谱构建工作流程9。要将 RUGGED 与自定义数据一起使用,请继续执行步骤 4。
3. 可解释的预测分析
注意:在图卷积网络模型上执行 GNNExplainer44 以预测知识图谱中的潜在边缘(关系),并提供对以前未知关联的见解。
4. 假设生成
这些代表性结果是按照本协议中概述的程序获得的。按照具有默认参数的 CaseOLAP LIFT 协议5 进行文本挖掘关联分析,研究八大类心血管疾病72 及其与线粒体蛋白 (GO:0005739) 的关联。截至 2024 年 5 月,共有 635,696 份报告被确定为与这些疾病相关;其中,确定了 4,655 个高置信度的蛋白质-疾病关联,为下游分析提供信息。使用 Know2BIO 的软件代码在 2024 年 9 月使用默认设置构建了生物医学知识图谱。生成的知识图谱包括 219,450 个节点、6,323,257 条边以及 189,493 个节点的节点特征,包括节点描述、蛋白质/基因序列、化学结构等(如果可用)。 表 1 列出了协议中所有步骤的计算时间估计值。
RUGGED 系统是通过为知识图谱节点和特征以及 CVD 相关出版物构建矢量数据库来初始化的。所有知识图谱节点、边和节点特征都使用 BART71 嵌入模型以 20 个标记大小的块大小进行处理,为 RAG 向量搜索做准备。同样,使用 500 个标记的块大小和 BART 嵌入模型处理原始贡献和评论文章,为 RAG 向量搜索做准备。对于文献检索,大于 500 个标记的全文出版物通过 BART 嵌入模型根据出版物的各个部分进行分层汇总。GPT-4o 模型用于系统中剩余的 LLM 代理。
这些代表性结果展示了一个示例用例,用于研究致心律失常性心肌病 (ACM) 和扩张型心肌病 (DCM) 的潜在药物治疗,分别确定为 MeSH_Disease:D019571 和 MeSH_Disease:D002311。 图 3 概述了一系列查询, 图 4 中显示了模型响应的突出显示示例,补充 文件 1 中报告了完整的响应 A 部分。调查方向根据研究者验证的回答进行了调整,根据先前回答的结果制作了后续查询。分析揭示了 11 种候选药物,归类为 β 受体阻滞剂和抗心律失常药。使用图卷积神经网络链接预测模型对完整知识图谱的子集进行评估,包括研究疾病和药物节点及其互连的 1 跳内的节点,评估指标如 表 4 所示。 图形可解释性模块 GNNExplainer 44 进一步检查了模型每个预测的前 10 条相关边,以分别确定对每个预测做出贡献的前10 个节点和边。在撰写本文时,对于此使用案例,在 RUGGED 协议的所有步骤中使用商业 LLM 的总成本估计为 1.50 USD。

图 1:在图形引导的可解释疾病区分 (RUGGED) 工作流程下进行检索。 RUGGED 由四个主要组件组成:(1) 汇编和处理来自合乎道德和专业管理的资源(例如 PubMed 和精选的生物医学知识库)的数据,(2) 将同行评审的研究结果整合到一个统一的知识图谱中,(3) 在数据库服务中构建文本和图形数据,(4) 建模和预测知识图谱中生物医学实体之间的可解释关系, (5) 通过检索增强生成 (RAG) 工作流程检索和合成知识(图 2),以验证复杂的分子关系并探索 AI 驱动的疾病预测。用户可以执行人机协同审查步骤,以提高输出的准确性。 请单击此处查看此图的较大版本。

图 2:检索架构和偏差缓解工作流程。 检索增强生成 (RAG) 框架使用多个 LLM 代理,每个代理执行特定任务,以支持根据用户查询访问相关信息。该系统为面向用户的基于 GPT 的推理代理提供书面证据,促进用户与代理的交互和知识的综合。(1) 生物医学文本检索:同行评审的原始贡献和评论文章根据它们与理解疾病关联的相关性进行过滤。为作者和编辑验证的文本证据构建了一个矢量数据库,分别根据出版物的相应部分进行加权:70% 摘要、10% 结果、10% 元数据和 10% 用于所有其他子部分。针对用户查询的文本嵌入的关键字搜索和相似性搜索一起标识相关文档。每个文档的摘要都是使用基于 BERT 的摘要器生成的,基于 GPT 的文本评估器代理会优化搜索以验证查询-文档的相关性。(2) 知识图谱检索:基于 BERT 的命名实体识别和基于 GPT 的关系提取模块将用户查询连接到知识图谱中的相关实体。向量数据库中的相似性搜索可识别相关的节点和边。数据通过基于 GPT 的 Cypher 查询代理生成的 Cypher 查询从 Neo4j 数据库中检索,并由 Query Verification Agent 进行优化。(3) 来自生物医学文本检索或知识图谱检索管道的单个响应被呈现给推理代理,推理代理合成一个简洁的响应,对用户查询的偏见最小。该系统旨在保持呈现事实信息的准确性和公正性。 请单击此处查看此图的较大版本。

图 3:通过查询级联进行知识合成和假设探索的用例。该图展示了一个突出的使用案例,重点关注研究人员和/或医疗保健专业人员可能向 RUGGED 系统提出的一系列相关问题和概念。用户的查询按数字顺序呈现给系统,箭头表示每个问题中推断出的逻辑推理和特定于领域的推理。系统从隐式和相关信息(源以蓝色显示)中检索,以响应查询。图 4 显示了系统响应的示例。请单击此处查看此图的较大版本。

图 4:心血管病理学用例:阐明 CVD 发病机制。 显示了用户和 RUGGED 系统之间的查询-响应对。在左上角的面板中,问题 1-6 通过从知识图谱数据库中提取信息来检索信息,以形成基于证据的回答。问题 7 采用可解释的图形链接预测来识别得分最高的疗法。查询会提示预测分析,该分析由系统自动执行和处理,并简要总结关键发现。问题 8 评估来自定义文本语料库的文献证据,该语料库作为相关证据检索,以验证、确认和证实预测结果。系统响应已通过人机协同检查流程进行审查,并进行了修改以提高可读性和简洁性。这些发现的完整记录详见 补充文件 1。 请单击此处查看此图的较大版本。
| 步骤 | 描述 | 时间 |
| 获取生物医学知识 | 总计 30% | |
| 准备生物医学文献语料库 | 连接到 PubMed 和 PubMed Central,下载并解析下游任务的出版物数据。 | 20% |
| 准备知识库数据 | 连接到生物医学知识库,下载并解析下游任务所需的信息。 | 5% |
| 信息提取 | 总计 30% | |
| CaseOLAP LIFT 文本挖掘分析 | 识别生物医学文本语料库中的高级别疾病-蛋白质关系。 | 25% |
| 知识图谱构建 | 将生物医学知识库中的不同信息连接并集成到一个统一的知识图谱中。 | 5% |
| 预测分析 | 总计 10% | |
| 训练图神经网络 | 在生物医学知识图谱数据上训练模型,以学习图中的隐藏模式。 | 5% |
| 相关性排名分析 | 应用可解释性模块来突出与研究疾病相关的最相关的节点和边缘。 | 2.5% |
| 链接预测 | 利用可解释性模块来识别影响新预测边缘的关键节点和边缘。 | 2.5% |
| 假设验证生成和/或验证 | 总计 30% | |
| 用于检索 Augmented Generation 的数据库设置 | 初始化图数据库以查询知识图谱,并初始化矢量数据库以检索文本。 | 25% |
| 假设探索 | 支持用户与 RUGGED 交互,以访问和仔细检查相关信息以进行假设探索。 | 5% |
表 1:工作流程和速率限制步骤。 此表提供了工作流每个阶段所需的计算时间的粗略估计值。速率限制步骤包括访问、提取和索引检索增强生成所需的生物医学知识。假设探索可以连续重复,而无需重新执行速率限制步骤。
| 疾病类别 | MeSH 树编号 | # PMID | # 原始贡献 | # 评论文章 |
| 心肌病 (CM) | 编号: C14.280.238 | 132,531 | 102,337 | 19,942 |
| 编号: C14.280.434 | ||||
| 心律失常 (ARR) | 货号 C14.280.067 | 125,286 | 92,374 | 13,854 |
| 编号: C23.550.073 | ||||
| 先天性心脏缺陷 (CHD) | 编号: C14.280.400 | 82,006 | 54,023 | 6,379 |
| 心脏瓣膜疾病 (VD) | 编号 C14.280.484 | 72,016 | 50,119 | 5,743 |
| 心肌缺血 (IHD) | 编号: C14.280.647 | 256,986 | 210,042 | 30,223 |
| 心脏传导系统疾病 (CCD) | 编号 C14.280.123 | 53,050 | 35,399 | 4,363 |
| 心室流出道梗阻 (VOO) | 编号: C14.280.955 | 22,244 | 15,504 | 1,686 |
| 其他心脏病 (OTH) | C14.280.195 C14.280.282 C14.280.383 C14.280.470 C14.280.945 C14.280.459 C14.280.720 | 114,085 | 77,302 | 11,799 |
| 总 | 635,696 | 478,404 | 69,690 |
表 2:生物医学文献统计。 该表详细介绍了研究疾病类别及其相应的 MeSH 树编号以及从 2024 年 5 月检索的 PubMed 文档数量,用作文本挖掘的语料库。这些出版物的子集(包括原始贡献研究文章和评论文章)被索引到向量数据库中,以便在假设生成期间由 RUGGED 检索。
| 类别 | 节点数 | 边数 | 数据源 |
| 解剖学 | 5,049 | 122,533 | Bgee、PubMed、MeSH、Uberon、 |
| 生物过程 | 27,047 | 108,106 | 基因本体论 |
| 蜂窝组件 | 4,057 | 52,238 | 基因本体论 |
| 复合 | 27,278 | 3,292,028 | DrugBank、MeSH、CTD、UMLS、KEGG、TTD、SIDER、Inxight Drugs、Hetionet、PathFX MyChem.info |
| 疾病 | 21,938 | 311,773 | PubMed、MeSH、DisGeNET、SIDER、ClinVar、ClinGen、PharmGKB、MyDisease.info、PathFX、UMLS、OMIM、Mondo、DOID、KEGG |
| 药品类别 | 5,721 | 8,283 | 空管 |
| 基因 | 29,810 | 943,419 | HGNC、GRNdb、KEGG、ClinVar、ClinGen、 |
| 分子功能 | 11,151 | 47,086 | SMPDB、DisGENET、PharmGKB MyGene.info |
| 路 | 52,012 | 234,944 | 基因本体论 |
| 蛋白 | 20,740 | 1,074,809 | Reactome、KEGG、SMPDB |
| 反应 | 14,647 | 128,038 | UniProt、Reactome、TTD、SMPDB、STRING、HGNC |
| 小计 | 219,450 | 6,323,257 | Reactome 公司 |
| 文本挖掘关联 | 8 | 4,670 | |
| 总 | 219,458 | 6,327,927 |
表 3:知识图统计。 该表详细介绍了构成构建的 Know2BIO 知识图谱的 11 个广泛的生物医学类别,并丰富了来自文本挖掘分析和预测分析的额外边缘。生成的知识图谱和预测由 Neo4j 图形数据库管理,以便在假设生成期间由 RUGGED 进行检索。
| 准确性 | 精度 | 召回 | F1 分数 | 奥洛克 | AUPRC | |
| 验证 | 0.7158 | 0.6639 | 0.8743 | 0.7547 | 0.8437 | 0.8637 |
| 测试 | 0.703 | 0.6367 | 0.9455 | 0.761 | 0.8961 | 0.9094 |
表 4:可解释的 AI 模型评估。 下表报告了使用两层图卷积神经网络进行知识图谱链接预测的评估指标。通过将图形边缘划分为 85% 的训练、5% 的验证和 10% 的测试数据集来评估指标。准确率表示正确分类的预测的比例。精度报告正确正预测在所有正预测中的比例。召回率衡量正确的正预测在实际正边缘中的比例。F1 分数是精确率和召回率的调和平均值,用于平衡这两个指标。AUROC 评估模型区分正预测和消极预测的能力。AUPRC 量化了不同阈值下精确率和召回率之间的权衡。对于所有指标,值越高表示模型性能越好。
补充文件 1: 该文件详细介绍了 RUGGED 的完整模型响应以及与 GPT-4o 的比较。 A 部分 介绍了 RUGGED 的完整人机交互,扩展了 图 3 中概述的查询链方法,并在 图 4 中突出显示的摘要之外提供了完整的响应。 B 部分 评估了 GPT-4o 的响应(无检索)与 RUGGED 的响应,评估了精度、深度、置信度评分、证据可靠性和成本等属性。 请点击此处下载此文件。
RUGGED 协议利用具有最新信息的现代语言模型,使研究人员能够动态探索不断发展的生物医学领域并发现新知识。这种人机交互推动了一个创新过程,体现了机器的效率 (RUGGED) 以及研究者的专业知识和判断力。该协议旨在按概述的顺序执行。步骤 1 详细说明软件安装。第 2 步和第 3 步对于准备生物医学文献和资源至关重要,而第 4 步为这些信息编制索引,以便进行检索增强生成和用户与 LLM 系统的交互。耗时的步骤可以同时和/或按顺序运行。例如,创建 Neo4j 图(步骤 4.2.2)可以在预测分析(步骤 3)期间开始,索引可以在构建知识图谱(步骤 2.3)和文本挖掘(步骤 2.1)后开始。必须重复这些步骤才能获得这些中间结果的最终结果。虽然该协议专为生物医学信息检索而设计,但经过细微修改后,还可以处理其他文本和图形数据,例如内部数据、临床记录或电子健康记录。数据格式详细信息位于步骤 4.2 中。
该平台的运行依赖于多种技术的正确安装和互连,包括语言模型、图形数据库和矢量数据库(参见 材料表)。为了验证这些服务是否已正确安装和连接,GitHub 存储库中的“test” 文件夹中提供了测试脚本。外部服务可能会产生费用,价格可能会因供应商而有所变化。这些可选服务还具有本地托管的替代方案,只需要足够的计算资源。但是,这些替代方案可能会影响模型性能和/或便利性,使其不适用于某些用例场景。
随着 LLM 领域的快速发展,新的地标模型和特定于任务的模型会定期发布。在本报告发布时,为该任务选择了最合适的模型。用户可以通过相应地更新配置文件来选择要使用的 LLM(请参阅步骤 1.3.2-1.3.4)。模型选择取决于它们与特定用例的相关性。例如,将专注于确保模型响应公平、经过审查且没有仇恨言论 73,74,75,76,77,78 的模型纳入此工作流程对于道德考虑至关重要。此外,及时的工程设计对于指导 LLM 79,80,81,82 的可靠和负责任的行为至关重要。为 RUGGED 工作流程制作的提示是根据所采用的模型和展示的用例量身定制的。要针对不同的用例微调提示,用户可以在 RUGGED 工作流程中“prompts.json”文件中的“configuration”文件夹中编辑提示。
虽然 RAG 系统旨在通过将反应建立在证据的基础上来减少 LLM 中的幻觉,但这些模型仍可能导致不准确的信息或通常正确的非特异性反应。补充文件 1 的 B 节提供了 RUGGED 与 GPT-4o 的基准比较。当检索到的信息超出模型的上下文窗口时,通常会发生模型幻觉,类似于记忆丧失和无法定位数据内容的痴呆,导致响应不准确 83,84,85.选择合适的 LLM 模型有助于缓解此问题。例如,GPT-4o 的上下文限制为 128k 令牌,明显高于 GPT-3.5 Turbo 的 16k 令牌限制,尽管用户的成本更高。此外,使用特定领域知识进行微调的 LLM 可能会提高生物医学应用中响应的准确性和特异性 86,87,88。尽管采取了这些措施,但在进行昂贵的湿实验室实验之前,必须交叉检查信息。
RUGGED 利用 RAG 管道中的可解释 AI 来仔细检查链接预测,识别可靠和以前未发现的关系。虽然传统的 RAG 系统依赖于基于批量相似性的检索,但这种方法将可解释性与有针对性的响应增强联系起来。 表 4 突出了该模型的强劲性能,展示了高召回率(验证:0.975 测试:0.976)和平衡的 F1 分数(验证:0.796,测试:0.797),表明识别真阳性的可靠性,尽管假阳性率更高。该模型的稳健性进一步得到了其 AUROC(验证:0.963,测试:0.964)和 AUPRC(验证:0.971,测试:0.972)值的支持。然而,精度(验证:0.673,测试:0.674)可以从阈值调整、结合详细的节点特征或改进类不平衡的处理中受益。模型的有效性在很大程度上取决于输入的知识图谱;较小的图形存在过度拟合的风险,而较大的图形需要更多的计算资源。但是,任何基于 RAG 的方法在很大程度上都取决于检索基础数据的质量。例如,由于原始图谱上的固有噪声,知识图谱的构建通常是耗时耗力的。这需要手动进行降噪和标记,以及数据库维护和更新的持续成本。
RUGGED 的主要用途是知识合成和假设探索。通过研究各种隐藏的关系,例如疾病机制和药物治疗,RUGGED 可以有效地进行文献分类。为了减少计算负担,大多数应用程序可以托管在服务器(例如 AWS 或计算服务器)上,并配置为定期更新最新信息。此外,此工作流可以适应完成特定于领域的应用程序,例如充当一个平台,将患者数据与本地模型一起包含,以维护安全性、隐私性和机密性。除了生物医学研究之外,RUGGED 的模块化设计使其能够通过定制 RAG 管道和针对目标领域量身定制的工程策略来支持信息检索、推理和摘要等任务。成功的适应需要仔细考虑特定于领域的挑战,例如对各种数据格式的预处理以及评估适合任务和领域需求的模型。
作者没有什么可披露的。
作者要感谢 Alex Bui 博士的指导和深思熟虑的讨论。此外,我们感谢 Ding Wang 博士的有益讨论。这项工作得到了 NIH 1U54HG012517-01 对 P.P.、KW 和 WW 的部分支持;NIH T32 HL13945 到 A.R.P.;美国国家科学基金会研究实习生 (NRT) 1829071 A.R.P.;以及加州大学洛杉矶分校 (UCLA) 的 TC Laubisch Endowment to P.P. 捐赠基金。
| Name | Company | Catalog Number | Comments |
|---|---|---|---|
| 硬件/软件 - 显卡和软件驱动程序 | Nvidia | https://www.nvidia.com | 强烈建议使用显卡及其相关的驱动程序软件来显着减少计算密集型任务(例如本地 LLM 和预测分析)的运行时间。对于配备 NVIDIA RTX GPU 的设备,请从 NVIDIA 网站 (https://developer.nvidia.com/cuda-downloads) 下载并安装必要的驱动程序和 CUDA 工具包。 |
| 软件 - 商业大语言模型服务 | OpenAI | https://openai.com | RUGGED 支持 GPT-3.5 和 GPT-4o 等模型的 OpenAI API。要使用 OpenAI 模型进行设置,请先获取 OpenAI API 密钥。前往 OpenAI 网站 (https://openai.com/blog/openai-api) 创建帐户、充值资金并获取 API 密钥。需要此 API 密钥才能使 RUGGED 使用 OpenAI 模型。从其文档 (https://platform.openai.com/docs/models) 中确定 RUGGED 系统中的哪些 LLM 代理将使用 OpenAI 模型。 注意:OpenAI API 是一项付费服务。在发布时,GPT-4o 的成本为每 100 万个输入代币 5.00 美元,每 100 万个输出代币 2.50 美元(欲了解更多信息,请访问 https://openai.com/pricing)。 |
| 软件 - 容器化Docker | https://www.docker.com | Docker 有助于维护一致的计算运行时环境,简化跨不同机器的软件安装和执行。要安装 Docker,请访问 Docker 网站 (https://www.docker.com/),单击"开始",下载并安装适合作系统的版本。通过在终端中键入"docker --version"来验证安装;成功安装报告已安装的 Docker 版本。 | |
| 软件 - 图数据库 | Neo4j | https://neo4j.com | Neo4j 是一款图数据库软件,可以高效管理和查询基于图的节点和关系。RUGGED 支持多种形式的 Neo4j:Docker 容器、Neo4j Desktop 或 Neo4j AuraDB 在线服务器。选择最适合用例的选项。 将 Neo4j 设置为 Docker 容器。运行这些命令以在 Docker 中设置 Neo4j,文件夹的文件路径(例如 /Users/username/RUGGED)为"PATH_TO_FOLDER"。有关故障排除的更多详细信息,请参阅 Neo4j Docker 网站 (https://hub.docker.com/_/neo4j)。 docker pull neo4j docker run –名称 neo4j --net rugged_network --publish=7474:7474 --publish=7687:7687 -d -v 'PATH_TO_FOLDER'\neo4j\data:/data neo4j 注意:通过设置用户名和密码,首次在 Docker 中初始化 Neo4j。运行 neo4j_setup.py 脚本(例如 python neo4j_setup.py)或通过 Web 界面运行 http://localhost:7474. 设置 Neo4j 桌面。如果使用 Neo4j Desktop,请从 Neo4j 网站 (https://neo4j.com/) 下载并安装。单击"新建"创建一个新项目,然后单击"添加"以创建新的数据库管理系统 (DBMS)。选择"本地DBMS",设置密码,点击"创建",然后点击"开始"。绿色的"ACTIVE"文本表示它正在运行。 设置 Neo4j AuraDB。访问 Neo4j 网站 (https://neo4j.com/cloud/aura-free/) 创建一个帐户并登录。选择"新建实例"创建一个空实例,并保存 URI 和初始密码以访问 bolt 接口(例如 bolt://myurl.neo4j.com)。单击播放按钮启动实例,这将在信息框中显示连接 URI。 注意:Neo4j AuraDB 提供高达 200,000 个节点和 400,000 个关系的免费套餐。如需更大的图表,请访问 Neo4j 定价 (https://neo4j.com/pricing)。 |
| 软件 - 本地大语言模型服务 | Ollama | https://ollama.com | RUGGED 支持使用 Ollama 使用本地模型(例如 Llama3)。要启用,请先在设备上安装 Ollama 或下载 Docker 容器。要安装 Ollama,请访问 Ollama 网站 (https://ollama.com/download) 并按照安装说明进行作。要在 Docker 上安装 Ollama,请运行以下命令: docker pull ollama/ollama 注意:在发布时,Windows作系统上没有 Ollama 的稳定版本。 |
| 软件 - 版本控制 | Git | https://www.git-scm.com | 版本控制软件可以高效地安装和更新软件。要安装 Git,请访问 Git 网站 (https://www.git-scm.com/),单击"下载",下载并安装适合作系统的版本。通过在终端中键入"git --version"来验证安装;安装成功将报告已安装的 Git 版本。 |
Request permission to reuse the text or figures of this JoVE article
Request Permission