Conformal Prediction for Dyadic Regression Under Complex Missingness
Reviewed by Pith T0 review T1 audit T2 compute T3 formal T4 kernel 2026-06-27 11:04 UTCgrok-4.3pith:46MCYOEKrecord.jsonopen to challenge →
The pith
A weighted conformal procedure achieves asymptotic validity for dyadic regression under a nonparametric graphon model for missingness.
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
Core claim
The authors establish asymptotic validity of a weighted conformal prediction procedure for missing elements in dyadic data under a nonparametric graphon model for the missingness mechanism. They also provide the first formal proof of asymptotic conditional validity for weighted conformal prediction under a missing-not-at-random assumption, for both continuous and discrete responses.
What carries the argument
A novel bijection argument constructing an explicit measure-preserving correspondence between events to handle random subsets of the index set, together with the weighted conformal procedure whose weights are derived from the graphon model.
If this is right
- The full, split, row-column, and selective conformal procedures achieve finite-sample validity under the stated invariance conditions for jointly exchangeable arrays.
- Asymptotic coverage guarantees apply to the weighted procedure when used on missing elements.
- Conditional validity holds for both continuous and discrete response variables.
- The methods apply directly to network data examples, both synthetic and real.
Where Pith is reading between the lines
- If the graphon model matches the true missingness process in a given network, these intervals can be used to quantify uncertainty without invoking a missing-at-random assumption.
- The bijection technique may apply to other array-valued or matrix-valued sampling problems that involve random subsets.
- Empirical tests of whether observed missingness patterns are consistent with a graphon structure would be a direct way to assess applicability.
Load-bearing premise
The missingness mechanism follows a nonparametric graphon model that permits derivation of the weights and the coverage guarantee.
What would settle it
An empirical check in which the missingness probabilities are generated from a non-graphon mechanism and the observed coverage rate of the weighted intervals falls materially below the nominal level would refute the asymptotic validity claim.
Figures
read the original abstract
We develop a framework for conformal prediction in dyadic regression problems under complex missingness mechanisms. At the theoretical level, we develop general technical tools for establishing finite-sample validity of conformal prediction under distributional invariance conditions weaker than exchangeability. A key result handles the case where the sample itself is a random subset of the index set, a setting not covered by existing theory, via a novel bijection argument that constructs an explicit measure-preserving correspondence between events. In addition, we propose conformal prediction procedures for jointly exchangeable arrays, including full conformal, split conformal, a row-column approach exploiting similarities within rows and columns, and a selective conformal procedure achieving mask-conditional validity. For missing elements, we establish asymptotic validity of a weighted conformal procedure under a nonparametric graphon model for the missingness mechanism. We further establish conditional validity results for both continuous and discrete responses; to the best of our knowledge, this is the first formal proof of asymptotic conditional validity for weighted conformal prediction under a missing-not-at-random assumption. The proposed methods are illustrated on synthetic and real network data.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper develops conformal prediction methods for dyadic regression under complex missingness. It introduces general tools for finite-sample validity under distributional invariance weaker than exchangeability, including a novel bijection argument for cases where the sample is a random subset of the index set. Procedures are proposed for jointly exchangeable arrays (full, split, row-column, and selective conformal). For missing elements under MNAR, it establishes asymptotic validity of a weighted conformal procedure under a nonparametric graphon model for the missingness mechanism and claims the first formal proof of asymptotic conditional validity for weighted conformal prediction under MNAR, illustrated on synthetic and real network data.
Significance. If the central results hold, the work provides new technical tools for conformal prediction beyond exchangeability and extends coverage guarantees to MNAR settings in dyadic data via graphon modeling. The bijection argument for random subsets and the conditional validity claim under MNAR are potentially significant contributions for network applications, though the nonparametric estimation requirements for the weights remain a key point of evaluation.
major comments (2)
- [Abstract / theorem on asymptotic validity under graphon model] The asymptotic validity result for the weighted conformal procedure (abstract and the relevant theorem establishing coverage under the graphon model): the claim relies on plug-in weights derived from the nonparametric graphon estimator, but the manuscript does not state regularity conditions (e.g., Hölder smoothness, lower bounds away from zero, or uniform convergence rates) that would ensure the estimation error is small enough for the asymptotic coverage to hold, particularly for the mask-conditional result.
- [Section establishing conditional validity results] The conditional validity claim for weighted conformal prediction under MNAR (abstract): while the bijection argument is used for the invariance, the transition from finite-sample to asymptotic conditional validity under the fully nonparametric graphon appears to require additional justification that the estimated weights preserve the necessary conditional independence or invariance properties at the required rate.
minor comments (2)
- [Introduction / notation section] Notation for the dyadic array and the missingness indicators could be clarified with an explicit definition early in the paper to aid readability.
- [Section on row-column conformal procedure] The description of the row-column approach would benefit from a small illustrative example or diagram showing how row and column similarities are exploited.
Simulated Author's Rebuttal
We thank the referee for the thoughtful and detailed report. The two major comments correctly identify gaps in the regularity conditions and justification for the asymptotic results. We will revise the manuscript to address both points by adding the necessary assumptions and expanding the proofs.
read point-by-point responses
-
Referee: [Abstract / theorem on asymptotic validity under graphon model] The asymptotic validity result for the weighted conformal procedure (abstract and the relevant theorem establishing coverage under the graphon model): the claim relies on plug-in weights derived from the nonparametric graphon estimator, but the manuscript does not state regularity conditions (e.g., Hölder smoothness, lower bounds away from zero, or uniform convergence rates) that would ensure the estimation error is small enough for the asymptotic coverage to hold, particularly for the mask-conditional result.
Authors: We agree that explicit regularity conditions are required for the plug-in estimator to yield the claimed asymptotic coverage. In the revision we will state Hölder smoothness of the graphon, a uniform lower bound away from zero on the missingness probabilities, and a uniform convergence rate for the nonparametric estimator. These assumptions will be added to the theorem statement together with a short argument showing that the resulting weight error is o_p(1) uniformly, which is sufficient for the coverage result to hold (including the mask-conditional version). revision: yes
-
Referee: [Section establishing conditional validity results] The conditional validity claim for weighted conformal prediction under MNAR (abstract): while the bijection argument is used for the invariance, the transition from finite-sample to asymptotic conditional validity under the fully nonparametric graphon appears to require additional justification that the estimated weights preserve the necessary conditional independence or invariance properties at the required rate.
Authors: The referee correctly notes that the passage from finite-sample invariance (via the bijection) to asymptotic conditional validity needs further detail once weights are estimated. We will expand the proof in the relevant section to show that, under the regularity conditions added in response to the first comment, the difference between the oracle and plug-in weights vanishes at a rate that preserves the conditional coverage property asymptotically. This will be done by controlling the total variation distance between the two weighted distributions conditionally on the mask. revision: yes
Circularity Check
No significant circularity; derivation introduces independent technical tools and relies on external modeling assumptions.
full rationale
The paper develops novel technical tools including a bijection argument for finite-sample validity under weaker distributional invariance conditions than exchangeability, and proposes multiple conformal procedures (full, split, row-column, selective) for jointly exchangeable arrays. Asymptotic validity for the weighted procedure is established under the nonparametric graphon model for missingness as an explicit modeling assumption used to derive weights and coverage guarantees, rather than any fitted parameter or self-referential reduction. No steps in the provided abstract or described claims reduce by construction to inputs via self-definition, fitted predictions, or load-bearing self-citations; the central results appear self-contained with independent mathematical content.
Axiom & Free-Parameter Ledger
axioms (2)
- domain assumption distributional invariance conditions weaker than exchangeability
- domain assumption nonparametric graphon model for the missingness mechanism
Reference graph
Works this paper leans on
-
[1]
Abbas, K., Abbasi, A., Dong, S., Niu, L., Yu, L., Chen, B., Cai, S.-M., and Hasan, Q. (2021). Application of network link prediction in drug discovery. BMC Bioinformatics, 22(1):187
2021
-
[2]
Aldous, D. J. (1981). Representations for partially exchangeable arrays of random variables.Journal of Multivariate Analysis, 11:581–598
1981
-
[3]
Angelopoulos, A. N. and Bates, S. (2021). A gentle introduction to conformal prediction and distribution-free uncertainty quantification. 32
2021
-
[4]
L., Marlowe, F
Apicella, C. L., Marlowe, F. W., Fowler, J. H., and Christakis, N. A. (2012). Social networks and cooperation in hunter-gatherers.Nature, 481(7382):497– 501
2012
-
[5]
Bao, Y., Huo, Y., Ren, H., and Zou, C. (2024). Selective conformal inference with false coverage-statement rate control.Biometrika, page asae010
2024
-
[6]
F., Cand` es, E
Barber, R. F., Cand` es, E. J., Ramdas, A., and Tibshirani, R. J. (2023). Con- formal prediction beyond exchangeability.The Annals of Statistics, 51(2):816 – 845
2023
-
[7]
Barber, R. F. and Tibshirani, R. J. (2026). Unifying different theories of conformal prediction.Electronic Journal of Statistics, 20(1):1428–1474
2026
-
[8]
Berlusconi, G., Calderoni, F., Parolini, N., Verani, M., and Piccardi, C. (2016). Link prediction in criminal networks: A tool for criminal intelligence analysis.PLOS ONE, 11(4):1–21
2016
-
[9]
and Chatterjee, S
Bhattacharya, S. and Chatterjee, S. (2022). Matrix completion with data- dependent missingness probabilities.IEEE Transactions on Information The- ory, 68(10):6762–6773
2022
-
[10]
Brody, S., Alon, U., and Yahav, E. (2022). How attentive are graph atten- tion networks? InInternational Conference on Learning Representations
2022
-
[11]
Chatterjee, S. (2015). Matrix estimation by universal singular value thresh- olding.The Annals of Statistics, 43(1):177–214
2015
-
[12]
Chernozhukov, V., Galichon, A., Hallin, M., and Henry, M. (2017). Monge– Kantorovich depth, quantiles, ranks and signs.The Annals of Statistics, 45(1):223 – 256
2017
-
[13]
Chernozhukov, V., W¨ uthrich, K., and Zhu, Y. (2021). Distributional con- formal prediction.Proceedings of the National Academy of Sciences, 118(48)
2021
-
[14]
Commenges, D. (2003). Transformations which preserve exchangeability and application to permutation tests.Journal of Nonparametric Statistics, 15(2):171–185
2003
-
[15]
and Dempsey, W
Crane, H. and Dempsey, W. (2018). Edge exchangeable models for interaction networks.Journal of the American Statistical Association, 113(523):1311–1326. PMID: 30467447
2018
-
[16]
Cuturi, M. (2013). Sinkhorn distances: lightspeed computation of optimal transport. InProceedings of the 27th International Conference on Neural Information Processing Systems - Volume 2, page 2292–2300, Red Hook, NY, USA. Curran Associates Inc
2013
-
[17]
Davezies, L., D’Haultfœuille, X., and Guyonvarch, Y. (2021). Empirical process results for exchangeable arrays.The Annals of Statistics, 49(2):845 – 862. 33
2021
-
[18]
and Verducci, J
Dean, A. and Verducci, J. (1990). Linear transformations that preserve majorization, Schur concavity, and exchangeability.Linear algebra and its applications, 127:121–138
1990
-
[19]
and Yu, M
Dobriban, E. and Yu, M. (2024). Symmpi: Predictive inference for data with group symmetries
2024
-
[20]
Du, W., Ma, W., Xia, D., Zhang, Y., and Zhou, W. (2025). Conformal link prediction with false discovery rate control
2025
-
[21]
and Wang, Y
Fang, K.-T. and Wang, Y. (1993). Number-theoretic methods in statistics
1993
-
[22]
and Klopp, O
Gaucher, S. and Klopp, O. (2021). Maximum likelihood estimation of sparse networks with missing observations.Journal of Statistical Planning and Inference, 215:299–329
2021
-
[23]
and Sen, B
Ghosal, P. and Sen, B. (2022). Multivariate ranks and quantiles using optimal transport: Consistency, rates and nonparametric testing.The Annals of Statistics, 50(2):1012 – 1037
2022
-
[24]
Graham, B. S. (2020). Chapter 2 - dyadic regression. InThe Econometric Analysis of Network Data, pages 23–40. Academic Press
2020
-
[25]
S., Niu, F., and Powell, J
Graham, B. S., Niu, F., and Powell, J. L. (2024). Kernel density estimation for undirected dyadic data.Journal of Econometrics, 240(2):105336
2024
-
[26]
F., and Ma, C
Gui, Y., Barber, R. F., and Ma, C. (2023). Conformalized matrix comple- tion
2023
-
[27]
Hallin, M., del Barrio, E., Cuesta-Albertos, J., and Matr´ an, C. (2021). Dis- tribution and quantile functions, ranks and signs in dimension d: A measure transportation approach.The Annals of Statistics, 49(2):1139 – 1165
2021
-
[28]
Hlavka, Z., Hlubinka, D., and Hudecova, S. (2023). Multivariate quantile- based permutation tests with application to functional data
2023
-
[29]
Hoover, D. N. (1979).Relations on probability spaces arrays of random variables. Institute for Advanced Study,RI
1979
-
[30]
Huang, K., Jin, Y., Candes, E., and Leskovec, J. (2023). Uncertainty quantification over graph with conformalized graph neural networks
2023
-
[31]
Huang, K. H. (2026). Data augmented bootstrap: Unifying confidence interval construction by approximate invariance
2026
-
[32]
and Candes, E
Jin, Y. and Candes, E. J. (2023). Selection by prediction with conformal p-values.Journal of Machine Learning Research, 24(244):1–41
2023
-
[33]
Klein, M., Bethune, L., Ndiaye, E., and Cuturi, M. (2025). Multivariate conformal prediction using optimal transport. 34
2025
-
[34]
Klopp, O., Tsybakov, A., and Verzelen, N. (2017). Oracle inequalities for network models and sparse graphon estimation.Annals of Statistics, 45:316– 354
2017
-
[35]
Kuchibhotla, A. K. (2021). Exchangeability, conformal prediction, and rank tests
2021
-
[36]
and Bickel, P
Lei, L. and Bickel, P. J. (2020). An assumption-free exact test for fixed- design linear models with exchangeable errors.Biometrika, 108(2):397–412
2020
-
[37]
Liang, Z., Xie, T., Tong, X., and Sesia, M. (2026). Structured confor- mal inference for matrix completion with applications to group recommender systems.Journal of the American Statistical Association, 0(ja):1–25
2026
-
[38]
Lunde, R. (2023). On the validity of conformal prediction for network data under non-uniform sampling
2023
-
[39]
Lunde, R., Levina, E., and Zhu, J. (2025). Conformal prediction for network-assisted regression.Journal of the American Statistical Association, 120(551):1633–1644
2025
-
[40]
Luo, R., Nettasinghe, B., and Krishnamurthy, V. (2023). Anomalous edge detection in edge exchangeable social network models. InProceedings of the Twelfth Symposium on Conformal and Probabilistic Prediction with Applica- tions, volume 204, pages 287–310. PMLR
2023
-
[41]
Marandon, A. (2024). Conformal link prediction for false discovery rate control.TEST, 33(4):1062–1083
2024
-
[42]
Marrs, F. W. and Fosdick, B. K. (2023). Regression of binary network data with exchangeable latent errors.Network Science, 11(3):502–535
2023
-
[43]
Mart´ ınez, V., Berzal, F., and Cubero, J.-C. (2016). A survey of link pre- diction in complex networks.ACM Comput. Surv., 49(4)
2016
-
[44]
McCallum, A. (2000). Cora dataset
2000
-
[45]
Oneal, J. R. and Russett, B. (1999). The kantian peace: The pacific benefits of democracy, interdependence, and international organizations, 1885–1992. World Politics, 52(1):1–37
1999
-
[46]
and Kuchibhotla, A
Paul, M. and Kuchibhotla, A. K. (2026). On a probability inequality for or- der statistics with applications to bootstrap, conformal prediction, and more
2026
-
[47]
Qi, Y., Bar-Joseph, Z., and Klein-Seetharaman, J. (2006). Evaluation of different biological data and computational classification methods for use in protein interaction prediction.Proteins, 63(3):490–500
2006
-
[48]
M., Romano, J
Ritzwoller, D. M., Romano, J. P., and Shaikh, A. M. (2025). Randomization inference: Theory and applications. 35
2025
-
[49]
Romano, Y., Sesia, M., and Candes, E. (2020). Classification with valid and adaptive coverage. InAdvances in Neural Information Processing Systems, volume 33, pages 3581–3591. Curran Associates, Inc
2020
-
[50]
and Zhang, Y
Shao, M. and Zhang, Y. (2023). Distribution-free matrix prediction under arbitrary missing pattern
2023
-
[51]
Conditional Predictive Inference for General Structured Data with Group Symmetries
Shen, Y. and Yu, M. (2026). Conditional predictive inference for general structured data with group symmetries.arXiv preprint arXiv:2605.17934
work page internal anchor Pith review Pith/arXiv arXiv 2026
-
[52]
J., Foygel Barber, R., Candes, E., and Ramdas, A
Tibshirani, R. J., Foygel Barber, R., Candes, E., and Ramdas, A. (2019). Conformal prediction under covariate shift. InAdvances in Neural Informa- tion Processing Systems, volume 32. Curran Associates, Inc
2019
-
[53]
Tinbergen, J. (1963). Shaping the world economy.The International Ex- ecutive, 5(1):27–30
1963
-
[54]
(2005).Algorithmic Learning in a Random World
Vovk, V., Gammerman, A., and Shafer, G. (2005).Algorithmic Learning in a Random World. Springer, New York
2005
-
[55]
Yunpeng Zhao, Yun-Jhong Wu, E. L. and Zhu, J. (2017). Link prediction for partially observed networks.Journal of Computational and Graphical Statistics, 26(3):725–733
2017
-
[56]
Zaffran, M., Dieuleveut, A., Josse, J., and Romano, Y. (2023). Confor- mal prediction with missing values. InProceedings of the 40th International Conference on Machine Learning, volume 202, pages 40578–40604. PMLR
2023
-
[57]
H., Antonelli, S., and Bojchevski, A
Zargarbashi, S. H., Antonelli, S., and Bojchevski, A. (2023). Conformal prediction sets for graph neural networks. InProceedings of the 40th Inter- national Conference on Machine Learning
2023
-
[58]
without network covariate
Zhang, Y., Levina, E., and Zhu, J. (2017). Estimating network edge prob- abilities by neighbourhood smoothing.Biometrika, 104(4):771–783. A Appendix A.1 Proofs for Section 5: Beyond Exchangeability Before we prove Proposition 5.1, we make an observation that simplifies argu- ments considerably. Proposition A.1.Let(W 1, . . . , Wn)be a vector of random var...
2017
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.