PiML-Toolbox
PiML-Toolbox 是一款专为可解释机器学习打造的 Python 集成工具箱,旨在让模型开发与诊断过程更加透明、可靠。它主要解决了传统机器学习模型“黑箱”操作难以理解、缺乏信任度的问题,帮助开发者深入洞察模型决策逻辑,并有效识别过拟合、公平性偏差及鲁棒性不足等潜在风险。
这款工具非常适合数据科学家、机器学习工程师以及需要应对高监管要求的研究人员使用。无论是构建新模型还是验证外部“黑箱”模型,PiML-Toolbox 都能提供强力支持。其独特亮点在于兼顾了低代码图形界面与高代码 API 两种使用方式,既方便快速上手,又满足深度定制需求。工具内置了多种天生具备可解释性的先进算法(如 EBM、GAMI-Net 及特定深度的梯度提升树),并提供了从全局到局部的全方位解释器(如 SHAP、LIME)。此外,它还涵盖了准确性评估、公平性测试、弱区识别及抗干扰能力分析等丰富的诊断功能,让用户能全面掌握模型性能,从而做出更自信的决策。
使用场景
某金融风控团队正在开发一款用于审批小微企业贷款的信用评分模型,面临严格的监管审查要求,必须清晰解释每一个拒贷决定的依据。
没有 PiML-Toolbox 时
- 模型如“黑盒”:团队使用复杂的集成算法虽提升了精度,但无法向监管机构解释为何特定特征(如现金流波动)导致拒贷,合规报告难以撰写。
- 盲区排查困难:缺乏系统化工具定位模型在特定细分群体(如初创科技企业)中的表现弱点,只能靠人工经验猜测,容易遗漏高风险区域。
- 验证流程割裂:评估准确性、公平性和鲁棒性需要拼接多个独立库,代码繁琐且结果不一致,导致模型上线前的验证周期长达数周。
- 过拟合难发现:难以直观区分模型是学到了真实规律还是在训练数据上“死记硬背”,常在部署后因环境变化导致性能骤降。
使用 PiML-Toolbox 后
- 白盒化决策:利用内置的 EBM 或 GAMI-Net 等原生可解释模型,直接生成可视化的特征贡献图,让每个评分理由都有据可查,轻松通过合规审计。
- 精准定位弱项:通过 WeakSpot 功能自动切片分析,迅速锁定模型在“高负债初创企业”群体的预测偏差,针对性地补充数据或调整策略。
- 一站式诊断:在一个框架内完成从准确率、公平性差异测试到抗噪声鲁棒性评估的全流程,将验证时间从数周缩短至几天。
- 防过拟合预警:利用训练集与测试集的性能差距分析,自动识别过拟合区域,并结合保形预测技术量化不确定性,确保模型在真实环境中稳健运行。
PiML-Toolbox 将晦涩的机器学习模型转化为透明、可信且易于诊断的决策工具,让高风险场景下的 AI 落地不再因“不可解释”而受阻。
运行环境要求
- 未说明
未说明
未说明

快速开始
用于可解释机器学习的集成式Python工具箱
2025年3月30日,由苏吉安托博士撰写:告别PiML,迎接MoDeVa!
在赋能模型开发者与验证者长达三年并取得显著成效之后,我们非常高兴地推出下一代工具——MoDeVa——MOdel DEvelopment & VAlidation(模型开发与验证)。
MoDeVa基于PiML的成功经验,将机器学习中的透明性、可解释性和稳健性提升至全新水平。无论您身处高风险的监管环境,还是正在探索前沿的模型架构,MoDeVa都能为您的工作提供有力支持。
为什么选择MoDeVa?
• 新一代模型:可解释的机器学习模型,如梯度提升树、专家混合模型和神经树,专为自信决策而设计。
• 模型分析新境界:帮助揭示模型的失效模式,分析其稳健性、可靠性和韧性。
• 交互式统计可视化:通过动态图表让模型“活”起来,超越传统的静态图表。
• 无缝验证:利用灵活的封装器轻松验证外部黑盒模型。
立即访问:https://modeva.ai/
pip install PiML
🎄 2023年12月1日: 发布V0.6.0版本,增强了数据处理与模型分析功能。
:rocket: 2023年5月4日: 发布V0.5.0版本,并附带PiML用户指南。
:rocket: 2022年10月31日: 发布V0.4.0版本,新增更多模型并强化诊断功能。
:rocket: 2022年7月26日: 发布V0.3.0版本,包含经典统计模型。
:rocket: 2022年6月26日: 发布V0.2.0版本,提供高代码API接口。
:loudspeaker: 2022年5月4日: 正式发布V0.1.0版本,配备低代码UI/UX界面。
PiML(或π-ML,/ˈpaɪ·ˈem·ˈel/)是一款全新的Python工具箱,用于可解释机器学习模型的开发与验证。通过低代码界面和高代码API,PiML支持一系列日益丰富的内在可解释性机器学习模型:
- GLM:带有L1或L2正则化的线性/逻辑回归
- GAM:使用B样条的广义加性模型
- Tree:用于分类与回归的决策树
- FIGS:快速可解释的贪婪树求和模型(Tan等,2022)
- XGB1:深度为1的极端梯度提升树,采用最优分箱技术(Chen和Guestrin,2016;Navas-Palencia,2020)
- XGB2:深度为2的极端梯度提升树,具备效应净化功能(Chen和Guestrin,2016;Lengerich等,2020)
- EBM:可解释梯度提升机(Nori等,2019;Lou等,2013)
- GAMI-Net:具有结构化交互作用的广义加性模型(Yang、Zhang和Sudjianto,2021)
- ReLU-DNN:利用Aletheia解包器和稀疏化技术的深层ReLU网络(Sudjianto等,2020)
PiML同样适用于回归和二分类任务下的任意监督学习模型。它支持全方位的结果评估,包括但不限于以下内容:
- 准确性:流行的指标,如回归任务中的MSE、MAE,以及二分类任务中的ACC、AUC、召回率、精确率和F1分数。
- 可解释性:事后全局解释器(PFI、PDP、ALE)和局部解释器(LIME、SHAP)。
- 公平性:通过集成solas-ai软件包进行差异性测试和分段分析。
- 薄弱点:利用切片技术识别残差较大的薄弱区域。
- 过拟合:根据训练集与测试集表现差距识别过拟合区域。
- 可靠性:通过分割共形预测技术评估预测不确定性。
- 稳健性:评估在协变量噪声扰动下的性能下降情况。
- 韧性:评估在不同分布外场景下的性能退化情况。
安装
pip install PiML
低代码示例
点击ipynb链接,在Google Colab中运行示例:
- 自行车共享数据:
ipynb - 加州住房数据:
ipynb - 台湾信贷数据:
ipynb - 公平性模拟研究1数据:
ipynb - 公平性模拟研究2数据:
ipynb - 以两种方式上传自定义数据:
ipynb - 处理外部模型:
ipynb
从这本演示笔记本开始您的PiML之旅:此演示笔记本。
高代码示例
同样的示例也可以通过高代码 API 运行:
在 Google Colab 上的低代码使用
第一阶段:初始化实验、加载并准备数据
from piml import Experiment
exp = Experiment()
exp.data_loader()
exp.data_summary()
exp.data_prepare()
exp.data_quality()
exp.feature_select()
exp.eda()
第二阶段:训练可解释模型
exp.model_train()
第三阶段:解释与解读
exp.model_explain()
exp.model_interpret()
第四阶段:诊断与比较
exp.model_diagnose()
exp.model_compare()
exp.model_fairness()
exp.model_fairness_compare()
任意黑盒建模
例如,训练一个深度为 7 的复杂 LightGBM 模型,并将其注册到实验中:
from lightgbm import LGBMClassifier
exp.model_train(LGBMClassifier(max_depth=7), name='LGBM-7')
然后,将其与固有可解释模型(如 XGB2 和 GAMI-Net)进行比较:
exp.model_compare()
引用文献
PiML、ReLU-DNN Aletheia 和 GAMI-Net
“用于可解释机器学习模型开发与诊断的 PiML 工具箱”(A. Sudjianto、A. Zhang、Z. Yang、Y. Su 和 N. Zeng,2023 年) arXiv 链接
@article{sudjianto2023piml,
title={PiML Toolbox for Interpretable Machine Learning Model Development and Diagnostics},
author={Sudjianto, Agus and Zhang, Aijun and Yang, Zebin and Su, Yu and Zeng, Ningzhou},
year={2023}
}
“设计内在可解释的机器学习模型”(A. Sudjianto 和 A. Zhang,2021 年) arXiv 链接
@article{sudjianto2021designing,
title={Designing Inherently Interpretable Machine Learning Models},
author={Sudjianto, Agus and Zhang, Aijun},
journal={arXiv preprint:2111.01743},
year={2021}
}
“揭开深度 ReLU 网络的黑箱:可解释性、诊断与简化”(A. Sudjianto、W. Knauth、R. Singh、Z. Yang 和 A. Zhang,2020 年) arXiv 链接
@article{sudjianto2020unwrapping,
title={Unwrapping the black box of deep ReLU networks: interpretability, diagnostics, and simplification},
author={Sudjianto, Agus and Knauth, William and Singh, Rahul and Yang, Zebin and Zhang, Aijun},
journal={arXiv preprint:2011.04041},
year={2020}
}
“GAMI-Net:基于具有结构化交互作用的广义加性模型的可解释神经网络”(Z. Yang、A. Zhang 和 A. Sudjianto,2021 年) arXiv 链接
@article{yang2021gami,
title={GAMI-Net: An explainable neural network based on generalized additive models with structured interactions},
author={Yang, Zebin and Zhang, Aijun and Sudjianto, Agus},
journal={Pattern Recognition},
volume={120},
pages={108192},
year={2021}
}
其他可解释机器学习模型
“快速可解释的贪婪树求和模型 (FIGS)”(Tan, Y.S., Singh, C., Nasseri, K., Agarwal, A. 和 Yu, B., 2022 年)
@article{tan2022fast,
title={Fast interpretable greedy-tree sums (FIGS)},
author={Tan, Yan Shuo and Singh, Chandan and Nasseri, Keyan and Agarwal, Abhineet and Yu, Bin},
journal={arXiv preprint arXiv:2201.11931},
year={2022}
}
“具有成对交互作用的准确且易懂的模型”(Y. Lou、R. Caruana、J. Gehrke 和 G. Hooker,2013 年)
@inproceedings{lou2013accurate,
title={Accurate intelligible models with pairwise interactions},
author={Lou, Yin and Caruana, Rich and Gehrke, Johannes and Hooker, Giles},
booktitle={Proceedings of the 19th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining},
pages={623--631},
year={2013},
organization={ACM}
}
“利用函数型方差分析净化交互效应:一种恢复可识别加性模型的有效算法”(Lengerich, B., Tan, S., Chang, C.H., Hooker, G. 和 Caruana, R., 2020 年)
@inproceedings{lengerich2020purifying,
title={Purifying interaction effects with the functional anova: An efficient algorithm for recovering identifiable additive models},
author={Lengerich, Benjamin and Tan, Sarah and Chang, Chun-Hao and Hooker, Giles and Caruana, Rich},
booktitle={International Conference on Artificial Intelligence and Statistics},
pages={2402--2412},
year={2020},
organization={PMLR}
}
“InterpretML:机器学习可解释性的统一框架”(H. Nori、S. Jenkins、P. Koch 和 R. Caruana,2019 年)
@article{nori2019interpretml,
title={InterpretML: A Unified Framework for Machine Learning Interpretability},
author={Nori, Harsha and Jenkins, Samuel and Koch, Paul and Caruana, Rich},
journal={arXiv preprint:1909.09223},
year={2019}
}
版本历史
V0.6.02023/11/30V0.5.12023/09/26V0.5.02023/05/03V0.4.32023/03/17V0.4.22022/12/15V0.4.12022/11/15V0.4.02022/10/31V0.3.32022/09/19V0.3.22022/08/19V0.3.12022/08/01V0.3.02022/07/25V0.2.22022/07/06V0.2.12022/06/29V0.2.02022/06/26V0.1.42022/05/27V0.1.32022/05/24V0.1.22022/05/16V0.1.12022/05/12V0.1.02022/05/03常见问题
相似工具推荐
openclaw
OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你
stable-diffusion-webui
stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。
everything-claude-code
everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上
ComfyUI
ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。
markitdown
MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器
LLMs-from-scratch
LLMs-from-scratch 是一个基于 PyTorch 的开源教育项目,旨在引导用户从零开始一步步构建一个类似 ChatGPT 的大型语言模型(LLM)。它不仅是同名技术著作的官方代码库,更提供了一套完整的实践方案,涵盖模型开发、预训练及微调的全过程。 该项目主要解决了大模型领域“黑盒化”的学习痛点。许多开发者虽能调用现成模型,却难以深入理解其内部架构与训练机制。通过亲手编写每一行核心代码,用户能够透彻掌握 Transformer 架构、注意力机制等关键原理,从而真正理解大模型是如何“思考”的。此外,项目还包含了加载大型预训练权重进行微调的代码,帮助用户将理论知识延伸至实际应用。 LLMs-from-scratch 特别适合希望深入底层原理的 AI 开发者、研究人员以及计算机专业的学生。对于不满足于仅使用 API,而是渴望探究模型构建细节的技术人员而言,这是极佳的学习资源。其独特的技术亮点在于“循序渐进”的教学设计:将复杂的系统工程拆解为清晰的步骤,配合详细的图表与示例,让构建一个虽小但功能完备的大模型变得触手可及。无论你是想夯实理论基础,还是为未来研发更大规模的模型做准备