跳到主要内容
AI News HubLIVE
站内改写1 分钟阅读

几何感知的表格扩散模型

文章摘要

本文提出GATD,通过向表格扩散去噪器显式注入列间成对角度与长度作为辅助监督,以更少参数在多个基准上达到SOTA,并证明收益源自监督而非额外输入或容量。

来源arXiv Machine Learning作者: David Turtora Zagardo
几何感知的表格扩散模型
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

表格合成在隐私保护的数据共享与数据增强中扮演着关键角色,但扩散模型在捕捉列与列之间的相互关系时,往往依赖隐式机制,缺少对结构化关联的显式约束。针对这一不足,研究者提出了几何感知表格扩散(Geometry-Aware Tabular Diffusion,简称 GATD)。该方法为表格扩散去噪器引入了由列值差异计算得到的成对角度与长度信息,并将这些几何量同时作为模型输入和辅助监督目标,从而把列间关系以显式方式注入扩散过程。

在具体实现上,作者给出了一种基于多层感知机(MLP)的实例化方案。实验结果显示,该方案在取得当前最优基准表现的同时,平均使用的参数量减少了约 3.5 倍;在分类任务中,参数量的缩减幅度最高可达 25 倍。在十个数据集的评测中,该模型在 8/10 的数据集上取得形状(Shape)指标最优,在 7/10 的数据集上取得趋势(Trend)指标最优,并在 9/10 的数据集上下游效用(以 F1 或 RMSE 衡量)表现最佳,同时将 Shape 与 Trend 的误差分别降低了 27% 和 20%。

值得注意的是,该方法并不局限于 MLP 架构。论文指出,其默认的损失权重可以直接迁移到图神经网络(GNN)与 Transformer 去噪器上,并在 27/30 的“架构—数据集”组合中改善了 Shape 指标,在 25/30 的组合中改善了 Trend 指标。为厘清性能提升的来源,作者还进行了匹配的消融实验,结果表明真正驱动增益的是显式的关系监督,而非额外输入或模型容量。这一发现说明,显式的关系型监督是一种可移植的归纳偏置,可推广到不同的表格扩散模型架构中。

该工作已被 ICML 2026 主赛道接收,论文共 24 页,包含 10 幅图和 22 张表格。研究主题涉及机器学习、人工智能以及密码学与安全等方向,论文编号为 arXiv:2606.02607,提交时间为 2026 年 5 月 23 日。整体来看,GATD 通过将列间的几何关系显式化,为表格扩散模型提供了一种更高效、更具泛化性的建模思路。

展开要点与分析

文章情报

工程师进阶

要点

  • GATD将列值差计算的成对角度和长度作为输入及辅助目标,增强表格扩散模型对列间关系的建模。
  • 其MLP版本在十个数据集上取得8/10形状、7/10趋势和9/10下游效用最优,同时参数量平均少3.5倍。
  • 形状和趋势误差分别降低27%和20%,默认损失权重可迁移至GNN和Transformer去噪器。
  • 匹配消融实验表明性能提升来自显式关系监督,而非额外输入或模型容量。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。