arXiv:2410.13203cs.LGcs.AI2024-10中稿 · presentation at th…被引 7

通过有序排列特征提升表格数据深度学习效果

TabSeq: A Framework for Deep Learning on Tabular Data via Sequential Ordering

  • 基于聚类的特征序列排序,融合局部与全局结构
  • 在抗体微阵列等3个生物医学数据集上提升模型性能
  • 适合处理异构特征的表格数据深度学习任务

表格数据的深度学习分析仍面临挑战,主要因其特征异质性及重要性不一。本文提出TabSeq框架,通过聚类实现特征的顺序排列,结合局部与全局排序机制,优化学习过程。该方法与去噪自编码器中的多头注意力机制协同工作,使模型更关注关键特征,减少冗余。在原始抗体微阵列及另外两个真实生物医学数据集上的实验表明,合理的特征排序可显著提升模型性能,验证了特征顺序对深度学习效果的重要影响。

原文摘要 · Abstract (English)

Effective analysis of tabular data still poses a significant problem in deep learning, mainly because features in tabular datasets are often heterogeneous and have different levels of relevance. This work introduces TabSeq, a novel framework for the sequential ordering of features, addressing the vital necessity to optimize the learning process. Features are not always equally informative, and for certain deep learning models, their random arrangement can hinder the model's learning capacity. Finding the optimum sequence order for such features could improve the deep learning models' learning process. The novel feature ordering technique we provide in this work is based on clustering and incorporates both local ordering and global ordering. It is designed to be used with a multi-head attention mechanism in a denoising autoencoder network. Our framework uses clustering to align comparable features and improve data organization. Multi-head attention focuses on essential characteristics, whereas the denoising autoencoder highlights important aspects by rebuilding from distorted inputs. This method improves the capability to learn from tabular data while lowering redundancy. Our research, demonstrating improved performance through appropriate feature sequence rearrangement using raw antibody microarray and two other real-world biomedical datasets, validates the impact of feature ordering. These results demonstrate that feature ordering can be a viable approach to improved deep learning of tabular data.

表格数据特征排序自编码器注意力机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。