Does tensorflow lite support on device training at now

Question

Because Google issued several paper on federated learning on mobile device, like Gboard, but I can not find docs on training on TFLite website, does TFLite support on device training at now?

Answer 1

We do not support general training (or federated training) on TensorFlow Lite. We however have released a tutorial on doing on-device personalization with TF lite. https://blog.tensorflow.org/2019/12/example-on-device-model-personalization.html

It basically goes through the process of exporting subgraphs that do training substeps and evaluation to stage training efforts manually. Personalization is the most practical on-device training mechanism right now, so hopefully this addresses your needed.

We are also working on a more comprehensive training solution that will eventually support things like federated learning and also more easy-to-use workflows.