Active Hypothesis Testing: Beyond Chernoff-Stein
Dhruva Kartik, Ashutosh Nayyar, Urbashi Mitra

TL;DR
This paper formulates an active hypothesis testing problem allowing a fixed number of experiments and inconclusive decisions, deriving bounds on misclassification probability and proposing a heuristic strategy for optimal decision-making.
Contribution
It introduces a new active hypothesis testing framework with bounds on misclassification, extending the Chernoff-Stein lemma and proposing a heuristic strategy.
Findings
Derived asymptotically tight bounds on misclassification probability.
Formulated a generalized Chernoff-Stein lemma for the problem.
Proposed a heuristic strategy with analyzed performance.
Abstract
An active hypothesis testing problem is formulated. In this problem, the agent can perform a fixed number of experiments and then decide on one of the hypotheses. The agent is also allowed to declare its experiments inconclusive if needed. The objective is to minimize the probability of making an incorrect inference (misclassification probability) while ensuring that the true hypothesis is declared conclusively with moderately high probability. For this problem, lower and upper bounds on the optimal misclassification probability are derived and these bounds are shown to be asymptotically tight. In the analysis, a sub-problem, which can be viewed as a generalization of the Chernoff-Stein lemma, is formulated and analyzed. A heuristic approach to strategy design is proposed and its relationship with existing heuristic strategies is discussed.
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Active Hypothesis Testing: Beyond Chernoff-Stein
Dhruva Kartik, Ashutosh Nayyar and Urbashi Mitra
Ming Hsieh Department of Electrical Engineering
University of Southern California, Los Angeles, CA, USA
Email: {mokhasun, ashutosh.nayyar, ubli}@usc.edu
Abstract
An active hypothesis testing problem is formulated. In this problem, the agent can perform a fixed number of experiments and then decide on one of the hypotheses. The agent is also allowed to declare its experiments inconclusive if needed. The objective is to minimize the probability of making an incorrect inference (misclassification probability) while ensuring that the true hypothesis is declared conclusively with moderately high probability. For this problem, lower and upper bounds on the optimal misclassification probability are derived and these bounds are shown to be asymptotically tight. In the analysis, a sub-problem, which can be viewed as a generalization of the Chernoff-Stein lemma, is formulated and analyzed. A heuristic approach to strategy design is proposed and its relationship with existing heuristic strategies is discussed.
I Introduction
We frequently encounter scenarios wherein we would like to deduce whether one of several hypotheses is true by gathering data or evidence. This problem is referred to as multi-hypothesis testing. If we have access to multiple candidate experiments or data sources, we can adaptively select more informative experiments to infer the true hypothesis. This leads to a joint control and inference problem commonly referred to as active hypothesis testing. There are numerous ways of formulating this problem and the precise mathematical formulation depends on the target application.
In this paper, we consider a scenario in which there is an agent that can perform a fixed number of experiments. Subsequently, the agent can decide on one of the hypotheses using the collected data. The agent is also allowed to declare the experiments inconclusive if needed. The objective is to minimize the probability of making an incorrect inference (misclassification probability) while ensuring that the true hypothesis is declared conclusively with moderately high probability. This formulation is of particular interest when the agent is time-constrained and the penalty for making an incorrect inference is significantly higher than the penalty for making no decision. In such cases, it is reasonable for the agent to abstain from drawing conclusions unless there is strong evidence supporting one of the hypotheses.
For example, consider a decentralized system in which an agent needs to perform experiments and convey its results to another decision-maker (such as a fusion center). Due to communication constraints, the agent can only communicate its estimate of the hypothesis or remain silent. The agent incurs heavy penalty for transmitting an incorrect hypothesis. However, the agent is also constrained to transmit the true hypothesis with moderately high probability. Thus, we would like to design an experiment selection strategy and an inference (transmission) strategy for the agent which minimize the misclassification probability while ensuring that the correct estimate is transmitted with sufficiently high probability.
Our contributions in this paper can be summarized as follows. We find lower and upper bounds on the optimal misclassification probabilities in our constrained problem. These bounds are asymptotically tight under some mild assumptions. In our analysis, we formulate a sub-problem and use the results from the sub-problem to solve our original problem. This sub-problem can be viewed as a generalization of the Chernoff-Stein lemma [1] to a setting with multiple hypotheses and multiple experiments. Thereby, we describe an alternate approach to finding a lower bound on the optimal error probability in the Chernoff-Stein lemma. Further, we show that the experiment selection strategy described in [2, 3] in a sequential setting is asymptotically optimal for our fixed horizon problem. We also describe an alternate heuristic approach to strategy design that might improve the performance in the non-asymptotic regime. This approach is based on, what we call, the expected confidence rate which naturally arises out of our analysis.
The rest of the paper is organized as follows. In Section I-A, we summarize key prior literature on hypothesis testing and discuss how our problem is related to various other formulations. In Section I-B, we describe our notation and in Section II, we formulate our problem. We state the main results in Section III and sketch the proof of our results in Section IV. In Section V, we discuss some heuristic approaches for strategy design. We conclude the paper in Section VI.
I-A Prior Work
Hypothesis testing is a long-standing problem and has been addressed in various settings. The classical formulations have been described in [1],[2],[4]. More recently, active hypothesis testing has been addressed in [3],[5]. The key difference between our formulation and the fixed horizon formulations in [1], [3] is that unlike our agent, the agents in these works are compelled to decide on a hypothesis after performing all the experiments. Our analysis shows that this modification significantly alters the optimal error exponents and the strategy design for inference and experiment selection. Another common formulation is the sequential setting in which the agent can perform experiments until sufficiently strong evidence is gathered [2], [3], [5]. The objective in the sequential setting is to minimize a combination of Bayesian error probability and expected stopping time. Interestingly, the analysis and results in our fixed horizon problem have a strong overlap with those in the sequential setting. As mentioned earlier, a sub-problem in our analysis is a generalization of the Chernoff-Stein lemma [1] and our original problem can be seen as a symmetric version of this lemma. To the best of our knowledge, our formulation has not been considered before. The analysis involved in obtaining the upper bound for our problem borrows from prior works [2],[3]. However, our approach for obtaining lower bounds is different from the approach used in all the aforementioned works.
I-B Notation
Random variables are denoted by upper case letters, their realization by the corresponding lower case letter. We use calligraphic fonts to denote sets (e.g. ) and is the probability simplex over a finite set . In general, subscripts denote time index unless stated otherwise. For time indices , is the short hand notation for the variables . For a strategy , we use and to indicate that the probability and expectation depend on the choice of . For an hypothesis , denotes the expectation conditioned on hypothesis . The Kullback-Leibler divergence between distributions and over a finite space is given by
[TABLE]
II Problem Formulation
Let be a finite set of hypotheses and let the random variable denote the true hypothesis. The prior probability on is . At each time , an agent can perform an experiment and obtain an observation . We assume that the sets and are finite. The observation at time is given by
[TABLE]
where is a collection of mutually independent and identically distributed primitive random variables. The probability of observing after performing an experiment under hypothesis is denoted by , that is,
[TABLE]
The time horizon, that is the total number of experiments performed, is fixed a priori to .
At time , the information available to the agent, denoted by , is the collection of all experiments performed and the corresponding observations up to time , i.e.
[TABLE]
At time , the agent selects a distribution over the set of actions according to an experiment selection rule and the action is randomly drawn from this distribution, that is
[TABLE]
The sequence is denoted by and referred to as the experiment selection strategy. Let the collection of all such strategies be .
After performing experiments, the agent can declare one of the hypotheses to be true or it can declare that its experiments were inconclusive. We refer to this final declaration as the agent’s inference decision and denote it by . The inference decision can take values in , where denotes the inconclusive declaration. is chosen according to an inference strategy , i.e.
[TABLE]
Let the set of all inference strategies be .
For an experiment selection strategy and an inference strategy , we define the following error probabilities.
Definition 1**.**
Let be the probability that the agent does not infer when the true hypothesis is indeed , i.e.
[TABLE]
Remark 1**.**
Note that when there are only two hypotheses and the agent is forced to decide on one of the two hypotheses, and are type I and type II errors, respectively. In this case, and .
In this paper, we will be interested in the event that the agent declares an incorrect hypothesis to be true. That is, we will consider the event . We refer to this event as the misclassification event. Let be the probability of this event. Using the definitions above, this probability can be written as
[TABLE]
We will consider the problem of designing the experiment selection and inference strategies to minimize (the probability of declaring an incorrect hypothesis) while satisfying constraints on the type- error probabilities. That is, we are interested in the following optimization problem:
[TABLE]
where . Let denote the infimum value of this optimization problem. We define if the optimization problem is infeasible.
The above formulation is intended for scenarios where the penalty for declaring an incorrect hypothesis to be true is much higher than the penalty for making no decision about the hypothesis. In such cases, it is reasonable for the agent to abstain from drawing conclusions when the evidence is not strong enough. The constraints on type- error probabilities ensure that the agent does not abstain from drawing conclusions too often. The optimization problem seeks to minimize the probability of declaring an incorrect hypothesis while satisfying the type- error probability constraints.
III Main Results
In this section, we will describe asymptotically tight lower and upper bounds on the optimal error probability in Problem (P1). We will first define some useful quantities and then state the assumptions we make to prove our results.
The posterior belief on the hypothesis based on information is given by
[TABLE]
Note that given a realization of the experiments and observations until time , the posterior belief does not depend on the experiment selection strategy .
Definition 2** (Bayesian Log-Likelihood Ratio & Expected Confidence Rate).**
The Bayesian log-likelihood ratio associated with an hypothesis is defined as
[TABLE]
The Bayesian log-likelihood ratio (BLLR) is the logarithm of the ratio of the probability that hypothesis is true versus the probability that hypothesis is not true. The BLLR can be interpreted as a confidence level on hypothesis . For a hypothesis and a strategy , we define the expected confidence rate as
[TABLE]
Assumption 1** (Full support).**
There exists a constant such that for every experiment , observation and pair of hypotheses , where
[TABLE]
Assumption 2**.**
For each experiment and any pair of hypotheses such that , we have
[TABLE]
Remark 2**.**
We make Assumption 2 for ease of exposition. Techniques for relaxing this Assumption have been discussed in [3] and [5].
For each hypothesis , define
[TABLE]
where . The equality of the min-max and max-min values follows from the minimax theorem [6] because the sets and are finite and the Kullback-Leibler divergences are bounded by .
Assumption 3**.**
For every , we have that the bound on the type- error satisfies . Further,
[TABLE]
Theorem 1** (Lower bound).**
There exists a positive constant that does not depend on such that for every the following statements are true.
- a)
For any experiment selection strategy and inference strategy that satisfy the constraints for every , we have the lower bound
[TABLE]
where is given by (12). 2. b)
The optimal misclassification probability in Problem (P1) satisfies
[TABLE]
where is given by (14).
Theorem 2** (Upper bound).**
For any , there exists an integer such that for every , we have
[TABLE]
Using Theorems 1 and 2, we can therefore conclude that
[TABLE]
IV Proof of Main Results
IV-A Supporting Lemmas
In this section, we describe some important properties of the confidence level which will be used in the proof of our main results.
Lemma 1**.**
For any experiment selection strategy , we have
[TABLE]
where
Proof.
We have
[TABLE]
∎
Corollary 1** (Bounded increments).**
For any experiment selection strategy , we have
[TABLE]
with probability 1.
Proof.
Using Lemma 1, we have
[TABLE]
Similarly,
[TABLE]
The same arguments can be used to show that for any experiment selection strategy and , we have
[TABLE]
with probability 1. ∎
Corollary 2**.**
If for every , for some , then
[TABLE]
Proof.
We have
[TABLE]
∎
Lemma 2**.**
For any experiment selection strategy ,
[TABLE]
Proof.
[TABLE]
The last inequality follows from the fact that the observations are independent conditioned on the experiment . ∎
Definition 3**.**
We define the following distributions:
[TABLE]
where .
Lemma 3**.**
For any experiment selection strategy , we have
[TABLE]
where
Proof.
For every , we have the following since is an increasing function
[TABLE]
Therefore,
[TABLE]
Further,
[TABLE]
∎
Lemma 4**.**
Let be an inference strategy in which hypothesis is decided if and only if . Then
[TABLE]
Proof.
In this proof, substitute with . Let be the region in which the inference policy selects hypothesis , that is
[TABLE]
We have
[TABLE]
Therefore,
[TABLE]
∎
IV-B Sub-problem vis-à-vis Chernoff-Stein
We formulate a sub-problem in this section that will be useful for analyzing Problem (P1). For hypothesis , consider the following optimization problem:
[TABLE]
Let the infimum value of this optimization problem be . Note that this problem is always feasible because the agent can trivially satisfy the type- error constraint by always declaring hypothesis .
Remark 3**.**
When there is only one experiment, two hypotheses and the inconclusive decision is not allowed, this formulation is identical to that of the Chernoff-Stein lemma [1].
We follow the proof methodology of the Chernoff-Stein lemma in [1], but with some important modifications. For each experiment selection strategy and inference strategy that satisfy the type- error constraint, we first establish a lower bound on the error probability based on the expected confidence rate . In [1], the lower bound is obtained using a typicality argument. However, such typicality properties may not hold for every experiment selection strategy . Thus, we use a different approach to obtain a similar lower bound. We then use Lemma 3 to obtain a lower bound on that does not depend on the strategies and . Further, we construct strategies that asymptotically achieve this strategy-independent lower bound. The construction of these strategies and the analysis thereof builds on the achievability proofs in [1] and [2].
Lemma 5**.**
Let be any experiment selection strategy and let be any inference strategy such that Then
[TABLE]
Proof.
In this proof, substitute with and with . Also, if the belief is formed using information , we denote with to emphasize dependence on . Let be the region in which the inference policy selects hypothesis , that is
[TABLE]
We have
[TABLE]
The function is convex in and thus, using Jensen’s inequality, we have
[TABLE]
Further, we have
[TABLE]
Therefore,
[TABLE]
The last inequality follows once again from Corollary 1. Hence using inequality (83), we have
[TABLE]
Therefore,
[TABLE]
∎
Lemma 6**.**
Let be any experiment selection strategy and let be any inference strategy such that Then there exist positive constants that do not depend on such that
[TABLE]
Proof.
This follows directly from Lemmas 3 and 5, and the fact that . ∎
Lemma 7**.**
There exists an integer such that for every
[TABLE]
Proof.
We prove this by constructing an experiment selection strategy and an inference strategy that achieve the rate and constraints. Let the agent select experiments randomly and independently from the distribution . Under strategy, we have for every
[TABLE]
Using Hoeffding’s inequality, we have
[TABLE]
Let the inference policy be as follows. If , decide hypothesis . Otherwise, declare . From Lemma 4, we have
[TABLE]
And thus,
[TABLE]
Now we need to show that . Notice that under the inference strategy, if and only if . Therefore , by Corollary 2, implies that for some
[TABLE]
Using the inequality established in (101) and a union bound, the probability of this event conditioned on hypothesis is at most . Therefore, . ∎
Using Lemmas 6 and 7 we can therefore conclude that
[TABLE]
IV-C Proof of Theorem 1
In problem (P1), the strategies are required to satisfy the constraints for every . For any pair of strategies that do satisfy all these constraints, we have the following because of Lemma 6.
[TABLE]
With defined as , this proves Theorem 1.
Remark 4**.**
We can obtain a tighter lower bound on using Lemma 5 without any restrictions on , that is
[TABLE]
IV-D Proof of Theorem 2
To prove Theorem 2, we construct appropriate experiment selection and inference strategies. We then show that these strategies achieve the desired bound on misclassification probability while satisfying the constraints in problem (P1). The construction is almost identical to the strategies in [2].
IV-D1 Experiment selection strategy
Let the maximum a posteriori (MAP) estimate at time be
[TABLE]
If , then an experiment is selected randomly with distribution . We denote this experiment selection strategy by .
IV-D2 Inference strategy
Consider the strategy where
[TABLE]
Let us assume that for every without loss of generality. This ensures that for large enough , the threshold condition above can be satisfied by at most one hypothesis.
Using Lemma 4, we can conclude that for each hypothesis , . Therefore, under these strategies, we have
[TABLE]
If we can show that the strategies also satisfy the type- error constraints in problem (P1), then clearly, . To prove that these strategies do satisfy the constraints for large values of , we use the arguments in [2] and the details of this proof are as follows.
Proof.
We now need to verify if the proposed strategy achieves the type- error constraints, that is for each hypothesis . Let us examine the evolution of the log likelihood ratio associated with a pair of hypotheses and under the hypothesis . Consider the following Doob decomposition
[TABLE]
Note that is a martingale difference sequence with respect to the filtration and with probability 1. Using Azuma’s inequality [7], we have
[TABLE]
We can choose such that
[TABLE]
Let be the smallest time index such that for every . Notice that is a random variable. Under Assumption 2, it was shown in [2] (Lemma 1) that there exist constants such that for every , we have . Notice that
[TABLE]
Therefore, if
[TABLE]
Therefore,
[TABLE]
We can pick a such that for large enough
[TABLE]
Using inequalities (114) and (119), we can conclude that
[TABLE]
Because of Assumption 3, for large enough , we have
[TABLE]
Using this fact and Corollary 2, we have
[TABLE]
Therefore, we can conclude that for every . ∎
V Discussion on Strategy Design
In Section IV-D, we described an experiment selection strategy . As discussed earlier, the agent starts with a prior belief on the set of hypotheses and as it performs experiments, its confidence on the true hypothesis improves. We refer to this initial phase of experimentation as the exploration phase. Soon enough, the MAP estimate . This implies that the agent starts selecting experiments using the distribution which rapidly improves its confidence on . We refer to this subsequent phase of experimentation as the verification phase.
According to Lemma 1 in [2], the exploration phase terminates in time with high probability under Assumption 2. We can relax Assumption 2 using the technique in [3] and show that the exploration phase terminates in sublinear time with high probability. Therefore, in the asymptotic analysis, the impact of exploration on the overall performance is negligible. However, in the non-asymptotic regime, the exploration performance may have a significant impact on the overall performance, especially in problems like dynamic search over trees. This issue was discussed in [5] and [8] in a stopping time setting and heuristic strategies were proposed to improve the exploration performance. One such heuristic is based on Extrinsic Jensen-Shannon (EJS) divergence [9]. Using our notation, the EJS divergence associated with an experiment and posterior belief is the expected increment in confidence level on , that is
[TABLE]
The heuristic strategy in [9] is to greedily select the experiment that maximizes at time .
We described a lower bound on the error probability in Remark 4. Using Jensen’s inequality, we can further weaken the bound (109) to obtain the following lower bound
[TABLE]
Therefore, as a heuristic, one can use the expected confidence level as a proxy for and try to maximize the confidence level instead of the error rate. This is equivalent to maximizing where
[TABLE]
It was shown in [10] that maximizing can be formulated as a Partially Observable Markov Decision Problem (POMDP). Using heuristics for solving POMDPs, we can approximately optimize and one such heuristic was presented in [10]. The strategy based on EJS divergence [9] happens to be a one-step greedy policy with respect to this POMDP.
VI Conclusions
We formulated a fixed horizon active hypothesis testing problem in which the agent can decide on one of the hypotheses or declare its experiments inconclusive. For analyzing this problem, we formulated a sub-problem which is a generalization of Chernoff-Stein lemma [1] to a setting with multiple hypotheses and multiple experiments. We obtained lower bounds on optimal error probability in the sub-problem and used them to obtain lower bounds on misclassification probability in our original problem. We also derived upper bounds by constructing appropriate strategies and analyzing their performance. We defined a quantity called expected confidence rate and based on it, we proposed a heuristic approach for strategy design.
Acknowledgments
This research was supported, in part, by National Science Foundation under Grant NSF CNS-1213128, CCF-1410009, CPS-1446901, Grant ONR N00014-15-1-2550, and Grant AFOSR FA9550-12-1-0215.
The reference list from the paper itself. Each links out to its DOI / PubMed record.
- 1[1] T. M. Cover and J. A. Thomas, Elements of information theory . John Wiley & Sons, 2012.
- 2[2] H. Chernoff, “Sequential design of experiments,” The Annals of Mathematical Statistics , vol. 30, no. 3, pp. 755–770, 1959.
- 3[3] S. Nitinawarat, G. K. Atia, and V. V. Veeravalli, “Controlled sensing for multihypothesis testing,” IEEE Transactions on Automatic Control , vol. 58, no. 10, pp. 2451–2464, 2013.
- 4[4] A. Wald, Sequential analysis . Courier Corporation, 1973.
- 5[5] M. Naghshvar, T. Javidi et al. , “Active sequential hypothesis testing,” The Annals of Statistics , vol. 41, no. 6, pp. 2703–2738, 2013.
- 6[6] M. J. Osborne and A. Rubinstein, A course in game theory . MIT press, 1994.
- 7[7] K. Azuma, “Weighted sums of certain dependent random variables,” Tohoku Mathematical Journal, Second Series , vol. 19, no. 3, pp. 357–367, 1967.
- 8[8] C. Wang, K. Cohen, and Q. Zhao, “Active hypothesis testing on a tree: Anomaly detection under hierarchical observations,” in Information Theory (ISIT), 2017 IEEE International Symposium on . IEEE, 2017, pp. 993–997.
