Matrix Completion, Counterfactuals, and Factor Analysis of Missing Data
Jushan Bai, Serena Ng
Introduction
Missing observations are prevalent in empirical work, and it is not surprising that solutions have been proposed by researchers in many disciplines. The classic econometric solution is some variant of the EM algorithm. In the case of factor analysis with missing data, the EM approach is to predict the missing values using initial estimates of the factors obtained from a balanced panel and iterate. While convergence of the algorithm can be established, the asymptotic properties of the converged estimates are not well understood.
Progress can be made if the panel of incompletely observed data is large in both dimensions and have a strong factor structure. This means in particular that has a common component of reduced rank , and whose population covariance matrix has eigenvalues that increase with the size of the panel. We show in this paper that in spite of missing values in , every entry of can be consistently estimated using a tall-wide (tw) algorithm that involves two applications of principal components. We provide an asymptotic characterization of the estimation error for each and show that there will be four convergence rates depending on observability of . The approach can be used to construct missing values of potential outcomes satisfying a factor structure.
Our factor imputation approach contrasts with those used in matrix completions. In that literature, the probabilistic structure of the data is not specified; nuclear norm regularization via singular-value thresholding is crucial, and successful matrix recovery typically requires that the low rank component is incoherent, and that the data are missing uniformly at random. See, for example, Cai et al. (2008). Instead, we impose moment conditions to ensure that the assumed factor structure is strong and identifiable. We also require that and , where is the number of units with data observed over the entire span, and is the length of the sample that data are available for all units. Exploiting the commonality between the observed and missing data allows recovery of the entire matrix using principal components with as few as observations. Provided that there are enough observed data to estimate the factors and the loadings consistently, a large fraction of the data can potentially be missing. And while regularization can yield fewer factors, it is not needed for consistent estimation of the missing values. In independent work, Jin et al. (2021) and Xiong and Pelger (2019) suggest alternative factor-based approaches. What makes our theory distinct is that we analyze factors estimated directly from the data without preliminary adjustment or iteration. The estimates that emerge from the tw algorithm are already consistent and asymptotically normal, though one re-estimation using imputed data will provide a faster convergence rate for one sub-block.
An immediate application of the tw approach is program evaluation in which the object of interest is the effect of treatment in the potential outcome model. The method of synthetic control pioneered in Abadie and Gardeazabal (2003) estimates potential outcomes from a weighted average of the control units, assuming that the difference between the treated and the control group is constant in the absence of treatment. Common variations between the treated and the control groups are crucial in estimation of counterfactual outcomes, and the factor model is a natural framework for modeling them. Treating potential outcomes as missing values, we develop a factor-based estimator of the individual as well as the average treatment effect. A distribution theory that permits tests of hypothesis is developed assuming that the size of the control group is large.
The rest of the paper is structured as follows. After presentation of the preliminaries in Section 2, Section 3 presents the least squares version of Algorithm tw and studies the asymptotic properties of the factor estimates that the algorithm delivers. Section 4 studies a factor-based estimation of treatment effect. Section 5 concludes.
Preliminaries
We use to index cross-section units and to index time series observations. Let be a vector of random variables and be a matrix. In practice, is transformed to be stationary, demeaned, and often standardized. The normalized data has singular value decomposition (svd)
In the above, is a diagonal matrix of singular values, are the corresponding left and right singular vectors respectively. Without loss of generality, the singular values in the diagonal entries of are ordered such that . Note that while the largest singular values of diverge and the remaining ones are bounded, the largest singular values of are bounded and the remaining ones tend to zero because the singular values of are those of divided by . The Eckart and Young (1936) theorem posits that the best rank approximation of is . The svd is also the goto algorithm for solving matrix factorization problems that seek to represent a matrix as a product of two low rank matrices. These results can be obtained without an assumed data generating process for .
We are interested in the principal components of viewed from the perspective of a factor model. Let be a matrix of common factors, be a matrix of factor loadings, and be a matrix of idiosyncratic errors . The data are assumed to have a factor structure
where are the true values of . The common component has reduced rank because and both have rank . The econometrics literature on large dimensional factor analysis has largely adopted as framework the approximate factor model due to Chamberlain and Rothschild (1983) according to which the largest population eigenvalues of will increase with and while the remaining ones are bounded. The following assumptions formalize these ideas in a statistical setting:
There exists a constant not depending on such that
(Factors and Loadings): (i) , , (ii) , and , and (iii) the eigenvalues of are distinct.
, , ;
, for some , and , ;
and ;
for every ;
.
(Central Limit Theorems): for each and , as , and as .
Assumption A ensures that the factor structure is strong and can be separated from the idiosyncratic errors. The requirements that and ensure that the eigenvectors of the low rank (common) component are sufficiently spread out and play the role analogous to incoherence conditions. Positive definiteness of and ensure that each factor has a non-trivial contribution to the low rank component. It is possible for A(a.i) to hold but not A(a.ii) and vice versa, and identification will fail. The assumption of distinct eigenvalues is used to separately identify the factors and factor loadings but is not needed to identify the common components. Part (b) of Assumption A allows the errors to be weakly correlated both in the time and cross-section dimensions. For example, b(ii) requires the sum of autocovariances be bounded, and b(iii) is an analogous assumption for cross-sectional weak dependence. Assumption A(vi) assumes weak dependence between the factors and the errors. Part (c) is needed for asymptotic distribution of the factor estimates.
We observe but not or , and and are not separately identifiable. We use as in Stock and Watson (2002); Bai and Ng (2002); Bai (2003) the normalizations and diagonal. The method of asymptotic principal components (APC) then constructs the factor estimates as
For each and for each , , consistently estimate , ) up to rotation matrices and where
When Assumption A is satisfied, the factors and loadings are consistently estimable up to a rotation, and the low rank component is recoverable. Furthermore, the number of factors can be consistently estimated using, for example, the criteria developed in Bai and Ng (2002, 2019). Hence the number of factors can be treated as known.
Missing Data
Missing data is a problem that researchers frequently encounter. As Zhu et al. (2019) points out, we can expect more occurrence of incomplete observations in the era of big data. Data can be missing for a variety of reasons: non-response in surveys, lack of economic activity, and staggered releases by statistical agencies to name a few. One can always work with a balanced panel but this effectively throws away information in many series and cannot be efficient. This has led to development of simple methods that replace the missing values with zero or the mean as well as sophisticated methods that fully specify the data generating process and the missing data mechanism. For example, Rubin (1987) suggests a Bayesian approach that fills in missing values by repeatedly sampling from the predictive distribution of the missing values. Kamakura and Wedel (2000) suggests a simulation based approach that is aimed at handling different types of missing data in survey responses within a likelihood setup. See Horton and Kieinman (2007) for a survey of the literature. Rubin (1976) obtains two sufficient conditions for unbiased estimation. First, missingness cannot depend on the missing values after conditioning on the observed data (a condition known as missing at random), and second, the parameters of the model must not depend on the missingness mechanism. While missing at random can be a reasonable characterization in observational studies and surveys, there are situations when the assumption is not appropriate. For example, high income survey respondents may be more likely to ignore questions with tax consequences.
The EM algorithm of Dempster et al. (1977) imputes missing values by alternating between an E-step that computes the expected log-likelihood using the most recent parameter estimates, and an M-step that maximizes the expected log-likelihood. In cases when the expected log-likelihood is difficult to compute, the Expectation-Conditional Maximization algorithm of Meng and Rubin (1993) can be considered. Schneider (2001) considers a ridge-regression based regularized EM algorithm for imputing missing values in climate data. But as Honaker and King (2010) noted, methods that work well in a cross-section setting tend not to work well in panel data that exhibit dependence across units and over time. Imputing the missing values using the Kalman filter such as discussed in Shumway and Stoffer (1982) remains a popular time series approach, but it is fully parametric. Dropping a series altogether because of partially missing data could lead to a huge loss of information. In the FRED-MD database of over 130 series for example, as many as 30 series can be discarded even though some are missing for only a handful of months (about 2% of the observations in the panel) over the sample 1960:01 to 2018:12.
For estimation of strict factor models with missing data, more options are available. Banbura and Modugno (2014), Jungbacker and Koopman (2011), Jungbacker et al. (2011) consider likelihood estimation which is conceptually appealing but non-linear filters are needed to compute the likelihood as and are both random. For approximate factor models, Stock and Watson (1998) suggests to fill missing values in with the most recent estimate of the common component. Though the precise implementation may differ, using estimates from the balanced panel as initial values is the most common, see Stock and Watson (2016) and Giannone et al. (2008).
While a variety of methods have been proposed for factor analysis with missing data, there are surprisingly few theoretical results until recently. Jin et al. (2021) puts zeros to observations assumed to be missing at random and rescales the asymptotic principal components by the probability of missing data. It is shown that these estimates are consistent but not asymptotically normal in general, though iteration can restore normality. Xiong and Pelger (2019) estimates the factors from a weighted covariance matrix. The estimates are robust to the unknown missing pattern at the expense of larger variances. Our objective is similar to that of Jin et al. (2021) and Xiong and Pelger (2019) and also use a factor model for imputation, but we re-organize the data instead of re-weigh them, and we do not make assumptions about the missing data mechanism.
Suppose we rearrange the data such that the observed ones are ordered first. Rearrangement is not necessary in practice but it makes the idea easier to grasp. Consider the following example:
The transformation from to shuffles the columns so that those that are observed at all times are ordered first. The transformation from to shuffles the rows so that time periods with complete data for all units are ordered first.
We will use ‘o’ to denote the size of the observed and ‘m’ for the size of the missing samples, respectively. The northwest block, labeled bal, is a subpanel of complete data of dimension . The wide block extends the bal block in the cross-section dimension to include data of all units with data for periods. The tall block extends the bal block in the time dimension to include all units with complete time series observations. The southeast block collects the missing data into a matrix where and . This block, labeled miss, is the sub-block bordered by the rows and columns . As drawn, miss is a “largest possible” block of missing data since some points in it are actually observed.
To give some economic content, we can think of the tall block as data for developed countries, the wide block for newly developed countries which have complete data over a shorter span, while the miss block consists of data for the less developed countries for which missing data are more prevalent. For financial data, acquisitions and mergers can yield a block structure. In macroeconomic settings, it is not uncommon for statistical agencies to stagger the release of data groups, but bunch the release of series within a group. In other cases, data may not collected in early years and terminated in later years due to attrition. As will be seen below, missing data also play a role in estimation of treatment effects, and as discussed in Athey et al. (2018), treatment may be given at the end of the sample, or it may be staggered or bunched by design of the experiment. In some of these cases, the assumption of missing at random may be inappropriate.
The estimation issue is that principal components cannot be directly applied when there are missing values. If we initialize using estimates from the balanced block, the factor estimate will always be spanned by factors that are originally in the balanced block. If this block is small in size, the information loss can be significant. Reorganizing data into four blocks reveals that we can make better use of the data to estimate the factors.
Our estimator is based on the idea that can be obtained from the tall block, while can be obtained from the wide block by APC, hence the acronym tall-wide, or tw for short. Results from our previous work can be used to show that
where and are unknown rotation matrices. The terms are uniform in and . To make further progress, some assumptions are needed. Our working assumptions are firstly, that there is an identifiable rank component in and in each of its four blocks so that there is commonality between blocks, and secondly that the observed blocks are sufficiently large so that consistent estimation by APC is possible.
(a) The conditions in Assumption A hold for the full matrix if it were observed, as well as the four sub-blocks: bal, tall, wide, and miss. (b) The order conditions and are satisfied for any . Furthermore, and as and .
Assumption C:
(strong sub-block factor and factor loadings)
Assumption D:
(block stationarity) Let , , and be defined as in Assumption A.
, and ,
, and .
Assumption B allows and as but requires an order condition to hold for the tall block, and one for the wide block. Assumption C requires the subsample moment matrices for the factor and factor loadings to be positive definite so that the factors from the subsamples are identifiable. Assumption D restricts these subsample limits to be the same as the limits for the whole sample. Results without Assumption D will also be stated.
Though there is no explicit restriction on the missing pattern, it is implicitly imposed through the factor model. If the miss block has nothing in common with the tall block, the missing values cannot be predicted. For example, if units are driven by a factor and the remaining units are driven by a different factor , our method will not help with the imputation. Our method is also not suited for situations when or is small. We certainly cannot handle the case of or . For example, if each entry is randomly observed with probability so that when , , will tend to zero and tall block will not be available for estimation.
Suppose that there are units with complete data in all rows, and periods when data are available for all units. Let and be obtained by applying the APC estimator to the tall and wide blocks respectively. Let for . Suppose that as and as . Under Assumptions A-D,
Lemma 2 is a direct implication of Lemma 1. The asymptotic variances are the same as defined in Lemma 1; the convergence rates differ because estimation is no longer based on the full sample.
Under our maintained assumptions, it immediately follows from Lemma 2 that
Obtaining an estimate of for in miss requires a bit more work because and are estimated from different blocks of data. However, if Assumptions A-D hold, the data in the block bal will contain information shared by both the tall and wide and can be used to re-rotate the data. Specifically, for any , it holds that , and
Define a new non-singular rotation matrix
This matrix can be estimated by regressing the matrix on the matrix , which is the sub-matrix of associated with the balanced block. As shown in the appendix, . This suggests the following;
Algorithm TW
Let be the matrix that is one in positions when the data are observed, i.e. if is observed and zero otherwise.
From the tall block of , obtain by APC where is .
From the wide block of , obtain by APC where is .
Let where is obtained by regressing on a submatrix of .
Output if and if
Steps (1) and (2) imply that a complete set of estimates of the low rank component can be obtained from data points. When the number of factors in tall and wide do not coincide, we let in Step (3).
Properties of C~~𝐶\widetilde{C} and Re-estimation
This section studies the properties of which will be used to replace the missing . The estimation error can be decomposed into four terms:
Proposition 1 shows that the estimates of the entire matrix are consistent and asymptotically normal without explicit restrictions on the nature of missingness. However, the convergence rate of depends on whether is observed. Note that the bal block can use estimates from tall or from wide block, so convergence rate for this block is the faster of the two rates, ie.
In contrast, the convergence rate of in the miss block is always the slowest possible. These convergence rates and asymptotic distributions are obtained without iteration. It is possible for other estimators to achieve a better convergence rate. But to our knowledge, few (if any) consistent estimator exists that does not require iteration. Algorithm tw is best suited for cases when a large tall blow is availble, and the number of missing values is similar across units. In the event that a few units have many more missing values, will be the smallest sample size possible and could result in information loss. In Cahan et al. (2021), we propose an alternative algorithm that estimates the loadings and the rotation matrix jointly by a series of projections.
While Algorithm tw produces factor estimates that are mutually orthogonal within the four blocks of data, they are not mutually orthogonal over the entire data matrix. Furthermore, the estimates in tall do not use all information available, and similarly for the estimates in wide. Re-estimation using provides an opportunity to use the imputed entries not previously available. However, embedded in are imputation errors which will propagate to other blocks in re-estimation because the APC is a weighted average of . A formal analysis is needed to determine whether re-estimation using can be justified.
Since was constructed using estimates of constructed from the tall block, and of constructed from the wide block, we partition the matrices as follows:
where and . We show in the Appendix that if is missing,
where uniformly in and
The dependence of these elements on is suppressed to simplify notation. Now
Imputation injects three errors into when is not observed:- a quantity that is negligible, an error from estimating , and one from estimating . As a consequence, will differ from the true error .
For , let be the APC estimates based on with the normalization . Let . Then, under Assumptions A and B,
Bai and Ng (2002) shows that in the complete data case, . Lemma 3 says that when the factors are estimated from , the convergence rate depends on the size of the balanced panel and , which is evidently slower than when all data are observed.
To obtain a distribution theory for the factor estimates, we also need the representation for and . We show in the Appendix that
where uniformly in . The first representation is for those estimates of when . Except for the term that is asymptotically negligible, the representation is the same as the case when all data are observed. More interesting is the case when has imputation error. In appendix, we show
where the matrix is defined as
Thus the convergence rate for is . Similar derivations show that
where , uniformly in and
Thus the convergence rate for is for and the rate is for . Note that under Assumption D,
Given the asymptotic representations, it is relatively easy to derive the limiting distributions.
Under Assumptions , the following holds as and , with .
The main implication of the proposition is that if is in the balanced block, then is consistent while is consistent. These are faster rates than those stated in Proposition 1 without re-estimation. Note that there is only a single rotation matrix for the factor estimates (instead of one for tall and one for wide) which are mutually orthogonal. This is a consequence of the fact that the factors are now estimated from the entire matrix instead of the sub-blocks.
The four convergence rates in Proposition 2 are derived under the assumption that no data from the miss block are available. Parts (a) and (c) already have the same rate of convergence as in complete data and are unaffected by this assumption. It can be shown that the convergence rates for parts (b) and (d) will be faster when some observations in the miss block are available. The assumption that no observations in the miss block is consistent with the usual counterfactual matrix in the program evaluation analysis to follow.
Let be the common component estimated from in which missing values of are replaced by the tw estimates of the common component . Under Assumptions , it holds that as and ,
The highlight of Proposition 3 is that three of the convergence rates are the same as in Proposition 1, but if is in bal, the convergence rate is now , the same as when were completely observed. This improvement is due to the simple fact that is based on data in the tall and wide blocks only, while also also exploits information in the miss block. Bai and Ng (2019) considers a robust principal components (rpc) estimator where , is a regualrization parameter. If we replace the apc part of Algorithm tw by rpc, will also have the four convergence rates, unaffected by regularization.
From the proof of Proposition 3 in Appendix, the asymptotic representation for implies the following error average rate in Frobenius norm (denoted ) for the four blocks:
For the block defined by : .
For the block defined by : .
For block defined by : .
For the block defined by : .
This in turn implies an average squared error for the entire common components matrix
where the weights are the proportions of block size:
The sum of the first four terms in the average squared error is where and . We have the following.
The first term in the square bracket is present even in the complete data case. The second term is due entirely to missing data, and the magnitude depends on the fraction of missing data but does not depend on the missing data mechanism.
Remark 1: In the machine learning literature, matrix completion problems are typically solved by nuclear norm regularization, and algorithmic errors bounds are given for fixed and . Athey et al. (2018) finds that for -sub-Gaussian data, the worse case bound for the average error (analogous to Corollary 1) depends on the regularization parameter and the unspecified distribution that generates . Our analysis complements the algorithmic error with a theory for asymptotic inference and shows that regularization is not needed for consistent estimation of . In fact, we are able to characterize the sampling error of each , not just the average over over and , while allowing . This is made possible by more fully exploiting the factor structure.
Compared to Jin et al. (2021) and Xiong and Pelger (2019), the missing pattern plays a less important role in our analysis because we look at the problem from the perspective of re-arranged data. Our results also differ in that our first step estimate is already consistent and asymptotically normal. Re-estimation in our setup accelerates the convergence rate of a sub-block rather than restores asymptotic normality. Furthermore, we do not need to assume that the number of missing data points as a fraction of is bounded away from zero. If the balanced block is of dimension , such an assumption would have implicitly required that and are of the comparable order, and likewise for and .
Lemma 2: the convergence rates are the same, but
Proposition 1: the convergence rates are the same, but for part i, replace by in ; for part ii, replace ) by in ; and for part iv, replace ) by ).
Proposition 2: parts and remain the same, the other two parts will read
where and are given in Assumption C, and
are the limits of and , respectively, is the limit of and is the limit of .
2 Finite Sample Properties
Simulations are used to compare the performance of tw with and without updating. For comparison, we also consider an iterative EM algorithm considered in Stock and Watson (2016), which will be denoted em. Using estimates from the balanced panel as initial values, the algorithm repeatedly regresses on and then on by ols till convergence. Note, however, that the converged factor estimates produced by em may not be mutually orthogonal.
Data are generated from and with , the diagonal entries in are equally spaced between 1 and , and . We report results for only as those for and are similar. For each replication, is computed for the four blocks. Also reported are results (labeled complete) for the infeasible case when all data are observable.
Our theory is silent about how to compute principal components. In the case of complete data, a common practice is to first standardize the data. But with missing data, it is unclear whether this is still desirable. Hence, we consider three versions of the estimator: one applied to the standardized data , one to the demeanend data, one to the raw data. These are labeled TW(0,1,2) in the tables reported. The mean and standard deviation used in the centering and normalization are computed using the observations available for each series.
Table 1 compares the mean error over 5000 replications, normalized by the size of the corresponding block. Not surprisingly, the error in estimating the low rank component is inflated by missing data. As seen from Table 1, re-estimation always reduces the error, and imputation of the raw data (ie. method (2)) always has smaller errors than re-estimation using standardized data (ie. method (0)). One possible explanation is that when there are few observations from which to estimate the sample means and standard deviations, noise could be injected into the factor estimates. The results for em also favor imputation of the raw data.
The Frobenius normed error strongly favors , but this is based on averaging the error over all estimates of . Table 2 reports the root-mean-square-error for four chosen pairs, one in each of the four blocks. Evidently, the estimation error is largest if is in the miss block and smallest when is in the bal block. As in Table 1, the error is smallest when the factors are re-estimated using . Both results are consistent with the theory.
In results not reported, the squared correlation between and averaged over is over 0.93 when all data are observed. For the four data points considered in Table 1, the squared correlations are 0.89, 0.85, 0.85, 0.81 using TW, and 0.92, 0.93, 0.90, and 0.86 upon re-estimation. Regardless of re-estimation, is well approximated by the normal distribution.
Factor Based Estimation of the Treatment Effects
If is a panel of data on GDP growth, a macroeconomist may be interested in a counterfactual prediction for some at some time . The theory in Bai and Ng (2006) can then be used to construct prediction intervals. We now show that Algorithm tw can also be used to estimate microeconomic type counterfactuals.
Program evaluation is widely used in economic analysis. Let denote the treated group and be the control group that is never exposed to treatment. To conform with the notation in this literature, we now index the treatment group by 1 and the control group by 0. The group size are and respectively with . Unit receives treatment in period and thus is number of pretreatment periods for unit . In this paper, we assume that for all and thus is also constant across .
Let be the potential outcome if individual receives the treatment, and be the potential outcome of individual without treatment in period . The individual treatment effect is
The sample average treatment effect on the treated at (often referred to as sample ) is:
A variety of methods have been proposed to estimate these quantities. As discussed in Athey et al. (2018), the unconfoundedness regressions literature tends to use a single-treated period to impute the missing potential outcomes in the last period from the control units with similar lagged outcomes. The synthetic control literature pioneered in Abadie and Gardeazabal (2003) uses a weighted average of the control units ( ) as estimate of the counterfactual outcome of the treated unit in the absence of treatment. The method can be specialized to produce difference-in-difference estimates under some conditions, see also Doudchenko and Imbens (2016). A ‘parallel-trend’ condition is needed to ensure that the sample path of the weighted average is parallel to the path of the treated in the absence of treatment.
Hsiao et al. (2012) assumes that potential outcome has a factor structure and considers estimation when the sample size is too small for estimation of the common factors. Their two step least squares-based estimator can be understood as replacing the latent factors by the outcome of the control units. To motivate the use of synthetic control methods in comparative case study research, Abadie et al. (2010) assumes that the outcome variable is driven by common factors. Increasingly, the synthetic control approach is studied from a factor model perspective. Gobillon and Magnac (2016) shows that if the true model is a linear factor model, synthetic controls are equivalent to factor models if the factor loadings and exogenous covariates for the treated are in the support of these variables for the control group. These conditions are analogous to requiring that the observed and missing blocks are driven by some factors in common.
In terms of factor-based estimation of treatment effects, Xu (2017) directly estimates the factors by principal components when and are large but the theoretical analysis is incomplete. Li (2018) suggests a procedure to determine and provides some asymptotic results for the treatment effect of a single treated unit in the absence of exogenous covariates. Amjad et al. (2018) analyses the mean-squared error of a robust synthetic control procedure. Xiong and Pelger (2019) allows the probability of missing data to depend on observed variables, but requires the fraction of observed data to bounded away from zero.
We will provide a distribution theory for estimates of the individual and average treatment effect without assuming a missing data mechanism. Indeed, when potential outcome is assumed to have a factor structure, estimation of treatment effect is very much related to factor analysis with missing data in which the outcomes for the treated group had they not been treated are the missing values to be recovered. Let be a vector of observed covariates. Let be the treatment indicator for individual is treated in period . Then
where is vector of latent common factors. This is precisely the interactive fixed effect model developed in Bai (2009) where is interactive fixed effect. The covariates also helps control for known sources of missingness and allows for more general missing patterns. We only observe on the treated and thus need to impute the corresponding counterfactual outcome in the absence of treatment. In the terminology of the previous section, we now have:
(IFE): Interactive fixed effect estimation of using observations in the control group. Let be matrix of residuals where
(tw): Estimate from and from the ; compute and
replace the th entry in by , OR
replace the th entry in by
Compute the average treatment effect .
The analysis to follow assumes re-estimation of the factors and so takes from (3b).
1 The Average Treatment Effect
Under the assumed factor structure, we see that for miss
There are three errors in the counterfactual , one from estimation of , one from estimation of interactive fixed effects, and an idiosyncratic noise . Since , it follows that
As , the convergence rate for is at most . Now is homogeneous across and by assumption, and from Bai (2009), . The first term is thus and is dominated. By (A.20) in Appendix (or equation (A.4) if Step (3a) is used)
where is the average of factor loadings in the treatment group. If is large, the first term on the right is which is also dominated. This leads to the asymptotic representation for the case when large:
The following shows that is asymptotically normal.
Suppose Assumptions A-D and those in Bai (2009) hold. Then as ,
The estimation of by tw presented above can be generalized to allow to vary with , as in Xu (2017). This approach was first considered in the unpublished dissertation of Cahan (2013), and which we analyze further in Cahan et al. (2021).
2 Treatment Effect on a Single Unit
Consider now the estimation of treatment effect on a single unit for some . Then
As before, we can ignore the error . Imputation error from factor estimation (that is, the second and third terms on the right) are and , respectively. But unlike , no averaging is taken over and, as a consequence, now dominates the composite estimation error. The distribution of thus depends on the distribution of . If one is willing to assume is identically distributed across and , its distribution can be estimated using the residuals The estimated individual treatment effect has variance
It is also of interest to consider the average treatment effect over the treatment period for a single unit, defined as . Let and . It can be shown that a result similar to Proposition 4 holds:
Table 3 reports the bias, root-mean-squared-error, and the probability that the true treatment effect is within two standard errors of the estimated effect (labeled covr). The first panel evaluates at , the second panel reports at . The estimation error is larger for than since averaging reduces noise. The error decreases with for given , but is quite insensitive to changes in for given because the convergence rate is . Except when and are both small, the coverage is quite close to 95%.
Conclusion
Missing data is prevalent in empirical work. There is a presumption that iteration is needed to impute missing values, and successful matrix recovery requires solving a regularized problem under a missing at random assumption. This paper shows that if we are willing to impose a strong factor structure, then the entire low rank component of the data can be consistently estimated by our proposed tw procedure without iteration or regularization. The methodology can be used within the potential outcomes framework to estimate the effect of treatment on the treated, and a distribution theory is provided.
References
Appendix A
This appendix provides proofs to the results in the main text along with more general results that are of independent interest. Throughout, denotes the Frobenius norm of matrix .
Recall the notation and
where is , and is ; is and is . The partitions of and are the same. We write if , and if .
Similar to the rotation matrix given in Section 2, let , then the tall estimator satisfies (see Bai (2003), Theorem 1)
where uniformly in . Similarly, there is a rotation matrix such that for each , the wide estimator satisfies (see Bai (2003), Theorem 2)
where uniformly in . Let , and define for
where is an estimator for obtained by regressing on for .
Under the assumptions of Proposition 1: (i)
where uniformly in .
Consider part (i). Rewrite the representation in (A.2) as
This follows from [see Bai (2003), p 166) and Bai and Ng (2019)],
Similarly, the Tall estimator has the asymptotic representation
This implies , where
Note converges in probability to a positive definite matrix. Consider the numerator,
Replace by (ignore higher orders), we see the two terms on the right hand side are each , thus dominated by . This proves (A.3).
We next proof part (ii). We can rewrite the representations in (A.1) by
This follows by multiplying (A.1) by and using
where . The second equality uses the definition of . Rewrite (A.5) as
Multiply (A.6) by and multiply (A.7) by , we have
where and are defined earlier. Thus
Proof of Lemma 2.
Proof of Proposition 1.
This proposition is also a direct consequence of Lemma 1. Applying Lemma 1 to the tall and wide blocks separately, we obtain the results in (i), (ii), and (iii). For the missing block, that is, for , the limiting distribution for the estimated common component follows from the asymptotic representation in (A.4). By Assumption C,
Analysis based on imputed data matrix
Let and so the missing values are replaced by the estimated common components . We have and Consider estimating the factor and factor loadings using the matrix . Let be the first eigenvectors corresponding to the first largest eigenvalues (arranged in decreasing order) of the matrix with the normalization , that is,
where is an diagonal matrix consisting of the eigenvalues. Let . Define the rotation matrix
all matrices, where , and are defined earlier. We further put
where are sub-blocks of , partitioned conformably, for example,
with representing the th row of with .
Let . We can write the matrix as
Let , we can write as
From , we have
The terms involving are dominated. We focus on the remaining terms. Expanding the preceding equation, ignoring the terms involving , we obtain
Proof of Lemma 3(i). We first collect some basic results. Notice
where is the th row of matrix (or .) Similarly,
Using (an by matrix), and (A.9),
(which can be much smaller than , depending on and ). Consider
The matrix satisfies . Thus
this term is dominated by others. Summarizing results, we have
Consider the first block. Let (a matrix of dimension ), then the first block is , which is a subblock of . Thus, from (A.10),
[in fact, ]. Next consider the off-diagonal block. Noticing , where is , and is , and \|{\cal E}_{11}{\cal E}_{21}^{\prime}\|^{2}=\sum_{t=1}^{T_{o}}\sum_{h=1}^{T_{m}}\Big{(}\sum_{j=1}^{N_{o}}e_{jt}e_{j,T_{o}+h}\Big{)}^{2}. Thus
Here for simplicity, we assume the non-overlapping errors are uncorrelated. The block is of the same order of magnitude as above. Next,
the last equality uses results (A.9). Similarly, is negligible.
Next consider . The dominating terms are and . We analyze each of them. Note , and , thus
where we use (A.9) and . Next
, and . Thus
Summarizing results gives us (A.11). This completes the proof of Lemma 3(i).
Left multiplying (A.8) by on each side and dividing by ,
By the argument of Bai (2003), it is sufficient to prove each of the last 3 terms on the left hand side converges in probability to zero. That is,
The first 3 terms are each . The last term is
After dividing by , the term involving is negligible, using Lemma 3(i). Notice is equal to the transpose of (A.13) (ignoring ), this proves (A.12). Similarly,
The term involving is negligible. It suffices to show . Bai and Ng (2002) proved to be . Given the difference between and , it remains to show
The dominating term is
(asymptotic representation for ). Under Assumptions of A-B
for , ;
for , , where uniformly in , and the matrix is defined as
Proof of Proposition A.1. Let denote the th entry of .
We can show that the first and the last terms on the right hand side are , the limiting distribution is determined by the second term. That is,
For , then for all . This gives part (a) of Proposition A.1. But for
Plugging in into the preceding formula we obtain
The first term on the right is equal to (note )
where and term is negligible because it can be rewritten as
Note is a scalar and is commutable with . Summarizing result, for ,
It is important to note that the limiting distribution of is determined by , and the convergence rate is .
Proof of Corollary A.1.
(asymptotic representation of ) Under Assumptions A-B,
for ;
for , \widetilde{\Lambda}^{+}_{i}-G^{+}\Lambda_{i}^{0}=H^{+\prime}\frac{1}{T}\Big{(}\sum_{t=1}^{T_{o}}F_{t}^{0}e_{it}+\sum_{t=T_{o}+1}^{T}F_{t}^{0}(u_{it}+v_{it})\Big{)}+\hat{\eta}_{NT,i}=H^{+\prime}\mathbf{B}_{F}\frac{1}{T_{o}}\Big{(}\sum_{t=1}^{T_{o}}F_{t}^{0}e_{it}\Big{)}+\hat{\eta}_{NT,i}+O_{p}((TN_{o})^{-1/2}), where uniformly in , and the matrix is defined as
Proof of Proposition A.2.
Given Proposition A.1, the proof of Proposition A.2 invokes some symmetry arguments, as in Bai (2003). The details are omitted. Part (b) of the Proposition uses
This implies for part (b).
Proof of Corollary A.2.
This follows from the asymptotic representation in Proposition A.2.
Proof of Proposition 2.
Parts (a) and (b) of the proposition are implied by Corollary A.1 and parts (c) and (d) of the proposition are implied by Corollary A.2.
Proof of Proposition 3.
where and .
To see this, rewrite the representations in part (a) of Proposition A.1 as
where again . This follows from
Similarly rewrite the representation in part (a) of Proposition A.2 as
Here we have used . Thus
where both and are . By assumption, , and , so and are dominated terms. Also by assumption, , it follows that, conditional on (if it is random),
The two limiting distributions are asymptotically independent, this implies (A.18) (see the proof of Theorem 3 in Bai, 2003).
Now consider , still with . From part (b) of Proposition A.1, rewrite the representation as
Using the same argument as in the proof of (A.18), we have
The limit of the first term was considered earlier. The second term, multiplying , is asymptotically normal , where
The two terms are asymptotically independent. Thus
The proof for the block is the same. The asymptotic representation becomes
This implies .
Proof of Corollary 1.
Consider the block defined by . From the representation in (Proof of Proposition 3.). Term I2 is for each and . Taking squares and then averaging over this block gives the rate . The square root of the average is . Similarly, averaging the squares of term I3 gives . The square root of this average is . Term I1 and term I4 are both uniformly bounded by . Thus its Frobenius norm over the corresponding blocks is still of this magnitude. This implies . The proofs for other blocks are the same. For example, for the block and , we use (A.20) to obtain . Corollary 1 is obtained by averaging the four blocks, the weight for each block corresponds to the block size.