Tutorials › Marginal Structural Models › The Mean Squared Error Trade-Off for Truncation

Weight Stabilization and Truncation · Tutorial 150 of 501

The Mean Squared Error Trade-Off for Truncation

Learn to compare the squared bias and variance of clipped and untruncated weighted estimators, and apply the exact condition for clipping to reduce mean squared error.

Advanced 10 min read

What You'll Learn

  • Decompose mean squared error into squared bias and variance.
  • Derive the exact finite-sample condition under which clipping improves on untruncated weights.
  • Calculate the bias, variance, and mean squared error in explicit weighted-mean examples.
  • Recognize why bounded weights do not guarantee lower variance for weighted contributions.
  • Identify the assumptions and diagnostics needed to interpret a clipping comparison.

Bias and variance answer different questions

Clipping limits the largest weights, but changing a weight also changes the weighted contribution it multiplies. The resulting estimator may be less variable across samples and, at the same time, systematically shifted away from its target. Whether clipping improves estimation therefore depends on the size of the bias squared relative to the variance saved, not on boundedness alone.

The earlier material on clipping bias establishes why clipping can change a weighted target. Here the goal is to combine that bias with sampling variance in a finite-sample mean squared error (MSE) comparison. We begin with a sample-mean estimator, so the calculation is exact rather than an approximation for a nonlinear estimator.

Definition: For an estimator \(\hat{\theta}\) of a fixed target \(\theta\), its mean squared error is $$ \operatorname{MSE}(\hat{\theta})=E[(\hat{\theta}-\theta)^2]. $$ Its bias is \(E[\hat{\theta}]-\theta\), and its variance is \(\operatorname{Var}(\hat{\theta})\).

Exact MSE decomposition for a clipped weighted mean

Suppose \((W_i,R_i)\), \(i=1,\ldots,n\), are independent and identically distributed pairs. Let \(W_i\geq 0\) be a weight and \(R_i\) the contribution multiplied by that weight. For a fixed cutoff \(c>0\), define \(C_i=\min(W_i,c)\). Consider the untruncated estimator \(\hat{\theta}_U=n^{-1}\sum_i W_iR_i\) and the clipped estimator \(\hat{\theta}_C=n^{-1}\sum_i C_iR_i\), both evaluated against the target \(\theta=E[WR]\).

Assume \(E[WR]=\theta\), \(E[(WR)^2]<\infty\), and \(E[(CR)^2]<\infty\). These conditions give finite means and variances for both contributions. The cutoff is fixed, and the pairs are identically distributed; this calculation does not account for uncertainty from estimating the weights.

Theorem: Under these conditions, define $$ b_c=E[CR]-\theta=E[(C-W)R],\qquad \sigma_U^2=\operatorname{Var}(WR),\qquad \sigma_C^2=\operatorname{Var}(CR). $$ Then $$ \operatorname{MSE}(\hat{\theta}_U)=\frac{\sigma_U^2}{n},\qquad \operatorname{MSE}(\hat{\theta}_C)=b_c^2+\frac{\sigma_C^2}{n}. $$ Consequently, clipping has strictly lower MSE exactly when $$ b_c^2<\frac{\sigma_U^2-\sigma_C^2}{n}. $$

Proof. By identical distribution and linearity of expectation, \(E[\hat{\theta}_U]=E[WR]=\theta\), while \(E[\hat{\theta}_C]=E[CR]=\theta+b_c\). Independence gives \(\operatorname{Var}(\hat{\theta}_U)=\sigma_U^2/n\) and \(\operatorname{Var}(\hat{\theta}_C)=\sigma_C^2/n\). For any square-integrable estimator, write \(\hat{\theta}-\theta=(\hat{\theta}-E[\hat{\theta}])+(E[\hat{\theta}]-\theta)\). Squaring and taking expectations eliminates the cross term because \(E[\hat{\theta}-E[\hat{\theta}]]=0\). Thus MSE equals variance plus squared bias. Applying this identity gives both displayed MSEs. Comparing them and multiplying by \(n>0\) gives the strict-improvement condition. \(\square\)

The condition has an important implication: if \(\sigma_U^2\leq\sigma_C^2\), clipping cannot strictly improve MSE in this comparison, because its squared bias is nonnegative. If \(\sigma_U^2>\sigma_C^2\), the variance reduction must still be large enough to exceed the squared bias. The theorem does not assert that clipping always reduces variance.

Worked calculations

Worked Example: A Variance Reduction That Pays Off at Small Sample Sizes

Let \(R=1\) always, and let \(W=1\) with probability \(0.9\) and \(W=10\) with probability \(0.1\). Use cutoff \(c=4\). Compare the two estimators for \(\theta=E[W]\).

The target is \(\theta=0.9(1)+0.1(10)=1.9\). The clipped weight is 1 with probability \(0.9\) and 4 with probability \(0.1\), so \(E[C]=0.9(1)+0.1(4)=1.3\). Hence \(b_c=1.3-1.9=-0.6\) and \(b_c^2=0.36\).

For the untruncated contribution, \(E[W^2]=0.9(1)+0.1(100)=10.9\), so \(\sigma_U^2=10.9-(1.9)^2=7.29\). For the clipped contribution, \(E[C^2]=0.9(1)+0.1(16)=2.5\), so \(\sigma_C^2=2.5-(1.3)^2=0.81\). The variance reduction is \(7.29-0.81=6.48\).

Clipping improves MSE precisely when \(0.36<6.48/n\), or \(n<18\). At \(n=10\), the clipped MSE is \(0.36+0.81/10=0.441\), compared with the untruncated MSE \(7.29/10=0.729\). At \(n=20\), the clipped MSE is \(0.36+0.81/20=0.4005\), while the untruncated MSE is \(7.29/20=0.3645\). The same cutoff helps in the first comparison and hurts in the second.

Worked Example: Clipping Bias Can Cancel Across Contributions

Suppose the pairs \((W,R)\) take values \((1,0)\), \((5,6)\), and \((10,-1)\) with probabilities \(0.5\), \(0.25\), and \(0.25\), respectively. Use \(c=4\).

The untruncated contributions \(WR\) are 0, 30, and \(-10\). Therefore \(\theta=E[WR]=0.5(0)+0.25(30)+0.25(-10)=5\). The clipped contributions \(CR\) are 0, 24, and \(-4\), with expectation \(0.5(0)+0.25(24)+0.25(-4)=5\). Thus \(b_c=0\): the changes in the two nonzero contributions cancel in expectation.

For the untruncated contribution, \(E[(WR)^2]=0.5(0)+0.25(900)+0.25(100)=250\), so \(\sigma_U^2=250-5^2=225\). For the clipped contribution, \(E[(CR)^2]=0.25(576)+0.25(16)=148\), so \(\sigma_C^2=148-5^2=123\). The MSEs are \(225/n\) and \(123/n\), respectively. Clipping strictly improves MSE for every positive \(n\) in this example because it introduces no bias and reduces the contribution variance.

Worked Example: Bounded Weights Need Not Reduce Contribution Variance

Let \(W=1\) or \(W=10\), each with probability \(0.5\), and let \(R=10\) when \(W=1\) and \(R=1\) when \(W=10\). Use \(c=4\).

The untruncated contribution \(WR\) is 10 in both cases. Thus \(\theta=10\), \(\sigma_U^2=0\), and the untruncated sample mean has MSE zero. After clipping, \(CR\) equals 10 or 4 with equal probability. Its mean is 7, so \(b_c=7-10=-3\). Its variance is \(0.5(10-7)^2+0.5(4-7)^2=9\).

The clipped MSE is \(9+9/n\), which exceeds zero for every \(n\). Although clipping lowers the weight 10 to 4, it breaks the equality of the original weighted contributions and creates both bias and variance. A bound on the weights alone does not establish a reduction in the variance of \(WR\).

What the comparison assumes—and what to check

The exact inequality is useful only when its quantities match the estimator and target under study. It compares estimators based on the same independent, identically distributed observations, a fixed cutoff, and a finite-variance sample mean. In a weighted marginal structural model, a coefficient may be obtained by solving a weighted estimating equation rather than averaging one fixed contribution. A ratio estimator also has a random denominator. For either case, the general identity \(\operatorname{MSE}=\operatorname{Var}+\operatorname{Bias}^2\) still holds when the estimator has a finite second moment, but the sample-mean variance formulas above do not automatically apply.

For a fitted coefficient or a ratio, the relevant comparison must use the bias and variance of that actual estimator, not just the variance of the weights. Likewise, estimated weights can introduce additional sampling variation, and dependence across repeated observations from the same person violates the independent-observation calculation unless the sampling structure is handled appropriately.

In practice, diagnostics should address both sides of the trade-off. Examine how much weight mass is clipped and how the weighted contributions change; then assess whether the resulting variance reduction is plausible relative to the induced shift in the target. A sample variance of observed contributions can help describe variability, but it cannot by itself measure clipping bias: the unobserved contribution beyond the cutoff is precisely what has been altered. Comparisons across plausible cutoffs can be informative, but choosing a cutoff from estimated MSE is a separate procedure.

Clipping, deleting observations with large weights, and leaving weights untruncated are not interchangeable. Clipping retains observations but changes their contributions; deletion removes observations and changes the empirical target; no truncation preserves the original contributions but may leave high variability. Weight normalization is another distinct operation: rescaling weights does not have the same effect as limiting their largest values. Each alternative requires evaluating the estimator’s bias and variance for the target of interest.

Key takeaway: For the sample-mean comparison, clipping improves MSE exactly when its squared bias is smaller than the variance reduction divided by sample size. The weights being bounded is not enough: calculate or justify the change in both bias and contribution variance.

Check Your Understanding

Answer each question using the exact finite-sample comparison. Solutions are included below.

  1. If the clipped estimator has bias \(0.2\), and the untruncated and clipped contribution variances are 5 and 2, respectively, for which positive integer sample sizes is clipping strictly better?
  2. Why does a smaller variance of the clipped estimator not, by itself, establish a smaller MSE?
  3. In the third worked example, what feature of the relationship between \(W\) and \(R\) makes the untruncated contribution constant?
  4. Name two conditions used to obtain the displayed sample-mean variance formulas.

Solutions. (1) The squared bias is \(0.2^2=0.04\), and the variance reduction is \(5-2=3\). Clipping is better when \(0.04<3/n\), equivalently \(n<75\); for positive integer \(n\), this means \(1\leq n\leq74\). (2) MSE adds squared bias to variance, so a variance decrease can be outweighed by clipping bias. (3) The outcome contribution \(R\) is inversely related to the two weight values in just the right way: \(1(10)=10(1)=10\), making \(WR=10\) in both cases. (4) The observations must be independent and identically distributed, and the contribution must have finite second moment; the cutoff is also fixed for the stated comparison.

Clinical Biostats