arXiv reaDer
DecisioNet:二分木構造のニューラルネットワーク
DecisioNet: A Binary-Tree Structured Neural Network
ディープニューラルネットワーク(DNN)とデシジョンツリー(DT)は、どちらも最先端の分類器です。 DNNは、表現学習機能により優れたパフォーマンスを発揮しますが、DTは、入力データに依存する1つのルート(ルートからリーフ)に沿って推論を実行するため、計算効率が高くなります。この論文では、二分木構造のニューラルネットワークであるDecisioNet(DN)を紹介します。既存のDNNをDNに変換して、元のモデルの軽量バージョンを作成する体系的な方法を提案します。 DecisioNetは、両方の長所を活用しています。ニューラルモジュールを使用して表現学習を実行し、ツリー構造を利用して計算の一部のみを実行します。さまざまなDNアーキテクチャを、FashionMNIST、CIFAR10、およびCIFAR100データセットの対応するベースラインモデルとともに評価します。 DNバリアントが、元のネットワークの計算コストを大幅に削減しながら、同様の精度を達成することを示します。
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.
updated: Sun Jul 10 2022 18:04:05 GMT+0000 (UTC)
published: Sun Jul 03 2022 21:47:03 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト