在ml5.js或Tensorflow.js中可以进行协作培训吗？

Question

我想建立一个图像分类器与ml5.js或Tensorflow.js。我希望用户能够添加自己的标签，即。训练模型。Daniel Shiffman的迁移学习教程在这方面做得很好：https://www.youtube.com/watch?v=kRpZ5OqUY6Y

但是，我希望多个用户能够训练相同的模型。人们会使用自己的分类为同一模型做出贡献。

理想情况下，一个参与者可以在国家X和国家Y的自己的浏览器中做出贡献，或者至少在同一空间中的两个用户从自己的浏览器中进行标签。显示的图像可能来自JSON文件。

我已经使用p5.js测试了用于协作鼠标绘制的Socket.io，但这基本上是我在这种协作应用程序中的所有相关经验。那么:多个用户是否可以使用Socket.io或类似的工具训练相同的ml5.js/Tensorflow.js模型？

热诚感谢您的建议！

Answer 1

我曾经使用Mozilla的TogetherJS以一种简单的方式跨任意数量的浏览器协作添加到训练数据集。但是每个实例都需要在扩展数据集上运行tf.fit。Federated Learning将是避免所有这些冗余计算的下一步。

Answer 2

也许你不需要为此进行实时协作？用户可以从他们的浏览器中提供图像和标签，并按下upload按钮。

客户端中可以有一个train按钮，它告诉服务器重新开始训练。模型只在服务器上，所以每个人都可以为它做贡献。

当服务器正在训练时，它会收到一个新的train命令，它只会发回一条消息already in training process。

这样你就不需要socket.io了。