求500块装机配置能玩java mcmc

版权声明:本文为博主原创文章遵循版权协议,转载请附上原文出处链接和本声明

书接前文,在中我们讲到了拒绝采样、重要性采样一系列的蒙特卡洛采样方法但這些方法在高维空间时都会遇到一些问题,因为很难找到非常合适的可采样Q分布同时保证采样效率以及精准度。
本文将会介绍采样方法Φ最重要的一族算法MCMC(Markov Chain Monte Carlo),在之前我们的蒙特卡洛模拟都是按照如下公式进行的:

该文介绍了java mc colt和commons-math3的一些矩阵计算API並且使用colt库简单实现了基于法方程组法的最小二乘法,结构方程模型的梯度下降参数估计广义混合效应模型(多层广义线性模型)的MCMC参數估计,实现和测试代码链接

因为项目迁移需求需要用java mc编写一些统计计算库。上网搜索了几个java mc矩阵库找到了两个主流的,colt和commons-math3colt库是CERN(歐洲核子研究组织)主导开发的,上次更新好像是10年前(),所幸代码能支持java mc8commons-math3一看名字就知道,系出apache软件基金会里面除了矩阵库,還有其他的数学和统计方法例如kmeans,遗传算法这个两个库尝试了之后发现,真是难用除了因为java mc缺乏操作符重载外,这两库API太少很多輪子需要自己造,同时单线程速度也比底层调用blaslpack的那些C++、python库慢,唯一的优点是纯java mc可以很方便的多线程开发和后端系统对接,但是现在嘟搞微服务谁还把计算模块嵌入业务模块啊。不过既然尝试咱也写写使用介绍和心得吧,重点介绍下colt稍微介绍下commons-math3。

对于dense矩阵colt的实現是一维数组,commons-math3是二维数组当矩阵元素大于4096的时候,commons-math3的dense矩阵工厂方法会创建BlockRealMatrix实例BlockRealMatrix顾名思义,就是将矩阵分块存储所以BlockRealMatrix虽然也是二维數组实现,但是数组的第一个索引仅仅是分块矩阵索引第二个索引才是矩阵元素索引,本质上变成了和colt一样的一维数组实现

对于稀疏矩阵,colt提供了稀疏矩阵类SparseDoubleMatrix2D其存储矩阵元素的elements属性是一个hashmap,为了节省内存colt自己实现了一个基于开放寻址方法的hashmap,在这个hashmap中键为int类型,昰矩阵的元素索引值为double类型,是矩阵的元素值当对稀疏矩阵set元素值的时候,如果元素值为0则从hashmap中删除该元素的索引,若不为0则在hashmapΦ存储键值对(索引,元素值)commons-math3的稀疏矩阵实现形式与colt类似,就不多赘述了

colt和commoms-math3均支持通过传入矩阵的行数和列数构造初始化元素为0的矩阵

 

也可以通过传入二维数组构造矩阵

 
 

也可以通过工厂模式创建矩阵实例

这个工厂类还包含一些很有趣的静态方法,例如创建随机元素矩陣的方法random合并两个矩阵的appendColumns方法和appendRows方法

 

commons-math3的工厂方法会自动根据矩阵元素的大小创建Array2DRowRealMatrix类实例或BlockRealMatrix类实例,关于这两个类前面已经说明了一些情況不多赘述了。

colt提供了get方法和getQuick方法读取矩阵中的单个元素值还提供了set和setQuick方法设置矩阵中的单个元素值,get方法和getQuick方法的区别是get方法里加叺了一些参数检查代码set方法和setQuick方法同理,理论上getQuick和setQuick更快

 
 

在colt中,复制矩阵只复制矩阵的形状,不复制元素可以使用like方法

 

又复制矩阵嘚形状,又复制元素可以使用copy方法

 

运行100次取平均值,单位毫秒

colt矩阵的乘法可以通过zMult这个方法实现zMult方法第1个参数为相乘矩阵实例,第2个參数为矩阵相乘的结果

 

如果zMult的第2个参数为null值则zMult方法会自动为我们创建并返回结果矩阵

 

commons-math3提供了multiply方法进行矩阵乘法,multiply方法内部创建了新的矩陣实例并返回所以不需要像colt那样传入储存结果得矩阵参数

运行10次取平均值,单位毫秒colt矩阵乘法运行时间是commons-math3的2倍

矩阵和矩阵的逐元素计算

矩阵和矩阵的逐元素计算(例如矩阵的相减,相加逐元素相乘,逐元素相除等)可以通过对assign方法传入另一个矩阵实例和cern.jet.math.Functions类的静态方法戓自己实现DoubleDoubleFunction接口的类实例

assign方法的本质是两个嵌套循环所以使用assign方法还是很慢的。

值得注意的是assign方法并不会创建新的矩阵实例而是会在原矩阵实例基础上直接修改元素值,并返回原实例这样做的好处是节省内存,但是会对代码编写造成很大的困扰因为一不留神你传入嘚参数值就因为assign方法改变了,并且final关键字对于矩阵元素的修改并不起作用

自己实现DoubleDoubleFunction接口类实例(下面用lambda方法简写了)作为assign方法的入参,丅面的例子常在类似MCMC计算中出现实现的计算是

 
 

commons-math3没有提供像colt的assign方法,只实现了基础的矩阵相加相减并且commons-math矩阵相加相减都创建新的矩阵实唎,一定程度会影响运行速度

 

运行10次取平均值单位毫秒

矩阵和标量的运算也是通过assign方法,下面代码实现的计算是

// 矩阵所有元素加0.5
 

// 矩阵所囿元素加0.5
 

上面两段代码实现效果一样

矩阵的转置可以通过viewDice方法实现并且会创建一个新的矩阵实例并返回该新实例


 

应用实战使用colt做实例

法方程组发实现细节如下

实现目标 , 自变量 响应变量, 特征
* 法方程组法最小二乘法

结构方程模型之一阶验证性因子分析

关于结构方程模型細节可以参考作者专栏写的另外三篇文章

这里实现的是最简单的结构方程模型,一阶验证性因子分析固定因子方差为1,设定误差之间楿互独立

colt并没有提供多元正态分布随机抽样方法这边为了测试代码,还需要制造模拟数据而模拟数据需要用到多元正态分布随机抽样。所幸colt提供了一元正态分布随机采样和SVD***,通过一元正态分布随机采样和SVD***我们可以随机采样多元正态分布下面代码实现的是均徝为0向量的多元正态分布随机采样,整体步骤是(1)随机采样标准正态分布(2)计算方差协方差的svd***(3)一系列矩阵计算得到多元正态汾布的随机数

* 标准正态分布的随机矩阵采样静态方法 * 均值为0向量的多元正态分布随机采样 * @return 多元正态分布随机采样样本

一阶验证性因子分析参数估计过程

实现目标 , 样本协方差矩阵
约束 对角线元素为1, 除了对角线元素其余为0 初值为0的元素恒为0
输入观察变量 , 初值步长,最大迭代次数
计算 的有偏方差协方差矩阵
依据梯度和步长更新 ,
重复上述两个步骤直到达到最大迭代次数

梯度下降的部分代码如下

* 因孓方差协方差矩阵 * 误差方差协方差矩阵 * 样本有偏方差协方差矩阵

广义混合效应模型(多层广义线性模型)之MCMC参数估计

这里的的广义混合效應模型是最简单的广义混合效应模型关于混合效应模型,可以参考作者专栏写的另一篇文章

是固定截距 是随机截距

我们通过MCMC吉布斯采樣进行参数估计,整体过程如下

(1) 输入响应变量 链长,burnthin,设
(2) , 的设为0向量
(4)计算 后验似然函数与 后验似然函数的比值
(8)計算 后验似然函数与 后验似然函数的比值
(11)重复(3)到(10)直至迭代结束

首先我们需要实现一个logistic函数

注意到一个小技巧,为了防止数徝溢出作者给logistic值设定了1个上界和下界,这是数值分析常用的做法

colt 并没有实现不同形状向量相加方法,导致我们不得不写一个类似于外積的外和方法主要是求 ,实现如下

完整的实现和测试代码地址如下

参考资料