Source-linked AI summary
Selection and Estimation for Mixed Graphical Models
Shizhe Chen, Daniela Witten, Ali Shojaie
TL;DR
The paper studies estimating pairwise graph structure when nodes have different conditional-distribution types, a setting that requires attention to whether the conditionals define a joint density. It develops an exponential-family mixed-model estimator and theory, then reports recovery and edge-selection results, including improved selection over union and intersection rules.
Problem
High-dimensional graph estimation has largely considered nodes with the same conditional parametric form, while mixed node types require conditions for a well-defined joint density.
Source: §1 · p. 2, §2.2 · p. 4Method
The paper specifies node-wise exponential-family conditionals for different types and uses neighbourhood selection to estimate graph structure.
Source: §1 · p. 2Results
The Section 4.2 selection rules outperform union and intersection rules, and Gaussian-node neighbourhoods are easier to recover than those of the other three types.
Source: §4.2 · p. 11, §6.4 · p. 17Takeaways & Limitations
For edges joining Gaussian and non-Gaussian nodes, using estimates from the Gaussian node's regression is recommended because its neighbourhood is easier to recover.
Source: §4.2 · p. 11Compatibility imposes substantial parameter restrictions, especially for unbounded variables, and the conditionally specified mixed-model class is not closed under marginalization.
Source: §2.1 · p. 4, §7 · p. 17
Abstract
from arXiv · showhide
We consider the problem of estimating the parameters in a pairwise graphical model in which the distribution of each node, conditioned on the others, may have a different parametric form. In particular, we assume that each node's conditional distribution is in the exponential family. We identify restrictions on the parameter space required for the existence of a well-defined joint density, and establish the consistency of the neighbourhood selection approach for graph reconstruction in high dimensions when the true underlying graph is sparse. Motivated by our theoretical results, we investigate the selection of edges between nodes whose conditional distributions take different parametric forms, and show that efficiency can be gained if edge estimates obtained from the regressions of particular nodes are used to reconstruct the graph. These results are illustrated with examples of Gaussian, Bernoulli, Poisson and exponential distributions. Our theoretical findings are corroborated by evidence from simulation studies.
1 Introduction
The paper addresses graph estimation when node types differ, extending high-dimensional graphical-model work beyond settings where all conditional distributions share one parametric form. It develops a parametric mixed model and theory for exponential-family conditionals, including compatibility requirements and edge-selection guidance.
1 Introduction
- The task is to reconstruct conditional-dependence edges from observations, including the high-dimensional setting where the number of variables exceeds the sample size.The graph may have p(p −1)/2 candidate edges to estimate from n observations.Source: §1 · p. 2
- When node types are known, parametric methods can be more efficient than flexible semiparametric or nonparametric approaches, motivating a parametric mixed model.The introduction notes that nonparametric methods are often less efficient than parametric counterparts in this setting.Source: §1 · p. 2
- The proposed parametric mixed graphical model allows nodes of different types, with each conditional distribution drawn from the exponential family and node types assumed known.The paper contrasts this setting with earlier approaches in which nodes share one parametric form.Source: §1 · p. 2
- The paper extends beyond mixed conditional Gaussian models and settings limited to two node types, allowing graphs with more than two types and giving specific compatibility requirements for common distributions.The cited parallel work treats exponential-family conditionals but limits the graph to two node types.Source: §1 · pp. 2–3
2 A Model for Mixed Data
The model specifies each node's conditional distribution using an exponential-family form and combines these conditionals into a pairwise graph. The section characterizes when such conditionals correspond to a joint distribution, including distribution-specific restrictions and the possibility of recovering graph structure without compatibility.
2.1 Conditionally-Specified Models for Mixed Data
- Conditionals are specified node by node because calculating the joint model's log-partition function is often intractable, but a conditional specification need not define a valid joint distribution.The model uses node potentials and symmetric pairwise edge potentials, with nonedges assigned zero parameters.Source: §2.1 · p. 3
- The model admits Gaussian, Bernoulli, Poisson, and exponential node conditionals within the exponential family, while the mixed class is not closed under marginalization.Integrating Bernoulli nodes from a graph containing Gaussian and Bernoulli nodes can yield a Gaussian mixture outside the exponential family.Source: §2.1 · pp. 3–4
2.2 Compatibility of Conditionally-Specified Models
- Compatibility means a shared generating function exists for the conditionals; strong compatibility additionally requires that function to be a density.For the specified exponential-family conditionals, symmetric edge parameters yield compatibility, and integrability yields strong compatibility.Source: §2.2 · pp. 4–5
- For the four common conditional families, compatibility and strong compatibility impose parameter restrictions, including prohibiting Gaussian–exponential and Gaussian–Poisson edges.For Poisson–Poisson edges, compatibility imposes no restriction, whereas strong compatibility requires negative edge potentials.Source: §2.2 · p. 5, Table 1 · p. 6
- The proposed estimator and theory also apply to other conditional densities of the specified form, and graph structure can be estimated even when compatibility requirements fail.The necessary-and-sufficient compatibility and strong-compatibility conditions for the four listed families are summarized in Table 1.Source: §2.2 · pp. 5–6, Table 1 · p. 6
3 Estimation Via Neighbourhood Selection
The section develops neighbourhood selection by maximizing penalized conditional likelihoods node by node, then describes coefficient weighting and BIC tuning. Asymmetric estimates across nodewise regressions can improve edge estimation between different node types.
3.1 Estimation
- Nodewise penalized conditional likelihood estimates each node’s neighbourhood from the nonzero coefficients in its fitted regression.Source: §3.1 · p. 6
- To accommodate variables on different scales, the ℓ1 penalty can be weighted using the empirical standard errors of the corresponding variables.Source: §3.1 · p. 6
- Although true edge parameters are symmetric, nodewise estimates can be asymmetric, and this asymmetry can be exploited for more efficient cross-type edge estimates.Source: §3.1 · p. 7
3.2 Tuning
- BIC selects the tuning parameter, with a separate value chosen for each node type by minimizing BIC summed over nodes of that type.The criterion combines nodewise log-likelihood with a penalty proportional to the number of nonzero coefficients.Source: §3.2 · p. 7
4 Neighbourhood Recovery and Selection With Strongly Compatible Conditional Distributions
Under strong compatibility and additional assumptions, neighbourhood selection can recover true node neighbourhoods, with graph sparsity delimiting the guarantee. The section then uses type-dependent recovery bounds to guide which endpoint's regression estimates an edge.
4.1 Neighbourhood Recovery
- Under strong compatibility and assumptions 1–7, neighbourhood selection exactly recovers a node’s true neighbourhood with probability approaching one exponentially in n.The guarantee is meaningful when the graph is sparse, with maximum degree d = o(n).Source: §4.1 · pp. 7, 9, 10
- As degree d increases, minimum edge-strength and sample-size requirements rise while the feasible tuning range shrinks, so the theorem requires d = o(n).Source: §4.1 · p. 9
- For Gaussian nodes, κ2 = 1 and κ3 = 0 relax derivative conditions and yield exact neighbourhood recovery under assumptions 1–5.Source: §4.1 · p. 10
4.2 Combining Neighbourhoods to Estimate the Edge Set
- For Gaussian–non-Gaussian edges, use the Gaussian-side neighbourhood estimate: it needs less sample size, allows broader tuning ranges, and generally has higher recovery probability.Source: §4.2 · p. 11
- When the comparison conditions are not met, there is no clear preference for which neighbourhood to use; same-type endpoints likewise offer no clear reason to favor one estimate.Source: §4.2 · pp. 10, 12
- For different-type non-Gaussian edges, choose the endpoint regression with the highest estimated correct-recovery probability, using consistent estimates of unknown bounds.The rules for choosing a neighbourhood are summarized in Table 2.Source: §4.2 · pp. 11–12, Table 2 · p. 12
5 Neighbourhood Recovery and Selection with Partially-Specified Models
When only some node conditionals follow form (3), the paper establishes recovery of those nodes’ neighbourhoods under revised assumptions. An edge is covered when at least one endpoint has a conditional density exactly of that form.
- If a node’s conditional density has form (3) and Assumptions 1–3 and 5–8 hold, the estimator exactly recovers its true neighbourhood with the stated probability bound.Source: §5 · p. 13
- Recovery of an edge between two nodes requires form (3) for only one endpoint, allowing relationships beyond those covered by Table 1, such as a Poisson node linked to a real-valued node.Source: §5 · p. 13
- The guarantee covers an edge only when at least one endpoint’s conditional density is exactly form (3); extending it to approximate conditionals is left for future work.Source: §5 · p. 13
6 Numerical Studies
The simulations examine neighbourhood recovery and edge selection in mixed Gaussian-Bernoulli and Poisson-Bernoulli graphs. They show gains from the proposed selection rules and from estimating mixed edges using regressions suited to the node types.
6.1 Data Generation
- The simulations generate sparse two-type graphs with same-type and cross-type edges, then sample from node conditionals using a Gibbs sampler.Gaussian-Bernoulli and Poisson-Bernoulli models use m = p/2 nodes of each type; samples are taken 500 iterations apart after a 3000-iteration burn-in.Source: §6.1 · p. 13, Figure 1 · p. 14
6.2 Probability of Successful Neighbourhood Recovery
- Recovery probability rises to 1 beyond the required scaled-sample-size threshold, and Gaussian-node regressions need fewer samples than Bernoulli-node regressions for a given recovery probability.The plots compare recovery probability against n/{3 log(p)} and separate estimates by node type regressed.Source: §6.2 · p. 14, Figure 2 · p. 15
6.3 Comparison to Competing Approaches
- The Gaussian-Bernoulli comparison evaluates seven approaches, including mixed-model methods and methods that ignore mixed node types.The computationally intensive Lee and Hastie method is run on 250 data sets from 50 graphs rather than 2000 data sets from 100 graphs.Source: §6.3 · p. 15
- The proposal generally outperforms the alternatives on the Gaussian-Bernoulli graph, with BIC-selected tuning yielding 52% precision and 95% recall.Gaussian graphical-model neighbourhood selection is closest to the proposal; Ising selection suffers from dichotomizing Gaussian variables, and graphical lasso performs poorly under its Gaussian assumption.Source: §6.3 · p. 16, Figure 3 · p. 16
6.4 Application of Selection Rules for Mixed Graphical Models
- Section 4.2’s selection rules outperform union and intersection rules for Poisson-Bernoulli edges; the graphical random forest leads only when few edges are estimated.Using estimated parameters produces a curve almost identical to using true parameters.Source: §6.4 · p. 17, Figure 4 · p. 17
7 Discussion
The discussion identifies compatibility restrictions and the pairwise interaction form as constraints on the model. It points to truncation and nonlinear edge potentials as possible ways to address these limits.
7 Discussion
- Compatibility and strong-compatibility requirements impose stringent restrictions, especially when modeling unbounded variables.The authors suggest truncation may circumvent some of these restrictions.Source: §7 · p. 17
- The pairwise interaction x_s x_t is a second-order approximation of the true edge potential.The authors describe semi-parametric penalized regressions with nonlinear edge potentials as a way to relax this assumption.Source: §7 · p. 18
- The paper provides R code for replicating its numerical results.The code is available at the stated GitHub repository.Source: §7 · p. 18
A.1 A Proof for Proposition 1
The proof derives the joint density compatible with the specified node conditionals by expanding interactions and showing that higher-order terms vanish. Symmetric edge parameters yield compatibility when the required integrability holds.
A.1 A Proof for Proposition 1
- Conditional densities of the specified form are compatible when edge parameters are symmetric, provided the relevant function is integrable with respect to each node variable.The proof relates this integrability to that of the conditional densities.Source: §A.1 · p. 21
- The proof constructs a log-density ratio and decomposes it into interactions of different orders to recover the joint-density form.It sets variables to zero to isolate node and pairwise interactions.Source: §A.1 · p. 21
- Setting all but the relevant variables to zero shows that third- and higher-order interactions vanish, yielding the pairwise form.The second-order interaction is simplified using the same isolation argument.Source: §A.1 · pp. 21–22
A.2 A Proof for Lemma 1
The proof derives necessary and sufficient restrictions for compatibility of the node-conditional distributions, considering non-degenerate conditionals. The restrictions depend on node types and edge potentials.
A.2 A Proof for Lemma 1
- For an exponential node, its rate must remain negative: interactions with unbounded Poisson or exponential nodes must be non-positive, interactions with Gaussian nodes must be zero, and Bernoulli interactions are bounded by the baseline rate.The Bernoulli interaction magnitudes must sum to less than −α1s.Source: §A.2 · p. 22
- Gaussian nodes require a negative-definite ΘJJ, while Bernoulli and Poisson conditionals impose no restrictions on their conditional parameters.The proof also states that Bernoulli nodes require no restrictions on edge potentials for a density.Source: §A.2 · pp. 22–23
- The restrictions in Table 1 are sufficient for strong compatibility: Gaussian nodes are isolated from Poisson and exponential nodes, and the remaining conditional densities are integrable under the stated restrictions.The proof uses the negative definiteness of ΘJJ for Gaussian integrability and bounds the joint density by an integrable no-interaction model for Poisson and exponential nodes.Source: §A.2 · pp. 22–23
A.3 A Proof for Theorem 1
The proof uses a primal-dual witness argument to establish exact neighbourhood recovery. Under its assumptions, the required events hold with high probability as sample size grows.
A.3 A Proof for Theorem 1
- The proof follows the primal-dual witness method, testing the sub-gradient condition of an oracle estimator to establish properties of the ℓ1-penalized estimator.It assumes familiarity with the method and cites Ravikumar et al. (2011) and Yang et al. (2012).Source: §A.3 · pp. 23–24
- The oracle construction sets coefficients for non-neighbours to zero, solves the restricted optimization, and then estimates the remaining dual variables from the sub-gradient condition.The proof verifies that the resulting pair satisfies the sub-gradient condition.Source: §A.3 · p. 24
- The constructed primal-dual pair recovers the true neighbourhood exactly when the required events hold; those events occur with high probability for large sample size.The proof also establishes sign consistency for all edges under the stated minimum-signal condition.Source: §A.3 · pp. 24–26
A.4 A Proof for Lemma 3
The proof bounds the concentration quantities needed for the theorem using exponential-family properties and probability inequalities. It applies the resulting bounds under specified parameter and regularization conditions.
A.4 A Proof for Lemma 3
- The concentration argument uses the exponential-family moment-generating function and a second-order Taylor expansion.The proof applies these steps to independent samples.Source: §A.4 · p. 26
- A Chernoff bound and the Bonferroni inequality are used to bound the probability of the required events.The argument sets δ = aλn/(8 −4a).Source: §A.4 · p. 27
- The argument requires λn to be large enough relative to √(log(2p)/n) and no larger than a bound determined by κ2, δ2, M, and a.These bounds ensure the parameter v used in the Chernoff argument is at most M/2.Source: §A.4 · p. 27
A.5 A Proof for Lemma 4
The appendix bounds the oracle-estimator error and residual, extends the argument to a Gaussian-node corollary, and describes data-generation adjustments used to satisfy compatibility conditions.
A.5 A Proof for Lemma 4
- For Lemma 4, convexity of F(u) and a Taylor expansion yield an oracle-estimator error bound of 10λn√(d + 1)/Λ1 and a bound on the residual term.The proof establishes the residual bound under its stated conditions.Source: §A.5 · pp. 27–30
A.6 A Proof for Corollary 1
- For the Gaussian-node corollary, the proof obtains a radius B = 5(d + 1)1/2λn/Λ1 and then uses Lemma 5 to verify the theorem’s conditions.It subsequently drops the requirement of ξ1 in Lemma 5’s condition.Source: §A.6 · pp. 30–31
A.7 Additional Details of Data-Generation Procedure
- The data-generation procedure adjusts the Gaussian-node interaction matrix to be negative definite with eigenvalues no larger than −0.1, then standardizes its diagonal to −1; Poisson–Poisson edge potentials are made negative.Source: §A.7 · p. 31