透明的神经网络:模型浏览器让您了解 AI 的“黑匣子”内部

新的 Google 工具可可视化神经网络并提高开发人员的工作效率。

来源:安全实验室新闻频道

透明的神经网络:模型浏览器让您了解 AI 的“黑匣子”内部

新的 Google 工具可可视化神经网络并提高开发人员的工作效率。

Google 推出了一款新的机器学习图形可视化工具 Model Explorer,可让您通过以直观的分层格式可视化来探索、调试和优化 ML 模型。

谷歌 提出

图可视化在模型开发过程中起着关键作用。它可以帮助您查找并修复转换和量化错误、识别性能瓶颈、找到优化模式以及更好地理解模型架构。当将模型部署到资源受限的设备(例如移动设备)时,此类功能特别有用。

然而,变压器和扩散器等现代模型的规模和复杂性不断增加,对现有可视化工具构成了重大挑战。大型 Transformer 模型通常会使传统渲染器过载,从而导致渲染失败或渲染复杂性过高。

Model Explorer 就是为了解决此类问题而设计的。该工具能够可视化大型模型并提供分层信息,例如函数名称和范围。

Model Explorer 支持多种图形格式,包括 JAX、PyTorch、TensorFlow 和 TensorFlow Lite 使用的图形格式。 Model Explorer 最初是为 Google 研究人员和工程师设计的,现在作为 Google AI Edge 产品线的一部分提供。

可用

模型浏览器在将大型模型部署到设备时特别有用,其中转换、量化和优化数据的可视化非常重要。该工具结合了 3D 游戏制作和动画中使用的图形技术,使其适用于图形可视化。这使您可以了解模型架构、调试转换错误和性能问题。