#259 · 主分类: 计算机视觉

Convolutional-KANs

cnn computer-vision deep-learning

该项目将Kolmogorov-Arnold网络(KAN)的创新架构扩展到卷积层,将卷积的经典线性变换改为每个像素上可学习的非线性激活。

项目最后更新:04/08/25

GitHub Stars

921

Forks数量

94

贡献者数量

7

许可证

MIT

收录理由

这个项目把 Kolmogorov-Arnold 网络(一种把可学习激活函数放在连接上、而非节点上固定激活的架构)重新构思并引入卷积层。传统卷积对输入图像块和卷积核做点积,而这里的卷积会对每个像素应用一个可学习的非线性激活,使得该层本身就是一个可训练的非线性映射,而不是线性滤波加固定激活的组合。代码基于 efficient-KAN 实现,想把它换成 PyTorch 里的标准 Conv2d 相当顺手,也方便做对比实验。随附的论文给出了准确率、参数量和每轮训练时间,让使用者能清楚看到其中的取舍:同样的参数量下表达能力更强,但训练更慢,而且每个卷积核的参数也更多。可以把它当作研究用的试验场,用来测试经典卷积的替代方案;当然,它还谈不上是可直接上生产的成熟软件。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目