KO | EN

GitLite

Minimalist Open Source Repository Search Engine

gitlite — search

#dsh #dsh-plugin #ai-agents #developer-tools #ai-agent #claude-code #mcp #typescript #codex #cli #rust #deepseek-harness

TFLite-Micro-Seq2Seq

★ 30 Open GitHub ↗

attentional sequence-to-sequence model (based on LSTMs) in TFLite Micro, tested on Arduino Nano 33 BLE

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# TFLite-Micro-Seq2Seq

Attentional sequence-to-sequence modeling (based on LSTMs) implemented in TFLite Micro, tested on Arduino Nano 33 BLE. Sequence-to-Sequence models can learn the mapping from a sequence of inputs to a sequence of outputs, and have been widely used in many applications including machine translation and speech recognition.

This repo provides the full training and deployment code for a number-to-word task, where the model is given a number represented as a sequence of digits, and the goal is to convert it to English words (this task can be solved by writing simple rules, but here we use it to demonstrate a learning-based approach).

## Model Architecture

![Model Diagram](https://raw.githubusercontent.com/da03/TFLite-Micro-Seq2Seq/main/img/encoder_decoder_attn_1layer.png)

## Demo on Arduino Nano 33 BLE

![Demo](https://raw.githubusercontent.com/da03/TFLite-Micro-Seq2Seq/main/img/demo.gif)

## Training

The code for training a model and converting it to TFLite Micro can be found under folder [notebooks](notebooks/TFLite_Micro_Seq2Seq_LSTM_w_attn_&_bidirectional_encoder.ipynb), or at this Google Colab Notebook: https://colab.research.google.com/drive/1d8Kz6P-7O0OAyFNSAoR0C2V48vldcjfA?usp=sharing. In the notebook we also provide script for directly running inference in TFLite, such that you can test your model in the notebook before deploying to MCUs.

After training, two files will be generated: `c_src/model.h` and `c_src/model.cpp`. If you want to deploy it to Arduino, download these two files and put them in the directory `Arduino_Project/src`.

## Dependency for Deployment to Arduino

If you want to deploy the model to Arduino, [this post](https://medium.com/tensorflow/how-to-get-started-with-machine-learning-on-arduino-7daf95b4157) shows how to setup the environment for running TFLite Micro on Arduino.

## Deployment to Arduino

The Arduino project can be found at [Arduino_Project](Arduino_Project). Compile and upload it to MCUs (we recommend using PlatformIO in VSCode, a tutorial can be found [here](https://maker.pro/arduino/tutorial/how-to-use-platformio-in-visual-studio-code-to-program-arduino)), and run. Note that there's already a pretrained model in [Arduino_Project/src](Arduino_Project/src) so you can directly use it for a simple number-to-word task.

Upon seeing "Input:", enter a number and hit Enter, and the model will run inference and output the result. Note that the program will not echo back the digits you typed in until Enter is hit.

## Video Presentation

An explanation about this project can be found at [bit.ly/tinyseq2seq](http://bit.ly/tinyseq2seq).

## Contact

Yuntian Deng (dengyuntian@seas.harvard.edu)

## Acknowledgments

This is a course project of [CS249r TinyML](https://scholar.harvard.edu/vijay-janapa-reddi/classes/cs249r-tinyml) at Harvard, and I am grateful to the teaching fellows for their help during this project.

TFLite-Micro-Seq2Seq

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README