REVIEW 2 major objections 5 minor 79 references
Learning Causality for Modern Machine Learning
T0 review · 2 major / 5 minor · reviewed 2026-08-07 · deepseek-v4-flash
Pith's one-line read Causal subgraph learning unlocks OOD, explainable, and attack-safe GNNs
desk verdict A thesis compiling six peer-reviewed papers into a causal-invariance narrative for graph learning; the GALA identification theorem has an unproven and likely false implication, making the 'provable' claim conditional on an unstated assumption. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The load-bearing object is the invariant subgraph $G_c$ generated by the latent causal part $C$ under a structural causal model, together with the information-theoretic property that $G_c$ maximizes the conditional mutual information of extracted subgraphs across environments while minimizing the spurious counterpart. The thesis operationalizes this through contrastive objectives (CIGA), an environment-assistant split of positive and negative graphs (GALA), the subgraph multilinear extension (SubMT) for interpretability (GMT), and homophily distribution regularization for adversarial unnoticeability (HAO). The optimization methods PAIR and FeAT target the ERM conflict via Pareto balancing of per-environment risks and feature-augmented training rounds.
What would settle it
Build the two-piece graph worlds $E_{tr}=\{(0.2,0.1),(0.2,0.3)\}$ and $E_{tr}'$ with the roles of $G_c$ and $G_s$ swapped so that $P(Y,G)$ is identical but the true causal subgraph differs; train CIGA and GALA on both. If either method outputs the same extracted subgraph in both worlds, the paper's no-free-graph-OOD-lunch result (Cor. 3.3.6) is falsified, while failure in one world confirms the unidentifiability claim.
Extended reading notes
Core claim
On its own terms, the thesis establishes that the independent-causal-mechanism principle can be lifted from tabular and image data to graphs: assume a graph $G$ is assembled from an invariant subgraph $G_c$ (controlled by latent $C$) and a spurious subgraph $G_s$ (affected by environment $E$); then a GNN that predicts the label $Y$ only from $G_c$ is invariant to distribution shifts. CIGA formalizes this and proves that an objective maximizing intra-class mutual information of extracted subgraphs identifies the true $G_c$ under invertibility and equal-environment assumptions, producing an invariant GNN. GALA strengthens the theory by showing that without environment labels, environment generation and inference both fail unless variation sufficiency and variation consistency hold, and gives a no-free-lunch result; under those assumptions, an ERM-trained environment assistant that separates correctly- versus incorrectly-predicted graphs enables provable identification. The interpretability chapter shows existing attention-based XGNNs fail to approximate the subgraph multilinear extension because they push a soft subgraph through a nonlinear GNN, and proposes GMT, which approximates SubMT via random subgraph sampling or a linearized variant and satisfies counterfactual fidelity. The adversarial-robustness chapter proves graph injection attacks can emulate and exceed modification attacks because of their flexibility, but that flexibility destroys homophily and makes them easily pruned; adding homophily unnoticeability restores their power. The optimization chapters, PAIR and FeAT, show the invariance objective conflicts with ERM and that Pareto balancing or feature augmentation with additional training rounds recovers causal invariance and richer feature learning.
Load-bearing premise
Graphs must be generated by a causal subgraph plus a spurious subgraph whose spurious correlations vary sufficiently across training environments while maintaining a consistent direction; the thesis itself proves that without these conditions no algorithm can learn the invariance.
Editorial extensions
If this is right
- If the CIGA/GALA identification guarantees hold, invariant subgraph learning can serve as a principled front-end for any GNN that must survive distribution shifts, without requiring environment labels.
- If GMT's SubMT approximation is correct, interpretable GNNs should be evaluated with counterfactual fidelity, and attention-based soft-subgraph explanations that skip SubMT are likely to mislead.
- If HAO's homophily unnoticeability is the right threat model, graph robustness evaluations that ignore homophily shift overstate the power of injection attacks.
- If PAIR and FeAT are correct, adding Pareto optimization or feature-augmented retraining to existing OOD algorithms can resolve the ERM conflict and improve OOD accuracy.
Reading between the lines
- A testable extension: use variation sufficiency and variation consistency as dataset diagnostics, and report a benchmark as unlearnable by invariant methods when they fail, which would discipline OOD benchmark construction.
- The counterfactual-fidelity measure suggests a practical rule for explainable models: an explanation's predictive power should be continuously sensitive to small subgraph perturbations, not merely rankable by attention scores.
- If homophily unnoticeability is adopted in standards, graph adversarial robustness comparisons should report both attack success and the induced homophily shift, since cheap pruning defenses should be neutralized before claims of attack power are made.
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The thesis argues that causal invariance learning, instantiated on graph data through invariant-subgraph identification, improves out-of-distribution (OOD) generalization, interpretability, and adversarial robustness in modern machine learning. It develops six methods across three parts: CIGA (Chapter 2) and GALA (Chapter 3) for invariant graph learning under distribution shifts; GMT (Chapter 4) and HAO (Chapter 5) for interpretability and adversarial robustness as implications of causal thinking; and PAIR (Chapter 6) and FeAT (Chapter 7) for optimization and feature learning in invariant risk minimization. The central theoretical claims include Theorem 2.3.1 (CIGA induces an invariant GNN under invertible graph generation and size assumptions), Theorem 3.4.1 (GALA provably identifies the invariant subgraph under 'minimal' assumptions), and Theorem 4.5.1 (GMT approximates the subgraph multilinear extension). Each chapter is based on previously published peer-reviewed papers and includes extensive experiments on synthetic and real benchmarks.
Significance. If the theoretical claims hold, the thesis makes a valuable contribution by extending invariant causal learning from Euclidean data to graphs, providing explicit identifiability results and negative results (e.g., Corollary 3.3.6), and demonstrating consistent empirical gains across many graph benchmarks. The thesis also ships detailed proofs in the appendices and is unusually explicit about limitations and future directions in several appendices. The empirical evaluation is broad, covering synthetic motif datasets, DrugOOD, WILDS, and geometric graph benchmarks. The main significance depends on whether GALA's provable identifiability truly follows from the stated 'minimal' assumptions, because that result anchors the thesis-level claim that causal invariance learning on graphs is feasible without environment labels. The empirical comparisons and the negative results are useful regardless, but the provable-identifiability claim needs repair.
major comments (2)
- [§3.4.3, Theorem 3.4.1] Assumption (iii) in Theorem 3.4.1 is load-bearing but is not shown to follow from Assumption 3.3.3. The theorem requires I(G_p_s; G_n_s|Y)=0 for spurious subgraphs extracted from the assistant's positive and negative sets, and the text asserts this 'is an implication of the variation sufficiency (Assumption 3.3.3)' without proof. Variation sufficiency only guarantees that for any spurious subgraph there exist training environments with different P(Y|G_s); it does not control the conditional dependence between the spurious subgraphs in {G_p} and {G_n}. Nothing in Assumptions 3.3.3 or 3.3.5 prevents the same spurious pattern from appearing in both sets, in which case I(G_p_s; G_n_s|Y)>0 and the GALA objective can select G_s instead of G_c. Moreover, condition (iii) is stated in terms of idealized argmax subgraphs, whereas Algorithm 1 defines {G_p} and {G_n} through the ERM-trained assistant's correctness; no proof links these empirical sets to the mutual-information condition. Until this implication is proved, or assumption (iii) is explicitly added as a premise and its necessity acknowledged, the claim that GALA provably identifies the invariant subgraph under the 'minimal' assumptions is unsupported.
- [§2.3.3, Theorem 2.3.1 / Assumption 2.2.1] Theorem 2.3.1(i) and (ii) both rely on invertibility of f_G_gen and f_Gc_gen in Assumption 2.2.1, together with balanced environment sizes. Invertibility is a strong condition on graph generation functions; it rules out many-to-one assemblies such as random attachment of G_s to G_c and aggregate node or edge features. The thesis does not test or discuss whether the SCMs behind its synthetic and real benchmarks satisfy this condition. Since Theorem 2.3.1 is the basis for the abstract's claim that CIGA can provably identify the underlying G_c, the scope of this guarantee should be stated precisely, and evidence or argument should be provided that at least the evaluated settings meet the invertibility condition.
minor comments (5)
- [Abstract] The text reads 'An Al-aided drug discovery system'; this should be 'An AI-aided drug discovery system'.
- [Table 2.2 caption] The phrase 'in the blankets' should be 'in the parentheses' (averaged rank is reported in the parentheses).
- [§3.5.1 and Table 3.2] The dataset is referred to as both 'Graph-SST' and 'Graph-SST2' in different places; the notation should be unified.
- [§3.4.1, Eq. (3.3)] The notation pG_c^s = g(G^s) with 'G^s ~ P(G|Y)' is confusing: the superscript 's' is not defined as a sample index and can be mistaken for the spurious subgraph. Consider using an index like G' or G_j.
- [Chapter 3, Table 3.1] The checkmark/cross table for DisC and CIGA is informative, but the table should define whether GALA is claimed to handle both H(S|Y)<H(C|Y) and H(S|Y)>H(C|Y) on the same data-generation process or under the additional condition (iii); this would clarify the scope of the claim.
Circularity Check
No significant circularity: the thesis compiles the author's prior peer-reviewed results, but the central derivations are conditional on explicit SCM assumptions and are not equivalent to their inputs by construction.
full rationale
I walked the main derivation chains: CIGA (Ch. 2) defines graph generation via SCMs and derives contrastive objectives whose identifiability theorems are proved in Appendices A.4.2/A.4.3; GALA (Ch. 3) establishes no-free-lunch results and proves its identifiability theorem in Appendix B.5.5 under stated assumptions; GMT (Ch. 4), HAO (Ch. 5), PAIR (Ch. 6), and FeAT (Ch. 7) similarly give explicit architectures/objectives and appendix proofs. None of the central claims is obtained by fitting a parameter to a target and then renaming the fit as a prediction. The heavy self-citation reflects the thesis's nature as a compilation of the author's own papers; the cited prior work is peer-reviewed and the assumptions, not the conclusions, are the load-bearing premises. The potential concern that Theorem 3.4.1's condition iii) is asserted to follow from variation sufficiency without proof is a rigor question rather than a circularity: the main text explicitly refers to Appendix B.5.5 for the proof, and even if that proof were incomplete the failure mode would be an unsupported implication, not an equation that reduces to its own input. I therefore find no constructional circularity and no fitted-input-called-prediction pattern.
Assumptions & free parameters
free parameters (4)
- sc (size of invariant subgraph) =
unknown; set by hyperparameter in CIGA
- alpha and beta (contrastive and hinge loss weights) =
tuned per dataset (figures 2.4, 2.5)
- lambda (information bottleneck weight) =
tuned per dataset in GMT
- penalty weight and upsampling times in GALA =
tuned per dataset
assumptions (7)
- domain assumption Independent Causal Mechanism (ICM) principle
- domain assumption Assumption 2.2.1: Graph generation SCM
- domain assumption Assumption 2.2.2 and 2.2.3: FIIF and PIIF SCMs
- domain assumption Assumption 2.2.4: Better Clustered Invariant Features
- domain assumption Assumption 3.3.3: Variation sufficiency
- domain assumption Assumption 3.3.5: Variation consistency
- domain assumption Invertibility of generation functions in Theorem 2.3.1
invented entities (3)
-
Latent invariant variable C
-
Latent spurious variable S
-
Invariant subgraph Gc and spurious subgraph Gs
Cite this review
Pith. "Pith review of Learning Causality for Modern Machine Learning." pith.science (2026). https://pith.science/paper/UCHIIZCF
@misc{pith2026250612226,
author = {Pith},
title = {Pith review of: Learning Causality for Modern Machine Learning},
year = {2026},
howpublished = {\url{https://pith.science/paper/UCHIIZCF}},
note = {Machine review of arXiv:2506.12226}
}
read the original abstract
In the past decades, machine learning with Empirical Risk Minimization (ERM) has demonstrated great capability in learning and exploiting the statistical patterns from data, or even surpassing humans. Despite the success, ERM avoids the modeling of causality the way of understanding and handling changes, which is fundamental to human intelligence. When deploying models beyond the training environment, distribution shifts are everywhere. For example, an autopilot system often needs to deal with new weather conditions that have not been seen during training, An Al-aided drug discovery system needs to predict the biochemical properties of molecules with respect to new viruses such as COVID-19. It renders the problem of Out-of-Distribution (OOD) generalization challenging to conventional machine learning. In this thesis, we investigate how to incorporate and realize the causality for broader tasks in modern machine learning. In particular, we exploit the invariance implied by the principle of independent causal mechanisms (ICM), that is, the causal mechanisms generating the effects from causes do not inform or influence each other. Therefore, the conditional distribution between the target variable given its causes is invariant under distribution shifts. With the causal invariance principle, we first instantiate it to graphs -- a general data structure ubiquitous in many real-world industry and scientific applications, such as financial networks and molecules. Then, we shall see how learning the causality benefits many of the desirable properties of modern machine learning, in terms of (i) OOD generalization capability; (ii) interpretability; and (iii) robustness to adversarial attacks. Realizing the causality in machine learning, on the other hand, raises a dilemma for optimization in conventional machine learning, as it often contradicts the objective of ERM...
Figures
Figures from the paper (21 more)
Reference graph
Works this paper leans on
-
[4]
Linear unit-tests for invariance discovery.arXiv preprint arXiv:2102.10867,
Aubin, B., Słowik, A., Arjovsky, M., Bottou, L., and Lopez-Paz, D. Linear unit-tests for invariance discovery.arXiv preprint arXiv:2102.10867,
-
[5]
Ba, L. J., Kiros, J. R., and Hinton, G. E. Layer normalization.arXiv preprint, arXiv:1607.06450,
-
[8]
Accounting for Unobserved Confounding in Domain Generalization
Bellot, A. and van der Schaar, M. Generalization and invariances in the presence of unobserved confounding.arXiv preprint, arXiv:2007.10653,
work page Pith review arXiv 2007
-
[10]
Bubeck, S., Chandrasekaran, V ., Eldan, R., Gehrke, J., Horvitz, E., Kamar, E., Lee, P ., Lee, Y. T., Li, Y., Lundberg, S. M., Nori, H., Palangi, H., Ribeiro, M. T., and Zhang, Y. Sparks of artificial general intelligence: Early experiments with GPT-4.arXiv preprint, arXiv:2303.12712,
-
[14]
Estimating generalization under distribution shifts via domain-invariant representations
Chuang, C., Torralba, A., and Jegelka, S. Estimating generalization under distribution shifts via domain-invariant representations. InInternational Conference on Machine Learning, volume 119, pp. 1984–1994. PMLR,
work page 1984
-
[17]
Deng, Y., Yang, Y., Mirzasoleiman, B., and Gu, Q. Robust learning with progressive data expansion against spurious correlation.arXiv preprint, arXiv:2306.04949,
-
[19]
Distributionally Robust Losses for Latent Covariate Mixtures
BIBLIOGRAPHY432 Duchin, J. C., Hashimoto, T., and Namkoong, H. Distributionally robust losses for latent covariate mixtures.arXiv preprint arXiv:2007.13982,
work page Pith review arXiv 2007
-
[20]
Dwivedi, V . P ., Joshi, C. K., Laurent, T., Bengio, Y., and Bresson, X. Benchmarking graph neural networks.arXiv preprint, arXiv:2003.00982,
arXiv 2003
Show all 79 references
-
[21]
Toy models of superposition.arXiv preprint arXiv:2209.10652,
Elhage, N., Hume, T., Olsson, C., Schiefer, N., Henighan, T., Kravec, S., Hatfield-Dodds, Z., Lasenby, R., Drain, D., Chen, C., Grosse, R., McCandlish, S., Kaplan, J., Amodei, D., Wattenberg, M., and Olah, C. Toy models of superposition.arXiv preprint arXiv:2209.10652,
-
[22]
Double equivariance for inductive link prediction for both new nodes and new relation types.arXiv preprint, arXiv:2302.01313,
Gao, J., Zhou, Y., Zhou, J., and Ribeiro, B. Double equivariance for inductive link prediction for both new nodes and new relation types.arXiv preprint, arXiv:2302.01313,
-
[23]
F., Peters, M
Gardner, M., Grus, J., Neumann, M., Tafjord, O., Dasigi, P ., Liu, N. F., Peters, M. E., Schmitz, M., and Zettlemoyer, L. Allennlp: A deep semantic natural language processing platform. arXiv preprint, arXiv:1803.07640,
-
[24]
Joint learning of label and environment causal independence for graph out-of-distribution generalization.arXiv preprint, arXiv:2306.01103, 2023a
Gui, S., Liu, M., Li, X., Luo, Y., and Ji, S. Joint learning of label and environment causal independence for graph out-of-distribution generalization.arXiv preprint, arXiv:2306.01103, 2023a. Gui, S., Liu, M., Li, X., Luo, Y., and Ji, S. Joint learning of label and environment...
-
[25]
Counterfactual learning on graphs: A survey.arXiv preprint, arXiv:2304.01391,
Guo, Z., Xiao, T., Aggarwal, C., Liu, H., and Wang, S. Counterfactual learning on graphs: A survey.arXiv preprint, arXiv:2304.01391,
-
[26]
SIGUA: Forgetting may make learning with noisy labels more robust
BIBLIOGRAPHY435 Han, B., Niu, G., Yu, X., Yao, Q., Xu, M., Tsang, I., and Sugiyama, M. SIGUA: Forgetting may make learning with noisy labels more robust. InProceedings of the 37th International Conference on Machine Learning, pp. 4006–4016, 2020a. Han, B., Yao, Q., Liu, T., Ni...
2011 arXiv
-
[27]
Does distributionally robust supervised learning give robust classifiers? InInternational Conference on Machine Learning, pp
BIBLIOGRAPHY436 Hu, W., Niu, G., Sato, I., and Sugiyama, M. Does distributionally robust supervised learning give robust classifiers? InInternational Conference on Machine Learning, pp. 2034–2042,
-
[28]
Graph neural networks prov- ably benefit from structural information: A feature learning perspective.arXiv preprint arXiv:2306.13926,
Huang, W., Cao, Y., Wang, H., Cao, X., and Suzuki, T. Graph neural networks prov- ably benefit from structural information: A feature learning perspective.arXiv preprint arXiv:2306.13926,
-
[29]
and Wallace, B
BIBLIOGRAPHY437 Jain, S. and Wallace, B. C. Attention is not explanation. InProceedings of the 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, pp. 3543–3556,
2019
-
[30]
Ji, Y., Zhang, L., Wu, J., Wu, B., Huang, L.-K., Xu, T., Rong, Y., Li, L., Ren, J., Xue, D., Lai, H., Xu, S., Feng, J., Liu, W., Luo, P ., Zhou, S., Huang, J., Zhao, P ., and Bian, Y. DrugOOD: Out-of-Distribution (OOD) Dataset Curator and Benchmark for AI-aided Drug Discovery ...
-
[31]
Empowering graph representation learning with test-time graph transformation.arXiv preprint, arXiv:2210.03561,
Jin, W., Zhao, T., Ding, J., Liu, Y., Tang, J., and Shah, N. Empowering graph representation learning with test-time graph transformation.arXiv preprint, arXiv:2210.03561,
-
[32]
E., Pillai, N
Johndrow, J. E., Pillai, N. S., and Smith, A. No free lunch for approximate mcmc.arXiv preprint, arXiv:2010.12514,
2010 arXiv
-
[33]
Kipf, T. N. and Welling, M. Variational graph auto-encoders.arXiv preprint, arXiv:1611.07308,
-
[34]
Kirichenko, P ., Izmailov, P ., and Wilson, A. G. Last layer re-training is sufficient for robustness to spurious correlations.arXiv preprint arXiv:2204.02937,
-
[35]
W., and Clark, B
Kochkov, D., Pfaff, T., Sanchez-Gonzalez, A., Battaglia, P . W., and Clark, B. K. Learn- ing ground states of quantum hamiltonians with graph networks.arXiv preprint, arXiv:2110.06390,
-
[36]
FLAG: adversarial data augmentation for graph neural networks.arXiv preprint, arXiv:2010.09891,
BIBLIOGRAPHY440 Kong, K., Li, G., Ding, M., Wu, Z., Zhu, C., Ghanem, B., Taylor, G., and Goldstein, T. FLAG: adversarial data augmentation for graph neural networks.arXiv preprint, arXiv:2010.09891,
2010 arXiv
-
[37]
and Yamaguchi, S
Koyama, M. and Yamaguchi, S. Out-of-distribution generalization with maximal invariant predictor.arXiv preprint arXiv:2008.01883, 2020a. Koyama, M. and Yamaguchi, S. Out-of-distribution generalization with maximal invariant predictor.arXiv preprint, arXiv:2008.01883, 2020b. Kr...
2008 arXiv
-
[39]
Deeper insights into graph convolutional networks for semi- supervised learning
Li, Q., Han, Z., and Wu, X. Deeper insights into graph convolutional networks for semi- supervised learning. InAAAI Conference on Artificial Intelligence, pp. 3538–3545, 2018a. Li, X., Gui, S., Luo, Y., and Ji, S. Graph structure and feature extrapolation for out-of- distribut...
-
[40]
Bayesian invariant risk minimization
Lin, Y., Dong, H., Wang, H., and Zhang, T. Bayesian invariant risk minimization. In IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 16000–16009, 2022b. Lin, Y., Zhu, S., Tan, L., and Cui, P . ZIN: When and how to learn invariance without environment partiti...
-
[41]
Z., Haghgoo, B., Chen, A
Liu, E. Z., Haghgoo, B., Chen, A. S., Raghunathan, A., Koh, P . W., Sagawa, S., Liang, P ., and Finn, C. Just train twice: Improving group robustness without training group information. InInternational Conference on Machine Learning, pp. 6781–6792, 2021a. Liu, G., Zhao, T., Xu...
-
[42]
Clear: Generative counterfactual explanations on graphs
Ma, J., Guo, R., Mishra, S., Zhang, A., and Li, J. Clear: Generative counterfactual explanations on graphs. InAdvances in Neural Information Processing Systems, pp. 25895–25907, 2022a. Ma, K., Yang, H., Yang, H., Jin, T., Chen, P ., Chen, Y., Kamhoua, B. F., and Cheng, J. Impr...
-
[43]
Towards better out-of-distribution generalization of neural algorithmic reasoning tasks.arXiv preprint arXiv:2211.00692,
BIBLIOGRAPHY445 Mahdavi, S., Swersky, K., Kipf, T., Hashemi, M., Thrampoulidis, C., and Liao, R. Towards better out-of-distribution generalization of neural algorithmic reasoning tasks.arXiv preprint arXiv:2211.00692,
-
[44]
Fisher discriminant analysis with kernels
Mika, S., Ratsch, G., Weston, J., Scholkopf, B., and Mullers, K.-R. Fisher discriminant analysis with kernels. InNeural networks for signal processing IX: Proceedings of the 1999 IEEE signal processing society workshop (cat. no. 98th8468), pp. 41–48,
1999
-
[45]
Towards stable backdoor purification through feature shift tuning.arXiv preprint arXiv:2310.01875,
Min, R., Qin, Z., Shen, L., and Cheng, M. Towards stable backdoor purification through feature shift tuning.arXiv preprint arXiv:2310.01875,
-
[46]
M., Bause, F., Kersting, K., Mutzel, P ., and Neumann, M
Morris, C., Kriege, N. M., Bause, F., Kersting, K., Mutzel, P ., and Neumann, M. Tudataset: A collection of benchmark datasets for learning with graphs.arXiv preprint, arXiv:2007.08663,
2007 arXiv
-
[47]
M., Grohe, M., Fey, M., and Borgwardt, K
Morris, C., Lipman, Y., Maron, H., Rieck, B., Kriege, N. M., Grohe, M., Fey, M., and Borgwardt, K. M. Weisfeiler and leman go machine learning: The story so far.arXiv preprint, arXiv:2112.09992,
-
[48]
and Rees, D
Murray, C. and Rees, D. The rise of fragment-based drug discovery.Nature chemistry, 1: 187–92, 06 2009a. Murray, C. W. and Rees, D. C. The rise of fragment-based drug discovery.Nature chemistry, 1 3:187–92, 2009b. Naganuma, H., Ahuja, K., Mitliagkas, I., Takagi, S., Motokawa, ...
-
[49]
M., Nicolicioiu, A
Nicolicioiu, A. M., Nicolicioiu, A. L., Alexe, B., and Teney, D. Learning diverse features in vision transformers for improved generalization.ICML 2023 Workshop on Spurious Correlations, Invariance, and Stability,
2023
-
[50]
Discovering environments with XRM.arXiv preprint, arXiv:2309.16748,
Pezeshki, M., Bouchacourt, D., Ibrahim, M., Ballas, N., Vincent, P ., and Lopez-Paz, D. Discovering environments with XRM.arXiv preprint, arXiv:2309.16748,
-
[51]
Fishr: Invariant gradient variances for out-of- distribution generalization.arXiv preprint arXiv:2109.02934,
BIBLIOGRAPHY449 Rame, A., Dancette, C., and Cord, M. Fishr: Invariant gradient variances for out-of- distribution generalization.arXiv preprint arXiv:2109.02934,
-
[52]
Model rata- touille: Recycling diverse models for out-of-distribution generalization.arXiv preprint arXiv:2212.10445,
Ramé, A., Ahuja, K., Zhang, J., Cord, M., Bottou, L., and Lopez-Paz, D. Model rata- touille: Recycling diverse models for out-of-distribution generalization.arXiv preprint arXiv:2212.10445,
-
[53]
Domain-adjusted regression or: Erm may already learn features sufficient for out-of-distribution generalization.arXiv preprint arXiv:2202.06856,
Rosenfeld, E., Ravikumar, P ., and Risteski, A. Domain-adjusted regression or: Erm may already learn features sufficient for out-of-distribution generalization.arXiv preprint arXiv:2202.06856,
-
[54]
Distilbert, a distilled version of BERT: smaller, faster, cheaper and lighter.arXiv preprint arXiv:1910.01108,
Sanh, V ., Debut, L., Chaumond, J., and Wolf, T. Distilbert, a distilled version of BERT: smaller, faster, cheaper and lighter.arXiv preprint arXiv:1910.01108,
1910 arXiv
-
[55]
Causality for machine learning.arXiv preprint, arXiv:1911.10500,
Schölkopf, B. Causality for machine learning.arXiv preprint, arXiv:1911.10500,
1911 arXiv
-
[57]
S., and Li, B
Sun, L., Dou, Y., Yang, C., Wang, J., Yu, P . S., and Li, B. Adversarial attack and defense on graph data: A survey.arXiv preprint, arXiv:1812.10528,
-
[58]
Sun, Y., Wang, S., Tang, X., Hsieh, T., and Honavar, V . G. Adversarial attacks on graph neural networks via node injections: A hierarchical reinforcement learning approach. In The Web Conference 2020, pp. 673–683,
2020
-
[61]
Evading the simplicity bias: Training a diverse set of models discovers solutions with superior OOD generalization
Teney, D., Abbasnejad, E., Lucey, S., and van den Hengel, A. Evading the simplicity bias: Training a diverse set of models discovers solutions with superior OOD generalization. arXiv preprint arXiv:2105.05612,
-
[62]
Evading the simplicity bias: Training a diverse set of models discovers solutions with superior OOD generalization
Teney, D., Abbasnejad, E., Lucey, S., and van den Hengel, A. Evading the simplicity bias: Training a diverse set of models discovers solutions with superior OOD generalization. In IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 16740–16751, 2022a. Teney, D....
-
[64]
Representation learning with contrastive predictive coding.arXiv preprint, arXiv:1807.03748,
van den Oord, A., Li, Y., and Vinyals, O. Representation learning with contrastive predictive coding.arXiv preprint, arXiv:1807.03748,
-
[65]
Wang, X., Cheng, M., Eaton, J., Hsieh, C., and Wu, S. F. Attack graph convolutional networks by adding fake nodes.arXiv preprint, arXiv:1810.10751,
-
[66]
Towards out-of- distribution generalizable predictions of chemical kinetics properties
Wang, Z., Chen, Y., Duan, Y., Li, W., Han, B., Cheng, J., and Tong, H. Towards out-of- distribution generalizable predictions of chemical kinetics properties. InNeurIPS 2023 AI for Science Workshop, 2023b. Wen, Z. and Li, Y. Toward understanding the feature learning process of...
2023
-
[67]
Recent advances in reliable deep graph learning: Adversarial attack, inherent noise, and distribution shift.arXiv preprint arXiv:2202.07114, 2022a
Wu, B., Li, J., Hou, C., Fu, G., Bian, Y., Chen, L., and Huang, J. Recent advances in reliable deep graph learning: Adversarial attack, inherent noise, and distribution shift.arXiv preprint arXiv:2202.07114, 2022a. Wu, F., Jr., A. H. S., Zhang, T., Fifty, C., Yu, T., and Weinb...
-
[68]
Optimization of graph neural networks: BIBLIOGRAPHY459 Implicit acceleration by skip connections and more depth
Xu, K., Zhang, M., Jegelka, S., and Kawaguchi, K. Optimization of graph neural networks: BIBLIOGRAPHY459 Implicit acceleration by skip connections and more depth. InInternational Conference on Machine Learning, pp. 11592–11602, 2021a. Xu, K., Zhang, M., Li, J., Du, S. S., Kawa...
-
[69]
Freeze then train: Towards provable representation learning under spurious correlations and feature noise.arXiv preprint arXiv:2210.11075,
Ye, H., Zou, J., and Zhang, L. Freeze then train: Towards provable representation learning under spurious correlations and feature noise.arXiv preprint arXiv:2210.11075,
-
[70]
XGNN: towards model-level explanations of graph neural networks
Yuan, H., Tang, J., Hu, X., and Ji, S. XGNN: towards model-level explanations of graph neural networks. InThe 26th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, pp. 430–438, 2020a. Yuan, H., Yu, H., Gui, S., and Ji, S. Explainability in graph neural networks: A...
2012 arXiv
-
[71]
Z., and Ravikumar, P
Zhai, R., Dan, C., Kolter, J. Z., and Ravikumar, P . Understanding why generalized reweight- ing does not improve over ERM.arXiv preprint arXiv:2201.12293,
-
[72]
and Bottou, L
Zhang, J. and Bottou, L. Learning useful representations for shifting tasks and distributions. arXiv preprint arXiv:2212.07346,
-
[73]
Rich feature construction for the optimization- generalization dilemma.arXiv preprint arXiv:2203.15516, 2022a
BIBLIOGRAPHY462 Zhang, J., Lopez-Paz, D., and Bottou, L. Rich feature construction for the optimization- generalization dilemma.arXiv preprint arXiv:2203.15516, 2022a. Zhang, M., Sohoni, N. S., Zhang, H. R., Finn, C., and Ré, C. Correct-n-contrast: A contrastive approach for i...
-
[74]
F., Tehrani, A
Zhang, X., Wang, L., Helwig, J., Luo, Y., Fu, C., Xie, Y., Liu, M., Lin, Y., Xu, Z., Yan, K., Adams, K., Weiler, M., Li, X., Fu, T., Wang, Y., Yu, H., Xie, Y., Fu, X., Strasser, A., Xu, S., Liu, Y., Du, Y., Saxton, A., Ling, H., Lawrence, H., Stärk, H., Gui, S., Edwards, C., G...
-
[75]
S., Gordon, G
Zhao, H., Dan, C., Aragam, B., Jaakkola, T. S., Gordon, G. J., and Ravikumar, P . Fundamental limits and tradeoffs in invariant representation learning.arXiv preprint arXiv:2012.10713,
2012 arXiv
-
[76]
Graph neural networks: A review of methods and applications.AI Open, 1:57–81, 2020a
Zhou, J., Cui, G., Hu, S., Zhang, Z., Yang, C., Liu, Z., Wang, L., Li, C., and Sun, M. Graph neural networks: A review of methods and applications.AI Open, 1:57–81, 2020a. Zhou, J., Bevilacqua, B., and Ribeiro, B. An ood multi-task perspective for link prediction with new rela...
-
[77]
Explaining and adapting graph conditional shift.arXiv preprint, arXiv:2306.03256,
Zhu, Q., Jiao, Y., Ponomareva, N., Han, J., and Perozzi, B. Explaining and adapting graph conditional shift.arXiv preprint, arXiv:2306.03256,
-
[78]
Understanding the generalization of adam in learning neural networks with proper regularization.arXiv preprint arXiv:2108.11371, 2021a
Zou, D., Cao, Y., Li, Y., and Gu, Q. Understanding the generalization of adam in learning neural networks with proper regularization.arXiv preprint arXiv:2108.11371, 2021a. Zou, D., Liu, S., Miao, S., Fung, V ., Chang, S., and Li, P . GDL-DS: A benchmark for geometric deep lea...
-
[79]
Adversarial attacks on neural networks for graph data
Zügner, D., Akbarnejad, A., and Günnemann, S. Adversarial attacks on neural networks for graph data. InProceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, pp. 2847–2856, 2018
2018
-
[1996]
Nuanced metrics for measuring unintended bias with real data for text classification
Borkan, D., Dixon, L., Sorensen, J., Thain, N., and Vasserman, L. Nuanced metrics for measuring unintended bias with real data for text classification. InCompanion of The 2019 World Wide Web Conference, pp. 491–500,
2019
-
[1998]
Towards better generalization with flexible representation of multi-module graph neural networks.arXiv preprint, arXiv:2209.06589,
Lee, H., Park, H., and Yoon, K. Towards better generalization with flexible representation of multi-module graph neural networks.arXiv preprint, arXiv:2209.06589,
-
[1999]
Ulyanov, D., Vedaldi, A., and Lempitsky, V . S. Instance normalization: The missing ingredient for fast stylization.arXiv preprint, arXiv:1607.08022,
-
[2005]
Understanding and improving graph injection attack by promoting unnoticeability
Chen, Y., Yang, H., Zhang, Y., Ma, K., Liu, T., Han, B., and Cheng, J. Understanding and improving graph injection attack by promoting unnoticeability. InInternational Conference on Learning Representations, 2022a. Chen, Y., Zhang, Y., Bian, Y., Yang, H., Ma, K., Xie, B., Liu,...
2024
-
[2006]
Crabbé, J
ISBN 0471241954. Crabbé, J. and van der Schaar, M. Evaluating the robustness of interpretability methods through explanation invariance and equivariance.arXiv preprint, arXiv:2304.06715,
-
[2008]
Idea: Invariant causal defense for graph adversarial robustness.arXiv preprint, arXiv:2305.15792,
Tao, S., Cao, Q., Shen, H., Wu, Y., Xu, B., and Cheng, X. Idea: Invariant causal defense for graph adversarial robustness.arXiv preprint, arXiv:2305.15792,
-
[2013]
E., Krizhevsky, A., Sutskever, I., and Salakhutdinov, R
BIBLIOGRAPHY453 Srivastava, N., Hinton, G. E., Krizhevsky, A., Sutskever, I., and Salakhutdinov, R. Dropout: a simple way to prevent neural networks from overfitting.Journal of Machine Learning Research, 15(1):1929–1958,
1929
-
[2014]
A., Hassani, K., Jayaraman, P
BIBLIOGRAPHY454 Taghanaki, S. A., Hassani, K., Jayaraman, P . K., Ahmadi, A. H. K., and Custis, T. Point- mask: Towards interpretable and bias-resilient point cloud processing.arXiv preprint, arXiv:2007.04525,
2007 arXiv
-
[2015]
Causality and independence enhancement for biased node classification
Chen, G., Wang, Y., Guo, F., Guo, Q., Shao, J., Shen, H., and Cheng, X. Causality and independence enhancement for biased node classification. InACM International Conference on Information and Knowledge Management, pp. 203–212, 2023a. Chen, H. S., Lee, Y., Setlur, A., Levine, ...
-
[2016]
W., Hamrick, J
Battaglia, P . W., Hamrick, J. B., Bapst, V ., Sanchez-Gonzalez, A., Zambaldi, V . F., Malinowski, M., Tacchetti, A., Raposo, D., Santoro, A., Faulkner, R., Gülçehre, Ç., Song, H. F., Ballard, A. J., Gilmer, J., Dahl, G. E., Vaswani, A., Allen, K. R., Nash, C., Langston, V ., ...
-
[2017]
and Li, Y
Allen-Zhu, Z. and Li, Y. Towards understanding ensemble, knowledge distillation and self-distillation in deep learning.arXiv preprint arXiv:2012.09816,
2012 arXiv
-
[2018]
The iwildcam 2020 competition dataset.arXiv preprint arXiv:2004.10340,
Beery, S., Cole, E., and Gjoka, A. The iwildcam 2020 competition dataset.arXiv preprint arXiv:2004.10340,
2020 arXiv
-
[2019]
A closer look at distribution shifts and out-of-distribution generalization on graphs
Ding, M., Kong, K., Chen, J., Kirchenbauer, J., Goldblum, M., Wipf, D., Huang, F., and Goldstein, T. A closer look at distribution shifts and out-of-distribution generalization on graphs. InNeurIPS 2021 Workshop on Distribution Shifts: Connecting Methods and Applications,
2021
-
[2020]
Learning a similarity metric discriminatively, with application to face verification
Chopra, S., Hadsell, R., and LeCun, Y. Learning a similarity metric discriminatively, with application to face verification. In2005 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR 2005), 20-26 June 2005, San Diego, CA, USA, pp. 539–546,
2005
-
[2021]
Invariant risk minimization.arXiv preprint arXiv:1907.02893,
Arjovsky, M., Bottou, L., Gulrajani, I., and Lopez-Paz, D. Invariant risk minimization.arXiv preprint arXiv:1907.02893,
1907 arXiv
-
[2022]
The evolution of out-of-distribution robustness throughout fine-tuning.arXiv preprint arXiv:2106.15831,
Andreassen, A., Bahri, Y., Neyshabur, B., and Roelofs, R. The evolution of out-of-distribution robustness throughout fine-tuning.arXiv preprint arXiv:2106.15831,
-
[2023]
W., Xu, R., Cranmer, K., Spergel, D
Cranmer, M., Sanchez-Gonzalez, A., Battaglia, P . W., Xu, R., Cranmer, K., Spergel, D. N., and Ho, S. Discovering symbolic models from deep learning with inductive biases.arXiv preprint, arXiv:2006.11287,
2006 arXiv
Reviewed August 7, 2026 · model on record in the stance chip above.
Discussion (0). Sign in to comment.