EconBase
← Back to paper

Inference methods for unit-specific coefficients in panel data models with latent group structure

Extracted main text — title through conclusion, appendix excluded. This is what our citation measures are computed over, published so the extraction can be checked by eye.

86,025 characters · 21 sections · 67 citation commands

Rendered from LaTeX for readability, not typeset faithfully. Citation keys are highlighted; maths is left as source; figures, tables and equation environments are summarised rather than reproduced; unrecognised commands are greyed out so nothing is silently dropped. Email addresses are removed.

Inference methods for unit-specific coefficients in panel data models with latent group structure

abstractThis paper introduces statistical inference procedures for unit-specific coefficients in panel data models, where the coefficients exhibit a latent group structure. The proposed methods achieve efficiency gains by clustering units into a small number of groups, while explicitly accounting for the statistical uncertainty of group assignments. The core idea is to integrate standard inference procedures, such as the $t$-test and Wald tests, with confidence sets for group membership. Two methods are proposed: the first takes the minimum of the test statistics over the confidence set for group membership, and the second corrects for bias caused by possible group misassignment. The former can produce shorter but possibly disconnected sets, while the latter guarantees connected, interpretable intervals at some cost in length. We also develop standard errors that are adjusted for possible group misassignment and valid even with short time periods, which may be of independent interest. Monte Carlo simulations demonstrate that our approach yields narrower confidence sets for units with relatively large error variances than unit-by-unit time-series methods. In contrast, ignoring statistical uncertainty in the group membership estimation leads to distortions in size and coverage. We illustrate the method with an empirical example that estimates the effect of the minimum wage in each U.S. state. Keywords: Panel data, latent group structure, confidence set, unit-specific coefficients JEL classifications: C23, C33, C38

Introduction

Heterogeneity plays a crucial role in panel data analysis, yet modeling and accounting for it in statistical inference pose numerous challenges. In the literature, various methods have been developed to address heterogeneity across units. One prominent approach is the latent group structure model, which posits that units can be partitioned into a few groups, with units within each group homogeneous and those across groups exhibiting distinct characteristics pionati2025latent. This framework is both flexible enough to capture complex heterogeneity and parsimonious enough to remain interpretable. However, inference methods for heterogeneous coefficients are still limited. Existing works mostly focus on group-level inferences.

In many applications, the objects of ultimate interest are the coefficients of individual units rather than those of the groups to which units are assigned. Researchers and policymakers typically need to draw conclusions about a particular unit, such as the effect of a policy in a specific state. In our empirical application, for instance, each U.S. state can, to some extent, set its own minimum wage, so the employment effect relevant to a given state's policy debate is that state's own coefficient. From this perspective, a concrete decision is guided by the parameter attached to the unit itself, and the group structure is primarily an estimation device that improves precision by pooling information across similar units. Reporting only group-level coefficients would fulfill these purposes if the group structure were known, but doing so can be misleading when a unit's group membership is uncertain. In contrast, an alternative of unit-by-unit time series regression provides unbiased inference but is inefficient. Furthermore, unit-by-unit regression may not be applicable in models such as the time fixed effects model.\footnote{Indeed, unit-by-unit regression cannot be applied in our empirical application due to the presence of time fixed effects and the lack of variation within units.} These observations motivate inference procedures that target unit-specific coefficients while still exploiting the efficiency gains afforded by the group structure.

In this paper, we develop new methods for statistical inference on unit-specific coefficients when these coefficients exhibit a latent group structure. If a unit's group membership were known, inference would simply involve the corresponding group-level coefficient. However, a central challenge in making valid statistical inferences on the coefficient for each unit is that group memberships are usually unobserved and must be estimated from the data, which introduces additional statistical uncertainty. This is particularly relevant in situations where group-specific parameters can be estimated precisely, but where uncertainty remains in the group membership structure. dzemskiokui2021convergence show that such situations occur when the group memberships of a sufficient number of units—not necessarily all—are consistently estimated, while other units with larger error variances have uncertain group assignments (namely their group memberships may not be consistently estimated). In such a scenario, making statistical inferences for units with relatively large error variances by relying solely on coefficient estimates for the estimated group can lead to misleading conclusions because it ignores statistical uncertainty.

To address this issue, our approach integrates standard inference procedures for group-specific coefficients, such as $t$-tests and Wald tests, with the confidence set methodology for group membership proposed by dzemski2024confidence. This combination ensures that our inference procedures properly account for the uncertainty in group membership estimation and, at the same time, enjoy the efficiency of group-wise estimation.

We propose two methods for unit-level statistical inference. The first approach relies on minimum-type test statistics. It constructs a confidence set by inverting the minimum of the $t$- or Wald test statistics for group-specific parameters, with the minimum taken over the groups included in a confidence set for the group membership. The underlying idea is that, under the true null hypothesis, the test statistic should be small when the group assignment is correct; therefore, it should be small for at least one group in the confidence set. This construction suggests that the resulting confidence set can be represented as a union of confidence sets built for each group. The minimum-type test statistics method may produce confidence sets that are disconnected. This possible disconnectedness has both advantages and disadvantages: it can reduce the volume of the confidence set and increase power, but in practice, a disconnected confidence set may be harder to interpret or communicate.

The second method employs bias correction. The coefficient for a given unit is estimated using its group-specific coefficient, but this estimator can be biased if the group assignment is inaccurately determined. Our approach addresses this bias by explicitly accounting for potential group misclassification. Specifically, we adjust the test statistic by considering the difference between the coefficient under the estimated group membership and those of other plausible groups to which the unit may belong, as indicated by the confidence set of dzemski2024confidence. The bias correction can be applied to the $t$-statistic for a single parameter and to the Wald test statistic for multiple parameters. In contrast to the minimum-type approach, the bias-correction approach necessarily produces a connected confidence interval for a single parameter. It is important to note that bias correction for the Wald test statistic is nontrivial, as the bias can affect the statistic nonlinearly. The bias correction ensures that the confidence set achieves the desired coverage probability. However, this approach is inherently conservative.

We also propose an asymptotic variance estimator valid under fixed-$T$ asymptotics, extending bonhomme2015grouped to models with heterogeneous coefficients. This asymptotic variance estimator is robust to group misallocation and better suited to our setting. We find that it achieves better coverage and does not necessarily make inference conservative. This contribution may also be of independent interest.

Our proposed procedures are straightforward to implement, both conceptually and computationally. They require only the group-specific coefficient estimates, their standard errors, and unit-level confidence sets for group memberships. Our Monte Carlo simulations indicate that the most computationally intensive step is the estimation of the group membership structure and group-specific coefficients. However, once these are obtained, the calculation of confidence sets for group memberships is rapid, as documented in dzemski2024confidence. With this preliminary information in hand, applying our procedure is seamless.

As an empirical illustration, we examine the effect of a minimum-wage increase in each US state. The impact of the minimum wage varies across states. It is therefore important to estimate state-specific effects. This is especially relevant since states can set their own policies. We use the data from dube2010minimum and extend the analysis by wang2019heterogeneous,dzemski2024confidence. In those studies, states are grouped into four based on similarities in coefficients. The results highlight the importance of accounting for group membership uncertainty. Specifically, when the groups differ mainly in covariate coefficients rather than the minimum wage effect itself, ignoring uncertainty in group assignments can be misleading. Our confidence sets address this issue by explicitly incorporating group assignment uncertainty.

The Monte Carlo simulations demonstrate the situations in which our proposed methods are most effective. We compare our methods to those that ignore statistical uncertainty in group membership estimation (naive) and to the unit-by-unit estimation approach. The naive method fails to provide adequate coverage for units whose group memberships remain uncertain, leading to severe undercoverage. The unit-by-unit method achieves adequate coverage, but its interval length can be large unless the error variance is small. In contrast, our methods maintain adequate coverage and, while our confidence intervals may be wider than those from the unit-by-unit approach for units with small error variance, they are significantly shorter when the error variance is moderate to large. This advantage is especially notable when confidence intervals are allowed to be unions of disjoint intervals. We therefore conclude that our methods are particularly beneficial for units with moderate to large error variances.

Our contributions are twofold. First, while earlier studies have primarily investigated asymptotic distributions and inference for group-specific coefficients, they do not address inference for unit-level coefficients. Second, our approach does not rely on the result that the estimated group structure converges to the true membership with probability approaching one (so-called super-consistency). Note that under super-consistency, inferences for group-specific and unit-specific coefficients become equivalent. Our method incorporates the statistical uncertainty associated with estimating group memberships.

\paragraph{Literature review}

This paper contributes to the recent econometrics literature on latent group structure. A partial list includes hahn2010panel, lin2012estimation, bonhomme2015grouped, sarafidis2015partially, ando2016panel,su_identifying_2016, wang2016homogeneity, liu2020identification, mehrabani2022estimation,chetverikov2022spectral,yu2024spectral, mugnier2025simple. pionati2025latent provides an excellent review. This paper addresses the underdeveloped area of inference methods for panel data models with latent group structure, specifically focusing on unit-specific coefficients. To our knowledge, this problem has not been previously tackled. wan2025conditionalselectiveinferenceselected,akgun2025robustinferencemethodslatent consider post-clustering inference in panel data, but their focus is again on group-specific coefficients. beyhum2024inference also consider post-clustering inference, but their interest is in the parameter common to all units.

Our approach combines standard statistical inference methods for the parameter of interest with a confidence set for the nuisance parameter, which, in our case, is the group membership. Our minimum-type approach builds on a long tradition of obtaining valid inference by combining over a confidence set for a nuisance parameter. Dufour1990 developed this approach for time series models. In statistics, BergerBoos1994 and Silvapulle1996 independently formalized computing a valid $p$-value by taking the supremum of $p$-values over a confidence set for the nuisance parameter. The same principle has been adapted to post-moment-selection inference by DiTraglia2016 in moment restriction models for GMM estimators. A related idea is used by kaji2025controlling for a financial application. A key distinction between the earlier works and the present paper is that the nuisance parameter in our model is discrete, while the earlier works considered continuous nuisance parameters; the discrete nature of the nuisance parameter may result in disconnected confidence intervals. The bias-correction approach, by contrast, relates to the literature on bias-aware confidence intervals built from bias bounds. For instance, ArmstrongWeidnerZeleneev2025 construct bias-aware confidence intervals for interactive fixed effects models based on a possible set of interactive fixed effects.

Our method is related to empirical Bayes in terms of purpose, but is fundamentally different from it in both approach and implementation. Empirical Bayes is a widely used approach for estimating and inferring unit-specific parameters and unit-specific predictions. Recent econometric articles on this topic include liu2020forecasting,liu2021panel,liu2023forecasting,kwon_optimal. The core principle of empirical Bayes methods is to utilize information from other units to improve the precision of these estimates. These methods typically begin with unit-level statistics and subsequently adjust them using information from the broader population, thereby enhancing statistical inference. In contrast, the proposed method adopts a different perspective. It assumes the presence of a latent group structure, enabling unit-specific parameters to be estimated as parameters shared among all units within a group. After estimating group-level parameters, the method incorporates information from the unit of interest to ensure robustness against statistical errors arising from the clustering process.

\paragraph{Roadmap} Section (ref) introduces the setting. Section (ref) describes our two new methods for inference on unit-specific parameters. Their asymptotic properties are given in Section (ref). Section (ref) develops asymptotic variance estimators, including one valid under fixed-$T$ asymptotics. Section (ref) provides an empirical illustration, and Section (ref) presents Monte Carlo simulation results. Section (ref) concludes the paper. The appendix contains the proofs of the main results, implementation details, and additional discussions.

Settings and preliminaries

We consider panel linear regression models with heterogeneous coefficients. Specifically, the coefficients exhibit a grouped pattern of heterogeneity. Our objective is to conduct statistical inference on the coefficients for each unit, accounting for this latent group structure.

Suppose we observe panel data $\{(y_{it}, x_{it}) : t = 1, \dots, T;\ i = 1, \dots, N\}$, where $y_{it}$ denotes the scalar outcome for unit $i$ at time $t$, and $x_{it}$ is a $p \times 1$ vector of regressors.

We consider the following linear regression model with heterogeneous coefficients:

align[align omitted — 60 chars of source]

where $\beta_i$ is the $p\times 1 $ coefficient vector for unit $i$, $\varepsilon_{it}$ is the error term, and $\beta_i^0$ denotes the true value of $\beta_i$.

The coefficients are assumed to follow a grouped pattern of heterogeneity: units are partitioned into $G$ distinct groups, denoted by $\mathbb{G} = \{1, \dots, G\}$. Let $g_i \in \mathbb{G}$ represent the group membership for unit $i$, with $g_i^0$ the true group assignment. All units within the same group share a common coefficient vector, while units in different groups have distinct coefficient vectors. Let $\theta_g\in\Theta$ denote the coefficient vector for group $g$, and $\theta_g^0$ its true value. Formally, we assume

align[align omitted — 45 chars of source]

Our objective is to conduct statistical inference on $\beta_i$. With sufficiently long panel data, one could, in principle, estimate $\beta_i$ for each unit individually using time-series regressions. However, this approach can be inefficient, especially when units share a grouped pattern of heterogeneity. By pooling cross-sectional information from units within the same group, we can make more precise inferences about $\theta_{g_i}$, the common coefficient for group $g_i$. The main challenge is that group memberships are unknown and must be estimated from the data. This introduces additional statistical uncertainty, making it non-trivial to fully leverage the cross-sectional information while properly accounting for the estimation of group structure. Indeed, ignoring this uncertainty may lead to biased inference as we argue below.

The source of bias

Suppose that some estimates $\hat{\theta}_g$ and $\hat{g}_i$ for the group-level slopes and the group memberships are available. We estimate $\beta_i$ using $\hat{\theta}_{\hat g_i}$. There are two main sources of statistical uncertainty: (1) the estimation error of $\hat{\theta}_{g}$ as an estimator of $\theta_g$, and (2) uncertainty in the estimated group membership $\hat g_i$. For the first, the literature provides valid procedures to evaluate the statistical uncertainty of group-specific coefficients. However, to make inferences for $\beta_i$, the second issue, namely an additional bias caused by the misclassification of group membership, must be addressed.

To be more specific, the estimation error of $\hat{\theta}_{\hat{g}_i}$ can be decomposed as

align[align omitted — 206 chars of source]

In the above decomposition, the first term converges in distribution to a normal distribution, whereas the second term represents the bias induced by potential misclassification. The literature often assumes that the probability of $\hat g_i = g_i^0$ converges to one, effectively treating the bias term as negligible. However, in finite samples, this bias may remain substantial and should not be ignored. Inference about the unit-specific parameters requires accounting for uncertainty in the estimation of $g_i$.

In what follows, we will explain how this statistical uncertainty can be quantified. We note that we are agnostic about the procedures used in the estimation of $\theta_g$ and the construction of confidence sets for $g_i$, as long as the conditions stated later are satisfied. For concreteness, we discuss the KMeans-type procedure given in bonhomme2015grouped and okui_heterogeneous_2021 for the estimation of the group-specific parameters and the procedure by dzemski2024confidence for the confidence set for group membership.

Estimation of the group-level parameters

The first step is to estimate the group-specific parameter $\theta_{g}$. It may be estimated by minimizing the least square objective function with respect to both the group-specific coefficients and group structure. We denote the group-specific coefficient estimator by $\hat{\theta}_g$ for $g\in \mathbb{G}$. Also, let $\hat{g}_{i}$ denote the estimates for $g_i^0$, for $i=1, \dots,N$. The estimator is defined as:

align[align omitted — 234 chars of source]

This minimization problem may be solved by extending the KMeans algorithm.

enumerate$s=0$. Let $\hat{g}_i^{(0)}$ be the initial group assignment. • Compute $\theta_{g}^{(s)}$ for $g \in \mathbb{G}$ using the observations belonging to group $g$: \begin{align} \hat{\theta}_{g}^{(s)} = \operatorname*{\arg\!\min}_{\theta_g} \sum_{\hat{g}_i^{(s)}=g} \sum_{t=1}^T (y_{it} - x_{it}'\theta_g)^2. \end{align} • Update the group membership assignment based on $\hat{\theta}_{g}^{(s)}$: \begin{align} \hat{g}_{i}^{(s+1)} = \operatorname*{\arg\!\min}_{g \in \mathbb{G}} \sum_{t=1}^T (y_{it} - x_{it}'\hat{\theta}_{g}^{(s)})^2. \end{align} • $s= s+1$. • Repeat 2-4 until convergence.

The algorithm extends KMeans clustering to panel-data regression models and is considered by bonhomme2015grouped in the context of models with grouped fixed effects and okui_heterogeneous_2021 for models with time-varying coefficients with group structure. Step 1 initializes the process by assigning each individual to an initial group. In Step 2, group-specific regression coefficients are estimated by minimizing the sum of squared residuals within each group, using only the observations currently assigned to that group. Step 3 reassigns each individual to the group whose estimated parameters best fit their data, specifically selecting the group that minimizes the squared prediction error aggregated across all time periods. Step 4 increments the iteration counter. Steps 2 through 4 are repeated until convergence, defined as the point at which group assignments stabilize between iterations. The algorithm alternates between estimating group-specific parameters and reassigning observations to groups, analogous to the standard KMeans algorithm but adapted for regression contexts. This algorithm always converges to a local minimum. To ensure the global minimum, we would need to run this algorithm with many initial values.

We also need two additional statistics that can be computed from the estimation results. Let $\widehat{N}_{g}=\Sigma_{i=1}^{N}1\left\{\hat{g}_{i}=g\right\}$ denote the estimates for $N_g=\Sigma_{i=1}^{N}1\left\{g_{i}^0=g\right\}$ (the number of units in group $g$). We also use $\widehat{\Sigma}_{g}$ to denote the estimates for the asymptotic variance-covariance matrices $\Sigma_g$ of $\sqrt{N_{g} T}\left(\hat{\theta}_g-\theta_{g}^{0}\right)$. Section (ref) discusses how to obtain $\widehat{\Sigma}_{g}$.

We note that many alternative estimators are available in the literature su_identifying_2016, wang2016homogeneity, mehrabani2022estimation. Our proposed inference procedures are agnostic about the choice of the estimator. All estimators that satisfy Assumptions (ref)--(ref) given below can be used.

Confidence set for group membership

Next, we construct a confidence set for the group membership parameter $g_i$ by following the procedure outlined in dzemski2024confidence. The central insight is that, for each unit, the sum of squared errors is minimized when the coefficient corresponds to the true group membership. To obtain the confidence set, the procedure inverts a test that examines whether this minimum is achieved under the parameter value associated with the group specified by the null hypothesis. We note that while dzemski2024confidence center on confidence sets for the entire group structure, our discussion here focuses on a marginal confidence set for each unit, as our interest is in inferring the coefficient for each unit.

The first step is to compute the adjusted difference between the squared losses under the two group membership assignments. Let

align*[align* omitted — 243 chars of source]

The first two terms on the right-hand side are squared residuals representing the fit of assigning unit $i$ to group $g$ and to group $h$, respectively. The third term is an adjustment that recenters the difference between the two squared residuals so that it is asymptotically mean zero under the null hypothesis $g_i^0 = g$. The $t$-statistic is then computed as follows:

align*[align* omitted — 128 chars of source]

where $\widehat{\Xi}_i (g,h,h)$ is a long-run variance estimator of $\hat{d}_{it}(g, h)$. Details regarding the calculation of $\widehat{\Xi}_i (g,h,h)$ are provided in Appendix (ref).

The test statistic is defined as the maximum $t$-statistic, taken over all $h \neq g$, where $g$ is the group membership assignment under the null hypothesis. The test statistic is given by:

align*[align* omitted — 100 chars of source]

If the null hypothesis $g_i^0=g$ is true, all values of $\widehat{D}_{i}(g, h)$ are unlikely to be large and positive. The null hypothesis is rejected when $\widehat{T}_i(g)$ exceeds the critical value. This test is one-sided.

The critical value is computed from the distribution of the maximum of a random vector following a multivariate $t$-distribution and is denoted as $\hat{c}_{i, \alpha}(g)$, where $\alpha$ is a prespecified size.\footnote{The discussion here is for a general $\alpha$. Later, we use $\alpha_1$ for the confidence set for group memberships.} In the case of two groups ($G=2$), the critical value is computed from the usual $t$-distribution with degrees of freedom $T-1$:

align*[align* omitted — 108 chars of source]

where $t_{T - 1}^{-1}(p)$ denotes the $p$-quantile of Student's $t$-distribution with $(T - 1)$ degrees of freedom. For $G \geq 3$, the critical value is given by

align*[align* omitted — 236 chars of source]

where $t_{\max, \Omega, T-1}$ denotes the distribution function of the maximal entry of a centered random vector with a multivariate $t$-distribution with scale matrix $\Omega$ and $(T-1)$ degrees of freedom. $\widehat{\Omega}_i (g)$ is an estimator of the correlation matrix of the moment inequalities, $\rho$ is a regularization function, and $\epsilon_N$ is a regularization parameter. The procedures for obtaining $\widehat{\Omega}_i (g)$, $\rho$, and $\epsilon_N$ are discussed in Appendix (ref).

The marginal confidence set for unit $i$ is obtained by inverting a test for group membership.

align*[align* omitted — 154 chars of source]

$\hat{g}_i$ is added to the confidence set to ensure that the set is not empty.

All necessary components for the proposed procedures have now been collected. Specifically, the confidence sets for $\beta_i$ can be constructed by combining $\hat{\theta}_{g}$, $\widehat{\Sigma}_g$, and $\mathrm{CS}_{i, \alpha} $. The following section discusses the proposed procedures.

Inference methods for unit-specific coefficients

In this section, we present inference methods for unit-specific coefficients $\beta_i$. We combine the group-level slope estimates and their standard errors with the confidence set for group membership to construct confidence sets for each unit’s coefficient. Specifically, let $1-\alpha$ be our intended coverage probability. We divide $\alpha$ into two parts: $\alpha = \alpha_1 +\alpha_2$. The confidence level for the group assignment is set to $1 - \alpha_1$. For each group membership assignment, we consider a confidence set for the slope coefficients with coverage of $1- \alpha_2$. Combining these two achieves the intended coverage $1-\alpha$. Two procedures are proposed for this purpose. Both methods address the bias in (ref). The minimum-type approach avoids it by considering various possible groups within which the test statistic is evaluated, and the bias-correction method keeps the estimated group but expands the confidence set to include the coefficient values in the other possible groups.

Confidence set based on minimum-type statistics

We first propose a procedure based on the minimum of the $t$- or Wald test statistics across groups in the confidence set for group membership. The idea is that, under the true value of the coefficient, $\beta_i^0=\theta_{g_i^0}^0$, the test statistic constructed from the group-level estimate, $\hat{\theta}_g$, should be small for some group $g$. The problem is that relying on $\hat{\theta}_{\hat{g}_i}$ incurs bias due to uncertainty in the point estimate $\hat{g}_i$, as shown in (ref). Therefore, it is natural to base inference on $\hat{\theta}_g$ for $g\in\mathrm{CS}_{i, \alpha_1}$, instead of relying solely on $\hat{\theta}_{\hat{g}_i}$, to avoid the bias identified in (ref).

Specifically, our proposal is to take the minimum of the (absolute) test statistics based on $\hat{\theta}_g$ over $g\in\mathrm{CS}_{i, \alpha_1}$, and then compare it with a standard critical value. The confidence set follows upon inverting the minimum test statistic. Suppose we want to construct a confidence set for $R\beta_i^0$ for some $r\times p$ matrix $R$ ($r\leq p$). Construction is discussed separately for $r=1$ and $r \geq 2$.

When $r=1$, the minimum absolute $t$-statistic is defined as

align[align omitted — 173 chars of source]

where $\operatorname{se}(g)\coloneqq\sqrt{R\widehat{\Sigma}_gR' / \widehat{N}_{g} T}$. The confidence interval based on $\tilde{t}_{i,NT, \alpha_1}(\theta)$ is defined as

align[align omitted — 165 chars of source]

with $z_{\alpha}$ the $\alpha$-quantile of $N(0,1)$.

Next, we consider inference for multiple parameters or multiple linear combinations of parameters, that is, when $r \geq 2$. In this case, we consider the Wald statistic, $W_{i,NT}(\theta, \hat{g}_i,\hat{g}_i)$, where

align[align omitted — 190 chars of source]

The second argument $g$ of $W_{i,NT}$ indicates the group used for scaling $\widehat{N}_{g}$ and $\widehat{\Sigma}_{g}$, and the third argument $f$ refers to the group for the coefficient vector. As in the case with $r=1$, $W_{i,NT}(\theta, \hat{g}_i,\hat{g}_i)$ suffers from bias due to uncertainty in $\hat{g}_i$, albeit in a nonlinear fashion because of its quadratic form. By the decomposition, $(\hat{\theta}_{g_i^0} - \theta^0_{g_i^0} ) + (\hat{\theta}_{\hat{g}_i} - \hat{\theta}_{g_i^0})$, in (ref), it holds that

align[align omitted — 958 chars of source]

We observe two sources of bias arising even when $\widehat{N}_{g_i^0}$ and $\widehat{\Sigma}_{g_i^0}$ are known. The first term represents the squared bias, which is necessarily positive. The second term is the cross-product between the bias and the estimation error, and its sign may be either positive or negative depending on the data. Furthermore, since $W_{i,NT}(\theta, \hat{g}_i,\hat{g}_i)$ uses $\widehat{N}_{\hat{g}_i}$ and $\widehat{\Sigma}_{\hat{g}_i}$ for scaling, the possible misclassification also affects inference through this channel.

The above bias can be avoided by using the minimum-Wald statistic:

align*[align* omitted — 143 chars of source]

The confidence set is defined as

align[align omitted — 175 chars of source]

where $\chi_{p,\alpha}^2$ is the $\alpha$-quantile of the chi-square distribution with $p$ degrees of freedom.

To practically compute these confidence sets, note that $\mathrm{CI}^{\mathrm{MT}}(i,\alpha)$ and $\mathrm{CS}^{\mathrm{MW}}(i, \alpha)$ can be calculated as the union of the confidence intervals (or confidence sets) based on $\hat{\theta}_g$, where the union is taken across $g\in\mathrm{CS}_{i, \alpha_1}$. More specifically, our confidence interval or set can be written as:

align[align omitted — 413 chars of source]

This formulation follows the ideas of Dufour1990,BergerBoos1994,Silvapulle1996, which are early works on constructing valid confidence sets that combine over nuisance parameter values.

The minimum-Wald approach may produce disjoint confidence sets.\footnote{We found through simulations that the minimum-type confidence interval is disjoint more frequently when $T$ is smaller and/or the error variance is larger.} This feature has both advantages and disadvantages. Compared with the bias-correction approach introduced below, which always yields connected intervals, this approach would yield smaller confidence sets and higher power. A drawback is that disjoint confidence intervals may be hard to interpret and explain. We illustrate these features in the empirical example.

Confidence set based on bias correction

Next, we introduce an alternative procedure that relies on bias correction. While the bias given in (ref) and (ref) cannot be directly estimated, we can bound its possible magnitude. We discuss cases with $r=1$ and $r\geq 2$ separately. The method for $r=1$ is for the scalar parameter and applies the upper and lower bounds of the bias to the confidence interval. The method for $r\geq 2$ corrects the quadratic forms of the bias and the cross-product term, corresponding to the formula in (ref).

When $r=1$, our $100(1-\alpha)\%$ bias-corrected confidence interval is defined as follows:

align[align omitted — 491 chars of source]

where

align[align omitted — 379 chars of source]

The bias-corrected confidence interval is constructed by explicitly bounding the magnitude of the bias, which arises from statistical uncertainty in $\hat g_i$. By considering a confidence set for group membership, we bound the bias from below by $\widehat{\mathrm{LBias}}_i$ and from above by $\widehat{\mathrm{UBias}}_i$.

It is also important to correct the standard error. To illustrate this, consider the decomposition of the $t$-statistic:

align[align omitted — 185 chars of source]

where $A_{i, N T}^*=\sqrt{\widehat{N}_{\hat{g}_{i}} T}R\left(\hat{\theta}_{g_i^0}-\theta_{g_i^0}^{0}\right) / \sqrt{R\widehat{\Sigma}_{\hat{g}_{i}}R'}$ and $B_{i, N T}^*=\sqrt{\widehat{N}_{\hat{g}_{i}} T}R\left(\hat{\theta}_{\hat{g}_{i}}-\hat{\theta}_{g_i^0}\right) / \sqrt{R\widehat{\Sigma}_{\hat{g}_{i}}R'}$. Notice that the denominator of $A_{i, N T}^*$ is $\sqrt{R\widehat{\Sigma}_{\hat{g}_{i}}R'}/ \sqrt{\widehat{N}_{\hat{g}_{i}} T}$, but this is not the standard error for the numerator $\hat{\theta}_{g_i^0}-\theta_{g_i^0}^{0}$ when $\hat g_i \neq g_i^0$. Therefore, it is necessary to account for possible group misassignment by using the maximum standard error across all groups in the confidence set for group membership.

We extend this approach to $r\geq2$. The bias formula given in (ref) motivates the following bias-corrected Wald statistic:

align[align omitted — 617 chars of source]

The bias-corrected Wald statistic, $\overline{W}_{i, N T, \alpha_1}(\theta)$, has three terms. The first, $\min _{g \in \mathrm{CS}_{i, \alpha_1}} W_{i, N T}(\theta, g, \hat{g}_{i})$, resembles the standard Wald statistic but may have bias if group membership is misclassified. We adjust by using variance-covariance matrices for each group in the confidence set (noting the minimization pertains only to $\widehat{N}_g$ and $\widehat{\Sigma}_{g}$). The second term is the maximum squared difference between the coefficients estimated for the assigned and true group. The third term is the cross-product between bias and estimation error; since its sign is indeterminate, we conservatively use its absolute value. This approach is robust, yet inherently conservative.

A confidence set is constructed by inverting the bias-corrected Wald statistic:

align[align omitted — 165 chars of source]

Confidence intervals based on the bias-corrected $t$ statistic are always connected, unlike those from the minimum $t$-statistic.\footnote{Note that confidence sets based on the bias-corrected Wald statistic are not necessarily connected, whereas the bias-corrected confidence interval is always connected.} However, this connectedness leads to wider, more conservative intervals. It is also important to note that these two methods differ in the choice of the variance or standard errors in the test statistics. The minimum-type approach can utilize the standard error for each group. On the other hand, the bias correction method must use the maximum standard error, since it fixes the center at $\hat{\theta}_{\hat{g}_i}$. Thus, the bias-correction method has two sources of conservativeness. We need to consider trade-offs: the bias-correction method offers connected, interpretable intervals, while the minimum-type approach yields smaller, more powerful ones. These trade-offs are illustrated in an empirical example in Section (ref) and in the simulations in Section (ref).

Asymptotic analysis

In this section, we study the theoretical properties of the confidence sets proposed in Section (ref). We first give a few assumptions and then provide the asymptotic justification for our methods.

Assumptions

We derive the asymptotic properties of our proposed inference methods under high-level assumptions about the estimators. These conditions are satisfied by the existing procedures under suitable conditions. We briefly discuss more primitive conditions, but the details can be found in the original papers.

The first assumption is the asymptotic normality of the group-specific slope estimators.

asmThere exists a permutation $\sigma:\mathbb{G}\to\mathbb{G}$ such that $\sqrt{N_{g} T}\left(\hat{\theta}_{\sigma(g)}-\theta_{g}^{0}\right) \xrightarrow{d} N\left(0, \Sigma_g\right)$ for some $\Sigma_g>0$ for all $g \in\mathbb{G}$.

The asymptotic normality of the group-specific slope estimators is established for a wide range of clustering methods su_identifying_2016, wang2016homogeneity, mehrabani2022estimation. dzemskiokui2021convergence show that Assumption (ref) can hold even when some units are potentially misclassified---a case where our proposed methods are particularly relevant. Following the convention in the literature, we take $\sigma(g)=g$ for all $g\in\mathbb{G}$ by relabeling groups.

asm$\widehat{N}_{g} / N_{g} \xrightarrow{p} 1$ for all $g$.

Assumption (ref) is typically implied by classification consistency; see, for example, Corollary 2.3 of su_identifying_2016. Furthermore, dzemskiokui2021convergence show that Assumption (ref) holds even if some units are potentially misclassified; see the proof of Theorem 1 of dzemskiokui2021convergence.

The next assumption requires that the group-specific variance-covariance matrix estimators are consistent.

asm$\widehat{\Sigma}_{g} \xrightarrow{p} \Sigma_g$ for all $g \in\mathbb{G}$.

How to obtain $\widehat{\Sigma}_{g} $ satisfying Assumption (ref) is discussed in Section (ref). In our setting, group memberships may not be consistently estimated across all units. We extend the existing analyses to accommodate possible inconsistencies in group membership estimation and show that the usual group-specific asymptotic variance estimator is consistent. We also develop an asymptotic variance estimator under fixed-$T$ asymptotics.

Lastly, we assume that the confidence sets for the group memberships have appropriate coverage.

asm\begin{align} \liminf_{N,T\to\infty}\min_{i=1,\ldots,N}P\left(g_i^0\in \mathrm{CS}_{i, \alpha_1}\right)\geq 1-\alpha_1. \end{align}

dzemski2024confidence extensively discuss sufficient conditions that ensure Assumption (ref); see Sections 3 and 4 of dzemski2024confidence for details.

Asymptotics for the minimum-type confidence set

We now establish the coverage properties of the confidence sets based on the minimum statistics.

thmSuppose Assumptions (ref)-(ref) hold. (i) When $r=1$, we have \begin{align} \liminf_{N,T\to\infty}\min_{i=1,\ldots,N}P\left(R\theta_{g_i^0}^{0} \in \mathrm{CI}^{\mathrm{MT}}(i, \alpha)\right) \geq 1 - \alpha. \end{align} (ii) When $r\geq2$, we have \begin{align} \liminf_{N,T\to\infty}\min_{i=1,\ldots,N}P\left(R\theta_{g_{i}^{0}}^{0} \in\mathrm{CS}^{\mathrm{MW}}(i, \alpha)\right) \geq 1 - \alpha. \end{align}
proof[Proof of Theorem (ref)] See Appendix (ref).

Asymptotics for the bias-corrected confidence set

The following theorem shows that the bias-corrected confidence sets attain nominal coverage.

thmSuppose Assumptions (ref)-(ref) hold. (i) When $r=1$, we have \begin{align} \liminf_{N,T\to\infty}\min_{i=1,\ldots,N}P\left(R\theta_{g_i^0}^{0} \in \mathrm{CI}^{\mathrm{BC}}(i, \alpha)\right) \geq 1 - \alpha. \end{align} (ii) When $r\geq2$, we have \begin{align} \liminf_{N,T\to\infty}\min_{i=1,\ldots,N}P\left(R\theta_{g_{i}^{0}}^{0} \in \mathrm{CS}^{\mathrm{BC}}(i, \alpha)\right) \geq 1 - \alpha. \end{align}
proof[Proof of Theorem (ref)] See Appendix (ref).
remTheorems (ref) and (ref) guarantee unit-wise (marginal) correct coverage by the bias-corrected and minimum-type confidence sets. The joint coverage across $N$ units can also be established under the assumption of joint coverage by the first step confidence set, $\mathrm{CS}_{i,\alpha}$, for group membership and a suitably adjusted nominal level for the second step confidence set construction; see Appendix (ref) for details.

Length analysis

In this subsection, we compare the length properties of our proposed method’s confidence interval with those of two existing alternatives: the unit-by-unit and naive approaches, defined below. Our minimum-type confidence interval is asymptotically shorter than the unit-by-unit approach. Although the naive procedure that ignores uncertainty in group membership estimation produces an even shorter interval, it is not guaranteed to be valid.

The first alternative, the unit-by-unit confidence interval, is constructed from the unit-wise time-series regression. Letting $\tilde{\beta}_i\coloneqq (\sum_{t=1}^Tx_{it}x_{it}')^{-1}\sum_{t=1}^Tx_{it}y_{it}$, the unit-by-unit confidence interval is defined as

align[align omitted — 202 chars of source]

where $\mathrm{se}_{i,\beta}$ is the standard error of $R\tilde{\beta}_i$. The second alternative is the clustering-based naive confidence interval, ignoring uncertainty in $\hat{g}_i$. This confidence interval, denoted by $\mathrm{CI}^{\mathrm{naive}}(i,\alpha)$, is constructed using $\hat{\theta}_{\hat{g}_i}$ and $\mathrm{se}(\hat{g}_i)$:

align[align omitted — 228 chars of source]

We study the length properties of four confidence intervals in total: the unit-by-unit, naive, bias-corrected, and minimum-$t$-based confidence intervals. Under the standard assumptions, $\tilde{\beta}_i$ is $\sqrt{T}$-consistent, which implies that $|\mathrm{CI}^{\mathrm{ubu}}(i,\alpha)|^{-1} = (2z_{1-\alpha/2}\mathrm{se}_{i,\beta})^{-1} = O_p(\sqrt{T})$, where $|\mathrm{CI}(i,\alpha)|$ denotes the length of $\mathrm{CI}(i,\alpha)$ excluding the gaps between disjoint intervals (namely, it is the Lebesgue measure of $\mathrm{CI}(i,\alpha)$). In contrast, the length of the naive confidence interval is $|\mathrm{CI}^{\mathrm{naive}}(i,\alpha)|=2z_{1-\alpha/2}\mathrm{se}(\hat{g}_i)=O_p(1/\sqrt{NT})$ under Assumption (ref). The length of the minimum-$t$-based confidence interval, $\mathrm{CI}^{\mathrm{MT}}(i,\alpha)$, is also of this order, since it is a finite union of the group-specific confidence intervals. Hence, we have

align[align omitted — 181 chars of source]

Note that $|\mathrm{CI}^{\mathrm{naive}}(i,\alpha)| \leq |\mathrm{CI}^{\mathrm{MT}}(i,\alpha)|$, because the naive procedure uses $\alpha$ for constructing the interval, while the minimum-type approach uses $\alpha_2<\alpha$ for each interval. We thus obtain $|\mathrm{CI}^{\mathrm{naive}}(i,\alpha)| \leq |\mathrm{CI}^{\mathrm{MT}}(i,\alpha)| \ll |\mathrm{CI}^{\mathrm{ubu}}(i,\alpha)|$.

The length of the bias-corrected confidence interval, $\mathrm{CI}^{\mathrm{BC}}(i,\alpha)$, is harder to study. It should be noticed that its length is $O_p(1)$ without additional assumptions. To see this, we write $|\mathrm{CI}^{\mathrm{BC}}(i,\alpha)| = 2z_{1-\alpha_2/2}\mathrm{se}_i + \max_{g,f\in\mathrm{CS}_{i,\alpha_1}}(\hat{\theta}_g - \hat{\theta}_f)$. Whereas the first term is $O_p(1/\sqrt{NT})$, the second term is $O_p(1)$ in general, as $\mathrm{CS}_{i,\alpha_1}$ may not be a singleton. That said, we recall that the advantages of the bias-corrected confidence interval are its connectedness and interpretability. We may establish conditions under which $\max_{g,f\in\mathrm{CS}_{i,\alpha_1}}(\hat{\theta}_g - \hat{\theta}_f)=O_p(1/g(N,T))$ for some $g(N,T)\to\infty$, although doing so requires analyzing the power of the confidence set $\mathrm{CS}_{i,\alpha_1}$ for the group membership. We leave this analysis as a topic for future research.

Asymptotic variance estimation

In this section, we consider the estimation of the asymptotic variance of the group-specific coefficients. Namely, we construct asymptotic variance estimators, $\hat{\Sigma}_g$, that satisfy Assumption (ref). Since we consider cases where the group memberships of some units are not consistently estimated, we need to extend the existing asymptotic analyses to accommodate such situations. We first consider the usual asymptotic variance estimator and show its consistency even when some units are misclassified. We also develop an asymptotic variance estimator that is valid when $T$ is fixed and only $N$ tends to infinity. This development may be of independent interest.

The cluster robust asymptotic variance estimator

We first consider the usual cluster robust asymptotic variance estimator and establish its consistency. The estimand is $\Sigma_g$, which in our setting is usually the asymptotic variance-covariance matrix of the within-group OLS estimator, i.e.,

align*[align* omitted — 51 chars of source]

where

align*[align* omitted — 281 chars of source]

We note that $M_g$ can be consistently estimated by $\widehat{M}_g \coloneqq (\widehat{N}_gT)^{-1}\sum_{i=1}^N\sum_{t=1}^T1\{\hat{g}_i=g\}x_{it}x_{it}'$. For $\Omega_g$, we often assume the independence across $i$, so that $\Omega_g$ is simplified and does not include the terms with $i\neq j$. Under such an assumption, a typical choice is arellano1987computing's (arellano1987computing) cluster robust variance estimator:

align[align omitted — 182 chars of source]

where $\hat{\varepsilon}_{it} \coloneqq y_{it}-x_{it}'\hat{\theta}_{\hat{g}_i}$.

The next lemma shows that, even if some units are misclassified, this estimator is still consistent under mild conditions.

lemSuppose that Assumptions (ref) and (ref) hold, and that \begin{itemize} • $\operatorname{liminf}_{N \rightarrow \infty}\left(N_{g} / N\right)>0$ for all $g \in \mathbb{G}$, • $\sum_{i=1}^{N} 1\left\{\hat{g}_{i}=g, g_{i}^{0} \neq g\right\}=o_p\left(N / T^{1+q}\right)$ for all $g \in \mathbb{G}$ for some $q>0$, • $\sup_{i,t} E[\|x_{it} \varepsilon_{it}\|^{2(1+q) / q+\varepsilon}] + \sup_{i,t}E[\|x_{it}\|^{2(1+q) / q}]<\infty$ for some $\varepsilon>0$, • $\{ (x_{it}, \varepsilon_{it}) \}_{t=1}^T$ is independent across $i$, • $(x_{it}, \varepsilon_{it})$ for each $i$ is a strong mixing sequence with mixing coefficients bounded uniformly across $i$ by size $(1-c) r /(r-c)$ for some $c \in 2 \mathbb{N}$, $c \geq 2(1+q) / q$, and $r>c$, • $\widetilde{\Omega}_g \xrightarrow{p} \Omega_g$ for all $g \in \mathbb{G}$, where \begin{align} \widetilde{\Omega}_{g}\coloneqq\frac{1}{N_{g} T} \sum_{i=1}^{N} \sum_{t=1}^{T} \sum_{s=1}^{T} 1\left\{g_{i}^{0}=g\right\} x_{it} x_{is}^{\prime} \tilde{\varepsilon}_{it} \tilde{\varepsilon}_{is}, \end{align} and $\tilde{\varepsilon}_{it}\coloneqq y_{it}-x_{it}^{\prime} \hat{\theta}_{g_i^0}$. \end{itemize} Then we have $\widehat{M}_g\xrightarrow{p} M_g$, and $\widehat{\Omega}_{g} \xrightarrow{p} \Omega_{g}$ for all $g \in \mathbb{G}$.
proof[Proof of Lemma (ref)] See Appendix (ref).

Lemma (ref) shows that a consistent estimation of the variance-covariance matrices is possible even if some units are misclassified.

remdzemskiokui2021convergence establish the asymptotic normality of the KMeans estimator in the presence of potential misclassification, under a condition that implies $\sum_{i=1}^{N} 1\left\{\hat{g}_{i}=g, g_{i}^{0} \neq g\right\}=o_p\left(N / T\right)$. Proving the consistency of the variance-covariance matrix estimators requires a bit stronger condition stated in (b). Assuming condition (d), hansen2007asymptotic shows that condition (f) holds if $(x_{it},\varepsilon_{it})$ is weakly serially dependent; see Theorem 3 of hansen2007asymptotic.

We may use other estimators for $\Omega_g$ proposed in the literature. An alternative approach is to use driscoll1998consistent's (driscoll1998consistent) kernel-based estimator, which is robust to both cross-sectional and temporal dependence. We note that, if classification based on $\{\hat{g}_i\}_{i=1}^N$ is consistent, the consistency of these estimators can be established following the same lines as hansen2007asymptotic and driscoll1998consistent. Formally establishing their theoretical properties under possible misclassification is beyond the scope of the current paper.

Fixed-$T$ adjustment for variance-covariance matrices

We next consider the asymptotic variance estimator under fixed-$T$ framework. The previous double asymptotic framework used there might be restrictive in that $T$ is assumed to diverge, so that misclassification is nullified in variance-covariance estimation. In contrast, if $T$ is short, then the variance-covariance estimation needs to incorporate possible misclassification, as pointed out by bonhomme2015grouped. In this section, we follow pollard1981strong, pollard1982central and bonhomme2015grouped to construct short-$T$ variance-covariance matrix estimators, assuming $T$ is fixed as $N\to\infty$.

For the purpose of exposition, define $\boldsymbol{\theta}\coloneqq\left(\theta_{1}^{\prime}, \ldots, \theta_{G}^{\prime}\right)^{\prime}$, $\hat{\boldsymbol{\theta}}\coloneqq\left(\hat{\theta}_{1}^{\prime}, \ldots, \hat{\theta}_{G}^{\prime}\right)^{\prime}, y_{i}\coloneqq\left(y_{i 1}, \ldots, y_{i T}\right)^{\prime}$, and $x_{i}\coloneqq\left(x_{i 1}, \ldots, x_{i T}\right)^{\prime}$. Let $\hat{g}_{i}(\boldsymbol{\theta})=\operatorname*{\arg\!\min}_{g \in \mathbb{G}}\left\|y_{i}-x_{i} \theta_{g}\right\|^{2}$ denote the optimal group assignment for unit $i$ conditional on $\boldsymbol{\theta}$. We allow the specified number of groups, $G$, to not coincide with the true number of groups, $G^0$, as long as $G$ is no larger than $G^0$. Following pollard1981strong and bonhomme2015grouped, we assume the following condition.

asm\begin{itemize} • $\theta_{g} \in \Theta$ for all $g \in \mathbb{G}$, where $\Theta$ is compact. • $\left(y_{i}, x_{i}\right)$ are i.i.d. across $i$. • $E\left[\left\|y_{i}\right\|^{2}\right]+E\left[\left\|x_{i}\right\|^{2}\right]<\infty$. • $\max _{i} \lambda_{\max }\left(x_{i}^{\prime} x_{i}\right) \leq M<\infty$ for some $M>0$, where $\lambda_{\max}(\cdot)$ is the maximum eigenvalue. • The solution \begin{align} \bar{\boldsymbol{\theta}}=\underset{\boldsymbol{\theta}}{\operatorname*{\arg\!\min}} E\left[\left\|y_{i}-x_{i} \theta_{\hat{g}_{i}(\boldsymbol{\theta})}\right\|^{2}\right] \end{align} is unique up to relabeling of groups for $G \leq G^{0}$. • $\left(y_{i}, x_{i}\right)$ has a continuous density, and $y_{i}$ has a continuous density given $x_{i}$. \end{itemize}

Adapting the argument by pollard1981strong, it can be shown that under Assumptions (ref) (a)--(e), $d_{H}(\hat{\boldsymbol{\theta}}, \bar{\boldsymbol{\theta}}) \xrightarrow{p} 0$, where $d_{H}$ is the Hausdorff distance. Furthermore, $\bar{\boldsymbol{\theta}}$ solves the following moment condition:

align[align omitted — 212 chars of source]

In this sense, $\bar{\boldsymbol{\theta}}$ may be viewed as the pseudo-true slope parameter that is identified based on the fixed-$T$ moment restrictions, (ref).

Following the argument by bonhomme2015grouped, we then have, under Assumption (ref), as $N\to \infty$ while $T$ is fixed,

align[align omitted — 138 chars of source]

where

align[align omitted — 253 chars of source]

$W_i(\boldsymbol{\theta})=e_{\hat{g}_{i}(\boldsymbol{\theta})} \otimes x_{i}^{\prime}$, $e_g$ is the $G\times 1$ vector whose $g$-th element is 1 and 0's elsewhere, and $\Gamma=\left(\Gamma_{g\widetilde{g}} \right) \in \mathbb{R}^{p G \times p G}$ with

align[align omitted — 312 chars of source]

The following representation of $\Gamma$ is useful in the estimation.

propSuppose Assumption (ref) holds. Then we have \begin{align} &\Gamma_{gg} = E\left[x_{i}^{\prime} x_{i} 1\left\{\hat{g}_{i}(\bar{\boldsymbol{\theta}})=g\right\}\right] -E\left[\sum_{h \neq g} \int_{\overline{S}_{g h}} \frac{x_{i}^{\prime}\left(y-x_{i} \bar{\theta}_{g}\right)\left(y-x_{i} \bar{\theta}_{g}\right)^{\prime} x_{i}}{\left\|x_{i}\left(\bar{\theta}_{h}-\bar{\theta}_{g}\right)\right\|} f\left(y \mid x_{i}\right) d y\right] \intertext{and} &\Gamma_{g\widetilde{g}} = E\left[\int_{\overline{S}_{g \widetilde{g}}} \frac{x_{i}^{\prime}\left(y-x_{i} \bar{\theta}_{g}\right)\left(y-x_{i} \bar{\theta}_{\widetilde{g}}\right)^{\prime} x_{i}}{\left\|x_{i}\left(\bar{\theta}_{\widetilde{g}}-\bar{\theta}_{g}\right) \right\|} f\left(y\mid x_{i}\right) d y\right] for g \neq \tilde{g}, \end{align} where \begin{align} S_{g h}=\left\{y \in \mathbb{R}^{T} ;\left\|y-x \theta_{g}\right\|^{2}=\left\|y-x \theta_{h}\right\|^{2}, and \left\|y-x \theta_{g}\right\|^{2} \leq\left\|y-x \theta_{\widetilde{g}}\right\|^{2} for all \tilde{g} \neq g, h\right\}, \end{align} and $\overline{S}_{gh}$ is $S_{gh}$ evaluated at $\boldsymbol{\theta}=\overline{\boldsymbol{\theta}}$.
proof[Proof of Proposition (ref)] See Appendix (ref).

To estimate $V$ and $\Gamma$, we follow bonhomme2015grouped. Specifically, we use the following estimators:

align[align omitted — 820 chars of source]

where $\hat{\varepsilon}_{i, g} \coloneqq y_i - x_i\hat{\theta}_g$, and

align[align omitted — 473 chars of source]

$\lambda(\cdot)$ is a kernel function, and $\eta_{N}$ is the bandwidth parameter such that $\eta_{N} \rightarrow 0$ and $\sqrt{N}{\eta_{N}} \rightarrow \infty$. We use the Gaussian kernel, $\lambda=\phi$, and

align[align omitted — 255 chars of source]

We then obtain $\hat \Sigma_g$ by taking the $g$-th block of $(\hat N_g / N) T \hat{\Gamma}^{-1} \hat{V}\hat{\Gamma}^{-1} $. This asymptotic variance estimator satisfies Assumption (ref) when we take both $N, T\to \infty$. Note that the fixed-$T$ asymptotic distribution is centered around $\overline{\boldsymbol{\theta}}$, which is not the same as the true value of $\boldsymbol{\theta}$. As $T\to \infty$, they tend to be identical. Because of this property, this estimator satisfies Assumption (ref) only under $T \to \infty$. Nonetheless, this fixed-$T$ version is expected to perform better than the usual cluster-robust variance estimator because it can accommodate misclassification of group memberships.

Empirical applications

As an empirical illustration, we investigate the impact of minimum wage increases across U.S. states. The impact of the minimum wage has been documented as heterogeneous, so it is interesting to estimate the effect for each state, given that each state can, to some extent, set its own minimum wage policy. Our analysis builds on the foundational work of dube2010minimum and uses similar data sources. We adopt the group structure framework explored by wang2019heterogeneous, and compute confidence sets for group memberships following dzemski2024confidence. Importantly, our approach extends dzemski2024confidence by enabling inference on the effect of minimum wage changes at the state level.

Our empirical setting closely follows dzemski2024confidence, using the same approaches for coefficient estimation and confidence set construction for group memberships. However, our interest is in the unit-specific coefficients. We also use fixed-$T$ standard errors. The dataset, described in dube2010minimum, comprises quarterly observations for 1,380 U.S. counties from 1990Q1 to 2006Q2. Consistent with wang2019heterogeneous, we set the number of groups to four, as determined by the information criterion of su_identifying_2016. Coefficient estimates are obtained using the CLasso method su_identifying_2016. The estimated model is as follows:

align*[align* omitted — 232 chars of source]

where $i = 1, \dotsc, 51$ indexes states, $c = 1, \dotsc, n_i$ indexes counties within state $i$, and $t = 1, \dotsc, T = 66$ indexes time periods, with $n_i$ denoting the number of counties in state $i$. Here, $\texttt{emp}_{ict}$ denotes employment in the restaurant sector, $\texttt{mw}_{ict}$ is the minimum wage, and $\texttt{emp}_{ict}^{\text{TOT}}$ represents total employment across all sectors. $\phi_c$ and $\tau_t$ capture county and time fixed effects, respectively, and $\sigma_i v_{ict}$ is an idiosyncratic error term. Table (ref) presents the estimated coefficients and associated fixed-$T$ standard errors. Our primary interest lies in $\theta_{g,1}$, which measures the effect of minimum wage changes for each group $g$. We subsequently compute the confidence sets for group memberships. Details of the computational procedure are provided in Appendix (ref). Table (ref) summarizes the point estimate and confidence set for group membership for each state. Notably, the confidence sets and point estimates for group memberships allow states to be classified into 11 categories as in Table (ref), with states within a given category sharing the same confidence interval.

table[table omitted — 730 chars of source]
table[table omitted — 1,381 chars of source]

The estimated effect of the minimum wage increase exhibits heterogeneity; see $\theta_{g,1} $ in Table (ref) and their 90% naive confidence intervals given in Figure (ref). There are two groups with estimated positive effects of the minimum wage on employment: one with a large coefficient and the other with a small effect. Similarly, two groups have negative effects, with one having a large effect and the other having a small effect. However, the interpretation of this result is more nuanced once we take into account the statistical uncertainty of the group membership estimation, as we discuss in detail below.

figure[figure omitted — 166 chars of source]
figure[figure omitted — 172 chars of source]
figure[figure omitted — 172 chars of source]

We calculate the 90% confidence intervals for each state with $\alpha_1=\alpha_2=0.05$. Figures (ref) and (ref) display the minimum-type and bias-corrected confidence intervals, respectively. The results of statistical inference, accounting for statistical uncertainty, reveal interesting patterns. Seven states in Category 1-\{1\} have a statistically significant positive effect of a minimum wage increase. Seven states in Category 4-\{4\} have a statistically significant negative effect. All other states have statistically insignificant effects. Note that all states with statistically significant effects have group memberships estimated precisely. While four states in Categories 4-\{3,4\} and 4-\{2,3,4\} (namely, Colorado, Nebraska, South Dakota, and Utah) are estimated to experience statistically significant negative effects from a minimum wage increase if possible misassignment is ignored, these effects are not statistically significant once statistical uncertainty is taken into account.

We also observe that the minimum-type procedure yields disconnected confidence intervals in certain cases. In particular, Categories 3-\{3,4\} and 4-\{3,4\} have wide gaps between two sub-intervals that constitute the confidence intervals. In contrast, the bias-corrected confidence intervals are always connected by construction. However, they can be much longer, particularly when the minimum-type confidence intervals are disconnected and have wide gaps between sub-intervals.

remWe note that state-by-state regression cannot be applied in our dataset. This is because, within each state, the cross-county variation of the minimum wage is almost absent in each quarter and thus absorbed in the time fixed effects $\tau_t$. In contrast, our proposed methods can be applied because they utilize cross-state variation of the minimum wage in each quarter.

Monte Carlo simulations

This section provides simulation results. Monte Carlo simulations are conducted to examine the finite-sample performance of the proposed methods in comparison with existing alternatives. In particular, we aim at quantifying the sensitivity of these confidence sets to unit-wise error variances. On the one hand, units having a large error variance are liable to misclassification, as pointed out by dzemskiokui2021convergence,dzemski2024confidence. This can lead to incorrect centering for the clustering-based naive confidence set. On the other hand, a large error variance implies that the parameter estimates have large standard errors, resulting in longer confidence intervals. In this section, we evaluate the coverage and length properties of several confidence sets.

Our simulation design is based on our empirical application and given by

align[align omitted — 168 chars of source]

where $g_i^0 \in\{1,2,3,4\}$, $x_{it,j} \sim N(0,1)$, $u_{it}\sim N(0,1)$, and $\sigma_{u,i} = 0.5 \sigma_{i}$ with $\sigma_i \sim \chi^2(4)/4$ generated from the chi-square distribution with 4 degrees of freedom.\footnote{We use one realized vector $(\sigma_1,\ldots,\sigma_N)$ for all simulations. This design allows us to easily analyze how the performance of the procedures is related to the magnitudes of the error variances. Although we unavoidably choose one particular set of error variances, the realized values are sufficiently diverse and informative.} $\theta_{g,1}, \theta_{g,2}, \theta_{g,3}$ are set equal to the estimated coefficients in Table (ref), and we set $g_i^0 = g$ for $i\in\{(g-1)\times N/4+1,\ldots,g\times N/4\}$.

We generate panels of size $N \in \{40,80,160\}$ and $T\in\{40,80\}$. In each simulation, we calculate 90% confidence intervals for $\theta_{g_i^0,1}$ and confidence sets for $(\theta_{g_i^0,1}, \theta_{g_i^0,2})$ for each $i$. For unit-wise confidence intervals, we calculate the unit-wise mean lengths and coverage ratios, averaging over 1000 simulations. For unit-wise confidence sets, we report the unit-wise coverage ratios. To save space, we only report the cases with $T=80$. The results for $T=40$ are presented in Appendix (ref). Thoughout the simulations, standard errors (or asymptotic variance estimators) are fixed-$T$ versions.\footnote{We also tried the usual cluster-robust standard errors or asymptotic variance estimators. We found that the fixed-$T$ versions achieve better coverage and do not necessarily make inference more conservative. This improvement is more visible when $T=40$.}

Performance of confidence intervals

We calculate four 90% confidence intervals for $\theta_{g_i^0,1}$: the unit-by-unit confidence interval, $\mathrm{CI}^{\mathrm{ubu}}(i,0.1)$, the naive confidence interval (ignoring uncertainty in $\hat{g}_i$), $\mathrm{CI}^{\mathrm{naive}}(i,0.1)$, the bias-corrected confidence interval, $\mathrm{CI}^{\mathrm{BC}}(i,0.1)$, and $\mathrm{CI}^{\mathrm{MT}}(i,0.1)$. We set $\alpha_1=\alpha_2=0.05$.

We first investigate coverage properties. Figure (ref) plots the empirical coverage fractions of the confidence intervals across different magnitudes of $\sigma_i$ (the double of the error standard deviation $\sigma_{u,i}$). When $N=40$, the unit-by-unit confidence interval generally attains the nominal 90% coverage rate for all $\sigma_i$, while the naive confidence interval suffers from severe undercoverage for moderate to large $\sigma_i$, because they do not account for potential misclassification, or uncertainty in $\hat{g}_i$. The confidence interval based on the minimum $t$-statistic can undercover for small $\sigma_i$ when $N=40$, yet as $N$ increases, it attains nominal coverage rate or becomes conservative. The bias-corrected confidence interval is conservative for moderate to large $\sigma_i$. These patterns are also observed for $N=80$ and $N=160$, with an exception that the minimum-statistic-based confidence interval is slightly conservative for all $\sigma_i$. Generally speaking, the proposed methods successfully overcome the severe undercoverage problem that the naive method suffers from by taking uncertainty in $\hat{g}_i$ into account.

figure[figure omitted — 928 chars of source]

Next, let us evaluate the length properties. We plot the unit-wise mean lengths against the corresponding values of $\sigma_i$. See Figure (ref). First, we look at the case with $N=40$. The naive confidence interval performs best in terms of mean length for all $\sigma_i$. This is because the naive confidence interval is based on a single, small group-specific standard error. The unit-by-unit confidence interval is short only for small $\sigma_i$, and its length linearly increases as $\sigma_i$ increases. The bias-corrected confidence interval is as long as the unit-by-unit confidence interval for almost all $\sigma_i$. This result is likely from the fact that the bias-correction method is connected and need to cover the range between multiple parameter values when the confidence set for group membership is not singleton. The minimum-type confidence interval is the second shortest on average. As $N$ increases, the naive confidence interval remains the shortest, the length of the bias-corrected confidence interval does not change much, and the minimum-type confidence interval becomes shorter than in the $N=40$ case. This is an advantage of the minimum-type confidence interval over the unit-by-unit confidence interval, whose power does not improve as $N$ increases.

figure[figure omitted — 912 chars of source]

Performance of confidence sets

In this subsection, we evaluate the coverage properties of four 90% confidence sets for bivariate parameters $(\theta_{g_i^0,1}, \theta_{g_i^0,2})$, namely, the unit-by-unit confidence set, the naive confidence set, $\mathrm{CS}^{\mathrm{BC}}(i,0.1)$, and $\mathrm{CS}^{\mathrm{MW}}(i,0.1)$ with $\alpha_1=\alpha_2=0.05$. Figure (ref) shows the result for $T=80$. The behaviors of the four confidence sets are qualitatively the same as those of the confidence intervals considered in Section (ref): the unit-by-unit confidence set has correct coverage, the naive confidence set severely undercovers when $N$ is small or $\sigma_i$ is moderate to large, and the bias-corrected confidence set tends to be conservative. The only exception is that the minimum-type confidence set tends to undercover for small to moderate $\sigma_i$ when $N=40$, but it becomes slightly conservative for larger $N$, as observed for the minimum-type confidence interval.

figure[figure omitted — 954 chars of source]

Conclusion

This paper introduces inference methods for unit-specific coefficients that exhibit a grouped pattern of heterogeneity. Our approach combines the efficiency gains from clustering units with accounting for statistical uncertainty in group membership. We propose two methods: one uses the minimum-type statistic within the confidence set for group membership, and the other corrects for bias due to possible group misassignment. The two methods present a deliberate trade-off. The minimum-type method yields shorter and more powerful confidence sets, but these may be disconnected and harder to communicate. The bias-correction method always returns a connected, readily interpretable interval for a scalar parameter, at the cost of being conservative and, in the worst case, no shorter than the unit-by-unit interval. Because both methods are valid, the choice can be made on the ground of a researcher's preference toward interpretability and power. These methods are particularly useful when group assignment is not perfectly certain. Ignoring statistical uncertainty in group membership estimation can severely distort test size and confidence set coverage. While unit-by-unit estimation is robust, it tends to be inefficient. Our proposed methods achieve adequate size and coverage while harnessing the efficiency of clustering. These tools enable empirical researchers to analyze unit-specific coefficients accurately, without distorting statistical properties.

Our inference procedures offer several opportunities for further development. This paper focuses on linear models, but extending these methods to nonlinear contexts would be beneficial. Empirical research often examines unit-specific coefficients in nonlinear models with latent group structures, a scenario common in experimental behavioral economics bruhin2010risk, conte2011mixture. In these studies, researchers classify participants into behavioral types and examine the magnitude of behavioral biases at the individual level. Most analyses use structural and nonlinear models.

The current paper focuses on situations in which the covariates are exogenous. We chose this focus because, in such situations, both consistent group-specific coefficient estimators and a valid confidence set for group membership have been developed. When the covariates are not exogenous, we may employ instrumental-variable estimators or generalized method-of-moments estimators su_identifying_2016,mehrabani2022estimation,choi2024latentgroupstructurelinear. In those cases, methods to construct confidence sets for group membership have not been developed. In the future, once such methods are developed, our proposed procedure can be readily applied.