arXiv · ML· Tianqi Zhao, Guanyang Wang, Yan Shuo Tan, Qiong Zhang·· 9 天前AI 评分21
TabClustPFN：用于表格数据聚类的先验数据拟合网络
TabClustPFN: A Prior-data Fitted Network for Tabular Data Clustering
AI 导读
TabClustPFN 提出了一种通过联合设计先验、目标和架构来解决表格数据聚类挑战的 PFN，在 44 个真实世界基准上达到 SOTA 性能。 其混合预训练先验捕捉异构几何，解耦的分区推断网络在单次推理中联合推断聚类分配和基数。 SoftARI 训练目标具有置换不变性，消除了标签顺序约定需求，运行时间可比拟高效基线，代码已开源。
来源：arXiv · ML · arxiv.org