EconBase
← Back to paper

Partial Identification in Moment Models with Incomplete Data--A Conditional Optimal Transport Approach

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.

95,715 characters · 20 sections · 123 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.

Partial Identification in Moment Models with Incomplete Data---A Conditional Optimal Transport Approach

abstractIn this paper, we develop a unified approach to study partial identification of a finite-dimensional parameter defined by a general moment model with incomplete data. We establish a novel characterization of the identified set for the true parameter in terms of a continuum of inequalities defined by conditional optimal transport. For the special case of an affine moment model, we show that the identified set is convex and that its support function can be easily computed by solving a conditional optimal transport problem. For parameters that may not satisfy the moment model, we propose a two-step procedure to construct its identified set. Finally, we demonstrate the generality and effectiveness of our approach through several running examples. Keywords: Algorithmic Fairness, Causal Inference, Data Combination, Linear Projection, Partial Optimal Transport, Support Function, Time Complexity

Introduction

Model and Motivation

Many parameters of interest in economic models satisfy a finite number of moment conditions. The seminal paper of hansen1982large developed a systematic approach to identification, estimation, and inference when all economic variables in the moment condition model are observable in one data set. Since then, the generalized method of moments has become an influential tool in empirical research in economics. However, researchers often do not have access to a single data set that contains observations on all variables and must rely on multiple data sources that cannot be matched; see the discussion in Section (ref). This paper develops the first systematic method to study the identification of parameters in moment condition models using multiple data sets.

Specifically, consider the following moment model,

equation[equation omitted — 129 chars of source]

where $Y_{1}\in\mathcal{Y}_{1}\subseteq\mathbb{R}^{d_{1}},Y_{0}\in\mathcal{Y}_{0}\subseteq\mathbb{R}^{d_{0}}$, and $X\in\mathcal{X}\subseteq\mathbb{R}^{d_{x}}$ denote three distinct random variables; $\theta^{\ast}\in\Theta\subseteq\mathbb{R}^{d_{\theta}}$ is the true unknown parameter; $m$ is a known $k$ dimensional moment function; and $\mathbb{E}_{o}$ denotes the expectation with respect to the true but unknown joint distribution $\mu_{o}$ of $\left(Y_{1},Y_{0},X\right)$.

This paper studies identification of $\theta^{\ast}$ (and known functions of $\theta^{\ast}$)\footnote{When $\delta^{\ast}=G\left(\theta^{\ast}\right)$ is the parameter of interest for some function $G\left(\cdot\right)$, we say $G\left(\cdot\right)$ is known if either its functional form is known or (point) identifiable.} in a prevalent case of incomplete data in empirical research, where the relevant information on $\left(Y_{1},Y_{0},X\right)$ is contained in two datasets, and the units of observations in different datasets cannot be matched: the first dataset identifies the distribution of $\left(Y_{1},X\right)$ and the second dataset identifies the distribution of $\left(Y_{0},X\right)$. We develop the first general unified methodology for studying the sharp identified set (identified set, hereafter) of $\theta^{\ast}$ (and known functions of $\theta^{\ast}$) in the general moment model ((ref)) with incomplete data. Our method allows for (non-additively separable) moment function of any finite dimension, parameter $\theta^{*}$ of any finite dimension, and two arbitrary random variables $Y_{1}$ and $Y_{0}$, each of any finite dimension.

Main Contributions

\paragraph*{General Contributions}

The available sample information in the incomplete data scenario that we study in this paper allows identification of the distribution functions of $\left(Y_{1},X\right)$ and $\left(Y_{0},X\right)$, but does not identify the distribution of $\left(Y_{1},Y_{0},X\right)$ without additional assumptions on the joint distribution of $\left(Y_{1},Y_{0},X\right)$. For the general moment model ((ref)), we establish a characterization of the identified set of $\theta^{\ast}$ in terms of a continuum of inequalities involving conditional optimal transport (COT hereafter).\footnote{Conditional optimal transport is an optimal transport between two conditional measures; see carlier2016vector and hosseini2025conditional for rigorous accounts of COT. For expositional simplicity, we often use optimal transport even though it is a conditional optimal transport in this paper.}

For an important class of affine moment models, where the moment function $m\left(\cdot;\theta\right)$ can be written as an affine transformation of $\theta$ with the expectation of the linear transformation matrix being identifiable, but not the translation vector, we show that by applying our general characterization, the identified set for $\theta^{\ast}$ in this class of models is characterized by a continuum of linear inequalities and is thus convex. Furthermore, we derive the expression of the support function of the convex identified set and show that the value of the support function in each direction can be obtained by solving a COT problem. As we demonstrate using several running examples, our novel characterization through COT allows us to make use of theoretical and computational tools for optimal transport (OT hereafter; e.g., rachev2006mass, villani2008optimal, santambrogio2015optimal, and peyre2019computational) to further simplify the characterization and develop computationally more tractable identified sets for specific models than the existing ones in the literature.\footnote{ Other works that have employed optimal transport in partial identification analysis in different set-ups from ours include galichon2011set, galichon2016optimal, and d2025partially. We refer interested readers to molinari2020microeconometrics for a recent survey.} The well-known Fr\'echet problem reviewed in Section (ref) is a simple example of an affine moment model to which our characterization of the identified set applies. We show in Section (ref) that bounding causal effect in meango2025combining is an example of the Fr\'echet problem and the COTs in our characterization provide closed-form expressions for the dual optimizations in Corollary 1 in meango2025combining.\footnote{Mathematically, the bounds problem in meango2025combining is similar to that in fan2014identifying, fan2016estimation. All are variations of the short and long regression in cross2002regressions; see also molinari2006generalization. }

To take full advantage of our results for affine moment models, we propose a two-step procedure to obtain the identified set of the parameter of interest. In the first step of this procedure, we construct an auxiliary parameter that satisfies an affine moment model and establish its identified set. The auxiliary parameter is chosen so that the parameter of interest can be written as a known transformation of the auxiliary parameter. In the second step, we compute the identified set of the parameter of interest from the known transformation of the convex identified set obtained in the first step. The two-step procedure substantially broadens the applicability of our results, since it applies to parameters of interest that may not even satisfy the moment model ((ref)) as we illustrate via the algorithmic measures in Examples (ref) and (ref).

\paragraph*{Results for Running Examples}

To demonstrate the versatility and advantages of our general method and the proposed two-step procedure, we study three examples of partially identified parameters with incomplete data and establish their identified sets using our approach. We show that our method leads to novel and significantly improved results in all examples compared with those in the literature.

Example (ref) is a linear projection model. We consider two different types of sample information, where the first extends that in d2024linear and hwang2023bounding; the second data structure extends that in kitagawa2023linear. The moment function is, in general, not affine, and the identified set of the parameter of interest may not be convex. We employ the proposed two-step procedure, where like hwang2023bounding, we express the parameter of interest as a known function of an unknown parameter $\theta^{\ast}$ satisfying an affine moment model. In the first step, we show that the identified set of $\theta^{\ast}$ is convex and its support function is given by an integral of COT with quadratic costs, the most studied OT problem; see brenier1991polar, santambrogio2015optimal, and peyre2019computational. In the second step, we obtain the identified set of the parameter of interest from that of $\theta^{*}$. Using Proposition 2.17 in santambrogio2015optimal, we show that when specialized to the model in d2024linear, our identified set reduces to that in d2024linear. When specialized to hwang2023bounding, our result leads to the identified set that is a proper subset of the superset proposed in hwang2023bounding. Based on our characterization of the identified set, we also propose a computationally simpler outer set that is also a subset of the superset in hwang2023bounding.

Examples (ref) and (ref) are two algorithmic measures first studied in kallus2022assessing (KMZ, hereafter) in the data combination setup. The measures themselves do not satisfy the moment model ((ref)). To apply our results, we first express them as known transformations of some auxiliary parameter $\theta^{*}$, such that $\theta^{*}$ satisfies model ((ref)) with an affine moment function. Consequently, the two-step procedure applies, and the results we establish for the identified set of affine models apply to $\theta^{*}$ in both examples. Specifically, for Example (ref) on the demographic disparity (DD) measure, we provide a closed-form expression for the support function of the identified set for any collection of DD measures and\emph\textit{\emph{any}} number of protected classes. Furthermore, we show that the identified set is a polytope with a closed-form expression for its vertices. For the specific collection of DD measures studied in KMZ, our closed-form expression solves the infinite dimensional linear programming formulation in KMZ. We analyze the theoretical time complexity of both our support function and vertex-based procedures and show that they are significantly faster than KMZ's method. Using the empirical example in KMZ, we demonstrate that the improvement can be over 15,000-fold.

For the true-positive rate disparity (TPRD) measure in Example (ref), KMZ propose to construct the convex hull of the identified set. They characterize the support function of the convex hull through a computationally costly, NP-hard non-convex optimization problem. Choosing the auxiliary parameter vector $\theta^{\ast}$ cleverly, we express the identified set for any finite number of TPRD measures as a continuous map of the identified set for $\theta^{\ast}$, where the identified set for $\theta^{\ast}$ is convex. Consequently, the identified set for any finite number of TPRD measures is connected, and any single TPRD measure is a closed interval regardless of the number of protected classes. For a single TPRD measure, we provide closed-form expressions for the lower and upper bounds of the closed interval extending the result in KMZ developed for two protected classes. For multiple TPRD measures, we demonstrate that the support function of the identified set for the auxiliary parameter $\theta^{\ast}$ can be computed through a conditional partial optimal transport problem. Technically, we show that the conditional partial OT problem always admits a solution with monotone support. Exploring the monotone support of the solution, we develop a novel algorithm, called Dual Rank Equilibration AlgorithM (DREAM), for computing the identified set for $\theta^{\ast}$ and for any collection of TPRD measures. We establish its theoretical time complexity and demonstrate that DREAM is faster than linear programming using the empirical example in KMZ.

Related Work

Data combination problems are pervasive in applied work. Examples include studies of the effect of age at school entry on the years of schooling by combining data from the US censuses in 1960 and 1980 (angrist1992effect); differentiated product demand using both micro and macro survey data (berry2004differentiated); studies of long-run returns to college attendance using PSID/NLSY and Addhealth data (fan2014identifying,fan2016estimation); repeated cross-sectional data in difference-in-differences designs (fanManzanares2017partial); evaluations of long-term treatment effects combining experimental and observational data (athey2019surrogate); algorithmic fairness analysis where decision outcomes and protected attributes are recorded in separate datasets (kallus2022assessing); study of neighborhood effects, where longitudinal residence information and information on individual heterogeneity are contained in separate datasets (hwang2023bounding); inter-generational income mobility, where linked income data across generations are often unavailable (santavirta2024name); consumption research, where income and consumption are often measured in two different datasets (crossley2022regression); and counterfactual analyses of actual choice using stated and revealed preference data (meango2025combining).

Existing works in the incomplete data case fall into two broad categories. The first category considers the case where the moment function is additively separable in $Y_{1}$ and $Y_{0}$ and establishes point identification of $\theta^{\ast}$ under weak conditions. See, e.g., angrist1992effect, hahn1998role, chen2008semiparametric, graham2016efficient, and athey2019surrogate.

The second category develops partial identification results for specific models. Except for the running examples we describe below, most existing work studying partial identification under data combination can be reformulated as examples of a special case of model ((ref)), where the moment function is

equation[equation omitted — 105 chars of source]

for a known function $h$ of dimension $k=1$. Equivalently, $\theta^{\ast}=\mathbb{E}_{o}\left[h\left(Y_{1},Y_{0}\right)\right]$. For univariate $Y_{1}$ and $Y_{0}$ with given marginals, computing lower and upper bounds for $\theta^{*}$ for a known function $h$ has a long history in probability literature, and is referred to as the (general) Fr\'{e}chet problem; see ridder2007econometrics and Section 3 in fan2014copulas for a detailed discussion of this problem, including early references and its relation to copulas. These bounds have been used to establish identified sets for $\theta^{*}$ in a broad range of applications, including distributional treatment effects in program evaluation and bivariate option pricing in finance; see e.g., fan2010sharp,fan2012confidence, fan2010partial, fan2017partial, and firpo2019partial. More recently, fan2023partial study identification in model ((ref)) with general moment function $m$ and univariate $Y_{1}$ and $Y_{0}$ via the copula approach and propose a valid inference procedure.\footnote{fan2023partial include a point-identified nuisance parameter $\gamma^{\ast}$ in their moment function to separate the parameter of interest $\theta^{\ast}$ from $\gamma^{\ast}$. Model ((ref)) can be accommodated to having the nuisance parameter by appending $\gamma^{\ast}$ to $\theta^{\ast}$.} However, the copula approach is limited to univariate $Y_{1}$ and $Y_{0}$. For multivariate $Y_{1}$ and $Y_{0}$, the lower and upper bounds on $\theta^{*}$ are defined by COTs with ground cost function $h$ or $-h$. lin2025estimation propose consistent estimation using the primal formulation of COT, while ji2023model study estimation and inference using their dual formulation.

\paragraph*{Organization} Section (ref) introduces our sample information and three motivating examples. In Section (ref), we first characterize the identified set for $\theta^{\ast}$ in our model ((ref)) in terms of a continuum of inequalities, and then study the properties of the identified set for the special class of affine moment models. Finally, we apply our characterization to the Fr\'{e}chet problem in ((ref)) and revisit meango2025combining. Sections (ref) to (ref) apply our general results to each of the motivating examples. The last section offers some concluding remarks. Technical proofs are relegated to Appendix (ref). Appendix (ref) contains additional results on each motivating example.

\paragraph*{Notation} We let $\boldsymbol{0}$ denote the zero vector and $I_{d}$ denote the identity matrix of dimension $d\times d$. We use $\mathds{1}\left\{ \cdot\right\} $ to denote the indicator function. For any vector $v\in\mathbb{R}^{d}$, we use $\left\Vert v\right\Vert $ to denote its Euclidean norm. We denote $\mathbb{S}^{d}$ as the unit sphere of dimension $d$. For any function $h$ and measure $\mu$, we denote $\int \vert h\vert d\mu < \infty$ by $h \in L^{1}(\mu)$. For any cumulative distribution function $F$ defined on $\mathbb{R}$, we let $F^{-1}\left(t\right)\equiv\inf\left\{ x:F\left(x\right)\geq t\right\} $ denote the quantile function. For any two random variables $W$ and $V$, we use $F_{W\mid v}^{-1}\left(\cdot\right)$ to denote the quantile function derived from the distribution of $W$ conditioning on $V=v$.

Model and Motivating Examples

Our model is defined by ((ref)) and Assumption (ref) below, where $\mu_{1X}$ and $\mu_{0X}$ denote probability distributions of $\left(Y_{1},X\right)$ and $\left(Y_{0},X\right)$, respectively.

assumption(i) $d_{1}\geq1$, $d_{0}\geq1$, and $d_{x}\geq0$. (ii) The distributions $\mu_{1X}$ and $\mu_{0X}$ are identifiable from the sample information contained in two separate datasets, but the joint distribution $\mu_{o}$ is not identifiable. (iii) The projections of $\mu_{o}$ on $\left(Y_{1},X\right)$ and $\left(Y_{0},X\right)$ are $\mu_{1X}$ and $\mu_{0X}$, respectively.

We maintain Assumption (ref) throughout the paper. To avoid the trivial case, we assume that $d_{1}\geq1$ and $d_{0}\geq1$, but $d_{x}\geq0$. If $d_{x}=0$, then there is no common $X$ in both datasets. Assumption (ref) (iii) can be interpreted as a comparability assumption. Consider, for example, the first dataset that contains only observations on $\left(Y_{1},X\right)$. Since $Y_{0}$ is missing, the distribution of $\left(Y_{0},X\right)$ is not identifiable. Assumption (ref) (iii) requires the underlying probability distribution of $\left(Y_{0},X\right)$ in the first dataset to be the same as the one of $\left(Y_{0},X\right)$ in the second dataset. The latter probability distribution $\mu_{0X}$ is identifiable because the second dataset contains observations on $\left(Y_{0},X\right)$. Furthermore, it implies that the distribution of $X$ in both datasets is the same.

In the rest of this section, we present three motivating examples and use them to illustrate our main results throughout the paper.

example[Linear Projection Model] Let $Y_{1}^{\top}\equiv\left(Y_{1s},Y_{1r}^{\top}\right)$ and $X^{\top}=\left(X_{p}^{\top},X_{np}^{\top}\right)$, where $Y_{1s}\in\mathbb{R}$, $Y_{1r}\in\mathbb{R}^{d_{1}-1}$, $X_{p}\in\mathbb{R}^{d_{x_{p}}}$, and $X_{np}\in\mathbb{R}^{d_{x_{np}}}$. Consider the following linear projection model: \[ Y_{1s}=\left(Y_{0}^{^{\top}},X_{p}^{\top},Y_{1r}^{\top}\right)\delta^{\ast}+\epsilon\text{ and }\mathbb{E}\left[\epsilon\left(Y_{0}^{^{\top}},X_{p}^{\top},Y_{1r}^{\top}\right)\right]=\boldsymbol{0}, \] where $\delta^{\ast}\in\mathbb{R}^{d_{0}+d_{x_{p}}+d_{1}-1}$ is the parameter of interest. We study the identification of $\delta^*$ under two different types of sample information. The first data type is motivated by the short and long regression in cross2002regressions and adopted in pacini2019two, d2024linear, and hwang2023bounding for linear projection models, where researchers observe $\left(Y_{1},X\right)$ and $\left(Y_{0},X\right)$ in two separate data sets.\footnote{We note that the variable $X_{np}$ is observed in both datasets, but does not enter the linear projection model. In the textbook complete data case, $X_{np}$ is an irrelevant variable. But as we discuss in Remark (ref), in the incomplete data case, $X_{np}$ may shrink the identified set and becomes relevant.} When $X=X_{p}$, i.e., there is no $X_{np}$, the model reduces to that in hwang2023bounding; When $d_{1}=1$, i.e., there is no $Y_{1r}$, the model becomes that in d2024linear which includes pacini2019two as a special case. d2024linear establish the identified set in their model, and note that their approach does not apply to the case in hwang2023bounding. On the other hand, hwang2023bounding construct supersets in her model. The second type of sample information is motivated by kitagawa2023linear, where $\left(Y_{1s},Y_{0},X_{p}\right)$ is available in one dataset and $\left(Y_{1s},Y_{1r},X_{p}\right)$ is in another. This fits into our model by relabeling the random variables. Assuming that only limited information is available, such as the projection coefficients of $Y_{1s}$ on $\left(Y_{0},X_{p}\right)$ and $Y_{1s}$ on $\left(Y_{1r},X_{p}\right)$ and variance-covariance matrices of the random variables within each dataset, kitagawa2023linear construct the identified set of $\delta^*$ (given their sample information) through a non-convex optimization. Instead, we establish the identified set of $\delta^*$ by making full use of the sample information, which identifies distributions of $\left(Y_{1s},Y_{0},X_{p}\right)$ and $\left(Y_{1s},Y_{1r},X_{p}\right)$ separately. Our approach is applicable to both data types. Same to hwang2023bounding, we characterize the identified set of $\delta^{*}$ in two steps. First we characterize the identified set of a newly defined parameter $\theta^{\ast}$ such that $\delta^{\ast}=G\left(\theta^{\ast}\right)$ for some continuous function $G\left(\cdot\right)$ and then deduce the identified set of $\delta^{*}$ from that of $\theta^{*}$. Specifically, we let $\theta^{*}=\left(\theta_{s}^{*\top},\theta_{r,1}^{*\top},\ldots,\theta_{r,\left(d_{1}-1\right)}^{*\top}\right)^{\top}\in\mathbb{R}^{d_{0}d_{1}}$, where $\theta_{s}^{*}=\mathbb{E}_{o}\left[Y_{0}Y_{1s}\right]$ and $\theta_{r,j}^{*}=\mathbb{E}_{o}\left[Y_{0}Y_{1r,j}\right]$ in which $Y_{1r,j}$ for $j=1,\ldots,d_{1}-1$ are the elements of $Y_{1r}$. It is easy to see that $\theta^{*}$ satisfies the moment condition: $\mathbb{E}_{o}\left[m\left(Y_{1},Y_{0},X;\theta^{\ast}\right)\right]=\boldsymbol{0}$, where \[ m\left(y_{1},y_{0},x;\theta\right)=\theta-\left(y_{0}^{\top}y_{1s},y_{0}^{\top}y_{1r,1},\ldots,y_{0}^{\top}y_{1r,\left(d_{1}-1\right)}\right)^{\top}\text{.} \] Let $\theta_{r}^{\ast}\equiv\left[\theta_{r,1}^{\ast},\ldots,\theta_{r,\left(d_{1}-1\right)}^{\ast}\right]\in\mathbb{R}^{d_{0} \times \left(d_{1}-1\right)}$. When $d_{1}>1$, $\delta^{\ast}$ can be expressed as \begin{equation} \delta^{\ast} = \begin{pmatrix} \mathbb{E}\left[Y_{0} Y_{0}^{^{\top}}\right] & \mathbb{E}\left[Y_{0}X_{p}^{\top}\right] & \theta_{r}^{\ast}\\ \mathbb{E}\left[X_{p}Y_{0}^{^{\top}}\right] & \mathbb{E}\left[X_{p}X_{p}^{\top}\right] & \mathbb{E}\left[X_{p}Y_{1r}^{\top}\right]\\ \theta_{r}^{\ast\top} & \mathbb{E}\left[Y_{1r}X_{p}^{\top}\right] & \mathbb{E}\left[Y_{1r}Y_{1r}^{\top}\right] \end{pmatrix}^{-1} \begin{pmatrix} \theta_{s}^{\ast}\\ \mathbb{E}\left[X_{p}Y_{1s}\right]\\ \mathbb{E}\left[Y_{1r}Y_{1s}\right] \end{pmatrix} \equiv G\left(\theta^{\ast}\right), \end{equation} where the expectations in the definition of $G\left(\cdot\right)$ are identifiable from both types of data. When $d_{1}=1$, we have that $Y_{1}=Y_{1s}$, $\theta^{*}=\theta_{s}^{*}$, and \begin{equation} \delta^{\ast} = \begin{pmatrix} \mathbb{E}\left[Y_{0}Y_{0}^{^{\top}}\right] & \mathbb{E}\left[Y_{0}X_{p}^{\top}\right]\\ \mathbb{E}\left[X_{p}Y_{0}^{^{\top}}\right] & \mathbb{E}\left[X_{p}X_{p}^{\top}\right] \end{pmatrix}^{-1} \begin{pmatrix} \theta^{\ast}\\ \mathbb{E}\left[X_{p}Y_{1}\right] \end{pmatrix} . \end{equation} Furthermore, for the data structure in kitagawa2023linear, $\theta_s^*=\mathbb{E}\left[Y_{0}Y_{1s}\right]$ is point identified from the first dataset $\left(Y_{1s},Y_{0},X_{p}\right)$. We can redefine $\theta_{r}^{\ast}$ as $\theta^{\ast}$ and express $\delta^{\ast}$ as \begin{equation} \delta^{\ast} = \begin{pmatrix} \mathbb{E}\left[Y_{0} Y_{0}^{^{\top}}\right] & \mathbb{E}\left[Y_{0} X_{p}^{\top}\right] & \theta^{\ast}\\ \mathbb{E}\left[X_{p}Y_{0}^{^{\top}}\right] & \mathbb{E}\left[X_{p}X_{p}^{\top}\right] & \mathbb{E}\left[X_{p}Y_{1r}^{\top}\right]\\ \theta^{\ast\top} & \mathbb{E}\left[Y_{1r}X_{p}^{\top}\right] & \mathbb{E}\left[Y_{1r}Y_{1r}^{\top}\right] \end{pmatrix}^{-1} \begin{pmatrix} \mathbb{E}\left[Y_{0}Y_{1s}\right]\\ \mathbb{E}\left[X_{p}Y_{1s}\right]\\ \mathbb{E}\left[Y_{1r}Y_{1s}\right] \end{pmatrix} . \end{equation}
example[Demographic Disparity in KMZ] We illustrate the applicability of our methodology in assessing algorithmic fairness through data combination studied in KMZ. We focus on the demographic disparity (DD) measure in this example and the true-positive rate disparity (TPRD) measure in the next example. Other measures, such as the true-negative rate disparity, can be studied in the same way. The assumptions on the data imposed in both examples align with KMZ. Let $Y_{1}\in\left\{ 0,1\right\} $ denote the binary decision outcome obtained from human decision making or machine learning algorithms. For instance, $Y_{1}=1$ represents approval of a loan application. Let $Y_{0}$ be the protected attribute, such as race or gender, that takes values in $\left\{ a_{1},...,a_{J}\right\} $. Researchers might be interested in knowing the disparity in within-class average loan approval rates. This measure is called demographic disparity and is defined as \[ \delta_{DD}^{\ast}\left(j,j^{\dagger}\right)=\Pr\left(Y_{1}=1\mid Y_{0}=a_{j}\right)-\Pr\left(Y_{1}=1\mid Y_{0}=a_{j^{\dagger}}\right) \] between classes $a_{j}$ and $a_{j^{\dagger}}$. Denote $X$ as the set of additional observed covariates. Assume that we observe $\left(Y_{1},X\right)$ and $\left(Y_{0},X\right)$ separately. KMZ study $\left[\delta_{DD}^{\ast}\left(1,J\right),\ldots,\delta_{DD}^{\ast}\left(J-1,J\right)\right]$ directly, where $a_{J}$ is treated as an advantage/reference group. For $J=2$, they provide a closed-form for the identified set for $\delta_{DD}^{\ast}\left(1,2\right)$. For $J>2$, they state that the identified set is convex and characterize its support function evaluated at each direction as an infinite dimensional linear programming. Noting that the DD measure itself does not satisfy the moment model ((ref)), we apply the two-step procedure by introducing an auxiliary parameter $\theta^{\ast}\equiv\left(\theta_{1}^{\ast},...,\theta_{J}^{\ast}\right)^{\top}$. Specifically, we define $\theta_{j}^{\ast}\equiv\Pr\left(Y_{1}=1\mid Y_{0}=a_{j}\right)$ for $j=1,\ldots,J$. The DD measure $\delta_{DD}^{\ast}\left(j,j^{\dagger}\right)$ for any $j\neq j^{\dagger}$ can be expressed as $\delta_{DD}^{\ast}\left(j,j^{\dagger}\right)\equiv\theta_{j}^{\ast}-\theta_{j^{\dagger}}^{\ast}$. We characterize $\theta^{\ast}$ via the moment model ((ref)) with the following moment function: \[ m\left(y_{1},y_{0},x;\theta\right) = \begin{pmatrix} \theta_{1}\mathds{1}\left\{ y_{0}=a_{1}\right\} -\mathds{1}\left\{ y_{1}=1,y_{0}=a_{1}\right\} \\ \vdots\\ \theta_{J}\mathds{1}\left\{ y_{0}=a_{J}\right\} -\mathds{1}\left\{ y_{1}=1,y_{0}=a_{J}\right\} \end{pmatrix} , \] where $\theta\equiv\left(\theta_{1},...,\theta_{J}\right)^{\top}$. It is easy to see that that $\mathbb{E}_{o}\left[m\left(Y_{1},Y_{0},X;\theta^{\ast}\right)\right]=\boldsymbol{0}$. Let $e_{+}\left(j\right)\in\mathbb{R}^{J}$ ($e_{-}\left(j\right)\in\mathbb{R}^{J}$) be a row vector such that the $j$-th element of $e_{+}\left(j\right)$ ($e_{-}\left(j\right)$) is $1$ ($-1$) and all the remaining elements are zero. Any DD measure $\delta_{DD}^{\ast}\left(j,j^{\dagger}\right)$ can be expressed as $\left[e_{+}\left(j\right)+e_{-}\left(j^{\dagger}\right)\right]\theta^{\ast}$. Suppose we are interested in $K$ different DD measures, where $K$ can be smaller than, greater than, or equal to $J-1$. The vector of DD measures can be written as $E\theta^{\ast}$, where $E\in\mathbb{R}^{K\times J}$ is a matrix such that each row of $E$ is of the form $e_{+}\left(j\right)+e_{-}\left(j^{\dagger}\right)$. The identified set for $E\theta^{\ast}$ follows from that of $\theta^{\ast}$.
example[True-Positive Rate Disparity in KMZ] Let $Y_{1s}\in\left\{ 0,1\right\} $ be the decision outcome and $Y_{1r}\in\left\{ 0,1\right\} $ be the true outcome. The true outcome is the target that justifies an optimal decision. $Y_{0}$ and $X$ denote the protected attribute and the proxy variable introduced in the previous section. True-positive rate disparity measures the disparity in the proportions of people who correctly get approved in loan applications between two classes, given their true non-default outcome. The TPRD measure between any two classes $a_{j}$ and $a_{j^{\dagger}}$ is defined as \[ \delta_{TPRD}^{\ast}\left(j,j^{\dagger}\right)\equiv\Pr\left(Y_{1s}=1\mid Y_{1r}=1,Y_{0}=a_{j}\right)-\Pr\left(Y_{1s}=1\mid Y_{1r}=1,Y_{0}=a_{j^{\dagger}}\right). \] Let $Y_{1}\equiv\left(Y_{1s},Y_{1r}\right)$. We assume that $\left(Y_{1},X\right)$ and $\left(Y_{0},X\right)$ can be observed separately. KMZ study the identified set for $\left[\delta_{TPRD}^{\ast}\left(1,J\right),\ldots,\delta_{TPRD}^{\ast}\left(J-1,J\right)\right]$. When $J=2$, they establish sharp bounds on $\delta_{TPRD}^{\ast}\left(1,2\right)$. For cases where $J>2$, KMZ state that the identified set is non-convex and provide the support function of its convex hull through rather complicated non-convex optimizations. As a result, it is difficult to directly analyze the properties of the identified set, such as its connectedness. Moreover, solving the optimizations can be computationally intense. Instead of directly analyzing the identified set for the TPRD measures which may be non-convex, we apply the two-step procedure by expressing the TPRD measures as a continuous nonlinear function of some auxiliary parameter $\theta^{\ast}$ such that $\theta^{*}$ satisfies model ((ref)) and its identified set is convex. Specifically, let $\theta^{\ast}\equiv\left(\theta_{1}^{\ast},...,\theta_{2J}^{\ast}\right)$, where for $j=1,...,J,$ we define \[ \theta_{j}^{\ast}\equiv\Pr\left(Y_{1s}=1,Y_{1r}=1,Y_{0}=a_{j}\right)\textrm{ and }\theta_{J+j}^{\ast}\equiv\Pr\left(Y_{1s}=0,Y_{1r}=1,Y_{0}=a_{j}\right). \] For any $j\neq j^{\dagger}$, define a nonlinear map $g_{j,j^{\dagger}}:\left[0,1\right]^{2J}\rightarrow\left[-1,1\right]$ as $ g_{j,j^{\dagger}}\left(\theta^{\ast}\right)=\frac{\theta_{j}^{\ast}}{\theta_{j}^{\ast}+\theta_{J+j}^{\ast}}-\frac{\theta_{j^{\dagger}}^{\ast}}{\theta_{j^{\dagger}}^{\ast}+\theta_{J+j^{\dagger}}^{\ast}}$. The TPRD measure $\delta_{TPRD}^{\ast}\left(j,j^{\dagger}\right)$ between classes $a_{j}$ and $a_{j^{\dagger}}$ can be expressed as $\delta_{TPRD}^{\ast}\left(j,j^{\dagger}\right)=g_{j,j^{\dagger}}\left(\theta^{\ast}\right)$. And we can represent any $K$ different TPRD measures as $G\left(\theta^{\ast}\right)$, where $G:\left[0,1\right]^{2J}\rightarrow\left[-1,1\right]^{K}$ is a multidimensional nonlinear map such that each row of $G\left(\cdot\right)$ takes the form of $g_{j,j^{\dagger}}\left(\cdot\right)$. Let $y_{1}\equiv\left(y_{1s},y_{1r}\right)$. For $\theta=\left(\theta_{1},...,\theta_{2J}\right)$, it holds that $\mathbb{E}_{o}\left[m\left(Y_{1},Y_{0},X;\theta^{\ast}\right)\right]=\boldsymbol{0}$, where \begin{align*} m\left(y_{1},y_{0},x;\theta\right)= & \left(\theta_{1}-\mathds{1}\left\{ y_{1}=\left(1,1\right),y_{0}=a_{1}\right\} ,\ldots,\theta_{J}-\mathds{1}\left\{ y_{1}=\left(1,1\right),y_{0}=a_{J}\right\} ,\right.\\ & \left.\theta_{J+1}-\mathds{1}\left\{ y_{1}=\left(0,1\right),y_{0}=a_{1}\right\} ,\ldots,\theta_{2J}-\mathds{1}\left\{ y_{1}=\left(0,1\right),y_{0}=a_{J}\right\} \right)^{\top}. \end{align*}

Identified Set for $\theta^{\ast}$ and its COT Characterization

Let $\Theta_{I}\subseteq\Theta$ denote the identified set for $\theta^{\ast}$. It is defined as

equation[equation omitted — 229 chars of source]

where $\mathbb{E}_{\mu}$ denotes the expectation taken with respect to some probability measure $\mu\in\mathcal{M}\left(\mu_{1X},\mu_{0X}\right)$, and $\mathcal{M}\left(\mu_{1X},\mu_{0X}\right)$ is the class of probability measures whose projections on $\left(Y_{1},X\right)$ and $\left(Y_{0},X\right)$ are $\mu_{1X}$ and $\mu_{0X}$.

Equivalently, the identified set $\Theta_{I}$ can be defined via conditional distributions. Let $\mu_{1\mid x}$ and $\mu_{0\mid x}$ be the conditional distributions of $Y_{1}$ on $X=x$ and $Y_{0}$ on $X=x$, respectively. Denote $\mu_{X}$ as the probability distribution of $X$. Recall that $\mathcal{Y}_{1}$, $\mathcal{Y}_{0}$, and $\mathcal{X}$ are the supports of random variables $Y_{1}$, $Y_{0}$, and $X$. For any $x\in\mathcal{X}$, let $\mathcal{M}\left(\mu_{1\mid x},\mu_{0\mid x}\right)$ denote the class of probability distributions whose projections on $Y_{1}$ and $Y_{0}$ are $\mu_{1\mid x}$ and $\mu_{0\mid x}$.\footnote{All the statement with “for any $x\in\mathcal{X}$” can be relaxed to “for almost any $x\in\mathcal{X}$ with respect to $\mu_{X}$ measure”. We ignore such mathematical subtlety to simplify the discussion.} The identified set $\Theta_{I}$ defined in ((ref)) can be equivalently expressed as \[\Theta_{I}=\left\{ \theta\in\Theta:

array[array omitted — 281 chars of source]

\right\} . \] To simplify the notation, from now on, we omit the integration variables when there is no confusion.

A General Characterization of $\Theta_{I}$ via COT

This section provides a COT approach to characterize the identified set for $\theta^{*}$ in model ((ref)). Consider the following set defined through a continuum of inequalities:

align[align omitted — 470 chars of source]

For any given $x\in\mathcal{X}$ and $\theta\in\Theta$, $\mathcal{KT}_{t^{\top}m}\left(\mu_{1\mid x},\mu_{0\mid x};x,\theta\right)$ is the COT cost between conditional measures $\mu_{1\mid x}$ and $\mu_{0\mid x}$ with (ground) cost function $t^{\top}m\left(y_{1},y_{0},x;\theta\right)$. Here we state the explicit dependence of $\mathcal{KT}_{t^{\top}m}\left(\mu_{1\mid x},\mu_{0\mid x};x,\theta\right)$ on $x$ and $\theta$ because the cost function $t^{\top}m\left(y_{1},y_{0},x;\theta\right)$ is allowed to depend on $x$ and $\theta$. With the following assumption, we show that $\Theta_{I}=\Theta_{o}$.

assumptionFor every $\theta\in\Theta$, the following two conditions hold. (i) For almost every $x\in\mathcal{X}$ with respect to $\mu_{X}$ measure, $m\left(y_{1},y_{0},x;\theta\right)$ is continuous in $\left(y_{1},y_{0}\right)$ on $\mathcal{Y}_{1}\times\mathcal{Y}_{0}$. (ii) There exist non-negative continuous functions $h_{0}\left(y_{0},x;\theta\right)\in L^{1}\left(\mu_{0X}\right)$ and $h_{1}\left(y_{1},x;\theta\right)\in L^{1}\left(\mu_{1X}\right)$ such that $ \left\Vert m\left(y_{1},y_{0},x;\theta\right)\right\Vert \leq h_{0}\left(y_{0},x;\theta\right)+h_{1}\left(y_{1},x;\theta\right)$.

Assumption (ref) (i) holds automatically when the supports of $\mu_{0\mid x}$ and $\mu_{1\mid x}$ are finite for every $x\in\mathcal{X}$, i.e., $\mu_{0\mid x}$ and $\mu_{1\mid x}$ are discrete measures, because any function on a finite set is continuous. Assumption (ref) (ii) is satisfied if $m$ is bounded. It also holds if $m$ does not depend on $x$, both $\mathcal{Y}_{1}$ and $\mathcal{Y}_{0}$ are bounded, and Assumption (ref) (i) holds. As a result, Examples (ref) and (ref) satisfy Assumption (ref). For Example (ref), the moment function satisfies Assumption (ref) (ii) if $\mathbb{E}\left[\left\Vert Y_{1}\right\Vert ^{2}\right]<\infty$ and $\mathbb{E}\left[\left\Vert Y_{0}\right\Vert ^{2}\right]<\infty$.

theoremIt holds that $\Theta_{I}\subseteq\Theta_{o}$. If Assumption (ref) holds, then $\Theta_{I}=\Theta_{o}$.

Proofs of theorems and propositions in this paper are provided in the appendix. The proof of Theorem (ref) relies on the minimax theorem derived in vianney2015minmax and the result on the weak compactness of $\mathcal{M}\left(\mu_{1\mid x},\mu_{0\mid x}\right)$ in the proof of Proposition 2.1 in villani2003topics.

Theorem (ref) characterizes the identified set for $\theta^{\ast}$ via a continuum of inequality constraints on the integral of the COT cost $\mathcal{KT}_{t^{\top}m}\left(\mu_{1\mid x},\mu_{0\mid x};x,\theta\right)$ with respect to $\mu_{X}$. This novel characterization allows us to explore both theoretical and computational tools, including duality theory developed in the OT literature, to study the identified set $\Theta_{I}$; see e.g., rachev2006mass, villani2008optimal, santambrogio2015optimal, and peyre2019computational.

\paragraph*{Covariate-Assisted Identified Set}

The conditioning variable $X$ provides information on the joint distribution of $\left(Y_{1},Y_{0}\right)$ by restricting the conditional distributions of $Y_{1}$ on $X=x$ and $Y_{0}$ on $X=x$ to be $\mu_{1\mid x}$ and $\mu_{0\mid x}$, which are identifiable from the datasets. The identified set obtained using only subvectors of $X$ can be potentially larger than the covariate-assisted identified set, which explores the information of the full vector $X$.

Partition $X\equiv\left(X_{p},X_{np}\right)$, where $X_{p}\in\mathbb{R}^{d_{x_{p}}}$ and $X_{np}\in\mathbb{R}^{d_{x_{np}}}$. Suppose that the moment function only depends on $X_{p}=x_{p}$ and denote it as $m\left(y_{1},y_{0},x_{p};\theta\right)$. This includes Fr\'{e}chet problem ((ref)) and Examples (ref)-(ref) for $d_{x_{p}}=0$. Denote by $\mu_{1\mid x_{p}}$ and $\mu_{0\mid x_{p}}$ the conditional distributions of $Y_{1}$ and $Y_{0}$ on $X_{p}=x_{p}$, respectively. Let $\mu_{X_{p}}$ be the probability distribution of $X_{p}$. Define the following set constructed using only $X_{p}$:

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

The following proposition shows that $\Theta_{I}$ is at most as large as $\Theta^{O}$.

propositionUnder Assumption (ref), it holds that $\Theta_{I}\subseteq\Theta^{O}$. Additionally, if $\mu_{0\mid x}$ is Dirac at $g_{0}(x)$, then $ \Theta_{I}=\left\{ \theta\in\Theta:\mathbb{E}\left[m\left(Y_{1},g_{0}\left(X\right),X;\theta\right)\right]=\boldsymbol{0}\right\} . $

Proposition (ref) extends Theorem 3.3 in fan2017partial established for Fr\'{e}chet problem ((ref)) with univariate $Y_{1}$ and $Y_{0}$. As long as $X_{np}$ is not independent of both $Y_{1}$ and $Y_{0}$, we expect the covariate-assisted identified set $\Theta_{I}$ to be smaller than $\Theta^{O}$. In general, the stronger $X$ and $Y_{1}$ (or $Y_{0}$) are dependent on each other, the smaller is the covariate-assisted identified set for $\theta^{\ast}$. The second part of the proposition shows that if the conditional measure $\mu_{0\mid x}$ is Dirac at $g_{0}(x)$, then $\Theta_{I}$ reduces to the identified set for the moment model with complete data $(Y_{1},X)$. Therefore, when one of $Y_{1}$ and $Y_{0}$ is perfectly dependent on $X$, $\theta^{\ast}$ is point identified under the standard rank condition. Without the common $X$ in both datasets, this would only be possible in the extreme case that one of $Y_{1}$ and $Y_{0}$ is constant almost surely.

Affine Moment Model

In this section, we show that if the moment function $m$ is affine in $\theta$, then $\Theta_{I}$ is convex with a simple expression for its support function. Throughout the discussion, we assume that Assumption (ref) holds so that Theorem (ref) applies.

assumptionLet $m_{a}\left(y_{0},x\right)$ and $m_{a}\left(y_{1},x\right)$ be matrix-valued functions of dimension $k\times d_{\theta}$ and $m_{b}\left(y_{1},y_{0},x\right)$ be a vector-valued function of dimension $k$. One of the following decompositions of $m$ holds: (i) $m\left(y_{1},y_{0},x;\theta\right)=m_{a}\left(y_{0},x\right)\theta+m_{b}\left(y_{1},y_{0},x\right)$; (ii) $m\left(y_{1},y_{0},x;\theta\right)=m_{a}\left(y_{1},x\right)\theta+m_{b}\left(y_{1},y_{0},x\right)$.

For each running example, we formulate the moment functions and $\theta$ in a way that ensures Assumption (ref) holds. Without loss of generality, we focus on the first decomposition in the following discussion. Define

equation[equation omitted — 248 chars of source]

Note that $\mathcal{KT}_{t^{\top}m_{b}}\left(\mu_{1\mid x},\mu_{0\mid x};x\right)$ does not depend on $\theta$, because the cost function $t^{\top}m_{b}\left(y_{1},y_{0},x\right)$ does not depend on $\theta$.

assumption$\Theta$ is compact and convex with a nonempty interior.
theoremSuppose Assumptions (ref) and (ref) hold. (i) $\Theta_{I}$ can be rewritten as \[ \Theta_{I}=\left\{ \theta\in\Theta:t^{\top}\mathbb{E}\left[m_{a}\left(Y_{0},X\right)\right]\theta\leq-\int\mathcal{KT}_{t^{\top}m_{b}}\left(\mu_{1\mid x},\mu_{0\mid x};x\right)d\mu_{X}\text{ for all }t\in\mathbb{S}^{k}\right\} . \] (ii) If Assumption (ref) also holds, then $\Theta_{I}$ is closed and convex.

Under Assumption (ref), the convexity of $\Theta_{I}$ follows immediately from Theorem (ref) (i), because the constraints in the expression of $\Theta_{I}$ in Theorem (ref) (i) are affine in $\theta$.

example[General Fr\'{e}chet Problem and Bounds on Causal Effect in meango2025combining] As a straightforward application of Theorem (ref), consider the Fr\'{e}chet problem in ((ref)). For this example, $k=1$ and $t\in\left\{ -1,1\right\} $. Under Assumptions (ref) and (ref), $\Theta_{I}$ reduces to the closed interval in ji2023model and lin2025estimation. In addition, the COT costs for a given $x\in\mathcal{X}$ are lower and upper bounds on the conditional parameter defined as $\theta^{\ast}\left(x\right)=\mathbb{E}_{o}\left[h\left(Y_{1},Y_{0}\right)\mid X=x\right]$ for a given $x\in\mathcal{X}$ and have closed-from expressions when $Y_{1}$ and $Y_{0}$ are univariate and $h\left(y_{1},y_{0}\right)$ is supermodular (see e.g., cambanis1976inequalities). Specifically, for each $x\in\mathcal{X}$, $\theta^{\ast}\left(x\right)\in\left[\theta_{L}\left(x\right),\theta_{U}\left(x\right)\right]$, where \begin{align} \theta_{L}\left(x\right) & =\int_{0}^{1}h\left(F_{1\mid x}^{-1}\left(u\mid x\right),F_{0\mid x}^{-1}\left(1-u\mid x\right)\right)du,\nonumber \\ \theta_{U}\left(x\right) & =\int_{0}^{1}h\left(F_{1\mid x}^{-1}\left(u\mid x\right),F_{0\mid x}^{-1}\left(u\mid x\right)\right)du \end{align} in which $F_{1\mid x}^{-1}\left(\cdot\mid x\right)$ and $F_{0\mid x}^{-1}\left(\cdot\mid x\right)$ are conditional quantile functions of $Y_{1}$ given $X=x$ and $Y_{0}$ given $X=x$, respectively. Recently, meango2025combining study a conditional parameter that can be shown to be of this form, where $h$ is a product function. Let $D$ denote the agent's binary decision based on an endogenous decision relevant attribute $X$. Using our notation, the parameter of interest in meango2025combining is $\theta^{\ast}\left(x\right)=\mathbb{E}\left[D\left(x\right)\right]$, where $D\left(x\right)$ is the potential decision when the decision relevant attribute is exogenously set to $x$. Let $P$ denote the vector of stated preference reports. Proposition 1 in meango2025combining shows that under their Assumptions 1 and 2, for each conditional distribution $F_{D,P\mid X}$ for $\left(D,P\right)$ given $X$, the average structural function $\theta^*\left(x\right)$ is identified through \[ \theta^*\left(x\right)=\int\mathbb{E}\left[D\mid X=x,P=p\right]dF_{P}\left(p\right). \] Since the sample information can only identify $F_{D\mid x}$ and $F_{P\mid x}$, which are the conditional distributions of $D$ and of $P$ given $X=x$, respectively, $\theta^*\left(x\right)$ is generally not point identified. Corollary 1 in meango2025combining expresses sharp lower and upper bounds on $\theta^*\left(x\right)$ for each fixed $x\in\mathcal{X}$ in terms of infinite dimensional optimization problems. We show below that the sharp lower and upper bounds can be obtained from ((ref)) similarly to fan2014identifying,fan2016estimation. Applying the law of iterated expectations to the first displayed equation in Section 1.4.2 in meango2025combining, we obtain an equivalent expression for $\theta^*\left(x\right)$ as \[ \theta^*\left(x\right)=\mathbb{E}_{o}\left[\left(\frac{f_{P}\left(P\right)}{f_{P\mid x}\left(P\mid x\right)}\right)D\mid X=x\right]. \] We observe separate samples on $\left(D,X\right)$ and $\left(P,X\right)$. Consequently, ((ref)) applies to $\theta^*\left(x\right)$ above with $h$ being a product function and the marginal distribution functions being respectively the conditional distribution functions of $\frac{f_{P}\left(P\right)}{f_{P\mid x}\left(P\mid x\right)}$ given $X=x$ and $D$ given $X=x$. This establishes explicit expressions for the value functions of the dual optimizations in Corollary 1 in meango2025combining.

The Support Function of $\Theta_{I}$

For a convex set $\Psi\subseteq\mathbb{R}^{d_{\theta}}$, its support function $h_{\Psi}\left(\cdot\right):\mathbb{S}^{d_{\theta}}\rightarrow\mathbb{R}$ is defined pointwise by $h_{\Psi}\left(q\right)=\sup_{\psi\in\Psi}q^{\top}\psi$. Our novel characterization of the identified set in Theorem (ref) provides a straightforward way to obtain its support function. Define \[ \Theta_{R}\equiv\left\{ \theta\in\mathbb{R}^{d_{\theta}}:t^{\top}\mathbb{E}\left[m_{a}\left(Y_{0},X\right)\right]\theta\leq-\int\mathcal{KT}_{t^{\top}m_{b}}\left(\mu_{1\mid x},\mu_{0\mid x};x\right)d\mu_{X}\text{ for all }t\in\mathbb{S}^{k}\right\} , \] where $\mathcal{KT}_{t^{\top}m_{b}}\left(\mu_{1\mid x},\mu_{0\mid x};x\right)$ is defined in ((ref)). We make the following two additional assumptions. The first assumption is on the parameter space $\Theta$.

assumptionIt holds that $\Theta_{I}=\Theta_{R}$.

By definition, we have that $\Theta_{R}\cap\Theta=\Theta_{I}$. Assumption (ref) requires that the parameter space $\Theta$ does not provide additional information on the identified set $\Theta_{I}$ given $\Theta_{R}$. It implies that the boundary of $\Theta_{I}$ is completely determined by the continuum of inequalities in Theorem (ref) (i). For all the running examples, we can easily choose $\Theta$ to make Assumption (ref) hold. The second assumption is on the moment function.

assumptionSuppose $k=d_{\theta}$ and $\mathbb{E}\left[m_{a}\left(Y_{0},X\right)\right]$ has full rank.

Assumption (ref) requires that the number of moments is the same as the number of parameters, and that the coefficient matrix is of full rank. It implies that if the joint distribution $\mu_{o}$ of $\left(Y_{1},Y_{0},X\right)$ is identifiable from the data, then $\theta^{\ast}$ is just identified but not overly-identified. For the examples in the paper, Assumption (ref) is satisfied either automatically because $\mathbb{E}\left[m_{a}\left(Y_{0},X\right)\right]$ is a diagonal matrix with positive diagonal entries or under very mild conditions.

Under Assumption (ref), the expression of $\Theta_{I}$ in Theorem (ref) (i) is equivalent to

align[align omitted — 598 chars of source]

As a result, we have that $q^{\top}\theta\leq s\left(q\right)$ for all $\theta\in\Theta_{I}$. Moreover, applying the result in rockafellar1997ConvexAnalysis, we establish that $s(q)$ is indeed the support function of $\Theta_{I}$, as it is a positively homogeneous, proper, closed, and convex function satisfying $s(0) = 0$.

propositionUnder Assumptions (ref)-(ref), it holds that $h_{\Theta_{I}}\left(q\right)=s\left(q\right)$.

Proposition (ref) shows that we can obtain the value of the support function for any direction $q$ by computing the OT cost $\mathcal{KT}_{q^{\top}\mathbb{E}\left[m_{a}\left(Y_{0},X\right)\right]^{-1}m_{b}}\left(\mu_{1\mid x},\mu_{0\mid x};x\right)$ for each $x$ and then integrating with respect to $\mu_{X}$. Moreover, if the parameter of interest is expressed as a linear map from $\theta^{\ast}$ through the linear operator $L$, then based on Proposition (ref), the support function of the identified set for the parameter of interest can be simply calculated as $s\left(L^{\star}q\right)$, where $L^{\star}$ is the adjoint operator with respect to the inner product.

remarkThe identification results in Section (ref) apply to the most general model defined in ((ref)). Even if any of Assumptions (ref)-(ref) fails, one can still use the original definition of $\Theta_{o}$ to study the identified set $\Theta_{I}$. However, if Assumptions (ref)-(ref) are satisfied, then the result in Section (ref) provides a mathematically and computationally attractive way to obtain the support function of $\Theta_{I}$. In both cases, the computational bottleneck lies in the evaluation of the OT cost: $\mathcal{KT}_{t^{\top}m}(\mu_{1\mid x},\mu_{0\mid x};x,\theta)$ in the general case and $\mathcal{KT}_{t^{\top}m_{b}}(\mu_{1\mid x},\mu_{0\mid x};x)$ in the affine case.

The Linear Projection Model

In this and the next two sections, we construct identified sets for the parameters in Examples (ref)-(ref) by applying our general results, and compare them with the existing results for each example. For Example (ref), we maintain the assumption that $\mathbb{E}\left[\left\Vert Y_{1}\right\Vert ^{2}\right]<\infty$ and $\mathbb{E}\left[\left\Vert Y_{0}\right\Vert ^{2}\right]<\infty$, an assumption that is also imposed in both d2024linear and hwang2023bounding.

Identified Sets for $\theta^{*}$ and $\delta^{*}$

Given the identified set $\Theta_{I}$ for $\theta^{\ast}$, the identified set for $\delta^{*}$ can be obtained by mapping each element in $\Theta_{I}$ through ((ref)), ((ref)), or ((ref)), depending on the model. We focus our discussion on constructing the identified set $\Theta_{I}$ in this section. We provide a detailed analysis for the first data type and summarize the identified set for the second data type in Remark (ref).

The moment function for $\theta^{*}$ is affine by its functional form:

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

Theorem (ref) implies that the identified set for $\theta^{\ast}$ is

equation[equation omitted — 235 chars of source]

Partitioning $t$ as $\left(t_{s}^{\top},t_{r,1}^{\top},\ldots,t_{r,\left(d_{1}-1\right)}^{\top}\right)^{\top}$ with $t_{s}\in\mathbb{R}^{d_{0}}$ and $t_{r,j}\in\mathbb{R}^{d_{0}}$ for $j=1,\ldots,d_{1}-1$, we can express $\mathcal{KT}_{t^{\top}m_{b}}\left(\mu_{1\mid x},\mu_{0\mid x};x\right)$ as a COT with a product cost function:

align[align omitted — 811 chars of source]

The first term on the right-hand side of Equation ((ref)) is half of the Wasserstein distance between the conditional distribution of $\left(t_{s}^{\top}Y_{0},t_{r,1}^{\top}Y_{0},\cdots,t_{r,\left(d_{1}-1\right)}^{\top}Y_{0}\right)$ and that of $Y_{1}$ given $X=x$. Wasserstein distance is the value function of the most studied and best understood OT problem. Recent breakthroughs in computational OT have contributed to the surge in its applications in diverse disciplines; see peyre2019computational for a comprehensive account of computational methods and applications. In addition, when both marginals are Gaussian, closed-form expressions for Wasserstein distance are known; see the discussion in santambrogio2015optimal and galichon2016optimal.

One important feature of the OT problem in ((ref)) is that the `effective' dimension of the marginal measures is equal to the dimension of $Y_{1}$ regardless of the dimension of $Y_{0}$, which can be high. Consequently, the computational burden only depends on $d_{1}$ and is independent of $d_{0}$. Moreover, in Appendix (ref), we show that by reordering elements of $\theta^{\ast}$, we can obtain a similar but alternative characterization of $\Theta_{I}$ such that the COT consists of a Wasserstein distance between the conditional distribution of $\left(t_{1}^{\top}Y_{1},\cdots,t_{d_{0}}^{\top}Y_{1}\right)$ and that of $Y_{0}$ given $X=x$ with $t_{j}\in\mathbb{R}^{d_{1}}$ for $j=1,\ldots,d_{0}$. For such a characterization, the computational burden would only depend on $d_{0}$.

When $d_{1}>1$ as in hwang2023bounding, the value function of Equation ((ref)) does not have a closed-from solution. However, a celebrated theorem of brenier1991polar implies that under mild regularity condition on $\mu_{1|x}$, the solution to ((ref)) concentrates on the gradient of a convex function, and the value function can be written as $\int-\nabla\phi_{t}\left(y_{1},x\right)y_{1}d\mu_{1\mid x},$ where $\nabla\phi_{t}\left(Y_{1},x\right)$ denotes the gradient of $\phi_{t}\left(\cdot,x\right)$, a convex function for each $t$ and $x$.

propositionSuppose $\mu_{1|x}$ is absolutely continuous with respect to the Lebesgue measure for almost every $x\in\mathcal{X}$ with respect to $\mu_{X}$ measure. Then the support function of $\Theta_{I}$ is $h_{\Theta_{I}}\left(q\right)=\mathbb{E}\left[\nabla\phi_{q}\left(Y_{1},X\right)Y_{1}\right]$ and the identified set $\Delta_{I}$ for $\delta^{\ast}$ is \[ \Delta_{I}=\left\{ \delta=G\left(\theta\right):q^{\top}\theta\leq\mathbb{E}\left[\nabla\phi_{q}\left(Y_{1},X\right)Y_{1}\right]\textrm{ for all }q\in\mathbb{S}^{d_{0}d_{1}}\right\} , \] where $G$ is defined in Equation ((ref)).

Due to the functional form of $G\left(\cdot\right)$, the resulting set $\Delta_{I}$ may not be convex. On the other hand, because $\Theta_{I}$ is convex and $G\left(\cdot\right)$ is continuous, the set $\Delta_{I}$ is connected and the projection of $\Delta_{I}$ onto any one-dimensional subspace is a connected interval.

remarkIn hwang2023bounding, $X=X_{p}$. By the definition of $\Theta_{I}$ in ((ref)), we obtain that \begin{align*} \Theta_{I}=\left\{ \theta\in\Theta:\theta-\mathbb{E}_{\mu}\left[m_{b}\left(Y_{1},Y_{0},X\right)\right]=\boldsymbol{0} for some \mu\in\mathcal{M}\left(\mu_{1X},\mu_{0X}\right)\right\} . \end{align*} Its support function is defined as $h_{\Theta_{I}}\left(q\right)=\sup_{\mu\in\mathcal{M}\left(\mu_{1X},\mu_{0X}\right)}\mathbb{E}_{\mu}\left[q^{\top}m_{b}\left(Y_{1},Y_{0},X\right)\right]$. This is the support function in Equation (7) in hwang2023bounding. hwang2023bounding proposes to discretize elements in $\mathcal{M}\left(\mu_{1X},\mu_{0X}\right)$ in order to compute the support function via linear programming. When the dimension of $Y_{1}$, $Y_{0}$, or $X$ is high, linear programming facilitated by discretization may be computationally challenging, which motivates hwang2023bounding to propose supersets of $\Theta_{I}$. On the other hand, depending on the formulation, the computational cost of our characterization of the identified set or support function can be independent of $d_{1}$, the dimension of $Y_{1}$; or of $d_{0}$, the dimension of $Y_{0}$. When both $d_{1}$ and $d_{0}$ are large and the computational burden is a concern, we can also construct outer sets for $\Theta_{I}$ from our novel characterization in ((ref)) by choosing specific values of $t$. In Appendix (ref), we provide one such outer set for $\Theta_{I}$, which is computationally attractive and, importantly, is a subset of the superset proposed in hwang2023bounding.
remarkFor the second data type, the identified set $\Theta_{I}$ for $\theta^{\ast}$ can be established in the same way as for the first data type with the COT $\mathcal{KT}_{t^{\top}m_{b}}\left(\mu_{1\mid x},\mu_{0\mid x};x\right)$ being $\inf_{\mu_{10\mid x}\in\mathcal{M}\left(\mu_{1\mid x},\mu_{0\mid x}\right)}\protect\int\protect\int-\left(t_{r,1}^{\top}y_{0},\cdots,t_{r,\left(d_{1}-1\right)}^{\top}y_{0}\right)y_{1}d\mu_{10\mid x}$, where $t_{r,j}\in\mathbb{R}^{d_{0}}$ for $j=1,\ldots,d_{1}-1$. The identified set for $\delta^{\ast}$ is then obtained by mapping $\Theta_{I}$ through ((ref)).

The Model in d2024linear

In d2024linear, $d_{1}=1$. Proposition 2.17 in santambrogio2015optimal implies that Equation ((ref)) has a closed-form solution and

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

Consequently, we derive the identified set $\Theta_{I}$ from ((ref)) as

equation[equation omitted — 243 chars of source]

Following d2024linear, we let $\Theta=\mathbb{R}^{d_{0}}$. Assumptions (ref) and (ref) are both satisfied. For any $q_{0}\in\mathbb{S}^{d_{0}}$, Proposition (ref) provides the support function of $\Theta_{I}$:

equation[equation omitted — 186 chars of source]

Assume that the inverse of the second moment matrix $\mathbb{E}\left[\left(Y_{0}^{\top},X_{p}^{\top}\right)^{\top}\left(Y_{0}^{\top},X_{p}^{\top}\right)\right]$ exists, and denote its block partition by $\left(A_{ij}\right)_{i,j\in\left\{ 0,p\right\} }$. The same assumption is imposed in d2024linear. It follows from Equation ((ref)) that $\delta^{\ast}=\left(A_{00}^{\top},A_{p0}^{\top}\right)^{\top}\theta^{\ast}+\left(A_{0p}^{\top},A_{pp}^{\top}\right)^{\top}\mathbb{E}\left[X_{p}Y_{1}\right]$, which is an affine map. The standard result on the support function provides the following proposition on the support function of $\Delta_{I}$, which coincides with Theorems 1 and 2 of d2024linear.

propositionIt holds that $\Delta_{I}=\left\{ \delta\in\mathbb{R}^{d_{0}+d_{x_{p}}}:q^{\top}\delta\leq h_{\Delta_{I}}\left(q\right)\text{ for all }q\in\mathbb{S}^{d_{0}+d_{x_{p}}}\right\}$, where for $q\equiv\left(q_{0}^{\top},q_{X_{p}}^{\top}\right)^{\top}$ such that $q_{0}\in\mathbb{R}^{d_{0}}$ and $q_{X_{p}}\in\mathbb{R}^{d_{x_{p}}}$, \[ h_{\Delta_{I}}\left(q\right)=\int\int_{0}^{1}F_{\left(q_{0}^{\top}A_{00}+q_{X_{p}}^{\top}A_{p0}\right)Y_{0}\mid x}^{-1}\left(u\right)F_{Y_{1}\mid x}^{-1}\left(u\right)dud\mu_{X}+\left(q_{0}^{\top}A_{0p}+q_{X_{p}}^{\top}A_{pp}\right)\mathbb{E}\left[Y_{1}X_{p}\right]. \]
remarkIt follows from Proposition (ref) that the identified set $\Theta_I$ in ((ref)) is generally tighter than the outer set using the distributions of $t^\top Y_0$ and $Y_1$ conditional on $X_p$ only. As a result, the `irrelevant variable' $X_{np}$ in the complete data case may tighten the identified set of $\delta^*$ and hence becomes `relevant' in the incomplete data set-up, see d2024linear for a similar discussion.
remarkIn the model studied in pacini2019two, there is no $X_{np}$. When $Y_{0}$ is a scalar, it can be shown that our $\Delta_{I}$ is equal to the set obtained in pacini2019two. On the other hand, when $Y_{0}$ is multivariate, the bound in pacini2019two is not tight; see d2024linear and Appendix (ref) for numerical comparisons. In fact, the set in pacini2019two is equivalent to the set mapped from an outer set of $\Theta_I$ defined by the same inequality constraint as in $\Theta_{I}$ in ((ref)) but for $t\in\mathbb{U}^{d_{0}}$ rather than for all $t\in\mathbb{S}^{d_{0}}$, where $\mathbb{U}^{d_{0}} \equiv\left\{ t\in\mathbb{S}^{d_{0}}:\textrm{only one element in }t\textrm{ is nonzero}\right\}$.

Demographic Disparity Measures in KMZ

Identified Sets for DD Measures and Its Vertex Representation

The moment function $m$ is affine with

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

The identified set $\Theta_{I}$ of $\theta^{\ast}$ follows from Theorem (ref) (i), where \[ \mathcal{KT}_{t^{\top}m_{b}}\left(\mu_{1\mid x},\mu_{0\mid x};x\right)=\inf_{\mu_{10\mid x}\in\mathcal{M}\left(\mu_{1\mid x},\mu_{0\mid x}\right)}\int\int\sum_{j=1}^{J}-t_{j}\mathds{1}\left\{ y_{1}=1,y_{0}=a_{j}\right\} d\mu_{10\mid x}. \] Let $\mathtt{\boldsymbol{d}}\left(y_{1}\right)=\mathds{1}\left\{ y_{1}=1\right\} $ and $\mathtt{\boldsymbol{d}}_{t}\left(y_{0}\right)=\sum_{j=1}^{J}t_{j}\mathds{1}\left\{ y_{0}=a_{j}\right\} $. Then $\mathcal{KT}_{t^{\top}m_{b}}\left(\cdot\right)$ equals to \[ \inf_{\mu_{10\mid x}\in\mathcal{M}\left(\mu_{1\mid x},\mu_{0\mid x}\right)}\int\int-\mathtt{\mathtt{\boldsymbol{d}}}\left(y_{1}\right)\times\mathtt{\mathtt{\boldsymbol{d}}}_{t}\left(y_{0}\right)d\mu_{10\mid x}\left(y_{1},y_{0}\right)=-\int_{\Pr\left(Y_{1}=0\mid x\right)}^{1}F_{D_{t}\mid x}^{-1}\left(u\right)du \] with $D_{t}\equiv \mathtt{\boldsymbol{d}}_{t}\left(Y_{0}\right)$. The last equality follows from the monotone rearrangement inequality.

propositionThe identified set of any $K$ different DD measures $E\theta^{\ast}$ is $\Delta_{DD}=\left\{ E\theta:\theta\in\Theta_{I}\right\} $, where \[ \Theta_{I}=\left\{ \theta\in\Theta:\sum_{j=1}^{J}t_{j}\theta_{j}\Pr\left(Y_{0}=a_{j}\right)\leq\int\int_{\Pr\left(Y_{1}=0\mid x\right)}^{1}F_{D_{t}\mid x}^{-1}\left(u\right)dud\mu_{X}\text{ for all }t\in\mathbb{S}^{J}\right\} . \] Moreover, if $\Pr\left(Y_{0}=a_{j}\right)>0$ for $j=1,\ldots,J$, then $\Delta_{DD}$ is convex. For any $p\in\mathbb{S}^{K}$, let $D_{E^{\top}p}\equiv\sum_{j=1}^{J}\left(E^{\top}p\right)_{j}\mathds{1}\left\{ Y_{0}=a_{j}\right\} \Pr\left(Y_{0}=a_{j}\right)^{-1}$, where $\left(E^{\top}p\right)_{j}$ denotes the $j$-th element of $E^{\top}p$. The support function of $\Delta_{DD}$ is \begin{equation} h_{\Delta_{DD}}\left(p\right)=\int\int_{\Pr\left(Y_{1}=0\mid x\right)}^{1}F_{D_{E^{\top}p}\mid x}^{-1}\left(u\right)dud\mu_{X}. \end{equation}

Given $p$, $D_{E^{\top}p}$ is a discrete random variable. As a result, we can easily obtain its quantile function. The closed-form expression of the support function of $\Delta_{DD}$ in ((ref)) provides an easy way to obtain the identified set for any single DD measure $\delta_{DD}^{\ast}\left(j,j^{\dagger}\right)$ for $J\geq2$; see Appendix (ref) for details. Furthermore, the analytical expression of the support function in ((ref)) allows us to obtain the vertex representation of the identified set $\Delta_{DD}$, which substantially simplifies its computation.

In the following, we prove that both $\Theta_{I}$ and $\Delta_{DD}$ are polytopes and provide their vertex representations. We first derive the vertices for $\Theta_{I}$ whose support function is

equation[equation omitted — 165 chars of source]

The ones for $\Delta_{DD}$ can be obtained by premultiplying $E$ to the vertices for $\Theta_{I}$.

We need to introduce some notation. For the set $M\equiv\left\{ 1,\ldots,J\right\} $, define $\mathfrak{S}_{M}$ as its symmetric group, which is the group of all permutations of $M$. Let $\sigma$ denote a generic element of $\mathfrak{S}_{M}$, i.e., a permutation of $M$. The order of $\mathfrak{S}_{M}$, denoted as $\left|\mathfrak{S}_{M}\right|$, equals $J!$. We can enumerate each element of $\mathfrak{S}_{M}$ as $\sigma_{s}$ for $s=1,\ldots,\left|\mathfrak{S}_{M}\right|$. The values of $\Pr\left(Y_{0}=a_{j}\right)$ for $j\in M$ are identifiable from the sample information. For any $\sigma_{s}\in\mathfrak{S}_{M}$, we define the following polyhedral region of $q$: \[ R_{s}\equiv\left\{ q\in\mathbb{R}^{J}:q_{\sigma_{s}\left(1\right)}\Pr\left(Y_{0}=a_{\sigma_{s}\left(1\right)}\right)^{-1}\leq\ldots\leq q_{\sigma_{s}\left(J\right)}\Pr\left(Y_{0}=a_{\sigma_{s}\left(J\right)}\right)^{-1}\right\} . \] Given $q\in R_{s}$, for any $s=1,\ldots,\left|\mathfrak{S}_{M}\right|$, the quantile function $F_{D_{q}\mid x}^{-1}\left(u\right)$ is a step function taking the value $\sum_{j=1}^{J}\frac{q_{\sigma_{s}\left(j\right)}}{\Pr\left(Y_{0}=a_{\sigma_{s}\left(j\right)}\right)}U\left(u,j,x\right)$, where \[ U\left(u,j,x\right)\equiv\mathds{1}\left\{ \sum_{j=1}^{j-1}\Pr\left(Y_{0}=a_{\sigma_{s}\left(j\right)}\mid x\right)<u<\sum_{j=1}^{j}\Pr\left(Y_{0}=a_{\sigma_{s}\left(j\right)}\mid x\right)\right\} \] and $\Pr\left(Y_{0}=a_{j}\mid x\right)$ denotes the conditional probability of $Y_{0}=a_{j}$ given $X=x$. This is a linear function of $q$. Since the integrations with respect to $u$ and $\mu_{X}\left(x\right)$ in Equation ((ref)) do not affect linearity, it holds that $h_{\Theta_{I}}\left(q\right)$ is linear in $q$ for $q\in R_{s}$. Because each $R_{s}$ is a convex cone and $\cup_{s=1}^{\left|\mathfrak{S}_{M}\right|}R_{s}=\mathbb{R}^{J}$, $h_{\Theta_{I}}\left(q\right)$ is a piecewise linear function in $q\in\mathbb{R}^{J}$ on a finite number of convex cones. Theorem 2.3.4 in hug2010course shows that $\Theta_{I}$ is a polytope. Furthermore, for each $s=1,\ldots,\left|\mathfrak{S}_{M}\right|$, because $h_{\Theta_{I}}\left(q\right)$ is linear in $q$ for $q\in R_{s}$, there is a vector $v_{s}$ such that $h_{\Theta_{I}}\left(q\right)=q^{\top}v_{s}$ for $q\in R_{s}$. Let $v_{s}\equiv\left(v_{s,1},\ldots,v_{s,J}\right)$. Then each element $v_{s,\sigma_{s}\left(j\right)}$ for $j=1,\ldots,J$ can be computed by \[ \int\int_{\Pr\left(Y_{1}=0\mid x\right)}^{1}\frac{1}{\Pr\left(Y_{0}=a_{\sigma_{s}\left(j\right)}\right)}U\left(u,j,x\right)dud\mu_{X}. \] There are $\left|\mathfrak{S}_{M}\right|$ number of vectors $v_{s}$ (not necessarily unique) such that $h_{\Theta_{I}}\left(q\right)=q^{\top}v_{s}$ for $q\in R_{s}$. The set $\left\{ v_{1},\ldots,v_{\left|\mathfrak{S}_{M}\right|}\right\} $ contains all the vertices of the identified set $\Theta_{I}$.\footnote{Not all $v_{s}$ for $s=1,\ldots,\left|\mathfrak{S}_{M}\right|$ are vertices. Some may lie inside the convex hull of others.} This proves the following proposition.

propositionLet $\textrm{conv}$ denotes the convex hull. Then the vertex representations of $\Theta_{I}$ and $\Delta_{DD}$ are $\textrm{conv}\left\{ v_{1},\ldots,v_{\left|\mathfrak{S}_{M}\right|}\right\} $ and $\textrm{conv}\left\{ Ev_{1},\ldots,Ev_{\left|\mathfrak{S}_{M}\right|}\right\} $, respectively.\footnote{Here we derive the vertex representation of $\Delta_{DD}$ from that of $\Theta_{I}$. Given the analytic expression of the support function of $\Delta_{DD}$, we can also obtain the vertex representation of $\Delta_{DD}$ directly from its support function.}

Numerical Comparison with KMZ

KMZ study a specific collection of DD measures: $E^{\star}\theta^{\ast}=\left[\delta_{DD}^{\ast}\left(1,J\right),\ldots,\delta_{DD}^{\ast}\left(J-1,J\right)\right]$. Denote $\Delta_{DD}^{\star}$ as the identified set for $E^{\star}\theta^{\ast}$. They propose to construct $\Delta_{DD}^{\star}$ using its support function as \[ \Delta_{DD}^{\star}\equiv\left\{ \delta :p^{\top}\delta\leq\int\varPhi_{K}\left(p,x\right)d\mu_{X}\left(x\right)\textrm{ for all }p\in\mathbb{S}^{J-1}\right\} , \] where $\int\varPhi_{K}\left(p,x\right)d\mu_{X}\left(x\right)$ is the value of the support function evaluated at $p$. Based on their Proposition 10, KMZ propose a linear programming approach to computing $\varPhi_{K}\left(p,x\right)$. For any given $p$ and $x$, there are $2J$ variables and about $5J$ (equality and inequality) constraints in their linear program. In Appendix (ref), we discuss the time complexity of the approach in KMZ and our methods, and show that both our support function approach and vertex representation approach are significantly more computationally efficient than the KMZ method.

For the numerical comparison, we use the empirical application in KMZ, which studies the identified set for demographic disparity in mortgage credit decisions using HMDA (Home Mortgage Disclosure Act) dataset. Following KMZ, we focus on three racial groups: White, Black, and Asian and Pacific Islander (API). We consider three sets of proxy variables for race: geolocation (county) only, annual income only, and both geolocation and annual income. To construct $\Delta_{DD}$, we apply the vertex representation $\textrm{conv}\left\{ Ev_{1},\ldots,Ev_{\left|\mathfrak{S}_{M}\right|}\right\} $ as discussed in Proposition (ref). In practice, this representation requires only the conditional probabilities $\Pr\left(Y_{1}=0\mid x_{i}\right)$ and $\Pr\left(Y_{0}=a_{j}\mid x_{i}\right)$ for $j=1,\ldots,J$, where $x_{i}$ for $i=1,\ldots,n$ are observations of $X$. Since these values are provided in the KMZ code, we use them directly instead of recomputing them from the raw data.\footnote{The computed conditional probabilities in Sections (ref) and (ref) can be downloaded from https://github.com/CausalML/FairnessWithUnobservedProtectedClass.} For the KMZ method, we rely on their published code and set the number of direction vectors to 100. As both methods require the same conditional probabilities, the comparison of computational time considers only the steps after these probabilities are obtained. Additional details on this empirical analysis are provided in Appendix (ref).

figure[figure omitted — 271 chars of source]

Figure (ref) shows that our method and the approach in KMZ yield the same identified set. However, our method is substantially faster. The vertex representation approach computes the identified set for each proxy variable in under 70 milliseconds, whereas the KMZ method requires approximately 20 minutes. This corresponds to a speed improvement of over 15,000-fold relative to the KMZ method.

True-Positive Rate Disparity Measures in KMZ

Identified Sets for TPRD Measures---Partial Optimal Transport Characterization

The moment function is affine with $m_{a}\left(y_{0},x\right)$ being the $2J\times2J$ identity matrix and

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

Consequently, the identified set for $\theta^{*}$ is \[ \Theta_{I}=\left\{ \theta\in\Theta:\sum_{j=1}^{2J}t_{j}\theta_{j}\leq-\int\mathcal{KT}_{t^{\top}m_{b}}\left(\mu_{1\mid x},\mu_{0\mid x};x\right)d\mu_{X}\text{ for all }t\in\mathbb{S}^{2J}\right\} \] with support function for any $q\in\mathbb{S}^{2J}$ given by

equation[equation omitted — 161 chars of source]

Let $\Delta_{TPRD}$ denote the identified set for $K$ different TPRD measures represented by $G\left(\theta^{\ast}\right)$. We can compute $\Delta_{TPRD}$ as a non-linear map from $\Theta_{I}$: \[ \Delta_{TPRD}=\left\{ \delta=G\left(\theta\right):\theta\in\Theta_{I}\right\} =\left\{ \delta=G\left(\theta\right):q^{\top}\theta\leq h_{\Theta_{I}}\left(q\right)\textrm{ for all }q\in\mathbb{S}^{2J}\right\} , \] where $h_{\Theta_{I}}\left(\cdot\right)$ is defined in ((ref)). Because $\Theta_{I}$ is convex and the map $G$ is continuous, we know that $\Delta_{TPRD}$ is connected, which implies that the identified set for any single TPRD measure is a closed interval. In Appendix (ref), we derive the closed-form expressions for the lower and upper endpoints of the interval, extending the result in KMZ established for $J=2$.

For multiple TPRD measures, we need to compute $\mathcal{KT}_{q^{\top}m_{b}}\left(\cdot\right)$, for which we propose a novel computationally efficient algorithm in Section (ref). Our algorithm solves the reformulation of $\mathcal{KT}_{q^{\top}m_{b}}\left(\cdot\right)$ as the value function of a linear programming problem in ((ref)).

Given $q\equiv\left(q_{1},\ldots,q_{2J}\right)$, define $c\left(i,j\right)\equiv-q_{\left(1-i\right)J+j}$ for $i=0,1$ and $j=1,\ldots,J$. Because $Y_{1}$ and $Y_{0}$ are discrete random variables, we define, with a slight abuse of notation,

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

The COT cost $\mathcal{KT}_{t^{\top}m_{b}}\left(\cdot\right)$ can then be rewritten as the following optimization problem with the argument $\mu_{10\mid x}\left(i,1,j\right)$:

align[align omitted — 630 chars of source]

Note that the second constraint is an inequality rather than an equality because the COT cost $\mathcal{KT}_{q^{\top}m_{b}}\left(\cdot\right)$ does not involve $\mu_{10\mid x}\left(y_{1},y_{0}\right)$ for $y_{1}=\left(y_{1s},0\right)$. Consequently, ((ref)) is actually a conditional partial optimal transport, where $\mu_{0\mid x}\left(j\right)$ and $\mu_{1\mid x}\left(i,1\right)$ are identified from the sample information. Furthermore, since $i$ in the optimization problem ((ref)) takes two values only, we can always make $c\left(i,j\right)$ submodular by relabeling the index $j$. Without loss of generality, we will assume from now on that the cost function $c\left(i,j\right)$ in ((ref)) is submodular.

The following lemma shows that this reformulation of the partial optimal transport problem admits a solution whose support is monotone.

lemmaThere is a solution $\left\{ \mu_{10\mid x}^{\ast}\left(i,1,j\right):i=0,1\textrm{ and }j=1,\ldots,J\right\} $ to ((ref)) with monotone support. That is, for some $J^{\ast}\in\left\{ 1,\ldots,J\right\} $, $\mu_{10\mid x}^{\ast}\left(1,1,j\right)=0$ for all $j<J^{\ast}$ and $\mu_{10\mid x}^{\ast}\left(0,1,j\right)=0$ for all $j>J^{\ast}$.

Dual Rank Equilibration Algorithm

Exploiting the structure of the solution, we propose Algorithm (ref) to compute $\mathcal{KT}_{q^{\top}m_{b}}\left(\cdot\right)$ and call it Dual Rank Equilibration AlgorithM (DREAM). DREAM is built on an equivalent characterization of ((ref)) which allows it to use the ranks of $c\left(i,j\right)$ to equilibrate between minimizing the partial transport cost and respecting the constraints in ((ref)). The equilibration is done twice by first separately adjusting $\mu_{10\mid x}^{\ast}\left(0,1,j\right)$ and $\mu_{10\mid x}^{\ast}\left(1,1,j\right)$ across index $j$ and then jointly adjusting $\mu_{10\mid x}^{\ast}\left(0,1,j\right)$ and $\mu_{10\mid x}^{\ast}\left(1,1,j\right)$ for a fixed $j$---Dual Rank Equilibration AlgorithM. Appendix (ref) provides a detailed description of DREAM. DREAM involves only basic arithmetic, comparison, and logical operations, making it straightforward to implement directly in code. In fact, it can be executed by hand. In Appendix (ref), we provide theoretical time complexity analysis of DREAM and show that it is faster than directly solving the linear programming ((ref)), particularly when $J$ is large. \RestyleAlgo{ruled} \SetKwComment{Comment}{/* }{ */}

algorithm[algorithm omitted — 2,093 chars of source]

\RestyleAlgo{ruled} \SetKwComment{Comment}{/* }{ */}

algorithm[algorithm omitted — 483 chars of source]

Different from our approach, KMZ study the convex hull of the identified set for a collection of TPRD measures instead of the identified set itself. They focus on $\left[\delta_{TPRD}^{\ast}\left(1,J\right),\ldots,\delta_{TPRD}^{\ast}\left(J-1,J\right)\right]$, which is one example of our TPRD measures with $G^{\star}\left(\theta^{\ast}\right)\equiv\left[g_{1,J}\left(\theta^{\ast}\right),\ldots,g_{J-1,J}\left(\theta^{\ast}\right)\right]$. In their Proposition 11, KMZ propose to compute the support function of the convex hull at each direction from a complicated double optimization problem, where the inner maximization problem is a linear program and the outer maximization is a non-convex optimization problem, which is known to be NP-hard.

We applied DREAM and gurobi, a general-purpose linear programming solver, to the empirical application studied in KMZ.\footnote{We tried and failed to replicate the figures in KMZ using their approach due to the computational challenge of the NP-hard non-convex optimization involved.} The application investigates TPRD in Warfarin dosing with the ClinPGx/PharmGKB dataset used in IWPC_2009.\footnote{The dataset can be downloaded from https://www.clinpgx.org/downloads.} The analysis focuses on three protected attributes (White, Black, and Asian) and three specifications of proxy variables: genetic factors alone, current medications alone, and both genetic factors and current medications jointly. Because values of $\mu_{0\mid x}\left(\cdot\right)$ and $\mu_{1\mid x}\left(\cdot\right)$ in $(\ref{eq:Partial transport problem})$ for each observed $X$ have been computed in KMZ, we use them directly. During implementation, we first sample vectors $q_{1},\ldots,q_{N_{q}}$ uniformly from the $2J$-dimensional unit sphere, along with vectors $\theta_{1},\ldots,\theta_{N_{\theta}}$ from $\prod_{j=1}^{2J}\left[\theta_{j}^{L},\theta_{j}^{U}\right]$, where $\theta_{j}^{L}$ and $\theta_{j}^{U}$ are lower and upper bounds for each $\theta_{j}$ that are defined in Appendix Corollary (ref). Then, we construct the set $\widehat{\Theta}_{I}=\left\{ \theta\in\left\{ \theta_{1},\ldots,\theta_{N_{\theta}}\right\} :q^{\top}\theta\leq h_{\Theta_{I}}\left(q\right)\textrm{ for all }q=q_{1},\ldots,q_{N_{q}}\right\}$. Finally, we apply the mapping to each $\theta\in\widehat{\Theta}_{I}$ to obtain an approximation of $\Delta_{TPRD}$: $\widehat{\Delta}_{TPRD}=\left\{ \delta=G\left(\theta\right):\theta\in\widehat{\Theta}_{I}\right\}$.\footnote{Based on DREAM, the solution to ((ref)) depends only on the ranks of linear combinations of $c\left(i,j\right)$ for $i=0,1$ and $j=1,\ldots,J$ (or equivalently elements of the direction vector $q$). As a result, one can show that $\Theta_{I}$ is a polytope, and obtain its vertex representation by following the similar analysis in Section (ref). However, because the representation is complex and we still need to sample $\theta$ to construct $\widehat{\Delta}_{TPRD}$, we recommend using $\widehat{\Theta}_{I}$.} Appendix (ref) provides more detail.

figure[figure omitted — 365 chars of source]

Figure (ref) plots the identified sets obtained from DREAM, which are identical to the sets produced by the gurobi solver up to a negligible numerical error. The runtime comparison shows that our approach is more than twice as fast as the commercial linear programming solver.

Concluding Remarks

In this paper, we have developed the first unified approach to study the identified set for a finite-dimensional parameter in a general moment equality model with incomplete data.\footnote{ The methods developed in this paper could, in principle, be extended to the case of more than two datasets using results in pass2015multi. We leave extensions for future work.} Through several examples, we have demonstrated the advantages and simplicity of our approach. By exploring recent developments in optimal transport, our method often leads to equivalent, yet computationally more tractable, identified sets for specific models than the existing ones in the literature.

This paper provides the first step towards developing a complete set of econometric techniques for moment models under data combination. Building on the identification analysis in this paper, the next step is to construct valid estimation and inference in the general moment model ((ref)) with incomplete data. For Example (ref) without common covariate $X$, d2024linear develops estimation and inference for $\theta^{*}$. For the special case that $k=1$ and $\theta^{*}=\mathbb{E}\left[h\left(Y_{1},Y_{0},X)\right)\right]$, ji2023model and lin2025estimation construct estimation and inference using primal and dual formulation, respectively. We leave the general case for future work.

Acknowledgment

We thank St\'{e}phane Bonhomme, Xiaohong Chen, Marc Henry, Ruixuan Liu, and participants of the BIRS Workshop on Optimal Transport and Distributional Robustness in March 2024, 2025 IAER Econometrics Workshop, and 2025 China Annual Conference of the Chinese Economists Association; seminar participants in the statistics department at the University of Washington, economics departments at Johns Hopkins University and Yale University, as well as Amazon in October 2024 for useful feedback. Brendan Pass is pleased to acknowledge the support of the Natural Sciences and Engineering Research Council of Canada Discovery Grant numbers 04658-2018 and 04864-2024.