TabDeco: A Comprehensive Contrastive Framework for Decoupled Representations in Tabular Data

Chen, Suiyao, Wu, Jing, Wang, Yunxiao, Ji, Cheng, Xie, Tianpei, Cociorva, Daniel, Sharps, Michael, Levasseur, Cecile, Brunzell, Hakan

arXiv.org Artificial Intelligence 

Representation learning is a fundamental aspect of modern artificial intelligence, driving substantial improvements across diverse applications. While selfsupervised contrastive learning has led to significant advancements in fields like computer vision and natural language processing, its adaptation to tabular data presents unique challenges. Traditional approaches often prioritize optimizing model architecture and loss functions but may overlook the crucial task of constructing meaningful positive and negative sample pairs from various perspectives like feature interactions, instance-level patterns and batch-specific contexts. To address these challenges, we introduce TabDeco, a novel method that leverages attention-based encoding strategies across both rows and columns and employs contrastive learning framework to effectively disentangle feature representations at multiple levels, including features, instances and data batches.