返回行业动态

AI2推出DiScoFormer模型:单Transformer统一估计分布密度与分数

2026/06/29 18:04
查看原文

OmniTools 6月30日消息,艾伦人工智能研究所(AI2)在Hugging Face博客发布新模型DiScoFormer,旨在以单一Transformer架构同时估计任意数据分布的密度与分数(score,即对数密度梯度)。该模型无需针对不同分布重新训练,仅需一次前向传播即可完成两项任务。

DiScoFormer采用共享主干与双输出头设计,利用分数与密度的数学关系构建无标签一致性损失,在推理时可对分布外输入进行即时自适应优化。研究表明,其交叉注意力机制可解析复现传统核密度估计(KDE),并进一步实现多尺度自适应建模。

实验显示,在100维高维空间中,DiScoFormer相较最优手工调参KDE,分数误差降低约6.5倍,密度误差降低超37倍;且在未见过的多模态混合分布及Laplace、Student-t等非高斯分布上保持强鲁棒性。模型基于高斯混合模型(GMM)生成海量监督数据进行训练,相关技术报告已发布于arXiv。

相关背景

想继续了解,可以看这些

从这条动态出发,继续查看相关分析、产品详情和同主题更新。

最新工具

刚收录的 AI 工具,适合顺手发现可用产品。

查看全部