arrow
返回

A distributed inference algorithm for Dirichlet process mixture models with exponential family components

delete2025-08-06
delete0
delete
OA
AI
R
Reda Khoufache *
M
Mustapha Lebbah
H
Hanene Azzag
É
Étienne Goffinet
D
Djamel Bouchaffra
DOI:10.1016/j.neucom.2025.131119delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
• 在联邦学习场景下,我们提出了一种基于主从架构(Master/Worker)的新型分布式推理框架用于Dirichlet Process Mixture Models(DPMMs)。数据在各工作节点间均匀划分以确保负载均衡。工作节点独立运行,彼此不共享信息,仅通过交换必要的最小统计量与主节点通信。 • 每个工作节点执行局部坍缩Gibbs采样器以发现局部簇并推断局部DPMM,然后将这些局部簇的充分统计量传输至主节点。 • 在主节点层面,仅利用充分统计量估计全局DPMM和聚类结构,无需访问各簇的原始数据。 • 我们在连续和离散数据上验证了该方法的有效性。对于连续数据,我们聚焦于多元高斯混合模型,在包含100 K个数据点的数据集上完成100次迭代仅需3分钟,而集中式坍缩Gibbs采样器需12小时。对于离散数据,我们引入了一个多项式DPMM并应用于文本聚类,展示了该框架的通用性。 • 此外,我们还展示了该方法如何推广至指数分布族并保持适用性,同时提供了计算细节。模型的整体工作流程如图1所示。
Keyword:
Horizontal federated learning
Distributed computing
Dirichlet process mixture models
Markov chain Monte Carlo
Bayesian non-parametric modeling

期刊

Neurocomputing 封面图
Neurocomputing
IF:
6.5
论文数:
2.5W
被引数:
6.5W

机构

S
Sorbonne Paris Nord University
学者数:
36
论文数: 25
被引数: 0
C
cerebras systems
学者数:
2
论文数: 2
被引数: 0
U
University of Versailles
学者数:
3
论文数: 1
被引数: 0
学者 查看更多机构
引用论文

引用论文

Variational inference for Dirichlet process mixtures
err2006-03-01
err0
errOAAI
errDavid M. Blei; Michael I. Jordan
err分享
err收藏
学者 查看更多内容