---
title: Partial Batch Label Sampling (ParBaLS)
url: https://www.emergentmind.com/topics/partial-batch-label-sampling-parbals
type: topic
---

# Partial Batch Label Sampling (ParBaLS)

Searching arXiv for the cited papers to ground the article in the current literature.
Partial Batch Label Sampling (ParBaLS) is a batch acquisition strategy for active learning that is derived from myopic Bayesian decision theory for Expected Predictive Information Gain (EPIG) and Expected Error Reduction (EER). It addresses pool-based batch selection by incrementally constructing a batch, maintaining a partial batch of already chosen but as-yet-unlabeled points, sampling pseudo-labels for that partial batch, and averaging single-point EPIG scores across multiple pseudo-labeled “universes” [2510.09877]. In a related but distinct usage, the same phrase is applied to the Global Optimal Transport Planner (GOP) component of "Holistic Optimal Label Selection for Robust Prompt Learning under Partial Labels," where batch-level optimal transport over partial candidate label sets functions as a form of partial batch label assignment under weak supervision [2604.06614].

## 1. Decision-theoretic foundation

ParBaLS is formulated in the setting of pool-based batch active learning for classification. The basic objects are an unlabeled pool \(D \subset \mathcal{X}\), a labeled set \(L \subset \mathcal{X} \times \mathcal{Y}\), a validation set \(V \subset \mathcal{X}\) sampled from the test distribution, a batch size \(B\), and a Bayesian predictive model \(M_L\) induced by a posterior \(p(w \mid L)\) [2510.09877]. At each active learning round, an algorithm selects a batch \(S_t \subset D\), queries their true labels, adds the labeled points to \(L\), and retrains or updates the model.

The derivation begins from myopic Bayesian decision theory. Under the myopic assumption, only one additional label is imagined to be acquired before making predictions on the evaluation distribution. With negative log-likelihood loss, the optimal predictions after acquiring a label are the posterior predictive distributions, and the resulting acquisition objective reduces to maximizing
\[
\argmax_{\hat{x} \in D} \sum_{x \in V} I(Y_x; Y_{\hat{x}} \mid L).
\]
This is exactly the EPIG objective, and in this formulation EPIG and EER coincide [2510.09877].

The significance of this derivation is that ParBaLS is not introduced as an ad hoc heuristic. It is obtained by extending a myopic BDT objective to the case where batch construction must account for dependencies among unlabeled points. This directly targets the known failure mode of naive top-\(B\) selection, namely redundancy among individually high-scoring points.

## 2. Partial-batch conditioning and pseudo-label universes

The defining idea of ParBaLS is the introduction of a *partial batch* \(S\): a set of points already committed to the current batch, but whose true labels are still unknown [2510.09877]. When selecting the next point \(\hat{x}\), the relevant objective is no longer the unconditional EPIG score, but the batch-aware criterion conditioned on the unknown labels of \(S\):
\[
\argmax_{\hat{x} \in D} \mathbb{E}_{Y_S \mid L} \Big[\sum_{x\in V} I(Y_x; Y_{\hat{x}} \mid Y_S, L)\Big].
\]
The difficulty is that the expectation is over \(|\mathcal{Y}|^{|S|}\) label configurations.

ParBaLS approximates this expectation by Monte Carlo label sampling. It samples \(m\) label configurations \(y^{(i)} \sim Y_D \mid L\), restricts each sample to the partial batch \(S\), treats \(y^{(i)}_S\) as if they were true labels in universe \(i\), updates a model \(M_i\) on \(L \cup \{(x,y_x^{(i)}):x\in S\}\), computes single-point EPIG scores in each universe, and averages them [2510.09877]. The resulting approximation is
\[
\argmax_{\hat{x} \in D} \frac{1}{m}\sum_{i=1}^m \sum_{x \in V} I\big(Y_x;Y_{\hat{x}} \mid Y_S = y^{(i)}_S, L\big).
\]

This construction is “partial” in a precise sense: it samples labels only for the already committed subset \(S\), not for the full future batch. That is the central distinction between ParBaLS and exact joint batch objectives. It is also the main reason its cost scales linearly rather than exponentially with batch size.

A frequent misconception is to equate ParBaLS with simple stochastic diversification. The formalism is narrower and more structured than that. Pseudo-labels are not used merely to perturb scores; they are used to approximate the conditional expectation appearing in the batch-aware myopic BDT objective [2510.09877].

## 3. Algorithmic form and computational properties

The operational form of "ParBaLS EPIG" is an iterative batch-construction procedure [2510.09877]. It starts from \(S \leftarrow \emptyset\), trains a Bayesian model \(M_L = M(L)\), samples \(m\) pseudo-label universes over the unlabeled pool, and then repeats \(B\) times: compute a ParBaLS score for each candidate point, select the maximizer, add it to \(S\), remove it from \(D\), and update each universe-specific model with the corresponding pseudo-labeled point.

The score used at each step is
\[
\text{score}(\hat{x}) = \sum_{i=1}^m \sum_{x \in V} I\big(Y_x; Y_{\hat{x}} \mid Y_S = y^{(i)}_S, L\big).
\]
Within each universe, earlier members of the partial batch are treated as labeled according to the sampled pseudo-labels. Later selections are therefore conditioned on what earlier selections would imply about the test distribution under that universe.

The paper states a round-level complexity of \(\mathcal{O}(T B m)\) in the number of EPIG score evaluations, ignoring training cost [2510.09877]. The crucial scaling statement is that complexity is linear in batch size \(B\) and in the number of label samples \(m\), and does not incur the exponential growth in \(B\) that affects exact joint-label objectives such as BatchBALD-style formulations.

A formal Monte Carlo error guarantee is also given. If \(x^*\) is the optimizer of the exact expectation and \(x_{\text{MC}}\) is the Monte Carlo choice, then with probability \(1-\delta\),
\[
\mathbb{E}_{y_S}[f(x^*, y_S)] - \mathbb{E}_{y_S}[f(x_{\text{MC}}, y_S)]
\le
2 \ln(|\mathcal{Y}|)\sqrt{\frac{\ln|D| + \ln(1/\delta)}{2m}},
\]
which is \(\mathcal{O}(1/\sqrt{m})\) and does not depend on batch size \(B\) [2510.09877]. This bound is one of the distinguishing theoretical features of ParBaLS.

The paper also defines a cheaper variant, ParBaLS-MAP, in which \(m=1\) and the pseudo-label for each unlabeled point is taken to be the MAP class under the current model [2510.09877]. This replaces Monte Carlo label sampling with a deterministic pseudo-label universe.

## 4. Empirical behavior in batch active learning

The empirical study evaluates ParBaLS on two tabular datasets from Kaggle—Airline Passenger Satisfaction and Credit Card Fraud—and four image datasets: CIFAR-10, CIFAR-100, iWildCam, and fMoW from WILDS [2510.09877]. For image data, two derived scenarios are used: one-vs-all binary classification and subpopulation-shift with three classes in the pool but only the first two in the test set.

The predictive model is Bayesian Logistic Regression on fixed embeddings. For WILDS image datasets, embeddings come from CLIP-ViT-B/32; for CIFAR-10/100, from DINOv2-ViT-S/14; PCA retains \(99\%\) variance; and posterior inference uses NUTS in PyMC with \(k = 400\) posterior samples [2510.09877]. Two main regimes are reported: an initial labeled set of 500 with \(T=10\) rounds and \(B=10\), and an initial labeled set of 100 with \(T=10\) and \(B=20\).

The main empirical findings reported are that ParBaLS EPIG has the highest mean performance in 4 of 10 datasets/settings and is “among the top” in 9, while ParBaLS-MAP EPIG has the highest mean in 2 and is among top in 7 [2510.09877]. The paper also reports specific examples in which ParBaLS EPIG or ParBaLS-MAP improves over top-\(B\) EPIG, including Credit Fraud, subpopulation-shift CIFAR-10, and subpopulation-shift iWildCam [2510.09877].

The comparisons emphasize three patterns. First, top-\(B\) EPIG is generally strong but is consistently slightly worse than ParBaLS EPIG. Second, BatchBALD is competitive on some image tasks but becomes too slow in the larger-batch regime. Third, stochastic batch heuristics such as PowerBALD, SoftmaxBALD, SoftRankBALD, and their EPIG variants show mixed performance and require tuning hyperparameters such as temperature \(\beta\), which the authors describe as impractical in active learning [2510.09877].

These results support the interpretation that the partial-batch conditioning mechanism improves label-efficiency by reducing redundancy while retaining the predictive-space rationale of EPIG.

## 5. Relation to top-\(B\), BatchBALD, and BAIT

ParBaLS occupies an intermediate position between cheap but myopically independent top-\(B\) selection and expensive joint batch objectives. Top-\(B\) EPIG simply ranks points by single-point EPIG and selects the \(B\) largest scores. This ignores dependencies among the labels of batch points and can yield highly redundant batches [2510.09877]. ParBaLS instead recomputes scores conditioned on pseudo-labeled versions of the already selected subset, so later points are chosen with awareness of earlier ones.

Relative to BatchBALD, the difference is both conceptual and computational. BatchBALD maximizes \(I(Y_{\hat{X}}; W \mid L)\), a parameter-space mutual information objective, whereas ParBaLS remains in the predictive space and approximates a batch-aware EPIG/EER objective through label-space sampling [2510.09877]. The paper states that BatchBALD requires many posterior weight samples as batch size grows and that its computation and memory become prohibitive for large batches. By contrast, ParBaLS has linear dependence on \(B\) and \(m\), and its Monte Carlo approximation error does not increase with \(B\) [2510.09877].

The same work also shows that BAIT can be derived from the myopic BDT viewpoint via asymptotic approximations involving Taylor expansion, Laplace approximation, and a V-optimal design criterion
\[
\mathrm{Tr}\Big([\nabla^2 \ell_{L \cup S}(\hat{w}_L)]^{-1}\nabla^2 \ell_D(\hat{w}_L)\Big)
\]
[2510.09877]. ParBaLS differs in that it does not rely on those asymptotic parameter-space approximations; it addresses batching for EPIG/EER directly via pseudo-label conditioning.

A plausible implication is that ParBaLS is best understood not as a replacement for all batch acquisition methods, but as a scalable mechanism for importing batch awareness into predictive mutual-information objectives that would otherwise be applied pointwise.

## 6. Partial batch label sampling in prompt learning under partial labels

In "Holistic Optimal Label Selection for Robust Prompt Learning under Partial Labels," the phrase *partial batch label sampling* is used in a different learning problem: partial-label prompt learning with a frozen CLIP-like vision-language model [2604.06614]. The training set is
\[
\mathcal{D} = \{(\mathbf{x}_i, S_i)\}_{i=1}^n,
\]
where each candidate set \(S_i \subseteq \{1,\dots,C\}\) contains exactly one true label and \(L-1\) confusing labels, with confusion rate \(\gamma_c=(L-1)/L\) [2604.06614]. Predictions are produced by a frozen vision encoder \(V\), a frozen text encoder \(T\), and learnable prompt parameters \(\mathbf{t}\).

The proposed method, HopS, has two components: a Local Density-based Filter (LDF) and a Global Optimal Transport Planner (GOP) [2604.06614]. The paper explicitly states that GOP is “the part that directly realizes partial batch label sampling.” For a mini-batch of size \(B\), GOP defines a source distribution \(\mathbf{r}\) that is uniform over batch instances and a candidate-aware target distribution \(\mathbf{c}\) over labels:
\[
c_l = \frac{1}{B}\sum_{i=1}^{B} \frac{s_{il}}{|S_i|}.
\]
It then solves an entropic optimal transport problem over a transport plan \(\mathbf{P}\in\mathbb{R}_+^{B\times C}\) with costs
\[
\mathbf{M}_{\text{cost}}[i,j] =
\begin{cases}
1 - p(s_{ij}\mid \mathbf{x}_i;\mathbf{t}), & \text{if } j\in S_i,\\
+\infty, & \text{otherwise},
\end{cases}
\]
subject to row-sum, column-sum, and candidate-set constraints [2604.06614].

The resulting transport plan is interpreted as a batch-level probabilistic assignment from instances to labels, and the hard pseudo-label for each instance is obtained by row-wise argmax:
\[
y_i^{\text{global}} = \arg\max_{j \in S_i} P^{\text{ot}}_{ij}.
\]
The paper characterizes this as “partial batch label sampling” because assignments are computed jointly over the batch, under partial candidate-label constraints, before being collapsed to discrete labels [2604.06614].

This usage is distinct from ParBaLS EPIG in active learning. In the active learning setting, ParBaLS samples hypothetical labels for already selected but unlabeled points in order to choose future query points [2510.09877]. In HopS, batch-level optimal transport assigns pseudo-label mass across instances and classes within a training batch in order to disambiguate partial labels during prompt learning [2604.06614]. The shared conceptual element is the joint treatment of ambiguous or unknown labels at batch level rather than independent per-instance selection.

The HopS paper reports that GOP, together with LDF, improves performance under partial supervision on eight benchmark datasets—Caltech-101, DTD, EuroSAT, FGVCAircraft, Food-101, Oxford Flowers-102, OxfordPets, and UCF101—and that combining local and global strategies yields higher accuracy and faster convergence than either alone [2604.06614]. It also notes that performance is sensitive when batch size \(B\) is close to the number of classes \(C\), with best performance typically occurring when \(B\) is near \(C\) [2604.06614]. This suggests that the batch-level allocation constraints central to GOP play a role analogous to the redundancy-reduction role of partial-batch conditioning in active learning, although the two mechanisms are mathematically different.

ParBaLS therefore denotes a family resemblance rather than a single invariant algorithmic form: in one case, Monte Carlo pseudo-label conditioning for batch active learning; in another, optimal-transport-based batch label assignment under partial labels. In both cases, the defining feature is that ambiguous label decisions are made jointly at batch level instead of independently per sample.

Source: https://www.emergentmind.com/topics/partial-batch-label-sampling-parbals