基于不确定性量化检测大语言模型幻觉的 Python 库,提供多类置信度评分器。
HowieHwong/TrustLLM
基于 ICML 2024 TrustLLM 基准的开源工具包,通过 CLI 和 Python 从六个维度评估大模型的可信度。
- Star
- 635
- Fork
- 69
- 未关闭 Issue
- 8
- 最近提交
- 上个月
这是什么?
TrustLLM 是一个研究型工具包,用于从真实性、安全性、公平性、鲁棒性、隐私和伦理六个维度衡量大语言模型的可信程度。它封装了 ICML 2024 基准的数据集,可以针对本地 Hugging Face 权重或任何兼容 OpenAI 接口的 API 运行测试,并将数据、配置和输出保存在同一个运行目录中。
流程分为生成与评分两步。生成步骤按任务收集模型回复,支持逐样本断点保存、有限次重试,并生成展示完成数量的 HTML 报告。评分步骤对完整的回复集进行打分,根据任务不同,会使用规则、下载的分类器、向量嵌入或 API 评判模型。基础安装包较轻量,可选依赖用于本地模型加载和评分。
项目还提供 CLI 与 Python 调用方式并输出 JSON 结果,便于用脚本或 AI 智能体自动化执行整个评测流程。
核心功能
- 01覆盖六个可信度维度,使用 ICML 2024 基准数据集。
- 02支持对本地 Hugging Face 因果语言模型或兼容 OpenAI 接口的 API 生成回复。
- 03通过 trustllm 命令行提供 download、tasks、generate、evaluate 等命令。
- 04逐样本保存断点,中断的运行可安全续跑。
- 05根据任务使用规则、分类器、向量嵌入或 API 评判模型进行打分。
适用场景
- 在统一基准下比较多个大模型的安全性与真实性。
- 在部署前检测微调或本地模型对越狱攻击的脆弱性。
- 为学术研究复现 ICML 2024 TrustLLM 的评测结果。
- 在完整评测前,用小样本快速验证模型 API 是否可用。
快速上手
- 1
从 GitHub 安装基础包(需要 Git)
python -m pip install "trustllm @ git+https://github.com/HowieHwong/TrustLLM.git@main#subdirectory=trustllm_pkg" - 2
下载基准数据集
python -m trustllm download --output data - 3
设置 API 密钥和端点,然后运行冒烟测试生成
export OPENAI_API_KEY="your-api-key" export OPENAI_BASE_URL="https://your-provider.example/v1" python -m trustllm generate \ --backend api --model your-model-id \ --task safety --data data/dataset \ --limit 5 --concurrency 4 --output runs/api-safety
使用前须知
- 完整运行的评分可能调用 API 评判模型,产生费用。
- 生成设置与原论文不同,新运行结果不能直接等同于论文结果。
- 安装需从 GitHub 主分支获取,而非已发布的正式包版本。
收录于专题
标签
#llm-evaluation#benchmark#ai-safety#trustworthy-ai#hallucination#responsible-ai#openai-compatible#python#ai#ai-agents
常见问题
- TrustLLM 是什么?
- 基于 ICML 2024 TrustLLM 基准的开源工具包,通过 CLI 和 Python 从六个维度评估大模型的可信度。 TrustLLM 是一个研究型工具包,用于从真实性、安全性、公平性、鲁棒性、隐私和伦理六个维度衡量大语言模型的可信程度。它封装了 ICML 2024 基准的数据集,可以针对本地 Hugging Face 权重或任何兼容 OpenAI 接口的 API 运行测试,并将数据、配置和输出保存在同一个运行目录中。
- TrustLLM 可以免费使用吗?用的什么许可证?
- TrustLLM 以 MIT 许可证开源。用于商业产品前,请先确认许可证条款。
- TrustLLM 还在维护吗?
- 半年内有更新。TrustLLM 最近一次提交是在 2026年10月4日。
- TrustLLM 适合用来做什么?
- 在统一基准下比较多个大模型的安全性与真实性。 在部署前检测微调或本地模型对越狱攻击的脆弱性。 为学术研究复现 ICML 2024 TrustLLM 的评测结果。 在完整评测前,用小样本快速验证模型 API 是否可用。
- TrustLLM 怎么安装和运行?
- 从 GitHub 安装基础包(需要 Git): python -m pip install "trustllm @ git+https://github.com/HowieHwong/TrustLLM.git@main#subdirectory=trustllm_pkg" 下载基准数据集: python -m trustllm download --output data 设置 API 密钥和端点,然后运行冒烟测试生成: export OPENAI_API_KEY="your-api-key" export OPENAI_BASE_URL="https://your-provider.example/v1" python -m trustllm generate \ --backend api --model your-model-id \ --task safety --data data/dataset \ --limit 5 --concurrency 4 --output runs/api-safety
- 在哪里获取 TrustLLM?
- 源代码和安装说明在 GitHub 上:https://github.com/HowieHwong/TrustLLM
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
AI 与机器学习#hallucination-detection#uncertainty-quantification#llm-evaluation#confidence-estimation#llm-safety
1217+2 本周140PythonApache-2.0维护活跃更新于 25天前
Python 库,用一行代码为 NLP 模型和大语言模型生成并运行鲁棒性、偏见、公平性等测试。
56153PythonApache-2.0维护活跃更新于 2个月前
用于评测大模型作为自主智能体能力的评测框架,包含 300 个人工验证任务。
906+39 本周96PythonMIT更新于 昨天
一款 Python 命令行工具,通过 61 项检测审计 AI 智能体是否按职责行事、能否抵御操纵、是否编造信息。
4.4万+4602 本周1785PythonApache-2.0更新于 24小时前
开源命令行工具,用于探测大语言模型在提示注入、数据泄露、越狱和有害输出等方面的漏洞。
9905+109 本周1444PythonApache-2.0维护活跃更新于 3天前
一个用于构建稳健强化学习环境的开源 Python 框架,在隔离虚拟机中运行智能体任务。
186+5 本周21PythonMIT维护活跃更新于 前天