Search code examples
deep-learningtensorflow-federated

Is it possible to TRAIN a neural network model with Tensoflow Lite/Or any other frameworks on smartphones?

Is it possible to TRAIN a neural network model with Tensoflow Lite/Or any other frameworks on smartphones?

Specifically in the context for federative learning?

Solution

  • You could check out Deeplearning4j which supports Android integration and also on-device training. For a Federated Learning setup, you may think of implementing the Federated Averaging algorithm by yourself on a Java based server, using the same DL4J framework as on mobile clients.

    Although, support for on-device training is already on the Tensorflow Lite roadmap for some time, so it is a matter of time until TFLite will provide it's own solution.