A DAPTIVE N EURAL T REES
Anonymous authors
Paper under double-blind review
A BSTRACT
Deep neural networks and decision trees operate on largely separate paradigms; typically, the former performs representation learning with pre-speciﬁed architectures, while the latter is characterised by learning hierarchies over pre-speciﬁed features with data-driven architectures.
We unite the two via adaptive neural trees (ANTs), a model that incorporates representation learning into edges, routing functions and leaf nodes of a decision tree, along with a backpropagation-based training algorithm that adaptively grows the architecture from primitive modules (e.g., convolutional layers).
ANTs allow increased interpretability via hierarchical clustering, e.g., learning meaningful class associations, such as separating natural vs. man-made objects.
We demonstrate this whilst achieving over 99% and 90% accuracy on the MNIST and CIFAR-10 datasets.
Furthermore, ANT optimisation naturally adapts the architecture to the size and complexity of the training data.
I NTRODUCTION
Neural networks (NNs) and decision trees (DTs) are both powerful classes of machine learning models with proven successes in academic and commercial applications.
The two approaches, however, typically come with mutually exclusive beneﬁts and limitations.
NNs are characterised by learning hierarchical representations of data through the composition of nonlinear transformations (Zeiler & Fergus, 2014; Bengio, 2013), which has alleviated the need for feature engineering, in contrast with many other machine learning models.
In addition, NNs are trained with stochastic optimisers, such as stochastic gradient descent (SGD), allowing training to scale to large datasets.
Consequently, with modern hardware, we can train NNs of many layers on large datasets, solving numerous problems ranging from object detection to speech recognition with unprecedented accuracy (LeCun et al., 2015).
However, their architectures typically need to be designed by hand and ﬁxed per task or dataset, requiring domain expertise (Zoph & Le, 2017).
Inference can also be heavy-weight for large models, as each sample engages every part of the network, i.e., increasing capacity causes a proportional increase in computation (Bengio et al., 2013).
Alternatively, DTs are characterised by learning hierarchical clusters of data (Criminisi & Shotton, 2013).
A DT learns how to split the input space, so that in each subset, linear models sufﬁce to explain the data.
In contrast to standard NNs, the architectures of DTs are optimised based on training data, and are particularly advantageous in data-scarce scenarios.
DTs also enjoy lightweight inference as only a single root-to-leaf path on the tree is used for each input sample.
However, successful applications of DTs often require hand-engineered features of data.
We can ascribe the limited expressivity of single DTs to the common use of simplistic routing functions, such as splitting on axis-aligned features.
The loss function for optimising hard partitioning is non-differentiable, which hinders the use of gradient descent-based optimization and thus complex splitting functions.
Current techniques for increasing capacity include ensemble methods such as random forests (RFs) (Breiman, 2001) and gradient-boosted trees (GBTs) (Friedman, 2001), which are known to achieve state-of-the-art performance in various tasks, including medical imaging and ﬁnancial forecasting (Sandulescu & Chiru, 2016; Kaggle.com, 2017; Le Folgoc et al., 2016; Volkovs et al., 2017).
The goal of this work is to combine NNs and DTs to gain the complementary beneﬁts of both approaches.
To this end, we propose adaptive neural trees (ANTs), which generalise previous work that attempted the same uniﬁcation (Su´arez & Lutsko, 1999; ˙Irsoy et al., 2012; Laptev & Buhmann, 2014; Rota Bulo & Kontschieder, 2014; Kontschieder et al., 2015; Frosst & Hinton, 2017; Xiao, 2017) and address their limitations (see Tab. 1).
ANTs represent routing decisions and root-to-leaf computational paths within the tree structures as NNs, which lets them beneﬁt from hierarchical representation learning, rather than being restricted to partitioning the raw data space.
In addition, we propose a backpropagation-based training algorithm to grow ANTs based on a series of decisions between making the ANT deeper—the central NN paradigm—or partitioning the data—the central DT paradigm (see Fig.
1 (Right)).
This allows the architectures of ANTs to adapt to the data available.
By our design, ANTs inherit the following desirable properties from both DTs and NNs:
• Representation learning : as each root-to-leaf path in an ANT is a NN, features can be learnt end-to-end with gradient-based optimisation.
This, in turn, allows for learning complex data partitioning.
The training algorithm is also amenable to SGD.
• Architecture learning : by progressively growing ANTs, the architecture adapts to the availability and complexity of data, embodying Occams razor.
The growth procedure can be viewed as architecture search with a hard constraint over the model class.
• Lightweight inference : at inference time, ANTs perform conditional computation, selecting a single root-to-leaf path on the tree on a per-sample basis, activating only a subset of the parameters of the model.
We empirically validate these beneﬁts through experiments on the MNIST (LeCun et al., 1998) and
CIFAR-10 (Krizhevsky & Hinton, 2009) datasets.
In terms of classiﬁcation accuracy, ANTs far outperform state-of-the-art random forest (RF) techniques (Zhou & Feng, 2017), with architectures achieving over 99% accuracy on MNIST and over 90% accuracy on CIFAR-10.
At the same time,
ANTs can learn meaningful hierarchical partitionings of data, e.g., grouping man-made and natural objects (see Fig. 2).
ANTs also have reduced time and memory requirements during inference, conferred by conditional computation.
In one case, we discover an architecture that achieves over 98% accuracy on MNIST using approximately the same number of parameters as a linear classiﬁer on raw image pixels, showing the beneﬁts of modelling a hierarchical structure that reﬂects the underlying data structure in enhancing both computational and predictive performance.
Finally, we demonstrate the beneﬁts of architecture learning by training ANTs on subsets of CIFAR-10 of varying sizes.
The method can construct architectures of adequate size, leading to better generalisation, particularly on small datasets.
R ELATED WORK
Our work is primarily related to research into combining DTs and NNs to beneﬁt from the power of representation learning.
Here we explain how ANTs subsumes a large body of such prior work as speciﬁc cases and address their limitations.
We include additional reviews of work in conditional computation and neural architecture search in Sec.
B in the supplementary material.
The very ﬁrst soft decision tree (SDT) introduced in (Su ´ arez & Lutsko, 1999) is a speciﬁc case where in our terminology the routers are axis-aligned features, the transformers are identity functions, and the routers are static distributions over classes or linear functions.
The hierarchical mixture of experts (HMEs) proposed by (Jordan & Jacobs, 1994) is a variant of SDTs whose routers are linear classiﬁers and the tree structure is ﬁxed.
More modern SDTs in (Rota Bulo & Kontschieder, 2014;
Laptev & Buhmann, 2014; Frosst & Hinton, 2017) used multilayer perceptrons (MLPs) or convolutional layers in the routers to learn more complex partitionings of the input space.
However, the simplicity of identity transformers used in these methods means that input data is never transformed and thus each path on the tree does not perform representation learning, limiting their performance.
More recent work suggested that integrating non-linear transformations of data into DTs would enhance model performance.
The neural decision forest (NDF) (Kontschieder et al., 2015), which held cutting-edge performance on ImageNet (Deng et al., 2009) in 2015, is an ensemble of DTs, each of which is also an instance of ANTs where the whole GoogLeNet architecture (Szegedy et al., 2015) (except for the last linear layer) is used as the root transformer, prior to learning tree-structured classiﬁers with linear routers.
Xiao (2017) employed a similar approach with a MLP at the root transformer, and is optimised to minimise a differentiable information gain loss.
The conditional network proposed in (Ioannou et al., 2016) sparsiﬁed CNN architectures by distributing computations on hierarchical structures based on directed acyclic graphs with MLP-based routers, and designed models with the same accuracy with reduced compute cost and number of parameters.
However, in all cases, the model architectures are pre-speciﬁed and ﬁxed.
[Table]
In contrast
[Table]
ANTs satisfy all criteria in Tab. 1; they provide a general framework for learning treestructured models with the capacity of representation learning along each path and within routing functions
[Table]
and a mechanism for learning its architecture
[Table]
Architecture growth is a key facet of DTs (Criminisi & Shotton, 2013), and typically performed in a greedy fashion with a termination criteria based on validation set error (Su ´ arez & Lutsko, 1999; ˙Irsoy et al., 2012).
Here we review previous attempts to improve upon this greedy growth strategy in the DT literature.
Decision jungles (Shotton et al., 2013) employ a training mechanism to merge partitioned input spaces between different sub-trees, and thus to rectify suboptimal “splits” made due to the locality of optimisation.
˙Irsoy
et al.
(2014) proposes budding trees, which are grown and pruned incrementally based on global optimisation of all existing nodes.
While our proposed training algorithm, for simplicity, grows the architecture by greedily choosing the best option between going deeper and splitting the input space (see Fig. 1), it is certainly amenable to the above advances.
Another related strand of work for feature learning is cascaded forests—stacks of RFs where the outputs of intermediate models are fed into the subsequent ones (Montillo et al., 2011; Kontschieder et al., 2013; Zhou & Feng, 2017).
It has been shown how a cascade of DTs can be mapped to
NNs with sparse connections (Sethi, 1990), and more recently Richmond et al.
(2015) extended this argument to RFs.
However, the features obtained in this approach are the intermediate outputs of respective component models, which are not optimised for the target task, and cannot be learned end-to-end, thus limiting its representational quality.
A DAPTIVE N EURAL T REES
We now formalise the deﬁnition of Adaptive Neural Trees (ANTs), which are a form of
DTs enhanced with deep, learned representations.
We focus on supervised learning, where the aim is to learn the conditional distribution p ( y | x ) from a set of N labelled samples ( x (1) , y (1) ) , ..., ( x ( N ) , y ( N ) ) ∈
X × Y as training data.
3. M ODEL T OPOLOGY AND O PERATIONS
In short, an ANT is a tree-structured model, characterized by a set of hierarchical partitions of the input space X , a series of nonlinear transformations, and separate predictive models in the respective component regions.
More formally, we deﬁne an ANT as a pair ( T , O ) where T deﬁnes the model topology, and O denotes the set of operations on it.
We restrict the model topology T to be instances of binary trees , deﬁned as a set of ﬁnite graphs where every node is either an internal node or a leaf, and is the child of exactly one parent node (apart from the parent-less root node).
We deﬁne the topology of a tree as T :=
{N , E} where N is the set of all nodes, and E is the set of edges between them.
Nodes with no children are leaf nodes, N leaf , and all others are internal nodes, N int .
Every internal node j ∈ N int has exactly two children nodes, represented by left( j ) and right( j ) .
Unlike standard trees, E contains an edge which connects input data x with the root node, as shown in Fig.1 (Left).
Every node and edge is assigned with operations which acts on the allocated samples of data (Fig.1).
Starting at the root, each sample gets transformed and traverses the tree according to the set of operations O .
An ANT is constructed based on three primitive modules of differentiable operations:
Routers , R : each internal node j ∈ N int holds a router module, r θ j : X j → [0 , 1] ∈ R , parametrised by θ , which sends samples from the incoming edge to either the left or right child.
Here X j denotes the representation at node j .
We use stochastic routing , where the binary decision ( 1 for the left and 0 for the right branch) is sampled from Bernoulli distribution with mean r θ j ( x j ) for input x j ∈ X j .
As an example, r θ j can be deﬁned as a small convolutional neural network (CNN).
Transformers , T : every edge e ∈ E of the tree has one or a composition of multiple transformer module(s).
Each transformer t ψ e ∈ T is a nonlinear function, parametrised by ψ , that transforms samples from the previous module and passes them to the next one.
For example, t ψ e can be a single convolutional layer followed by ReLU Nair & Hinton (2010).
Unlike in standard DTs, edges transform data and are allowed to “grow” by adding more operations (Sec.4), learning “deeper” representations as needed
Solvers , S : each leaf node l ∈ N leaf is assigned to a solver module, s φ l : X l → Y ∈ S ,
parametrised by φ , which operates on the transformed input data and outputs an estimate for the conditional distribution p ( y | x ) .
For classiﬁcation tasks, we can deﬁne, for example, s φ as a linear classiﬁer on the feature space X l , which outputs a distribution over classes.
Deﬁning operations on the graph T amounts to a speciﬁcation of the triplet O =
( R , T , S ) .
For example, given image inputs, we would choose the operations of each module to be from the set of operations commonly used in CNNs (examples are given in Tab. 2).
In this case, every computational path on the resultant ANT, as well as the set of routers that guide inputs to one of these paths, are given by CNNs.
In Sec. 4, we discuss methods for constructing such tree-shaped NNs end-toend from simple building blocks.
Lastly, many existing tree-structured models (Su´arez & Lutsko, 1999; ˙Irsoy et al., 2012; Laptev & Buhmann, 2014; Rota Bulo & Kontschieder, 2014; Kontschieder et al., 2015; Frosst & Hinton, 2017; Xiao, 2017) are instantiations of ANTs with limitations which we will address with our model (see Sec.
2 for a more detailed discussion).
3. P ROBABILISTIC M ODEL AND I NFERENCE
An ANT ( T , O ) models the conditional distribution p ( y | x ) as a HME (Jordan & Jacobs, 1994), each of which is deﬁned as a NN and corresponds to a particular root-to-leaf path in the tree.
The key difference with traditional HMEs is that the input is not only routed but also transformed within the tree hierarchy.
Each input x stochastically traverses the tree based on decisions of routers and undergoes a sequence of selected transformations until it reaches a leaf node where the corresponding solver module predicts the label y .
Supposing we have L leaf nodes, the full predictive distribution
[Table]
is given by
[Equation]
[Table]
[Equation]
[Table]
[Equation]
[Table]
where z ∈ { 0 , 1 } L is an L -dimensional binary latent variable such that (cid:80) Ll =
z l = 1 , which describes the choice of leaf node (e.g. z l = 1 means that leaf l is used).
Here θ , ψ , φ summarise the parameters of router, transformer and solver modules in the tree.
The mixing coefﬁcient π θ , ψ l ( x ) :
= p ( z l = 1 | x , ψ , θ ) quantiﬁes the probability that x is assigned to leaf l and is given by a product of decision probabilities over all router modules on the unique path P l from the root to leaf node l :
[Equation]
where l (cid:46) j is a binary relation and is only true if leaf l is in the left subtree of internal node j , and x ψ j is the feature representation of x at node j .
Let T j =
{ t ψ e 1 , ..., t ψ e n } denote the ordered set of the n transformer modules on the path from the root to node j , then the feature vector x ψ j is given by
[Equation]
On the other hand, the leaf-speciﬁc conditional distribution p φ , ψ l ( y ) : = p ( y | x [Equation] =
[Equation]
, ψ ) in eq.
equation 1 yields an estimate for the distribution over target y for leaf node l and is given by its solver’s output s φ l ( x ψ parent( l ) ) .
We consider two schemes of inference, based on a trade-off between accuracy and computation.
Firstly, the full predictive distribution given in eq.
equation 1 is used as the estimate for the target conditional distribution p ( y | x ) .
However, averaging the distributions over all the leaves, weighted by their respective path probabilities, involves computing all operations at all nodes and edges of the tree, which makes inference expensive for a large ANT.
We therefore consider a second scheme which uses the predictive distribution at the leaf node chosen by greedily traversing the tree in the directions of highest conﬁdence of the routers.
This approximation constrains computations to a single path, allowing for more memory- and time-efﬁcient inference.
O PTIMISATION
Training of an ANT proceeds in two stages: 1) growth phase during which the model architecture is learned based on local optimisation, and 2) reﬁnement phase which further tunes the parameters of the model discovered in the ﬁrst phase based on global optimisation.
We include pseudocode for the joint training algorithm in Sec.
A in the supplementary material.
4. L OSS FUNCTION : OPTIMISING PARAMETERS FOR FIXED ARCHITECTURE
For both phases, we use the negative log-likelihood (NLL) as the common objective function to minimise, which is given by − log p ( Y | X , θ , ψ , φ )
= − (cid:80)
Nn =1 log ( (cid:80) Ll =
1 π θ , ψ l ( x ( n ) )
p φ , ψ l ( y ( n ) ))
where X = { x (1) , ..., x ( N ) } , Y = { y (1) , ..., y ( N ) } denote the training inputs and targets.
As all component modules (routers, transformers and solvers) are differentiable with respect to their parameters Θ
= ( θ , ψ , φ ) , we can use gradient-based optimisation.
Given an ANT with ﬁxed topology T , we use backpropagation (Rumelhart et al., 1986) for gradient computation and use gradient descent to minimise the NLL for learning the parameters.
4. G ROWTH PHASE : LEARNING ARCHITECTURE T
We next describe our proposed method for growing the tree T to an architecture of adequate complexity for the availability of training data.
Starting from the root, we choose one of the leaf nodes in breadth-ﬁrst order and incrementally modify the architecture by adding extra computational modules to it.
In particular, we evaluate 3 choices (Fig.
(Right))
at each leaf node; (1).“split data” extends the current model by splitting the node with an addition of a new router; (2) “deepen transform” increases the depth of the incoming edge by adding a new transformer; (3) “keep” retains the current model.
We then locally optimise the parameters of the newly added modules in the architectures of (1) and (2) by minimising NLL via gradient descent, while ﬁxing the parameters of the previous part of the computational graph.
Lastly, we select the model with the lowest validation
NLL if it improves on the previously observed lowest NLL, otherwise we execute (3) and keep the original model.
This process is repeated to all new nodes level-by-level until no more “split data” or “deepen transform” operations pass the validation test.
The rationale for evaluating the two choices is to the give the model a freedom to choose the most effective option between “going deeper” or splitting the data space.
Splitting a node is equivalent to a soft partitioning of the feature space of incoming data, and gives birth to two new leaf nodes (left and right children solvers).
In this case, the added transformer modules on the two branches are identity functions.
Deepening an edge on the other hand does not change the number of leaf nodes, but instead seeks to learn richer representation via an extra nonlinear transformation, and replaces the old solver with a new one.
Local optimisation saves time, memory and compute.
Gradients only need to be computed for the parameters of the new peripheral parts of the architecture, reducing the amount of time and computation needed.
Forward activations prior to the new parts do not need to be stored in memory, saving space.
4. R EFINEMENT PHASE : GLOBAL TUNING OF O
Once the model topology is determined in the growth phase, we ﬁnish by performing global optimisation to reﬁne the parameters of the model, now with a ﬁxed architecture.
This time, we perform gradient descent on the NLL with respect to the parameters of all modules in the graph, jointly optimising the hierarchical grouping of data to paths on the tree and the associated expert NNs.
The reﬁnement phase can correct suboptimal decisions made during the local optimisation of the growth phase, and empirically improves the generalisation error (see Sec. 5.3).
E XPERIMENTS
We evaluate the abilities of ANTs using the MNIST (LeCun et al., 1998) and CIFAR-10 (Krizhevsky & Hinton, 2009) object recognition datasets.
Firstly, we show ANTs learn hierarchical structures in the data, while still achieving favourable classiﬁcation accuracies against relevant DT and NN models.
Next, we examine the effects of reﬁnement phase on ANTs, and show that it can automatically prune the tree.
Finally, we demonstrate that our proposed training procedure adapts the model size appropriately under varying amounts of labelled data.
It should be noted that we focus our efforts on these widely used datasets, but our method is applicable to other problems, such as structured prediction.
All of our models are constructed using the PyTorch framework (Paszke et al., 2017).
5. P ERFORMANCE ON IMAGE CLASSIFICATION
We train ANTs with a range of primitive modules (Tab. 2) and compare against relevant DT and
NN models (Tab. 3).
In general, DT methods without feature learning, such as RFs (Breiman, 2001; Zhou & Feng, 2017) and GBTs (Ponomareva et al., 2017), perform poorly on complex image
[Table]
data (Krizhevsky & Hinton, 2009).
In comparison with CNNs without shortcut connections (LeCun et al., 1998; Goodfellow et al., 2013; Lin et al., 2014; Springenberg et al., 2015), different ANTs balance between strong performance with comparable numbers of trainable parameters
[Table]
and reasonable performance with a relatively small amount of parameters.
At the other end of the spectrum, state-of-the-art NNs (Sabour et al., 2017; Huang et al., 2017) contain signiﬁcantly more parameters.
For simplicity, we deﬁne primitive modules based on three types of NN layers: convolutional, global-average-pooling (GAP) and fully-connected (FC).
Solver modules are ﬁxed as linear classiﬁers (LC) with a softmax output.
Router modules are binary classiﬁers with a sigmoid output.
All convolutional and FC layer are followed by ReLUs, except in the last layers of solvers and routers.
We also apply 2 × 2 max-pooling to feature maps after every d transformer modules where d is the downsample frequency.
We balance the number of parameters in the router and transformer modules to be of the same order of magnitude to avoid favouring either partitioning the data or learning more expressive features.
We hold out 10% of training images as a validation set, on which the best performing model is selected.
Full training details, including training times, are provided in the supplementary material.
Two inference schemes: for each ANT, classiﬁcation is performed in two ways: multi-path inference with the full predictive distribution (eq. equation 1), and single-path inference based on the greedily-selected leaf node (Sec. 3.2).
We observed that with our training scheme the splitting probabilities in the routers tend to be very conﬁdent, being close to 0 or 1 (see histograms in blue in Fig.
2(b)).
This means that single path inference gives a good approximation of the multi-path inference but is more efﬁcient to compute.
We show this holds empirically in Tab. 3, where the largest difference between Error (Full) and Error (Path) is 0 .
06% while number of parameters is reduced from
Params (Full) to Params (Path) across all ANT models.
Patience-based local optimisation: in the growth phase the parameters for the new modules are trained until convergence, as determined by patience-based early stopping on the validation set.
We observe that very low or high patience levels result in new modules underﬁtting or overﬁtting locally, respectively, thus preventing meaningful further growth.
We tuned this hyperparameter using the validation sets, and set the patience level to 5 , which produced consistently good performance on both MNIST and CIFAR-10 datasets across different speciﬁcations of primitive modules.
A quantitative evaluation is given in the supplementary (Sec. E).
MNIST digit classiﬁcation: we observe that ANT-MNIST-A outperforms state-of-the-art GBT (Ponomareva et al., 2017) and RF (Zhou & Feng, 2017) methods in accuracy.
This performance is attained despite the use of a single tree, while RF methods operate with ensembles of classiﬁers (the size shown in Tab. 2).
In particular, the NDF (Kontschieder et al., 2015) has a pre-speciﬁed architecture where LeNet-5 (LeCun et al., 1998) is used as the root transformer module, and 10 trees of ﬁxed depth 5 are constructed from this base feature extractor.
On the other hand, ANT-MNISTA is constructed in a data-driven manner from primitive modules, and displays an improvement over the NDF both in terms of accuracy and number of parameters.
In addition, reducing the size of convolution kernels (ANT-MNIST-B) reduces the total number of parameters by 25% and the path-wise average by almost 40% while only increasing absolute error by < 0 .
We also compare against the LeNet-5 CNN (LeCun et al., 1998), comprised of the same types of operations used in our primitive modules (i.e. convolutional, max-pooling and FC layers).
For a fair comparison, the network is trained with the same protocol as that of the ANT reﬁnement phase, achieving an error rate of 0 .
82% (lower than the reported value of 0 .
87% ) on the test set.
Both
ANT-MNIST-A and ANT-MNIST-B attain better accuracy with a smaller number of parameters than LeNet-5.
The current state-of-the-art, capsule networks (CapsNets) (Sabour et al., 2017), have more parameters than ANT-MNIST-A by almost two orders of magnitude.
Notably, CapsNets also feature a routing mechanism, but with a signiﬁcantly different mechanism and motivation.
Lastly, we highlight the observation that ANT-MNIST-C, with the simplest primitive modules, achieves an error rate of 1 . 68% with single-path inference, which is signiﬁcantly better than that of the linear classiﬁer ( 7 . 91% ), while engaging almost the same number of parameters ( 7 , 956 vs.
7 , 840 ) on average.
To isolate the beneﬁt of convolutions, we took one of the root-to-path CNNs on
ANT-MNIST-C
and increased the number of kernels to adjust the number of parameters to the same value.
We observe a higher error rate of 3 .
55% , which indicates that while convolutions are beneﬁcial, data partitioning has additional beneﬁts in improving accuracy.
This result demonstrates the potential of ANT growth protocol for constructing performant models with lightweight inference.
See Sec.
G in the supplementary materials for the architecture of ANT-MNIST-C.
CIFAR-10 object recognition: we see that variants of ANTs outperform the state-of-the-art DT method, gcForest (Zhou & Feng, 2017) by a large margin, achieving over 90% accuracy, demonstrating the beneﬁt of representation learning in tree-structured models.
Secondly, with fewer number of parameters in single-path inference, ANT-CIFAR-A achieves higher accuracy than CNN models without shortcut connections (Goodfellow et al., 2013; Lin et al., 2014; Springenberg et al., 2015) that held the state-of-the-art performance at the time of publication.
With simpler primitive modules we learn more compact models (ANT-MNIST-B and -C) with a marginal compromise in accuracy.
In addition, initialising the parameters of transformers and routers from a pre-trained single-path
CNN further reduced the error rate of ANT-MNIST-A by 20% (see ANT-MNIST-A* in Tab. 3), which indicates room for improvement in our proposed optimisation method.
Shortcut connections (Fahlman & Lebiere, 1990) have recently lead to leaps in performance in deep
CNNs (He et al., 2016; Huang et al., 2017).
We observe that our best network, ANT-MNIST-A*, has a comparable error rate and half the parameter count (with single-path inference) to the bestperforming residual network, ResNet-110 (He et al., 2016).
Densely connected networks leads to substantially better accuracy, but with an order of magnitude more parameters (Huang et al., 2017).
We expect that shortcut connections could also improve ANT performance, and leave integrating them to future work.
Ablation study: we lastly compare the classiﬁcation errors of different variants of ANTs in cases where the options for adding transformer or router modules are disabled (see Tab. 4).
In this experiment, patience levels are tuned separately for respective models.
In the ﬁrst case, the resulting models are equivalent to SDTs (Su´arez & Lutsko, 1999) or HMEs (Jordan & Jacobs, 1994) with locally grown architectures, while the second case is equivalent to standard CNNs, grown adaptively layer by layer.
We observe that either ablation consistently leads to higher classiﬁcation errors across different module conﬁgurations.
[Table]
[Figure]
5. I NTERPRETABILITY
The proposed training procedure in the growth phase of ANTs is capable of discovering hierarchical structures in the data that are useful to the end task.
Learned hierarchies often display strong specialisation of paths to certain classes or categories of data on both the MNIST and CIFARdatasets.
Fig.
2 (a) displays an example with particularly “human-interpretable” partitions e.g. manmade versus natural objects, and road vehicles versus other types of vehicles.
It should, however, be noted that human intuitions on relevant hierarchical structures do not necessarily equate to optimal representations.
Rather, what needs to be highlighted is the ability of ANTs to learn when to share or separate the representation of data to optimise end-task performance, which gives rise to automatically discovering such hierarchies.
To further attest that the model learns a meaningful routing strategy, we also present the test accuracy of the predictions from the leaf node with the smallest reaching probability in Supp.
Sec.
F. We observe that using the least likely “expert” leads to a substantial drop in classiﬁcation accuracy.
In addition, we observe that most learned trees are unbalanced (see Supp.
Sec.
G for more examples).
This property of adaptive computation is plausible since certain types of images may be easier to classify than others, as seen in prior work (Figurnov et al., 2017).
5. E FFECT OF GLOBAL REFINEMENT
We observe that global reﬁnement phase improves the generalisation error.
Fig.
3 (Right) shows the generalisation error of various ANT models on CIFAR-10, with vertical dotted lines indicating the epoch when the models enter the reﬁnement phase.
As we switch from optimising parts of the ANT in isolation to optimising all parameters, we shift the optimisation landscape, resulting in an initial drop in performance.
However, they all consistently converge to higher test accuracy than the best value attained during the growth phase.
This provides evidence that reﬁnement phase remedies suboptimal decisions made during the locally-optimised growth phase.
In many cases, we observed that global optimisation polarises the decision probability of routers, which occasionally leads to the effective “pruning” of some branches.
For example, in the case of the tree shown in
Fig. 2(b), we observe that the decision probability of routers are more concentrated near 0 orafter global reﬁnement, and as a result, the empirical probability of visiting one of the leaf nodes, calculated over the validation set, reduces to 0.09%—meaning that the corresponding branch could be pruned without a negligible change in the network’s accuracy.
The resultant model attains lower generalisation error, showing that the pruning has resolved a suboptimal partioning of data.
We emphasise that this is a consequence of global ﬁne-tuning, and does not involve additional algorithms that would be used to prune or compress standard NNs.
5. A DAPTIVE MODEL COMPLEXITY
Overparametrised models, trained without regularization, are vulnerable to overﬁtting on small datasets.
Here we assess the ability of our proposed ANT training method to adapt the model complexity to varying amounts of labelled data.
We run classﬁcation experiments on CIFAR-10 and train three variants of ANTs, the baseline All-CNN (Springenberg et al., 2015) and linear classiﬁer on subsets of the dataset of sizes 50, 250, 500, 2.5k, 5k, 25k and 45k (the full training set).
We choose All-CNN as the baseline as it reports the lowest error among the comparison targets and is the closest in terms of constituent operations (convolutional, GAP and FC layers).
Fig.3 (Left) shows the corresponding test performances.
The best model is picked based on the performance on the same validation set of 5k examples as before.
As the dataset gets smaller, the margin between the test accuracy of the ANT models and All-CNN/linear classiﬁer increases (up to 13% ).
Fig.
3 (Middle) shows the model size of discovered ANTs as the dataset size varies.
It can be observed that for different settings of primitive modules, the number of parameters generally increases as a function of the dataset size.
All-CNN has a ﬁxed number of parameters, consistently larger than the discovered ANTs, and suffers from overﬁtting, particularly on small datasets.
The linear classiﬁer, on the other hand, underﬁts to the data.
Our method constructs models of adequate complexity, leading to better generalisation.
This shows the added value of our tree-building algorithm over using models of ﬁxed-size structures.
C ONCLUSION
We introduced Adaptive Neural Trees (ANTs), a holistic way to marry the architecture learning, conditional computation and hierarchical clustering of decision trees (DTs) with the hierarchical representation learning and gradient descent optimization of deep neural networks (DNNs).
Our proposed training algorithm optimises both the parameters and architectures of ANTs through progressive growth, tuning them to the size and complexity of the training dataset.
Together, these properties make ANTs a generalisation of previous work attempting to unite NNs and DTs.
Finally, we validated the claimed beneﬁts of ANTs on standard object classiﬁcation datasets, whilst still achieving high classiﬁcation accuracies.
A CKNOWLEDGEMENTS
We would like to thank Konstantinos Kamnitsas, Sebastian Tschiatschek, Jan St¨uhmer, Katja Hofmann and Danielle Belgrave for their insightful feedback and discussions.
A T RAINING ALGORITHM
Initialise topology T and parameters O (cid:46) T is set to a root node with one solver and one transformer
Optimise parameters in O via gradient descent on NLL (cid:46) Learning root classiﬁer
Set the root node “suboptimal” while true do (cid:46) Growth of T begins
Freeze all parameters O
Pick next “suboptimal” leaf node l ∈ N leaf in the breadth-ﬁrst order Add (1) router to l and train new parameters (cid:46) Split data
Add (2) transformer to the incoming edge of l and train new parameters (cid:46)
Deepen transform
Add (1) or (2) permanently to T if validation error decreases, otherwise leaf is set to “optimal”
Add any new modules to O if no “suboptimal” leaves remain then Break
Unfreeze and train all parameters in O (cid:46)
Global reﬁnement with ﬁxed T
B A DDITIONAL RELATED WORK
The tree-structure of ANTs naturally performs conditional computation.
We can also view the proposed tree-building algorithm as a form of neural architecture search.
Here we provide surveys of these areas and their relations to ANTs.
Conditional Computation:
In NNs, computation of each sample engages every parameter of the model.
In contrast, DTs route each sample to a single path, only activating a small fraction of the model.
Bengio Bengio (2013) advocated for this notion of conditional computation to be integrated into NNs, and this has become a topic of growing interest.
Rationales for using conditional computation ranges from attaining better capacity-to-computation ratio (Bengio et al., 2013; Davis &
Arel, 2013; Bengio et al., 2015; Shazeer et al., 2017) to adapting the required computation to the difﬁculty of the input and task (Bengio et al., 2015; Almahairi et al., 2016; Teerapittayanon et al., 2016; Graves, 2016;
Figurnov et al., 2017; Veit & Belongie, 2017).
We view the growth procedure of ANTs as having a similar motivation with the latter—processing raw pixels is suboptimal for computer vision tasks, but we have no reason to believe that the hundreds of convolutional layers in current state-of-the-art architectures (He et al., 2016; Huang et al., 2017) are necessary either.
Growing ANTs adapts the architecture complexity to the dataset as a whole, with routers determining the computation needed on a per-sample basis.
Neural Architecture Search:
The ANT growing procedure is related to the progressive growing of
NNs (Fahlman & Lebiere, 1990; Hinton et al., 2006; Xiao et al., 2014; Chen et al., 2016; Srivastava et al., 2015; Lee et al., 2017; Cai et al., 2018; ˙Irsoy & Alpaydın, 2018), or more broadly, the ﬁeld of neural architecture search (Zoph & Le, 2017; Cortes et al., 2017; Brock et al., 2017).
This approach, mainly via greedy layerwise training, has historically been one solution to optimising NNs (Fahlman & Lebiere, 1990; Hinton et al., 2006).
However, nowadays it is possible to train NNs in an end-toend fashion.
One area which still uses progressive growing is lifelong learning, in which a model needs to adapt to new tasks while retaining performance on previous ones (Xiao et al., 2014; Lee et al., 2017).
In particular, (Xiao et al., 2014) introduced a method that grows a tree-shaped network to accommodate new classes.
However, their method never transforms the data before passing it to the children classiﬁers, and hence never beneﬁt from the parent’s representations.
Whilst we learn the architecture of an ANT in a greedy, layerwise fashion, several other methods search globally.
Based on a variety of techniques, including evolutionary algorithms (Stanley &
Miikkulainen, 2002; Real et al., 2017), reinforcement learning (Zoph & Le, 2017) and sequential optimisation (Liu et al., 2017), these methods ﬁnd extremely high-performance yet complex architectures.
In our case, we constrain the search space to simple tree-structured NNs, retaining desirable properties of DTs such as data-dependent computation and interpretable structures, while keeping the space and time requirement of architecture search tractable thanks to the locality of our growth procedure.
C T RAINING DETAILS
We perform our experiments on the MNIST digit classiﬁcation task (LeCun et al., 1998) and CIFAR10 object recognition task (Krizhevsky & Hinton, 2009).
The MNIST dataset consists of 60 ,training and 10 , 000 testing examples, all of which are 28 × 28 grayscale images of digits fromto 9 ( 10 classes).
The dataset is preprocessed by subtracting the mean, but no data augmentation is used.
The CIFAR-10 dataset consists of 50 , 000 training and 10 , 000 testing examples, all of which are 32 × 32 coloured natural images drawn from 10 classes.
We adopt an augmentation scheme widely used in the literature (Goodfellow et al., 2013; Lin et al., 2014; Springenberg et al., 2015;
He et al., 2016; Huang et al., 2017) where images are zero-padded with 4 pixels on each side, randomly cropped and horizontally mirrored.
For both datasets, we hold out 10% of training images as a validation set.
The best model is selected based on the validation accuracy over the course of ANT training, spanning both the growth phase and the reﬁnement phase, and its accuracy on the testing set is reported.
The hyperparameters are also selected based on the validation performance alone.
Both the growth and reﬁnement phase of ANTs takes up to 2 hours on a single Titan X GPU on both datasets.
For all the experiments in this paper, we employ the following training protocol:
(1) optimize parameters using Adam (Kingma & Ba, 2014) with initial learning rate of 10 − 3 and β = [0 . 9 , 0 . 999] , with minibatches of size 512 ; (2) during the growth phase, employ early stopping with a patience of 5 , that is, training is stopped after 5 epochs of no progress on the validation set; (3) during the reﬁnement phase, train for 100 epochs for MNIST and 200 epochs for CIFAR-10, decreasing the learning rate by a factor of 10 at every
multiple of 50 .
[Table]
D T RAINING TIMES
Tab. 6 summarises the time taken on a single Titan X GPU for the growth phase and reﬁnement phase of various ANTs, and compares against the training time of All-CNN (Springenberg et al., 2015)
[Table]
Local optimisation during the growth phase means that the gradient computation is constrained to the newly added component of the graph, allowing us to grow a good candidate model under 2 hours on a single GPU.
[Table]
) – – 1.1 (hr)
ANT-CIFAR10-A 1.3 (hr) 1.5 (hr)
ANT-CIFAR10-B 0.8 (hr) 0.9 (hr)
ANT-CIFAR10-C 0.7 (hr) 0.8 (hr)
E
Fig.
4 compares the validation accuracies of the same ANT-CIFAR-C model trained on the CIFAR10 dataset with varying levels of patience during early stopping in the growth phase.
A higher patience level corresponds to more training epochs for optimising new modules in the growth phase.
When the patience level is 1, the architecture growth terminates prematurely and plateaus at low accuracy at 80% .
On the other hand, a patience level of 15 causes the model to overﬁt locally with 87% .
In between these, the patience level of 5 gives the best results with 91% validation accuracy.
F T EST THE SPECIALISATION OF EXPERTS
We investigate if the learned routing strategy is meaningful by comparing the classiﬁcation accuracy of our default path-wise inference against that of the predictions from the leaf node with the smallest reaching probability.
Tab.
7 shows that using the least likely “expert” leads to a substantial drop in classiﬁcation accuracy, down to close to that of random guess or even worse for large trees (ANTMNIST-C and ANT-CIFAR10-C).
This demonstrates that ANTs have the capability to split the input space in a meaningful way.
[Table]
G V ISUALISATION OF OTHER DISCOVERED ARCHITECTURES
Fig.
5 shows ANT architectures discovered on the MNIST (i-iii) and CIFAR-10 (iv-vi) datasets.
We observe two notable trends.
Firstly, most architectures learn a few levels of features before resorting to primarily splits.
However, over half of the architectures (ii-v) still learn further representations beyond the ﬁrst split.
Secondly, all architectures are slightly unbalanced.
This reﬂects the fact that some groups of samples may be easier to classify than others.
This property is reﬂected by traditional DT algorithms, but not “neural” tree-structured models that stick to pre-speciﬁed architectures (Laptev & Buhmann, 2014; Frosst & Hinton, 2017; Xiao, 2017; Kontschieder et al., 2015; Ioannou et al., 2016).
[Figure]