arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.02676stat.ME
收藏

协变量依赖的分层狄利克雷过程

Covariate-dependent hierarchical Dirichlet processes

Huizi Zhang, Sara Wade, Natalia Bochkina

AI总结:

本文提出一种结合协变量信息的分层贝叶斯非参数模型,用于密度估计和跨组聚类,通过数据增广实现高效推断,并在单细胞测序和钙成像数据上验证了其有效性。

摘要:

贝叶斯分层建模是有效整合数据并在各组之间借用信息的自然框架。在本文中,我们通过提出一种纳入额外协变量信息的分层贝叶斯方法,解决了与密度估计和识别相关组间聚类相关的问题。为了实现灵活性,我们的方法基于贝叶斯非参数思想,将分层狄利克雷过程与依赖狄利克雷过程相结合。所提出的模型具有广泛的适用性,通过适当的核函数适应多种和混合类型的协变量,并通过合适的特定于组分的似然函数适应不同的输出类型。这扩展了我们辨别协变量与聚类之间关系的能力,同时有效地借用信息并量化各组之间的差异。通过采用数据增广技巧,我们能够处理棘手的归一化权重,并构建用于后验推断的马尔可夫链蒙特卡罗算法。所提出的方法在模拟数据以及两个真实数据集(单细胞RNA测序(scRNA-seq)和钙成像)上进行了说明。对于scRNA-seq数据,我们表明细胞动力学的纳入有助于发现额外的细胞亚群。在钙成像数据上,我们的方法识别出具有相似神经活动的时间帧的可解释聚类,与观察到的动物行为一致。

英文摘要:

Bayesian hierarchical modeling is a natural framework to effectively integrate data and borrow information across groups. In this paper, we address problems related to density estimation and identifying clusters across related groups, by proposing a hierarchical Bayesian approach that incorporates additional covariate information. To achieve flexibility, our approach builds on ideas from Bayesian nonparametrics, combining the hierarchical Dirichlet process with dependent Dirichlet processes. The proposed model is widely applicable, accommodating multiple and mixed covariate types through appropriate kernel functions as well as different output types through suitable component-specific likelihoods. This extends our ability to discern the relationship between covariates and clusters, while also effectively borrowing information and quantifying differences across groups. By employing a data augmentation trick, we are able to tackle the intractable normalized weights and construct a Markov chain Monte Carlo algorithm for posterior inference. The proposed method is illustrated on simulated data and two real data sets on single-cell RNA sequencing (scRNA-seq) and calcium imaging. For scRNA-seq data, we show that the incorporation of cell dynamics facilitates the discovery of additional cell subgroups. On calcium imaging data, our method identifies interpretable clusters of time frames with similar neural activity, aligning with the observed behavior of the animal.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑