神经网络图形可视化

Question

我想使用Graphcore Poplar生成类似于this的神经网络可视化(PyTorch或ONNX模型)。我已经在documentation中找过了，但我找不到这个可视化功能在哪里。

我怎样才能完成这样的任务呢？还有没有其他的库？

Answer 1

这种可视化并不是Graphcore Poplar软件的一部分。它是由GraphCore团队生成的“数据艺术”。

这是一项艰巨的工作，需要很多时间才能达到这样好的质量，但如果你决定了，我建议开始寻找图形可视化工具，寻找“图形网络可视化”(并从https://cytoscape.org/screenshots.html等图库中获得灵感)。

NN架构可以转换为通用的图形格式(神经元作为节点，连接作为边)，然后您可以开始尝试。

一些想法:从一个简单的NN开始，有三层。将输入层放在外圈，隐藏层有一个内圈，输出层放在中心。每个神经元都是一个点，半径相对于权重，颜色带有偏差，您可以根据权重将其朝向/远离前一层中的神经元。如果你正在寻找一种“生物”风格，请查看这张图片的灵感：https://cytoscape.org/images/screenshots/edge_bundling3_1400px.png