用Transformer解决高斯混合模型,兼具理论与实证优势。
Transformers as Unsupervised Learning Algorithms: A study on Gaussian Mixtures
- 设计TGMM框架,共享主干网络统一处理多个GMM任务。
- 实验证明其优于传统EM和谱方法,且对分布偏移更鲁棒。
- 首次证明Transformer可逼近EM算法与张量迭代核心步骤。
Transformer架构在现代人工智能中展现出卓越能力,尤其在推理时隐式学习内部模型的能力,被认为是理解预训练大语言模型的关键。然而,现有研究多集中于有监督学习如上下文学习,对无监督学习关注不足。本文从统计估计视角,探究Transformer求解高斯混合模型(GMM)的能力。提出基于Transformer的TGMM框架,通过共享主干网络同时学习多个GMM任务。实验表明,该模型有效克服了经典方法(如期望最大化算法EM或谱方法)的局限性,且对分布偏移具有合理鲁棒性。理论上,证明了Transformer可逼近EM算法及谱方法的核心组件——三次张量幂迭代。这些成果弥合了实践成功与理论理解之间的鸿沟,将Transformer定位为通用的无监督学习工具。
原文摘要 · Abstract (English)
The transformer architecture has demonstrated remarkable capabilities in modern artificial intelligence, among which the capability of implicitly learning an internal model during inference time is widely believed to play a key role in the under standing of pre-trained large language models. However, most recent works have been focusing on studying supervised learning topics such as in-context learning, leaving the field of unsupervised learning largely unexplored. This paper investigates the capabilities of transformers in solving Gaussian Mixture Models (GMMs), a fundamental unsupervised learning problem through the lens of statistical estimation. We propose a transformer-based learning framework called TGMM that simultaneously learns to solve multiple GMM tasks using a shared transformer backbone. The learned models are empirically demonstrated to effectively mitigate the limitations of classical methods such as Expectation-Maximization (EM) or spectral algorithms, at the same time exhibit reasonable robustness to distribution shifts. Theoretically, we prove that transformers can approximate both the EM algorithm and a core component of spectral methods (cubic tensor power iterations). These results bridge the gap between practical success and theoretical understanding, positioning transformers as versatile tools for unsupervised learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。