Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

TIP: Tabular-Image Pre-training for Multimodal Classification with Incomplete Data

About

Images and structured tables are essential parts of real-world databases. Though tabular-image representation learning is promising to create new insights, it remains a challenging task, as tabular data is typically heterogeneous and incomplete, presenting significant modality disparities with images. Earlier works have mainly focused on simple modality fusion strategies in complete data scenarios, without considering the missing data issue, and thus are limited in practice. In this paper, we propose TIP, a novel tabular-image pre-training framework for learning multimodal representations robust to incomplete tabular data. Specifically, TIP investigates a novel self-supervised learning (SSL) strategy, including a masked tabular reconstruction task for tackling data missingness, and image-tabular matching and contrastive learning objectives to capture multimodal information. Moreover, TIP proposes a versatile tabular encoder tailored for incomplete, heterogeneous tabular data and a multimodal interaction module for inter-modality representation learning. Experiments are performed on downstream multimodal classification tasks using both natural and medical image datasets. The results show that TIP outperforms state-of-the-art supervised/SSL image/multimodal algorithms in both complete and incomplete data scenarios. Our code is available at https://github.com/siyi-wind/TIP.

Siyi Du, Shaoming Zheng, Yinsong Wang, Wenjia Bai, Declan P. O'Regan, Chen Qin• 2024

Related benchmarks

TaskDatasetResultRank
ClassificationCAD 1% labels
AUC81.13
27
ClassificationDVM 1% labels
Accuracy88.93
27
ClassificationDVM 10% labels
Accuracy98.75
27
ClassificationInfarction 10% labels
AUC0.8031
27
ClassificationCAD 10% labels
AUC83.82
27
ClassificationInfarction 1% labels
AUC75.05
27
Alzheimer's disease diagnosisADNI
AUC94.6
24
Multi-class classificationDVM (test)
Accuracy95.31
18
Multi-Label ClassificationUK Biobank (test)
AUC0.8049
18
RegressionUK Biobank (test)
MAE2.13
18
Showing 10 of 18 rows

Other info

Follow for update