REVIEW 4 major objections 7 minor 43 references
Geodesic Flow Kernels for Semi-Supervised Learning on Mixed-Variable Tabular Dataset
T0 review · 4 major / 7 minor · reviewed 2026-08-11 · deepseek-v4-flash
Pith's one-line read The paper proposes GFTab, a semi-supervised method that corrupts continuous and categorical variables differently and measures the corruption with a geodesic flow kernel, and claims it outperforms existing ML/DL methods on mixed-variable…
desk verdict A useful but sloppy empirical paper: the geodesic similarity loss is not well-defined as printed and no code hash is provided, so the central claims cannot yet be attributed to the proposed kernel. 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 geodesic flow kernel, a positive semi-definite matrix $A$ defined by integrating the projections of the corrupted representations $z_{hard}$ and $z_{soft}$ onto every intermediate subspace along the geodesic between their PCA subspaces $P_{hard}$ and $P_{soft}$ on the Grassmann manifold $G(D, d_{lin})$ (the manifold of $D$-dimensional linear subspaces). The generalized SVD of $P_{hard}^\top P_{soft}$ and $R_{hard}^\top P_{soft}$ supplies the principal angles, and $A$ has a closed form, so the similarity loss $L_{sim} = 1 - z_{soft}^\top A z_{hard} / (\sqrt{z_{soft}^\top A z_{soft}} \sqrt{z_{hard}^\top A z_{hard}})$ can be computed without sampling the path. This loss is what turns 'how much did the corruption change this representation' into a manifold-aware quantity, and the paper's ablation attributes its advantage over InfoNCE, Barlow Twins, and uniformity-alignment losses to that geometric sensitivity. Two supporting pieces are the variable-specific corruption (permutation mixing for continuous columns, neighborhood shifts for categorical columns) and the tree-based embedding of labeled rows from a gradient-boosted decision tree, which enters the supervised cross-entropy term.
What would settle it
Train GFTab with the geodesic-flow similarity loss replaced by ordinary cosine similarity between $z_{hard}$ and $z_{soft}$, keeping the corruption methods and tree embedding unchanged; if the F1 scores on the 21 datasets do not drop materially, the geodesic kernel is not the component carrying the claimed improvement.
Extended reading notes
Core claim
The central claim is that the geometric relationship between a soft and a hard corrupted view of the same tabular instance, integrated along a geodesic on the Grassmann manifold, is a better similarity signal for mixed-variable tabular data than Euclidean losses such as InfoNCE, Barlow Twins, or alignment-uniformity. GFTab corrupts continuous variables by a variable-selection network followed by weighted row and column shuffles, and categorical variables by neighborhood shifts sized by a corruption rate; a Transformer encodes both views; PCA of mini-batch representations produces a subspace for each view; and the geodesic flow kernel between those subspaces defines the similarity loss. The paper reports that, with 20% labeled training data, GFTab achieves the highest F1 score on 10 of 21 curated mixed-variable datasets and ranks in the top three on 7 more, that the wins concentrate on categorical-dominant datasets, and that the ordering is largely preserved when 20% label noise is added.
Load-bearing premise
The load-bearing premise is that the relationship between the soft and hard corrupted versions of a data point is faithfully captured by low-dimensional linear subspaces of the learned representations, so the curved path between those subspaces is a meaningful similarity signal.
Editorial extensions
If this is right
- Learning from mixed-variable tabular data in low-label regimes can be improved by corrupting continuous and categorical columns differently rather than applying the same noise to every feature.
- A similarity loss defined by a geodesic path between subspaces can replace Euclidean contrastive losses as the self-supervised signal for tabular representation learning.
- The reported gains are concentrated on categorical-dominant datasets, so variable-type-aware geometry matters most where tabular data are most heterogeneous.
- The method maintains its ranking when labels are noisy, implying the geometric similarity signal is not simply memorizing label information from the small labeled set.
Reading between the lines
- The paper compares its full loss against other similarity losses but does not isolate the geodesic geometry from the two-view corruption scheme; replacing the kernel with ordinary cosine similarity on the same views would show how much of the gain is geometric.
- Since the kernel is built from PCA bases of mini-batch representations, its usefulness plausibly depends on batch size and subspace dimension $D$; sensitivity to those two hyperparameters is a natural next experiment.
- Dataset curation intentionally excluded purely continuous or purely categorical tables, so the contribution of the variable-specific corruption on those edge cases remains an open question.
- The tree-based embedding consumes labeled examples only; an unlabeled-data extension via pseudo-labels would be a natural way to scale the supervised component as more data are encountered.
Signed reviews
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes GFTab, a semi-supervised learning framework for mixed-variable tabular data. The method combines variable-specific corruption of continuous and categorical features, a geodesic-flow-kernel similarity loss that aligns clean and corrupted representations on a Grassmann manifold, and tree-based embeddings to exploit labeled structures. The authors evaluate GFTab on 21 curated datasets at 10% and 20% labeled fractions, with and without label noise, comparing against seven ML/DL baselines and reporting F1 scores. The paper claims that GFTab outperforms existing models on many datasets, especially under limited labels, and includes ablations of each component (corruption, loss function, VSN, tree embedding, trade-off parameter β).
Significance. If the method were precisely specified, the paper would constitute a useful empirical study of semi-supervised tabular learning: the benchmark suite is broader than in many prior works, baselines are tuned, and component-level ablations are provided. The paper's central formal contribution, the geodesic flow kernel loss, is, however, not well-defined as printed, so the current version does not permit independent verification of the method or attribution of the reported gains to the proposed kernel. The empirical results themselves would also benefit from statistical tests that account for paired dataset comparisons. The paper's strengths (broad evaluation, ablations, effort toward mixed-variable realism) are substantial, and the formal issues appear fixable, so the appropriate course is a major revision.
major comments (4)
- [The GFTab Framework II: Geodesic Flow Kernel, Eqs. (5)-(9)] The geodesic flow kernel loss is not well-defined as printed. Eq. (7) writes z_hard^T A z_soft but Def. 2 declares A ∈ R^{D×D}, whereas z_hard and z_soft are both in R^{dlin}; for the quadratic form to make sense A must be dlin×dlin. In Eq. (5), the block matrix [[U1Γ(π), 0],[0, -U2Σ(π)]] cannot multiply [P_hard R_hard] (dlin×dlin) to produce a dlin×D matrix; the correct geodesic flow formula from Gong et al. uses a vertical stack [[U1Γ(π)],[-U2Σ(π)]] and no zero blocks. Eq. (8) further mixes R_hard and R_soft (Eq. (5) uses R_hard but Eq. (8) uses R_soft) and does not specify the dimensions of U2, D, E, and G. Finally, Eq. (9) divides by sqrt(A z_soft) and sqrt(A z_hard), which are vectors, not scalar norms; the loss should be normalized by sqrt(z_soft^T A z_soft) sqrt(z_hard^T A z_hard). These are not mere typos: because L_sim is contribution (2) of the paper and the basis for Figure 3, the equations must be corrected and the computation of P_hard, P_soft, U1, U2, Γ, Σ, and D precisely specified.
- [The GFTab Framework II / Implementation details] The procedure for obtaining the Grassmannian subspaces is omitted. The paper states 'we leverage the assumption that data lie in a low-dimensional linear subspace' and models z_soft and z_hard with bases P ∈ R^{dlin×D}, but does not say how P_hard and P_soft are computed (e.g., PCA of the minibatch embeddings, PCA of the full training set, recomputed each iteration), how D is chosen, or how differentiation through the PCA/SVD steps is handled. Appendix B lists d_emb, dim_head, layers, and depths for GFTab but not D, γ, λ, or β (β is later fixed to 1.0 after the analysis in Figure 4). Without this information, the method cannot be reimplemented, and the loss ablation in Figure 3 cannot be attributed to the geodesic kernel rather than to the other components.
- [Categorical Variable Corruption, Remark] The Remark states that to achieve a corruption rate of at least γ, the minimum neighborhood size s must satisfy s ≥ ⌈2n(1−γ)−1⌉. This inequality is not derived and appears inconsistent with the paper's own definition of corruption rate. Since s controls the size of the neighborhood from which a replacement category is sampled, larger s should increase the chance of corruption; however, the inequality is easiest to satisfy for γ near 1 and impossible for small γ (e.g., γ=0.2, n=10 requires s≥15, exceeding the number of categories). The authors should define the corruption rate explicitly (probability that a category is changed, or expected fraction of altered categories) and give a correct derivation.
- [Experiments / Tables 1, 2, 7, 8] The central empirical claim of superiority over baseline models is based on point estimates of mean F1 with standard deviations over (presumably) a small number of runs. Many differences in Tables 1 and 2 are within one standard deviation of the baseline, and the win matrices in Figures 2-4 do not account for the paired structure of the comparison. I recommend adding a paired test across datasets (e.g., Wilcoxon signed-rank test comparing per-dataset scores) or at least reporting the number of seeds and a corrected resampled t-test, so that the reader can assess whether the reported wins reflect a reliable difference.
minor comments (7)
- [The GFTab Framework II, Eq. (9)] Eq. (9) uses z_soft^T A z_hard while Eq. (7) uses z_hard^T A z_soft; since A is symmetric (if it is the GFK matrix), the order is fine, but the asymmetric notation should be unified for readability.
- [Appendix B, Table 4] The hyperparameter table header 'GKSMT' appears to be a typo for 'GFTab'; please correct it.
- [The GFTab Framework II, Eq. (8)] The symbols 'd1i', 'e2i', and 'g3i' below Eq. (8) should be 'd_i', 'e_i', and 'g_i', and the definitions should be checked for typographical errors (e.g., the sign in e2i).
- [Categorical Variable Corruption] The sentence 'Unlike continuous variables, Because ordered categories allow for the concept of relative closeness, ...' is grammatically broken and should be rephrased.
- [The GFTab Framework III: Tree-Based Embedding] The tree-based embedding section is under-specified: it is unclear whether the GBDT is trained on the labeled subset, which target is used, and how the leaf embedding is combined with the soft/hard representations in the total loss. A short pseudocode or an architecture figure with tensor dimensions would help.
- [The GFTab Framework I, Eq. (2)] The text says 'randomly shuffling the row and columns within the Ξcont' but the permutation matrix P appears to permute columns; please clarify the exact permutation operation.
- [Implementation details / Code availability] The code is stated to be publicly available, but the manuscript itself contains no code or commit hash; please provide an anonymized repository or detailed pseudocode in the revision so that reviewers can verify the implementation.
Circularity Check
No circularity: the geodesic-flow kernel is imported from external sources and the central claims are benchmark measurements, not constructed outputs.
full rationale
The paper's derivation chain is self-contained in the relevant sense. The geodesic-flow kernel is not derived from GFTab's own outputs; Eq. (7) defines the kernel as an integral over projections onto Grassmannian points, and Eq. (8) is explicitly cited to Gong et al. (2012) and Simon, Koniusz, and Harandi (2021), both external works, so the closed-form matrix A is imported independent support rather than a self-citation. The 'low-dimensional linear subspace' premise is stated as an assumption ('we leverage the assumption that data lie in a low-dimensional linear subspace to derive features from both z_soft and z_hard'), not as a conclusion obtained from the method. The empirical claims in Tables 1 and 2 and the ablations in Figures 2 and 3 are benchmark measurements; hyperparameters are tuned and reported in Appendix B, not hidden as predictions. The dimensional inconsistency in Eqs. (7)-(9) (A declared as D x D while z_hard and z_soft are d_lin-dimensional, and Eq. (9) normalizing by vector-valued quantities) is a formal specification gap, but it is not circular: the loss is not equivalent by construction to the data used to fit it. No self-citation chain is load-bearing, and no fitted parameter is renamed as a prediction.
Assumptions & free parameters
free parameters (5)
- beta (balance between L_sim and L_ce) =
1.0
- corruption rate gamma for categorical corruption =
not stated in paper
- lambda in continuous soft/hard views =
not stated in paper
- Grassmann subspace dimension D =
not stated in paper
- architecture hyperparameters (d_emb, dim_head, layers, depths) =
grid-searched values
assumptions (3)
- domain assumption Learned representations of soft and hard views lie in a low-dimensional linear subspace.
- domain assumption Benchmark data samples are i.i.d.
- standard math Standard gSVD and Grassmann geodesic flow formulas from prior work.
Cite this review
Pith. "Pith review of Geodesic Flow Kernels for Semi-Supervised Learning on Mixed-Variable Tabular Dataset." pith.science (2026). https://pith.science/paper/P7OKCDDQ
@misc{pith2026241212864,
author = {Pith},
title = {Pith review of: Geodesic Flow Kernels for Semi-Supervised Learning on Mixed-Variable Tabular Dataset},
year = {2026},
howpublished = {\url{https://pith.science/paper/P7OKCDDQ}},
note = {Machine review of arXiv:2412.12864}
}
read the original abstract
Tabular data poses unique challenges due to its heterogeneous nature, combining both continuous and categorical variables. Existing approaches often struggle to effectively capture the underlying structure and relationships within such data. We propose GFTab (Geodesic Flow Kernels for Semi- Supervised Learning on Mixed-Variable Tabular Dataset), a semi-supervised framework specifically designed for tabular datasets. GFTab incorporates three key innovations: 1) Variable-specific corruption methods tailored to the distinct properties of continuous and categorical variables, 2) A Geodesic flow kernel based similarity measure to capture geometric changes between corrupted inputs, and 3) Tree-based embedding to leverage hierarchical relationships from available labeled data. To rigorously evaluate GFTab, we curate a comprehensive set of 21 tabular datasets spanning various domains, sizes, and variable compositions. Our experimental results show that GFTab outperforms existing ML/DL models across many of these datasets, particularly in settings with limited labeled data.
Figures
Reference graph
Works this paper leans on
-
[1]
Akiba, T.; Sano, S.; Yanase, T.; Ohta, T.; and Koyama, M. 2019. Optuna: A next-generation hyperparameter optimization framework. In Proceedings of the 25th ACM SIGKDD international conference on knowledge discovery & data mining, 2623--2631
2019
-
[2]
Arik, S. \"O .; and Pfister, T. 2021. Tabnet: Attentive interpretable tabular learning. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 35, 6679--6687
work page 2021
-
[3]
Bahri, D.; Jiang, H.; Tay, Y.; and Metzler, D. 2022. Scarf: Self-supervised contrastive learning using random feature corruption. Proceedings of the 38th International Conference on Machine Learning
work page 2022
-
[4]
Borisov, V.; Leemann, T.; Sessler, K.; Haug, J.; Pawelczyk, M.; and Kasneci, G. 2022. Deep Neural Networks and Tabular Data: A Survey. IEEE Transactions on Neural Networks and Learning Systems , 1--21
work page 2022
-
[5]
Chen, T.; and Guestrin, C. 2016. Xgboost: A scalable tree boosting system. In Proceedings of the 22nd acm sigkdd international conference on knowledge discovery and data mining, 785--794
2016
-
[6]
Chen, T.; Kornblith, S.; Norouzi, M.; and Hinton, G. 2020. A simple framework for contrastive learning of visual representations. In International conference on machine learning, 1597--1607. PMLR
2020
-
[7]
A.; Srivastava, A.; Liu, X.; and Van Dooren, P
Gallivan, K. A.; Srivastava, A.; Liu, X.; and Van Dooren, P. 2003. Efficient algorithms for inferences on Grassmann manifolds. In IEEE Workshop on Statistical Signal Processing, 2003, 315--318. IEEE
work page 2003
-
[8]
Gijsbers, P.; Bueno, M. L. P.; Coors, S.; LeDell, E.; Poirier, S.; Thomas, J.; Bischl, B.; and Vanschoren, J. 2022. AMLB: an AutoML Benchmark
work page 2022
Show all 43 references
-
[9]
Gong, B.; Shi, Y.; Sha, F.; and Grauman, K. 2012. Geodesic flow kernel for unsupervised domain adaptation. In 2012 IEEE conference on computer vision and pattern recognition, 2066--2073. IEEE
2012
-
[10]
Gorishniy, Y.; Rubachev, I.; Khrulkov, V.; and Babenko, A. 2021. Revisiting deep learning models for tabular data. Advances in Neural Information Processing Systems, 34: 18932--18943
2021
-
[11]
Grinsztajn, L.; Oyallon, E.; and Varoquaux, G. 2022. Why do tree-based models still outperform deep learning on tabular data? arXiv preprint arXiv:2207.08815
2022 arXiv
-
[12]
Guo, X.; Quan, Y.; Zhao, H.; Yao, Q.; Li, Y.; and Tu, W. 2021. TabGNN: Multiplex graph neural network for tabular data prediction. arXiv preprint arXiv:2108.09127
2021 arXiv
-
[13]
K.; M \"u ller, T.; Piccinno, F.; and Eisenschlos, J
Herzig, J.; Nowak, P. K.; M \"u ller, T.; Piccinno, F.; and Eisenschlos, J. M. 2020. TaPas: Weakly supervised table parsing via pre-training. arXiv preprint arXiv:2004.02349
2020 arXiv
-
[14]
Hollmann, N.; M \"u ller, S.; Eggensperger, K.; and Hutter, F. 2023. Tab PFN : A Transformer That Solves Small Tabular Classification Problems in a Second. In The Eleventh International Conference on Learning Representations
2023
-
[15]
Huang, X.; Khetan, A.; Cvitkovic, M.; and Karnin, Z. 2020. Tabtransformer: Tabular data modeling using contextual embeddings. arXiv preprint arXiv:2012.06678
2020 arXiv
-
[16]
Iida, H.; Thai, D.; Manjunatha, V.; and Iyyer, M. 2021. Tabbie: Pretrained representations of tabular data. arXiv preprint arXiv:2105.02584
2021 arXiv
-
[17]
Jing, L.; and Tian, Y. 2020. Self-supervised visual feature learning with deep neural networks: A survey. IEEE transactions on pattern analysis and machine intelligence, 43(11): 4037--4058
2020
-
[18]
Kim, S.; Tsai, Y.-C.; Singh, K.; Choi, Y.; Ibok, E.; Li, C.-T.; and Cha, M. 2020. DATE: Dual attentive tree-aware embedding for customs fraud detection. In Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, 2880--2890
2020
-
[19]
N.; Rainforth, T.; and Gal, Y
Kossen, J.; Band, N.; Lyle, C.; Gomez, A. N.; Rainforth, T.; and Gal, Y. 2021. Self-attention between datapoints: Going beyond individual input-output pairs in deep learning. Advances in Neural Information Processing Systems, 34: 28742--28756
2021
-
[20]
Lee, K.; Chang, M.-W.; and Toutanova, K. 2019. Latent retrieval for weakly supervised open domain question answering. arXiv preprint arXiv:1906.00300
2019 arXiv
-
[21]
\"O .; Loeff, N.; and Pfister, T
Lim, B.; Ar k, S. \"O .; Loeff, N.; and Pfister, T. 2021. Temporal fusion transformers for interpretable multi-horizon time series forecasting. International Journal of Forecasting, 37(4): 1748--1764
2021
-
[22]
Loshchilov, I.; and Hutter, F. 2017. Decoupled weight decay regularization. arXiv preprint arXiv:1711.05101
2017 arXiv
-
[23]
Mandolesi, A. L. 2019. Grassmann angles between real or complex subspaces. arXiv preprint arXiv:1910.00147
2019 arXiv
-
[24]
Marton, S.; L \"u dtke, S.; Bartelt, C.; and Stuckenschmidt, H. 2024. GRANDE : Gradient-Based Decision Tree Ensembles. In The Twelfth International Conference on Learning Representations
2024
-
[25]
Oord, A. v. d.; Li, Y.; and Vinyals, O. 2018. Representation learning with contrastive predictive coding. arXiv preprint arXiv:1807.03748
2018 arXiv
-
[26]
Popov, S.; Morozov, S.; and Babenko, A. 2020. Neural oblivious decision ensembles for deep learning on tabular data. ICLR
2020
-
[27]
V.; and Gulin, A
Prokhorenkova, L.; Gusev, G.; Vorobev, A.; Dorogush, A. V.; and Gulin, A. 2018. CatBoost: unbiased boosting with categorical features. Advances in neural information processing systems, 31
2018
-
[28]
Qiu, X.; Sun, T.; Xu, Y.; Shao, Y.; Dai, N.; and Huang, X. 2020. Pre-trained models for natural language processing: A survey. Science China Technological Sciences, 63(10): 1872--1897
2020
-
[29]
Ruder, S.; and Plank, B. 2018. Strong baselines for neural semi-supervised learning under domain shift. arXiv preprint arXiv:1804.09530
2018 arXiv
-
[30]
Simon, C.; Koniusz, P.; and Harandi, M. 2021. On learning the geodesic path for incremental learning. In Proceedings of the IEEE/CVF conference on Computer Vision and Pattern Recognition, 1591--1600
2021
-
[31]
B.; and Goldstein, T
Somepalli, G.; Goldblum, M.; Schwarzschild, A.; Bruss, C. B.; and Goldstein, T. 2021. Saint: Improved neural networks for tabular data via row attention and contrastive pre-training. arXiv preprint arXiv:2106.01342
2021 arXiv
-
[32]
Song, K.; Tan, X.; Qin, T.; Lu, J.; and Liu, T.-Y. 2020. Mpnet: Masked and permuted pre-training for language understanding. Advances in Neural Information Processing Systems, 33: 16857--16867
2020
-
[33]
Song, W.; Shi, C.; Xiao, Z.; Duan, Z.; Xu, Y.; Zhang, M.; and Tang, J. 2019. AutoInt . In Proceedings of the 28th ACM International Conference on Information and Knowledge Management . ACM
2019
-
[34]
Ucar, T.; Hajiramezanali, E.; and Edwards, L. 2021. SubTab: Subsetting Features of Tabular Data for Self-Supervised Representation Learning. Advances in Neural Information Processing Systems, 34
2021
-
[35]
Van Loan, C. F. 1976. Generalizing the singular value decomposition. SIAM Journal on numerical Analysis, 13(1): 76--83
1976
-
[36]
N.; Kaiser, L
Vaswani, A.; Shazeer, N.; Parmar, N.; Uszkoreit, J.; Jones, L.; Gomez, A. N.; Kaiser, L. u.; and Polosukhin, I. 2017. Attention is All you Need. In Guyon, I.; Luxburg, U. V.; Bengio, S.; Wallach, H.; Fergus, R.; Vishwanathan, S.; and Garnett, R., eds., Advances in Neural Infor...
2017
-
[37]
Wang, T.; and Isola, P. 2020. Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere. In III, H. D.; and Singh, A., eds., Proceedings of the 37th International Conference on Machine Learning, volume 119 of Proceedings of Machine L...
2020
-
[38]
Yin, P.; Neubig, G.; Yih, W.-t.; and Riedel, S. 2020. TaBERT: Pretraining for joint understanding of textual and tabular data. arXiv preprint arXiv:2005.08314
2020 arXiv
-
[39]
Yoon, J.; Zhang, Y.; Jordon, J.; and van der Schaar, M. 2020. Vime: Extending the success of self-and semi-supervised learning to tabular domain. Advances in Neural Information Processing Systems, 33: 11033--11043
2020
-
[40]
Zbontar, J.; Jing, L.; Misra, I.; LeCun, Y.; and Deny, S. 2021. Barlow Twins: Self-Supervised Learning via Redundancy Reduction. CoRR, abs/2103.03230
2021 arXiv
-
[41]
Zhai, X.; Oliver, A.; Kolesnikov, A.; and Beyer, L. 2019. S4l: Self-supervised semi-supervised learning. In Proceedings of the IEEE/CVF international conference on computer vision, 1476--1485
2019
-
[42]
, " * write output.state after.block = add.period write newline
ENTRY address archivePrefix author booktitle chapter edition editor eid eprint howpublished institution isbn journal key month note number organization pages publisher school series title type volume year label extra.label sort.label short.list INTEGERS output.state before.all...
-
[43]
write newline
" write newline "" before.all 'output.state := FUNCTION n.dashify 't := "" t empty not t #1 #1 substring "-" = t #1 #2 substring "--" = not "--" * t #2 global.max substring 't := t #1 #1 substring "-" = "-" * t #2 global.max substring 't := while if t #1 #1 substring * t #2 gl...
Reviewed August 11, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.