AIX360

AIX360

全面支持多种数据类型和机器学习模型解释的开源工具包

工具包提供全面支持机器学习模型和数据集的可解释性,涵盖多种解释算法和指标,包括表格、文本、图像及时间序列数据。该项目提供互动体验、教程和示例,并提供指导材料和分类树帮助选择算法。工具包易于扩展,欢迎贡献新的算法、指标和用例。

AI Explainability 360解释性工具包机器学习模型算法AIX360Github开源项目

AI可解释性360 (v0.3.0)

Build Documentation Status PyPI version

AI可解释性360工具包是一个开源库,支持数据集和机器学习模型的可解释性和可解释性。AI可解释性360 Python包包括一整套算法,涵盖了不同维度的解释以及代理可解释性指标。AI可解释性360工具包支持表格数据、文本、图像和时间序列数据。

AI可解释性360互动体验通过不同消费者角色的示例用例,提供了一个温和的介绍。 教程和示例笔记本提供了更深层次的数据科学家导向的介绍。完整的API也可用。

没有一种单一的可解释性方法是最好的。有许多解释方式:数据 vs 模型,直接可解释 vs 事后解释,局部 vs 全局等。因此,可能很难确定哪些算法最适合特定的用例。为此,我们创建了一些指导材料和一个分类树供参考。

我们在开发这个包时考虑了可扩展性。这个库还在开发中。我们鼓励您贡献您的可解释性算法、指标和用例。如需作为贡献者入门,请通过这里请求邀请后加入AI可解释性360社区。请查看贡献代码和Python笔记本的说明

支持的可解释性算法

数据解释

局部事后解释

时间序列局部事后解释

  • 用集成梯度生成的时间序列显著性图(Inspired by Sundararajan等
  • 时间序列LIME(时间序列适配经典论文Ribeiro等,2016
  • 时间序列个体条件期望(时间序列适配个体条件期望图Goldstein等

局部直接解释

认证的局部解释

全局直接解释

全局事后解释

支持的可解释性指标

设置

支持的配置:

安装关键字解释器操作系统Python版本
cofrnetcofrnetmacOS, Ubuntu, Windows3.10
contrastivecem, cem_mafmacOS, Ubuntu, Windows3.6
dipvaedipvaemacOS, Ubuntu, Windows3.10
gcegcemacOS, Ubuntu, Windows3.10
ecertifyecertifymacOS, Ubuntu, Windows3.10
imdimdmacOS, Ubuntu3.10
limelimemacOS, Ubuntu, Windows3.10
matchingmatchingmacOS, Ubuntu, Windows3.10
nncontrastivenncontrastivemacOS, Ubuntu, Windows3.10
profwtprofwtmacOS, Ubuntu, Windows3.6
protodashprotodashmacOS, Ubuntu, Windows3.10
rbmbrcg, glrmmacOS, Ubuntu, Windows3.10
rule_inductionrippermacOS, Ubuntu, Windows3.10
shapshapmacOS, Ubuntu, Windows3.6
tedtedmacOS, Ubuntu, Windows3.10
tsicetsicemacOS, Ubuntu, Windows3.10
tslimetslimemacOS, Ubuntu, Windows3.10
tssaliencytssaliencymacOS, Ubuntu, Windows3.10

(可选)创建虚拟环境

AI Explainability 360 需要特定版本的许多 Python 包,这些包可能与系统上的其他项目冲突。强烈建议使用虚拟环境管理器以确保依赖关系能够安全安装。如果您在安装工具包时遇到问题,请首先尝试此方法。

Conda

推荐使用 Conda,但虚拟环境通常也可以互换使用。Miniconda 就足够了(有关 Anaconda 和 Miniconda 之间的区别,请参见此处),如果您尚未安装,可以从这里下载。

然后,根据您希望使用的可解释性算法,通过参考上表,创建一个新的 Python 环境。例如,对于 Python 3.10,使用以下命令:

conda create --name aix360 python=3.10 conda activate aix360

命令提示符应该现在显示 (aix360) $。要退出环境,运行:

(aix360)$ conda deactivate

提示符将返回到 $ (base)$

注意: 较旧版本的 conda 可能使用 source activate aix360source deactivate(在 Windows 上使用 activate aix360deactivate)。

安装

克隆本仓库的最新版本:

(aix360)$ git clone https://github.com/Trusted-AI/AIX360

如果您想运行示例和教程笔记本,现在下载数据集并将其放在各自的文件夹中,新详细描述见 aix360/data/README.md

然后,导航到包含 setup.py 文件的项目根目录并运行:

(aix360)$ pip install -e .[<algo1>,<algo2>, ...]

上述命令安装特定算法所需的程序包。这里 <algo> 指的是上面中的安装关键词。例如,要安装 BRCG、DIPVAE 和 TSICE 算法需要的包,可以使用

(aix360)$ pip install -e .[rbm,dipvae,tsice]

默认命令 pip install . 仅安装默认依赖

请注意,您可能无法在同一环境中安装需要不同版本的 python 的两个算法(例如 contrastiverbm)。

如果您遇到任何问题,请尝试升级 pip 和 setuptools 并卸载任何以前版本的 aix360,然后再尝试上述步骤。

(aix360)$ pip install --upgrade pip setuptools (aix360)$ pip uninstall aix360

AI Explainability 360 的 PIP 安装

如果您想快速开始使用 AI explainability 360 工具包而不明确克隆此存储库,可以使用以下选项之一:

  • 通过存储库链接安装 v0.3.0
(your environment)$ pip install -e git+https://github.com/Trusted-AI/AIX360.git#egg=aix360[<algo1>,<algo2>,...]

例如,使用 pip install -e git+https://github.com/Trusted-AI/AIX360.git#egg=aix360[rbm,dipvae,tsice] 安装 BRCG、DIPVAE 和 TSICE。如果环境中尚未安装 cmake,可以使用 conda install cmake 安装。

  • 通过 pypi 安装 v0.3.0(或以前版本)
(your environment)$ pip install aix360

如果您遵循上述任一选项,您需要单独下载 examples 文件夹中提供的笔记本。

处理安装错误

AI Explainability 360 工具包在 Windows、MacOS 和 Linux 上进行了测试。然而,如果您仍然因包依赖问题遇到安装问题,请尝试通过 conda 安装相应的包(例如 conda install package-name),然后按照常规步骤继续安装工具包。例如,如果您在安装过程中与 pygraphviz 相关的问题,使用 conda install pygraphviz,然后安装工具包。

请根据上方中的信息使用正确的 python 环境。

在 Docker 中运行

  • AIX360 目录下,通过 docker build -t aix360_docker . 从 Dockerfile 构建容器镜像
  • 使用命令 docker run -it -p 8888:8888 aix360_docker:latest bash 启动容器镜像,假设您的机器上端口 8888 是空闲的
  • 在容器内使用命令 jupyter lab --allow-root --ip 0.0.0.0 --port 8888 --no-browser 启动 jupyter lab
  • 使用 URL localhost:8888 在您的机器上访问示例教程

使用 AI Explainability 360

examples 目录包含一系列使用 AI Explainability 360 各种方式的 jupyter 笔记本。示例和教程笔记本都展示了使用该工具包的实际代码。教程提供了额外的讨论,详细介绍了笔记本的各个步骤。请参阅此处的教程和示例详细信息。

引用 AI Explainability 360

如果您在工作中使用 AI Explainability 360,我们鼓励您

  • 引用以下论文。bibtex 条目如下:
@misc{aix360-sept-2019,
title = "One Explanation Does Not Fit All: A Toolkit and Taxonomy of AI Explainability Techniques",
author = {Vijay Arya and Rachel K. E. Bellamy and Pin-Yu Chen and Amit Dhurandhar and Michael Hind
and Samuel C. Hoffman and Stephanie Houde and Q. Vera Liao and Ronny Luss and Aleksandra Mojsilovi\'c
and Sami Mourad and Pablo Pedemonte and Ramya Raghavendra and John Richards and Prasanna Sattigeri
and Karthikeyan Shanmugam and Moninder Singh and Kush R. Varshney and Dennis Wei and Yunfeng Zhang},
month = sept,
year = {2019},
url = {https://arxiv.org/abs/1909.03012}
}

AIX360 视频

  • Vijay Arya 和 Amit Dhurandhar 介绍 AI Explainability 360 的视频,2019 年 9 月 5 日(35 分钟)

致谢

AIX360 是在若干开源包的帮助下构建的。这些包都列在 setup.py 中,其中包括:

许可信息

请查看根目录中 LICENSE 文件和文件夹 supplementary license 了解许可证信息。

编辑推荐精选

商汤小浣熊

商汤小浣熊

最强AI数据分析助手

小浣熊家族Raccoon,您的AI智能助手,致力于通过先进的人工智能技术,为用户提供高效、便捷的智能服务。无论是日常咨询还是专业问题解答,小浣熊都能以快速、准确的响应满足您的需求,让您的生活更加智能便捷。

imini AI

imini AI

像人一样思考的AI智能体

imini 是一款超级AI智能体,能根据人类指令,自主思考、自主完成、并且交付结果的AI智能体。

Keevx

Keevx

AI数字人视频创作平台

Keevx 一款开箱即用的AI数字人视频创作平台,广泛适用于电商广告、企业培训与社媒宣传,让全球企业与个人创作者无需拍摄剪辑,就能快速生成多语言、高质量的专业视频。

即梦AI

即梦AI

一站式AI创作平台

提供 AI 驱动的图片、视频生成及数字人等功能,助力创意创作

扣子-AI办公

扣子-AI办公

AI办公助手,复杂任务高效处理

AI办公助手,复杂任务高效处理。办公效率低?扣子空间AI助手支持播客生成、PPT制作、网页开发及报告写作,覆盖科研、商业、舆情等领域的专家Agent 7x24小时响应,生活工作无缝切换,提升50%效率!

TRAE编程

TRAE编程

AI辅助编程,代码自动修复

Trae是一种自适应的集成开发环境(IDE),通过自动化和多元协作改变开发流程。利用Trae,团队能够更快速、精确地编写和部署代码,从而提高编程效率和项目交付速度。Trae具备上下文感知和代码自动完成功能,是提升开发效率的理想工具。

AI工具TraeAI IDE协作生产力转型热门
蛙蛙写作

蛙蛙写作

AI小说写作助手,一站式润色、改写、扩写

蛙蛙写作—国内先进的AI写作平台,涵盖小说、学术、社交媒体等多场景。提供续写、改写、润色等功能,助力创作者高效优化写作流程。界面简洁,功能全面,适合各类写作者提升内容品质和工作效率。

AI辅助写作AI工具蛙蛙写作AI写作工具学术助手办公助手营销助手AI助手
问小白

问小白

全能AI智能助手,随时解答生活与工作的多样问题

问小白,由元石科技研发的AI智能助手,快速准确地解答各种生活和工作问题,包括但不限于搜索、规划和社交互动,帮助用户在日常生活中提高效率,轻松管理个人事务。

热门AI助手AI对话AI工具聊天机器人
Transly

Transly

实时语音翻译/同声传译工具

Transly是一个多场景的AI大语言模型驱动的同声传译、专业翻译助手,它拥有超精准的音频识别翻译能力,几乎零延迟的使用体验和支持多国语言可以让你带它走遍全球,无论你是留学生、商务人士、韩剧美剧爱好者,还是出国游玩、多国会议、跨国追星等等,都可以满足你所有需要同传的场景需求,线上线下通用,扫除语言障碍,让全世界的语言交流不再有国界。

讯飞智文

讯飞智文

一键生成PPT和Word,让学习生活更轻松

讯飞智文是一个利用 AI 技术的项目,能够帮助用户生成 PPT 以及各类文档。无论是商业领域的市场分析报告、年度目标制定,还是学生群体的职业生涯规划、实习避坑指南,亦或是活动策划、旅游攻略等内容,它都能提供支持,帮助用户精准表达,轻松呈现各种信息。

AI办公办公工具AI工具讯飞智文AI在线生成PPTAI撰写助手多语种文档生成AI自动配图热门
下拉加载更多