← 返回信息流

Reddit r/MachineLearning短讯

我构建了 MaRN:一个通过低维参数映射训练神经网络的 PyTorch 库 [P]

reddit.com作者:/u/Less_Dream_6331开源AI评分:50/100

作者开发了 MaRN(Mapping Networks),这是一个基于 PyTorch 的库,允许用户优化紧凑的潜在表示而非直接训练所有模型参数。基准测试显示,MNIST CNN 的可训练参数量从 107,998 减少至 1,872(减少 57.7 倍),准确率达 91.80%;LSTM 预测任务参数量从 12,051 降至 2,048,验证 MSE 为 0.00006。该方法在大幅降低参数量的同时保持了性能,但存在训练权衡。

阅读原文