神经网络画图神器 ML Visuals 正在持续更新。
大家好,我是辰哥
去年 5 月,曾向大家推荐一款名为 ML Visuals 的机器学习画图模板,该项目受到广泛关注,迄今已收获 2.2K Star。ML Visuals 专为解决神经网络画图问题设计,最近,这一模板进行了更新。
项目地址:https://github.com/dair-ai/ml-visuals
ML Visuals 现在包含了 100 多个可用的自定义图形,使用者可以在任何论文、博客、PPT 中使用这些资源。
这份 101 页的模板共包含几个部分:
- 基础组件
- 架构
- 机器学习概念
- 抽象背景
- 渐变背景
- 机器学习 & 健康
- 其他
- 机器学习系统设计
基础组件
这套画图模板首先提供了多种基础组件,比如表示过程、操作或转换的圆角矩形,表示神经元或任意操作的小圆圈,表示向量的一排小方块以及表示多维数组的网格等。
架构
架构部分的模板数量最多,共有 32 张。对于复杂的模型架构来说,套模板显然事半功倍,比亲手画图要便捷得多。
比如,画出卷积操作的示意图:
使用该模板重现一些经典架构也是得心应手,比如下图是使用该模板绘制的 Transformer 架构图:
机器学习概念
该模板还可以用来表示机器学习中的一些基本概念,比如 DropOut、归一化:
除此之外,目前这套模板还包含多张关于生物、医疗领域的内容。比如疼痛强度预估、疼痛位置预估:
这张代表「疼痛指数」的图,稍加修改就能用于别处:
最后再来看下使用该模板绘制的神经网络:
此外,该模板还可用于机器学习系统设计,例如:
作者:蛋酱、小舟,编辑:机器之心