Transformers for Tabular Data: A Training Perspective of Self-Attention via Optimal Transport
Antonio Candelieri
Alessandro Quadrio
- OT
Main:4 Pages
9 Figures
16 Tables
Appendix:41 Pages
Abstract
This thesis examines self-attention training through the lens of Optimal Transport (OT) and develops an OT-based alternative for tabular classification. The study tracks intermediate projections of the self-attention layer during training and evaluates their evolution using discrete OT metrics, including Wasserstein distance, Monge gap, optimality, and efficiency. Experiments are conducted on classification tasks with two and three classes, as well as on a biomedical dataset.
View on arXivComments on this paper
