Efficient on-device learning for resource-constrained devices - #326
Draft
christophe0606 wants to merge 17 commits into
Draft
Efficient on-device learning for resource-constrained devices#326christophe0606 wants to merge 17 commits into
christophe0606 wants to merge 17 commits into
Conversation
…s of fully connected
Use this operator in backward pass of matrix multiply in autodiff
christophe0606
marked this pull request as draft
August 11, 2026 13:52
Added an example of quantization aware training for a fully connected layer.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Experiment with an optimized on-device machine learning framework that could be used to:
The framework leverages the work done on the CMSIS-DSP C++ extension and its loop-fusion technology to implement efficient backward passes.
The framework focuses on:
Fine-tuning often involves fully connected layers, so the framework is intentionally focused on the operations needed to train these layers efficiently.
This framework is not intended to be a full-featured machine learning framework. In particular, it will not provide layers such as convolutions or pooling. These operations are expected to be handled by existing inference frameworks such as CMSIS-NN or NPU like Ethos.