Efficient Learning by Directed Acyclic Graph For Resource Constrained Prediction

Joseph Wang, Kirill Trapeznikov, Venkatesh Saligrama

Introduction

Many scenarios involve classification systems constrained by measurement acquisition budget. In this setting, a collection of sensor modalities with varying costs are available to the decision system. Our goal is to learn adaptive decision rules from labeled training data that, when presented with an unseen example, would select the most informative and cost-effective acquisition strategy for this example. In contrast, non-adaptive methods attempt to identify a common sparse subset of sensors that can work well for all data. Our goal is an adaptive method that can classify typical cases using inexpensive sensors while using expensive sensors only for atypical cases.

We propose an adaptive sensor acquisition system learned using labeled training examples. The system, modeled as a directed acyclic graph (DAG), is composed of internal nodes, which contain decision functions, and a single sink node (the only node with no outgoing edges), representing the terminal action of stopping and classifying (SC). At each internal node, a decision function routes an example along one of the outgoing edges. Sending an example to another internal node represents acquisition of a previously unacquired sensor, whereas sending an example to the sink node indicates that the example should be classified using the currently acquired set of sensors. The goal is to learn these decision functions such that the expected error of the system is minimized subject to an expected budget constraint.

First, we consider the case where the number of sensors available is small (as in ), though the dimensionality of data acquired by each sensor may be large (such as an image taken in different modalities). In this scenario, we construct a DAG that allows for sensors to be acquired in any order and classification to occur with any set of sensors. In this regime, we propose a novel algorithm to learn node decisions in the DAG by emulating dynamic programming (DP). In our approach, we decouple a complex sequential decision problem into a series of tractable cost-sensitive learning subproblems. Cost-sensitive learning (CSL) generalizes multi-decision learning by allowing decision costs to be data dependent . Such reduction enables us to employ computationally efficient CSL algorithms for iteratively learning node functions in the DAG. In our theoretical analysis, we show that, given a fixed DAG architecture, the policy risk learned by our algorithm converges to the Bayes risk as the size of the training set grows.

Next, we extend our formulation to the case where a large number of sensors exist, but the number of distinct sensor subsets that are necessary for classification is small (as in where the depth of the trees is fixed to 55). For this regime, we present an efficient subset selection algorithm based on sub-modular approximation. We treat each sensor subset as a new “sensor,” construct a DAG over unions of these subsets, and apply our DP algorithm. Empirically, we show that our approach outperforms state-of-the-art methods in both small and large scale settings.

Related Work: There is an extensive literature on adaptive methods for sensor selection for reducing test-time costs. It arguably originated with detection cascades (see and references therein), a popular method in reducing computation cost in object detection for cases with highly skewed class imbalance and generic features. Computationally cheap features are used at first to filter out negative examples and more expensive features are used in later stages.

Our technical approach is closely related to Trapeznikov et al. and Wang et al. . Like us they formulate an ERM problem and generalize detection cascades to classifier cascades and trees and handle balanced and/or multi-class scenarios. Trapeznikov et al. propose a similar training scheme for the case of cascades, however restrict their training to cascades and simple decision functions which require alternating optimization to learn. Alternatively, Wang et al. attempt to jointly solve the decision learning problem by formulating a linear upper-bounding surrogate, converting the problem into a linear program (LP).

Conceptually, our work is closely related to Xu et al. and Kusner et al., who introduce Cost-Sensitive Trees of Classifiers (CSTC) and Approximately Submodular Trees of Classifiers (ASTC), respectively, to reducing test time costs. Like our paper they propose a global ERM problem. They solve for the tree structure, internal decision rules and leaf classifiers jointly using alternative minimization techniques. Recently, Kusner et al. propose Approximately Submodular Trees of Classifiers (ASTC), a variation of CSTC which provides robust performance with significantly reduced training time and greedy approximation, respectively. Recently, Nan et al. proposed random forests to efficiently learn budgeted systems using greedy approximation over large data sets.

The subject of this paper is broadly related to other adaptive methods in the literature. Generative methods pose the problem as a POMDP, learn conditional probability models, and myopically select features based information gain of unknown features. MDP-based methods encode current observations as state, unused features as action space, and formulate various reward functions to account for classification error and costs. He et al. apply imitation learning of a greedy policy with a single classification step as actions. Dulac-Arnold et al. and Karayev et al. apply reinforcement learning to solve this MDP. Benbouzid et al. propose classifier cascades with an additional skip action within an MDP framework. Nan et al. consider a nearest neighbor approach to feature selection, with classification confidence driven by the classification margin.

Adaptive Sensor Acquisition by DAG

In this section, we present our adaptive sensor acquisition DAG that during test-time sequentially decides which sensors should be acquired for every new example entering the system.

Before formally describing the system and our learning approach, we first provide a simple illustration for a 3 sensor DAG shown in Fig. 1. The state indicating acquired sensors is represented by a binary vector, with a 0 indicating that a sensor measurement has not been acquired and a 1 representing an acquisition. Consider a new example that enters the system. Initially, it has a state of T^{T} (as do all samples during test-time) since no sensors have been acquired. It is routed to the policy function π0\pi_{0}, which makes a decision to measure one of the three sensors or to stop and classify. Let us assume that the function π0\pi_{0} routes the example to the state T^{T}, indicating that the first sensor is acquired. At this node, the function π1\pi_{1} has to decide whether to acquire the second sensor, acquire the third, or classifying using only the first. If π1\pi_{1} chooses to stop and classify then this example will be classified using only the first sensor.

Such decision process is performed for every new example. The system adaptively collects sensors until the policy chooses to stop and classify (we assume that when all sensors have been collected the decision function has no choice but to stop and classify, as shown for π7\pi_{7} in Fig. 1).

A data instance, x∈Xx\in{\cal X}, consists of MM sensor measurements, x={x1,x2,…,xM}x=\{x^{1},x^{2},\ldots,x^{M}\}, and belongs to one of LL classes indicated by its label y∈Y={1,2,…L}y\in\mathcal{Y}=\{1,2,\ldots L\}. Each sensor measurement, xmx^{m}, is not necessarily a scalar but may instead be multi-dimensional. Let the pair, (x,y)(x,y), be distributed according to an unknown joint distribution D{\cal D}. Additionally, associated with each sensor measurement xmx^{m} is an acquisition cost, cmc_{m}.

To model the acquisition process, we define a state space S={s1,…,sK,sSC}\mathcal{S}=\{s_{1},\ldots,s_{K},s_{SC}\}. The states {s1,…,sK}\{s_{1},\ldots,s_{K}\} represent subsets of sensors, and the stop-and-classify state sSCs_{SC} represents the action of stopping and classifying with a current subset. Let Xs{\cal X}_{s} correspond to the space of sensor measurements in subset ss. We assume that the state space includes all possible sensor subsetsWhile enumerating all possible combinations is feasible for small MM, for large MM this problem becomes intractable. We will overcome this limitation in Section 3 by applying a novel sensor selection algorithm. For now, we remain in the small MM regime., K=2MK=2^{M}. For example in Fig. 1, the system contains all subsets of 3 sensors. We also introduce the state transition function, T:S→S\mathcal{T}:\mathcal{S}\rightarrow\mathcal{S}, that defines a set of actions that can be taken from the current state. A transition from the current sensor subset to a new subset corresponds to an acquisition of new sensor measurements. A transition to the state sSCs_{SC} corresponds to stopping and classifying using the available information. This terminal state, sSCs_{SC}, has access to a classifier bank which is used to predict the label of an example. Since classification has to operate on any sensor subset, there is one classifier for every sks_{k}: fs1,…,fsKf_{s_{1}},\ldots,f_{s_{K}} such that fs:Xs→Yf_{s}:{\cal X}_{s}\to\mathcal{Y}. We assume such classifier bank is given and pre-trained. Practically, the classifiers can be either unique for each subset or a missing feature (i.e. sensor) capable classification system as in . We overload notation and use node, subset of sensors, and path leading upto that subset on the DAG interchangeably. In particular we let S\mathcal{S} denote the collection of subsets of nodes. Each subset is associated with a node on the DAG graph. We refer to each node as a state since it represents the “state-of-information” for an instance arriving at that node.

Next, we define the loss associated with classifying an example/label pair (x,y)(x,y) using the sensors in sjs_{j} as

Using this convention, the loss is the sum of the empirical risk associated with classifier fsjf_{s_{j}} and the cost of the sensors in the subset sjs_{j}. The expected loss over the data is defined

Our goal is to find a policy which adaptively selects subsets for examples such that their average loss is minimized

where π:X→S\pi:{\cal X}\rightarrow\mathcal{S} is a policy selected from a family of policies Π\Pi and π(x)\pi(x) is the state selected by the policy π\pi for example xx. We denote the quantity LD\mathcal{L}_{\mathcal{D}} as the value of (3) when Π\Pi is the family of all measurable functions. LD\mathcal{L}_{\mathcal{D}} is the Bayes cost, representing the minimum possible cost for any function given the distribution of data. In practice, the distribution D{\cal D} is unknown, and instead we are given training examples (x1,y1),…,(xn,yn)(x_{1},y_{1}),\ldots,(x_{n},y_{n}) drawn I.I.D. from D{\cal D}. The problem becomes an empirical risk minimization:

Recall that our sensor acquisition system is represented as a DAG. Each node in a graph corresponds to a state (i.e. sensor subset) in S\mathcal{S}, and the state transition function, T(sj)\mathcal{T}(s_{j}), defines the outgoing edges from every node sjs_{j}. We refer to the entire edge set in the DAG as EE. In such a system, the policy π\pi is parameterized by the set of decision functions π1,…,πK\pi_{1},\ldots,\pi_{K} at every node in the DAG. Each function, πj:X→T(sj)\pi_{j}:{\cal X}\rightarrow\mathcal{T}(s_{j}), maps an example to a new state (node) from the set specified by outgoing edges. Rather than directly minimizing the empirical risk in (4), first, we define a step-wise cost associated with all edges (sj,sk)∈E(s_{j},s_{k})\in E

C(⋅)C(\cdot) is either the cost of acquiring new sensors or is the classification error induced by classifying with the current subset if sk=sSCs_{k}=s_{SC}. Using this step-wise cost, we define the empirical loss of the system w.r.t a path for an example xx:

where path(x,π1,…,πK)\text{path}\left(x,\pi_{1},\ldots,\pi_{K}\right) is the path on the DAG induced by the policy functions π1,…,πK\pi_{1},\ldots,\pi_{K} for example xx. The empirical minimization equivalent to (4) for our DAG system is a sample average over all example specific path losses:

Next, we present a reduction to efficiently learn the functions π1,…,πK\pi_{1},\ldots,\pi_{K} that minimize the empirical loss in (7).

2 Learning Policies in a DAG

Learning the functions π1,…,πK\pi_{1},\ldots,\pi_{K} that minimize the cost in (7) is a highly coupled problem. Learning a decision function πj\pi_{j} is dependent on the other functions in two ways: (a) πj\pi_{j} is dependent on functions at nodes downstream (nodes for which a path exists from πj\pi_{j}), as these determine the cost of each action taken by πj\pi_{j} on an individual example (the cost-to-go), and (b) πj\pi_{j} is dependent on functions at nodes upstream (nodes for which a path exists to πj\pi_{j}), as these determine the distribution of examples that πj\pi_{j} acts on.

Consider a policy πj\pi_{j} at a node corresponding to state sjs_{j} such that all outgoing edges from jj lead to leaves. Also, we assume all examples pass through this node πj\pi_{j} (we are ignoring the effect of upstream dependence b). This yields the following important lemma:

Given the assumptions above, the problem of minimizing the risk in (6) w.r.t a single policy function, πj\pi_{j}, is equivalent to solving a k-class cost sensitive learning (CSL) problem.We consider the k-class CSL problem formulated by Beygelzimer et al. , where an instance of the problem is defined by a distribution DD over X×[0,inf⁡)k\mathcal{X}\times[0,\inf)^{k}, a space of features and associated costs for predicting each of the kk labels for each realization of features. The goal is to learn a function which maps each element of X\mathcal{X} to a label {1,…,k}\{1,\ldots,k\} s.t. the expected cost is minimized.

Consider the risk in (6) with πj\pi_{j} such that all outgoing edges from jj lead to a leaf. Ignoring the effect of other policy functions upstream from jj, the risk w.r.t πj\pi_{j} is:

Minimizing the risk over training examples yields the optimization problem on the right hand side. This is equivalent to a CSL problem over the space of “labels” T(sj)\mathcal{T}(s_{j}) with costs given by the transition costs C(x,y,sj,sk)C(x,y,s_{j},s_{k}). ∎

In order to learn the policy functions π1,…,πK\pi_{1},\ldots,\pi_{K}, we propose Algorithm 1, which iteratively learns policy functions using Lemma 2.1. We solve the CSL problem by using a filter-tree scheme for LearnLearn, which constructs a tree of binary classifiers. Each binary classifier can be trained using regularized risk minimization. For concreteness we define the LearnLearn algorithm as

where the binary classifiers in the filter tree are trained using an appropriately regularized calibrated convex loss function. Note that multiple schemes exist that map the CSL problem to binary classification.

A single iteration of Algorithm 1 proceeds as follows: (1) A node jj is chosen whose outgoing edges connect only to leaf nodes. (2) The costs associated with each connected leaf node are found. (3) The policy πj\pi_{j} is trained on the entire set of training data according to these costs by solving a CSL problem. (4) The costs associated with taking the action πj\pi_{j} are computed for each example, and the costs of moving to state jj are updated. (5) Outgoing edges from node jj are removed (making it a leaf node), and (6) disconnected nodes (that were previously connected to node jj) are removed. The algorithm iterates through these steps until all edges have been removed. We denote the policy functions trained on the empirical data using Alg. 1 as π1n,…,πKn\pi_{1}^{n},\ldots,\pi_{K}^{n}.

3 Analysis

Our goal is to show that the expected risk of the policy functions π1,…,πK\pi_{1},\ldots,\pi_{K} learned by Alg. 1 converge to the Bayes risk. We first state our main result:

Alg. 1 is universally consistent, that is

where π1n,…,πKn\pi_{1}^{n},\ldots,\pi_{K}^{n} are the policy functions learned using Alg. (1), which in turn uses LearnLearn described by Eq. 2.2.

Alg. 1 emulates a dynamic program applied in an empirical setting. Policy functions are decoupled and trained from leaf to root conditioned on the output of descendant nodes.

To adapt to the empirical setting, we optimize at each stage over all examples in the training set. The key insight is the fact that universally consistent learners output optimal decisions over subsets of the space of data, that is they are locally optimal. To illustrate this point, consider a standard classification problem. Let X′⊂X{\cal X}^{\prime}\subset{\cal X} be the support (or region) of examples induced by upstream deterministic decisions. d∗d^{*} and f∗f^{*}, Bayes optimal classifiers w.r.t the full space and subset, respectively, are equal on the reduced support:

From this insight, we decouple learning problems while still training a system that converges to the Bayes risk. This can be achieved by training universally consistent CSL algorithms such as filter trees that reduce the problem to binary classification. By learning consistent binary classifiers , the risk of the cost-sensitive function can be shown to converge to the Bayes risk .

(Theorem 2.2) The proof can be broken down into two steps. First, we show that training the policy function with no downstream policy functions decouples from other policies. Next, we show that sequentially learning policy functions from leaf to root leads to an optimal policy.

Consider first the node associated with state sjs_{j} whose outgoing edges lead to leaves. Alg. 1 trains the policy πjn\pi_{j}^{n} over the entire training set using (2.2). As (2.2) is a universally consistent algorithm, the πjn\pi_{j}^{n} converges to the optimal policy as the data grows:

where the infimum is over any measurable function πj∗\pi^{*}_{j}. As this infimum is over any measurable function, we point out that this convergence holds for any realization x∈Xx\in\mathcal{X}

where D(x)\mathcal{D}(x) is the distribution of yy conditioned on xx. As the outgoing edges of node sjs_{j} contain only leaves, other policy functions π1n,…,πj−1n,πj+1n,…,πKn\pi_{1}^{n},\ldots,\pi_{j-1}^{n},\pi_{j+1}^{n},\ldots,\pi_{K}^{n} do not affect the conditional distribution D(x)\mathcal{D}(x). Instead, they only reduce the support of X\mathcal{X} observed by πjn\pi_{j}^{n}, and therefore πjn\pi_{j}^{n} converges to the Bayes optimal function independent of the other policy functions, and therefore the learned policy πjn\pi_{j}^{n} is fixed.

Alg. 1 updates the edge costs (costs-to-go) of edges directed to sjs_{j}. These costs-to-go do not vary as we train new policy functions in ancestor nodes of sjs_{j} and these values can be viewed as fixed when training the next policy function, πkn\pi_{k}^{n}. The dependence on πjn\pi_{j}^{n} is well captured when learning πkn\pi_{k}^{n}. As πjn\pi_{j}^{n} converges to the Bayes optimal function, the costs-to-go converge to the Bayes optimal values, πkn\pi_{k}^{n} is trained on the Bayes optimal costs-to-go as n→∞n\rightarrow\infty.

By recursion, this implies that every decision function is learned on costs-to-go approaching the Bayes optimal, and therefore each function in the learned decision functions approach the point-wise Bayesian optimal decision. Consequently, the learned system approaches the Bayesian optimal system. ∎

Computational Efficiency: Alg. 1 reduces the problem to solving a series of O(KM)O(KM) binary classification problems, where KK is the number of nodes in the DAG and MM is the number of sensors. Finding each binary classifier is computationally efficient, as it reduces to solving a convex problem with O(n)O(n) variables. In contrast, nearly all previous approaches require solving a non-convex problem and resort to alternating optimization or greedy approximation . Alternatively, convex surrogates proposed for the global problem require solving large convex programs with θ(n)\theta(n) variables, even for simple linear decision functions. Furthermore, existing off-the-shelf algorithms cannot be applied to train these systems, often leading to less efficient implementation in practice.

4 Generalization to Other Budgeted Learning Problems

Although, we presented our algorithm in the context of supervised classification and a uniform linear sensor acquisition cost structure, the above framework holds for a wide range of problems. In particular, any loss-based learning problem can be solved using the proposed DAG approach by generalizing the cost function

Adaptive Sensor Acquisition in High-Dimensions

So far, we considered the case where the DAG system allows for any subset of sensors to be acquired, however this is often computationally intractable as the number of nodes in the graph grows exponentially with the number of sensors. In practice, these complete systems are only feasible for data generated from a small set of sensors ( 10 or less).

Although constructing an exhaustive DAG for data with a large number of sensors is computationally intractable, in many cases this is unnecessary. Motivated by previous methods , we assume that the number of “active” nodes in the exhaustive graph is small, that is these nodes are either not visited by any examples or all examples that visit the node acquire the same next sensor. Equivalently, this can be viewed as the system needing only a small number of sensor subsets to classify all examples with low acquisition cost.

Rather than attempt to build the entire combinatorially sized graph, we instead use this assumption to first find these “active” subsets of sensors and construct a DAG to choose between unions of these subsets. The step of finding these sensor subsets can be viewed as a form of feature clustering, with a goal of grouping features that are jointly useful for classification. By doing so, the size of the DAG is reduced from exponential in the number of sensors, 2M2^{M}, to exponential in a much smaller user chosen parameter number of subsets, 2t2^{t}. In experimental results, we limit t=8t=8, which allows for a diverse subsets of sensors to be found while preserving computational tractability and efficiency.

Our goal is to learn sensor subsets with high classification performance and low acquisition cost: subsets of sensors with empirically low cost as defined in (1). Ideally, our goal is to jointly learn the subsets which minimize the empirical risk of the entire system as defined in (4), however this presents a computationally intractable problem due to the exponential search space. Rather than attempt to solve this difficult problem directly, we minimize classification error over a collection of sensor subsets σ1,…,σt\sigma_{1},\ldots,\sigma_{t} subject to a cost constraint on the total number of sensors used. We decouple the problem from the policy learning problem by assuming that each example is classified by the best possible subset. For a constant sensor cost, the problem can be expressed as a set constraint problem:

where BB is the total sensor budget over all sensor subsets and δ\delta is the cost of a single sensor.

Although minimizing this loss is still computationally intractable, consider instead the equivalent problem of maximizing the “reward” (the event of a correct classification) of the subsets, defined as

This problem is related to the knapsack problem with a non-linear objective. Maximizing the reward in (12) is still a computationally intractable problem, however the reward function is structured to allow for efficient approximation.

The objective of the maximization in (12) is sub-modular with respect to the set of subsets, such that adding any new set to the reward yields diminishing returns.

This follows directly from the fact that maximization over a set of objects is a submodular function. ∎

Given that the empirical risk of each classifier fσkf_{\sigma_{k}} is submodular and monotonically decreasing w.r.t. the elements in σk\sigma_{k} and uniform sensor costs, the strategy in Alg. 2 is an O(1)O(1) approximation of the optimal reward in (12).

Consider adding a sensor kk to any subset σj\sigma_{j}. By assumption, the empirical risk of each classifier is monotonically decreasing and therefore the reward is monotonically increasing. Additionally, note that the reward for any training point xix_{i} using σj\sigma_{j} is less than the reward from using σj∪k\sigma_{j}\cup k and therefore the objective is equal to the objective without replacement of σj\sigma_{j} by σj∪k\sigma_{j}\cup k:

As a result, we can view adding a sensor to a subset as adding an entirely new subset without changing the objective in (12). From the above lemma, adding a new subset results in a submodular function, and therefore the reward in (12) is submodular with respect to adding sensors to each subset. Applying a greedy strategy therefore yields a 1−1e1-\frac{1}{e} approximation of the optimal strategy . ∎

2 Constructing DAG using Sensor Subsets

Alg. 2 requires computation of the reward GG for only O(BδtM)O\left(\frac{B}{\delta}tM\right) sensor subsets, where MM is the number of sensors, to return a constant-order approximation to the NP-hard knapsack-type problem. Given the set of sensor subsets σ1,…,σt\sigma_{1},\ldots,\sigma_{t}, we can now construct a DAG using all possible unions of these subsets, where each sensor subset σj\sigma_{j} is treated as a new single sensor, and apply the small scale system presented in Sec. 2. The result is an efficiently learned system with relatively low complexity yet strong performance/cost trade-off. Additionally, this result can be extended to the case of non-uniform costs, where a simple extension of the greedy algorithm yields a constant-order approximation .

A simple case where three subsets are used is shown in Fig. 2. The three learned subsets of sensors are shown on the bottom left of Fig. 2, and these three subsets are then used to construct the entire DAG in the same fashion as in Fig. 1. At each stage, the state is represented by the union of sensor subsets acquired. Grouping the sensors in this fashion reduces the size of the graph to 88 nodes as opposed to 6464 nodes required if any subset of the 66 sensors can be selected. This approach allows us to map high-dimensional adaptive sensor selection problems to small scale DAG in Sec. 2.

Experimental Results

To demonstrate the performance of our DAG sensor acquisition system, we provide experimental results on data sets previously used in budgeted learning. Three data sets previously used for budget cascades are tested. In these data sets, examples are composed of a small number of sensors (under 4 sensors). To accurately compare performance, we apply the LP approach to learning sensor trees and construct trees containing all subsets of sensors as opposed to the fixed order cascades previously applied .

Next, we examine performance of the DAG system using 3 higher dimensional sets of data previously used to compare budgeted learning performance . In these cases, the dimensionality of the data (between 50 and 400 features) makes exhaustive subset construction computationally infeasible. We greedily construct sensor subsets using Alg. 2, then learn a DAG over all unions of these sensor subsets. We compare performance with CSTC and ASTC .

For all experiments, we use cost sensitive filter trees , where each binary classifier in the tree is learned using logistic regression. Homogeneous polynomials are used as decision functions in the filter trees. For all experiments, uniform sensor cost were were varied in the range [0,M][0,M] achieve systems with different budgets. Performance between the systems is compared by plotting the average number of features acquired during test-time vs. the average test error.

We compare performance of our trained DAG with that of a complete tree trained using an LP surrogate on the landsat, pima, and letter datasets. To construct each sensor DAG, we include all subsets of sensors (including the empty set) and connect any two nodes differing by a single sensor, with the edge directed from the smaller sensor subset to the larger sensor subset. By including the empty set, no initial sensor needs to be selected. 3rd3^{rd}-order homogeneous polynomials are used for both the classification and system functions in the LP and DAG.

As seen in Fig. 3, the systems learned with a DAG outperform the LP tree systems. Additionally, the performance of both of the systems is significantly better than previously reported performance on these data sets for budget cascades . This arises due to both the higher complexity of the classifiers and decision functions as well as the flexibility of sensor acquisition order in the DAG and LP tree compared to cascade structures. For this setting, it appears that the DAG approach is superior approach to LP trees for learning budgeted systems.

2 Large Sensor Set Experiments

Next, we compare performance of our trained DAG with that of CSTC and ASTC for the MiniBooNE, Forest, and CIFAR datasets. We use the validation data to find the homogeneous polynomial that gives the best classification performance using all features (MiniBooNE: linear, Forest: 2nd2^{nd} order, CIFAR: 3rd3^{rd} order). These polynomial functions are then used for all classification and policy functions. For each data set, Alg. 2 was used to find 7 subsets, with an 8th8^{th} subset of all features added. An exhaustive DAG was trained over all unions of these 8 subsets.

Fig. 4 shows performance comparing the average cost vs. average error of CSTC, ASTC, and our DAG system. The systems learned with a DAG outperform both CSTC and ASTC on the MiniBooNE and Forest data sets, with comparable performance on CIFAR at low budgets and superior performance at higher budgets.

References