Skip to content
/ eden Public

Efficient Decision tree Ensembles libary for IoT edge nodes

License

Notifications You must be signed in to change notification settings

eml-eda/eden

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Eden: Efficient Decision tree Ensembles

Installation

Install this library as a package either by cloning locally the repository or directy from the github url.

Running the code

Examples on how to quantize, export and benchmark adaptive models can be found in the examples/ folder.

Supported ensembles

RandomForests, GradientBoosting and DecisionTrees should be easily exportable in C, while ExtraTrees and other flavours of BoostingEnsembles will be supported in future. Eden supports both classifiers with logits in the leaves (see the DecisionTreeClassifier implementation of Sklearn) and with only class labels stored (see the examples/ folder).

Supported architectures

  • GAP8
  • Pulpissimo
  • Gcc (no custom instructions)

Citing

If you use our code, please cite our paper:

@article{daghero2023dynamic,
  title={Dynamic Decision Tree Ensembles for Energy-Efficient Inference on IoT Edge Nodes},
  author={Daghero, Francesco and Burrello, Alessio and Macii, Enrico and Montuschi, Paolo and Poncino, Massimo and Pagliari, Daniele Jahier},
  journal={IEEE Internet of Things Journal},
  year={2023},
  publisher={IEEE}
}