Abstract
Deep neural networks (DNNs) and decision trees (DTs) are both state-of-the-art classifiers. DNNs perform well due to their representational learning capabilities, while DTs are computationally efficient as they perform inference along one route (root-to-leaf) that is dependent on the input data. In this paper, we present DecisioNet (DN), a binary-tree structured neural network. We propose a systematic way to convert an existing DNN into a DN to create a lightweight version of the original model. DecisioNet takes the best of both worlds - it uses neural modules to perform representational learning and utilizes its tree structure to perform only a portion of the computations. We evaluate various DN architectures, along with their corresponding baseline models on the FashionMNIST, CIFAR10, and CIFAR100 datasets. We show that the DN variants achieve similar accuracy while significantly reducing the computational cost of the original network.
Original language | English |
---|---|
Title of host publication | Computer Vision – ACCV 2022 - 16th Asian Conference on Computer Vision, 2022, Proceedings |
Editors | Lei Wang, Juergen Gall, Tat-Jun Chin, Imari Sato, Rama Chellappa |
Publisher | Springer Science and Business Media Deutschland GmbH |
Pages | 556-570 |
Number of pages | 15 |
ISBN (Print) | 9783031263187 |
DOIs | |
State | Published - 2023 |
Event | 16th Asian Conference on Computer Vision, ACCV 2022 - Macao, China Duration: 4 Dec 2022 → 8 Dec 2022 |
Publication series
Name | Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics) |
---|---|
Volume | 13841 LNCS |
ISSN (Print) | 0302-9743 |
ISSN (Electronic) | 1611-3349 |
Conference
Conference | 16th Asian Conference on Computer Vision, ACCV 2022 |
---|---|
Country/Territory | China |
City | Macao |
Period | 4/12/22 → 8/12/22 |
Bibliographical note
Funding Information:Thanks to the ISF (1439/22) and the DFG for funding.
Publisher Copyright:
© 2023, The Author(s), under exclusive license to Springer Nature Switzerland AG.
Keywords
- Decision trees
- Neural network optimization