Summary
This paper proposes a new mechanism for performing computationally-efficient adversarial attacks on decision tree ensembles. Specifically, it considers the setting where an attacker wants to attack *many* samples in a dataset at the same time, and considers the *average* time to attack each sample. (The paper focuses specifically on the L_infinity threat model, but could be applied more generally.)
The proposed method is inspired by the empirical observation that, when attacking various different samples on the same trained model, attacks on decision tree ensembles tend to focus on only a subset of the total space of features, and consistently leave many features un-perturbed for all samples. The key idea is to limit the search space of the attack on later samples to only those features which are likely to be necessary to perturb, based on earlier attacks. This speeds up the attacks on the later samples.
This paper therefore proposes an algorithm that, after fully attacking only a small number of samples (using a "base" attack algorithm that can either be a full-verification MILP solver or a heuristic technique) identifies the features most likely to be involved in an attack; specifically, it ranks all of the features by how often they are perturbed. Then, using several additional small subsets of samples, the algorithm determines, coarsely, the minimum number of features that must be considered in the search space (i.e., the cutoff in the ranking) in order for the attack success rate, perturbing only these features, to be close, with high probability, to the attack success rate when perturbing all features. Finally, the attacker attacks the rest of the samples, only perturbing the identified subset of features.
The paper considers two attack variants: the "pruned" variant which, in the final stage, *only* performs the attack on the subset of features, and the "mixed" variant, which, if the attack fails in the specified L_infinity ball when considering only the subset of features, will then "fall back" on the full-feature-space search attack. The "mixed" variant is guaranteed to have the same success rate as the full search, so only time comparisons are relevant when evaluating its success. Over a wide variety of standard datasets and decision tree ensemble models, the proposed methods are shown to produce significant speedups in attacks.
Strengths
- The presentation is extremely clear and precise, and the paper appears to be very technically sound.
- The problem setting is interesting, and the results are compelling.
- Assuming that this is in fact the first paper, as claimed, to consider the "high-throughput attack" setting for adversarial attacks on decision tree ensembles, then it seems to be a highly impactful result. (However, this is not my area of expertise, so there may be prior work that I am not aware of.)
Weaknesses
- The scope of interest in this work is perhaps somewhat limited: it is fairly specific to the problem of attacking many samples on a decision tree ensemble in a batch.
- The algorithm itself is simple, and empirically (rather than theoretically) motivated. However, it appears to be highly effective, so this is not a necessarily a problem.
- The Limitations section could use some work, or be omitted: it does not mention any limitations.
Questions
I have two suggestions for improvements to the algorithm; perhaps you could try these?
1. It seems odd to use effectively a "sparse" (L_0 constrained) adversarial attack when attacking under the the L_infinity threat model. Have you considered randomly perturbing (or perhaps maximally perturbing), within the L_infinity ball, all of the "other" features, that are not part of the identified sensitive-feature subset? This may give a somewhat-higher access rate on some classifiers, to the extent that the classifier is sensitive to random noise, without significantly increasing attack time.
2. In the "mixed" setting defined in the paper, the entire objective of feature pruning is to reduce runtime: there is no trade-off between final success rate and runtime. Therefore, it occurs to me that during the ExpandFeatureSet loop in algorithm 2, rather than selecting the smallest feature set that is below an arbitrary FNR threshold, one could instead just select the subset size that directly minimizes the average runtime, in the "mixed" case with fall-backs to full attacks. Note that we are already running full attacks and subset attacks on these samples, so this shouldn't take any additional time. For the sake of sample-efficiency, one elegant way to approach this would be as a stochastic multi-arm bandit problem, where the "arms" are the choices of feature set sizes, and the "reward" is the negative runtime. Any existing bandit algorithm could then be applied.
Limitations
The limitations section is perfunctory, and could be expanded: for example, the limits on the situations in which this attack is relevant could be discussed further. Societal impacts are addressed.