【机器学习-无监督学习】概率图模型

机器学习是一门人工智能的分支学科，通过算法和模型让计算机从数据中学习，进行模型训练和优化，做出预测、分类和决策支持。Python成为机器学习的首选语言，依赖于强大的开源库如Scikit-learn、TensorFlow和PyTorch。本专栏介绍机器学习的相关算法以及基于Python的算法实现。

【GitCode】专栏资源保存在我的GitCode仓库：https://gitcode.com/Morse_Chen/Python_machine_learning。

  本文讨论无监督学习中的数据分布建模问题。当我们需要在一个数据集上完成某个任务时，数据集中的样本分布显然是最基本的要素。面对不同的数据分布，我们可能针对同一任务采用完全不同的算法。例如，如果样本有明显的线性相关关系，我们就可以考虑用基于线性模型的算法解决问题；如果样本呈高斯分布，我们可能会使用高斯分布的各种性质来简化任务的要求。上一篇文章介绍的数据降维算法，也是为了在数据分布不明显的前提下，尽可能提取出数据的关键特征。因此，如何建模数据集中样本关于其各个特征的分布，就成了一个相当关键的问题。

  我们从图1的场景中看起。一群人在秋天出游，银杏满地，风光无限。但是，在同一个温度下，不同人的衣着也有差异。有人穿了厚厚的大衣，有人穿了长袖长裤，还有人穿着短袖或者裙子。可以设想，如果气温再高一些，穿短袖的人会增加；如果气温再低一些，恐怕就不会有人再穿短袖了。因此我们可以认为，人群穿衣选择的概率分布受到天气的影响。

图1 不同的人的衣着背后有一个数据分布

  我们先从最简单的表格数据看起。假设表1中是天气和人群中衣服选择的部分数据，我们可以直接从中写出最简单的数据分布：

表1 不同天气和穿衣选择的概率

  以此类推，我们可以把表中的每一行都写出来，得到样本的分布。但是，这样的做法显然过于低效。当特征的数目增加时，我们按此建模的复杂度将成指数增长。因此，我们需要设法寻找不同特征之间的相关性，降低模型的复杂度。例如，根据生活常识，我们可以认为人们选择衣服的概率应该是和天气有关的。天气热时，人们更倾向于选择衬衫，而天气冷时倾向于大衣。这样，我们可以将上面的分布转化为条件概率：

  通过这种方式，我们可以建立起样本不同特征之间的关系。如果用随机变量

表示天气，

表示衣服，那么上述的关系可以表示为图2中的结构。

图2 天气与衣服之间的概率模型

  图2中，从

指向

的箭头表示随机变量

依赖于

。把样本的所有特征按依赖关系列出来，每个特征作为一个顶点，每对依赖关系作为一条边，就形成了一张概率图。我们可以通过概率图中体现的不同特征之间的关系，推断出数据的概率分布。像上面这样，由依赖关系构成的概率图是有向图，称为贝叶斯网络（Bayes networks）。如果我们只知道两个特征之间相关，但没有明确的单向依赖关系，就可以用无向图来建模，称为马尔可夫网络（Markov networks）。下面，我们就来介绍这两种概率图模型的具体内容。

一、贝叶斯网络

  贝叶斯网络又称信念网络（belief network），与概率中的贝叶斯推断有很大关联。由于网络中的有向关系清楚地表明了变量间的依赖，我们可以根据网络结构直接写出变量的分布。例如，设变量

、

和

构成的贝叶斯网络如图3所示。

图3 3个变量的贝叶斯网络

从图中可以看出，

依赖于

，

同时依赖于

和

，于是其联合分布

可以写为

对于一般的有

个节点

的贝叶斯网络，记

为所有有边指向

的节点，也即是

在图上的父亲节点集，那么其概率分布为

  根据贝叶斯网络中，我们可以清晰地判断变量之间是否独立。如图4所示，我们以3个变量

、

和

为例来说明3种不同的依赖关系。在图4(a)中，

和

分别依赖于

，但是

与

之间没有直接关联。直接根据图写出三者之间的联合分布为

从上式中我们并不能直接得到这些变量间的独立性。但是，如果给定变量

，上式就变为

这说明变量

与

在给定

的条件下独立，记作

。本文最开始所举的天气和穿衣的例子中，把穿什么衣服看作变量

，天气看作变量

，如果在加上一个是否打伞的变量

，他们之间就满足这样尾对尾的依赖关系。条件独立（conditional independence）是概率图模型中最重要的基本概念之一，它直接刻画了一个多变量联合分布中变量之间的依赖关系，从而影响建模的方式。

图4 3个变量的依赖关系

  接下来我们考虑图4(b)展示的头对头关系。同样，可以根据图直接写出三者的联合分布

而根据条件概率公式，有

，于是我们得到

因此，在这一关系中，

和

是天然独立的，不需要条件，记作

。然而，一旦

给定，

与

就会由

产生联系，用条件概率写为

这种情况似乎有些反直觉，为何原本独立的

和

在观测到

后反而不独立了？这是因为

引入了额外信息，我们用带概率的逻辑与来解释这一现象。假设

、

和

的取值都是0或1，且

与

的取值相互独立，都满足

。

的取值

与

的关系如表2所示。

表2 带概率的逻辑与真值表

000100.2010.2111

  假如我们观测到

，那么

和

取0的概率是多少？我们可以进行如下简单的计算：

由于

和

是对称的，同样也有

。但是，此时

和

已经不独立了，我们可以通过计算

来验证这一观点。考虑进一步观测到

的前提下

的概率，计算可得

  这一结果说明，当观测到

时，已经有很大概率导致

，

是否等于0就变得没有那么重要了。也就是说，在给定

的条件下，

的结果会影响

取值的概率，从而

。

  最后，我们来看图4

的头对尾关系，这一关系比较好理解。首先，

可以通过

影响

，所以

与

之间不独立。用数学语言来说，它们的联合分布为

当给定

后，

与

的联系就被切断了，这时考察

与

的联合分布

，有

因此，

与

关于

是条件独立的，即

。这3种依赖关系是贝叶斯网络中各种复杂依赖的基础，通过这3种依赖的拆解就可以分析更加复杂的贝叶斯网络。

二、最大后验估计

  在逻辑斯谛回归一文中，我们曾用最大似然估计的思想推导出了逻辑斯谛回归的优化目标。由于贝叶斯网络中的有向边清晰地表明了先验（prior）与后验（posterior）的关系，我们可以通过它推导出类似的结果。设数据集

，

表示所有样本，

表示所有标签。假设样本与标签服从参数为

的分布，但是

是未知的，我们的目标就是根据观测到的数据计算

的后验分布

。例如，在上面天气与衣服选择的例子中，我们可以把每个人看成样本

，选择的衣服看成标签

，天气看成参数

。因此，参数

、样本

和样本标签

的依赖关系可以用贝叶斯网络表示为图5的结构。

图5 线性模型的贝叶斯网络

  这里，图5(a)是完整的贝叶斯网络。可以看出，对每个样本和标签来说，它们之间的依赖关系都是相同的，在网络中表现为大量重复的结构。对于这样的重复，我们通常将其简写为图5(b)的形式，用一组

的关系代表所有相似的节点，并在外面加一个方框，框里的

表示该重复的个数。此外，由于样本

对标签

有影响，但我们无法控制或者通常不关心其分布，因此在模型中将其看作定值，图5中用橙色节点表示。此外，根据尾对尾的依赖法则，我们可以判定，不同样本的标签

之间相对模型参数

条件独立。这样，

和

的概率分布为

其中，参数自身的分布

就是先验分布，而

项之间之所以可以连乘，就是因为条件独立的性质。进一步，根据贝叶斯公式，模型参数

的后验分布可以表示为

上式中的分母

同样被数据集完全确定，属于常数。通过贝叶斯网络，我们就确定了待求参数

的后验分布

的表达式。

  如果要进一步求解该后验分布，我们必须对参数空间和样本与标签之间的关系做出假设，以写出

和

的具体形式。我们以线性回归为例，假设参数

的先验分布为高斯分布，即

，其中

的每个维度相互独立，都服从均值为0、方差为

的高斯分布，我们用

来表示。样本的标签

和样本

满足

，用

表示。由于引入了新的量

和

，我们将概率图更新为图6。

图6 高斯假设下线性模型的贝叶斯网络

  两个新引入的量都可以视为常量，因此在图中也用橙色节点表示。把上述的表达式代入后验分布，就得到

设样本的维度是

，那么参数

。上式中高斯分布的具体表达式为

  与最大似然估计的思路类似，这里我们希望

让后验最大化，称为最大后验（maximum a posteriori，MAP）估计。此外，我们对后验两边取对数，让连乘变为连加，并代入高斯分布的表达式，可以计算出

因此，我们最终要解决的优化问题是

  可以发现，该目标函数与线性回归中用MSE损失函数、约束强度为

的

正则化约束所得到的的结果完全一致。如果把标签

关于样本

的分布从高斯分布改为二项分布等其他形式，就可以得到其他的朴素贝叶斯模型。对于更复杂的变量依赖结构和分布模型，我们也可以用贝叶斯网络建模，再用最大化后验的思路求解。

  最大似然估计与最大后验估计是机器学习中常用的两种求解模型参数的方式，但两者有所不同。设数据集为

，模型参数为

。MLE考虑哪个参数生成当前数据集的概率最大，因此求解最大化的对数似然：

另一方面，MAP考虑最大化参数的后验分布

。利用贝叶斯公式，可以得到

  两者对比可以发现，MAP的优化目标比MLE多了一项

，也即参数

的先验分布的对数。因此MAP相当于在MLE的基础上引入了我们对参数的先验假设，对参数的分布添加了一定限制。从MLE的角度来考虑，这种做法等价于引入正则化约束。所以，上面我们用MAP推导出的线性回归是自然带有

正则化约束的。贝叶斯模型给了我们一种理解正则化约束的更自然的视角。

三、用朴素贝叶斯模型完成文本分类

朴素贝叶斯（naive Bayes）是贝叶斯公式和贝叶斯网络模型的最简单应用。顾名思义，朴素贝叶斯模型只使用基本的条件独立假设和计数方法，统计各个变量的先验分布，再由贝叶斯公式反推出参数的后验。应用到分类模型中，待求的参数就是每个样本的类别。设样本的特征是

，类别是

，那么它们之间的依赖关系可以用贝叶斯网络表示为图7中的结构。可以看出，朴素贝叶斯模型是一个生成模型（generative model），从每个样本的类别标签生成整个样本的特征。

图7 朴素贝叶斯分类

  这里，我们事实上隐含了各个特征之间相互独立的假设。具体来说，朴素贝叶斯模型假设：给定每个样本的类别

，样本特征变量

之间是相互独立的，也即是

这样就有可以将特征变量的联合概率拆解为独立概率的乘积：

虽然这一假设在实际中不甚合理，但是作为最简单的朴素模型，这一假设可以大大简化我们计算的难度。下面，我们按与上节相同的步骤，在样本特征

给定的前提下最大化类别

的后验概率，作为对样本类别的预测：

其中

是总的类别数量，由于

是离散的类别，我们直接把分布

写成概率

。因此，朴素贝叶斯分类模型的核心就是统计

的先验概率

和各个特征的概率

。

  下面，我们在新闻数据集“The 20 Newsgroups”上用朴素贝叶斯分离器完成新闻分类任务。该数据集包含了20个类别的大概20000篇新闻，涵盖了计算机、科学、体育等等主题，每个主题下的新闻数量大致相等。我们的目标是根据新闻中出现的单词判断出新闻属于哪个类别。首先，我们导入必要的库和数据集，该数据集可以直接从sklearn中获取到。

  考虑到文本处理不属于重点讲述范围，我们直接从sklearn中下载处理好的数据集，仅在这里简要介绍数据处理的方法。首先，我们将文本按照空白字符分隔成一个个单词，再将所有长度在2以下的单词（如I，a，单独的数字和符号）去除。对于剩下的单词，我们建立起词汇表，设其大小为

，那么每个单词都可以按照它在词汇表中的位置，用一个独热向量表示。假设“the”是词汇表中的第一个单词，那么它的向量表示就是

，共有

维。把一篇新闻中的所有单词都用独热向量表示后，我们把这些独热向量相加，就得到表示该文章的向量。例如，词汇表中有3个单词“the”、“dog”和“cat”，其向量表示分别为

、

和

，那么“the cat”的向量表示就是

。这一表示方法完全忽略了单词之间的先后顺序和文章的整体上下文结构，是一种非常粗略的表示方法，但对我们的朴素贝叶斯模型来说已经足够。由于一篇文章中包含的单词相比于整个词汇表来说非常有限，其向量表示也很稀疏，因此我们获得的数据集也是用稀疏矩阵表示的。fetch_20newsgroups_vectorized函数存储所用的是SciPy库的稀疏矩阵，输出中(i,j) k表示矩阵第

行

列的值为

。在本例中，其具体含义为新闻

中单词

出现了

次（省略部分程序输出内容）。

import numpy as np
from sklearn.datasets import fetch_20newsgroups_vectorized
from tqdm import trange

# normalize表示是否对数据归一化，这里我们保留原始数据
# data_home是数据保存路径
train_data = fetch_20newsgroups_vectorized(subset='train', normalize=False, data_home='20newsgroups')
test_data = fetch_20newsgroups_vectorized(subset='test', normalize=False, data_home='20newsgroups')
print('文章主题：', '\n'.join(train_data.target_names))
print(train_data.data[0])

  接下来，我们统计训练集中的

和

，这里

是新闻的主题，

是单词。对于给定离散数据，我们直接用频率代替概率，因此

是不同主题新闻的频率，

是主题为

的新闻下单词

出现的频率。在上一节中，我们认为数据集中的样本先验概率与模型无关，但对于文本中的单词来说，虽然一篇文章中不可能出现所有单词，但这并不代表没有出现的单词概率就是零。一般来说，我们会为所有单词设置一个先验计数

，通常取

，真实计数将在先验计数的基础上累加。

# 统计新闻主题频率
cat_cnt = np.bincount(train_data.target)
print('新闻数量：', cat_cnt)
log_cat_freq = np.log(cat_cnt / np.sum(cat_cnt))

# 对每个主题统计单词频率
alpha = 1.0
# 单词频率，20是主题个数，train_data.feature_names是分割出的单词
log_voc_freq = np.zeros((20, len(train_data.feature_names))) + alpha
# 单词计数，需要加上先验计数
voc_cnt = np.zeros((20, 1)) + len(train_data.feature_names) * alpha
# 用nonzero返回稀疏矩阵不为零的行列坐标
rows, cols = train_data.data.nonzero()
for i in trange(len(rows)):
    news = rows[i]
    voc = cols[i]
    cat = train_data.target[news] # 新闻类别
    log_voc_freq[cat, voc] += train_data.data[news, voc]
    voc_cnt[cat] += train_data.data[news, voc]

log_voc_freq = np.log(log_voc_freq / voc_cnt)

  至此，统计的信息已经足够我们判断其他新闻的类型了。我们遍历所有主题

，计算对数后验，并返回使对数后验最大的主题。

def test_news(news):
    rows, cols = news.nonzero()
    # 对数后验
    log_post = np.copy(log_cat_freq)
    for row, voc in zip(rows, cols):
        # 加上每个单词在类别下的后验
        log_post += log_voc_freq[:, voc]
    return np.argmax(log_post)

  最后，我们在测试集的所有新闻上查看我们的分类准确率，只有当我们判断的类别与真实类别相同时才算作正确。

preds = []
for news in test_data.data:
    preds.append(test_news(news))
acc = np.mean(np.array(preds) == test_data.target)
print('分类准确率：', acc)

  在sklearn中也提供了朴素贝叶斯分类器，对于本例中的离散特征多分类任务，我们选用MultinomialNB进行测试，并与我们自己实现的效果比较。该分类器同样有默认参数

，大家可以通过调整

的值，观察分类效果的变化。

from sklearn.naive_bayes import MultinomialNB

mnb = MultinomialNB(alpha=alpha)
mnb.fit(train_data.data, train_data.target)
print('分类准确率：', mnb.score(test_data.data, test_data.target))

四、马尔可夫网络

  与贝叶斯网络不同，某些情况下，变量之间的关联是双向的。例如在KNN一文中，我们曾用到了照片中某一像素和周围像素大概率很接近这一假设。如果把每个像素看作一个变量，那么它和周围的像素就是互相影响、互相依赖的。这时，我们可以把贝叶斯网络的有向图变为无向图，构造出马尔可夫网络。

  图8展示了一个简单的马尔可夫网络，其中包含5个变量

、

、

、

和

，每个变量在图中为一个节点，两个节点之间的边代表这两个变量之间存在直接依赖关系。与贝叶斯网络的有向图不同，由于依赖是双向的，我们无法将这些变量的联合分布通过条件概率直接拆分。和贝叶斯网络不同，马尔可夫网络更多地用于描述几个变量之间是如何相互依赖的，在物理上更像是一种“场”的体现，因此它又称为马尔可夫随机场（Markov random fields）。例如穿衣时上衣、裤子和鞋子的搭配，三者是相互依赖的，并没有明确哪个是因、哪个是果。因此，我们需要通过其他方式分析马尔可夫网络中变量之间的依赖方式。通过观察可以发现，如果两个变量之间所有路径上的变量全部已知，那么这两个变量相互独立。例如在图8中，假如我们给定了

和

，它们就不再被视为变量，与其相连的边也失去了意义。这时，变量

和

之间不再有路径可以互相到达，它们之间的关联也就不存在了。我们记为

，表示在给定

与

时，

和

条件独立。

图8 简单的马尔可夫网络

  我们可以从这一性质出发，继续考虑马尔可夫网络的拆分。对于网络中的任意两个不同节点

和

，记

为除去这两个节点以外的所有节点。当

给定时，如果

与

没有直接相连，那么它们之间的所有路径上的变量必然都是固定的，两者条件独立。因此，这两个变量的条件联合分布可以拆分为：

如果变量

与

之间有边相连，那么无论其他变量如何，它们之间必然存在依赖关系，无法通过中间变量拆分。所以，我们可以按照变量之间的连接关系，将网络拆分成一些内部紧密连接、相互条件独立的部分。

  在图论中，如果一张无向图中的某些节点之间两两互相连接，我们就称这些节点组成了一个团（clique）。在图8中，大小为2的团有6个，分别是

、

、

、

、

和

；大小为3的团有1个，是

。需要注意，

不是团，因为

和

之间没有边连接。根据上面的推导，我们可以将团视为马尔可夫网络的基本单位，每个团之内的变量互相之间存在无法拆分的依赖关系。然而，有些较小的团是被较大的团所包含的，例如上面的团

中就包含了3个大小为2的团

、

和

。如果我们用函数来描述一个团中变量之间的关联，那么较小团中的关联是被较大团中的关联所包含的。因此，我们定义不被其他团包含的团为极大团（maximal clique）。在上面的例子中，极大团共有4个，分别是

、

、

和

。我们只需要将马尔可夫网络拆分成极大团，就可以涵盖所有存在的变量关联。

  记团

中所有变量为

，设其在整个网络的联合分布

拆分后的因子可以用函数

表示。由上所述，我们只考虑极大团的因子，将

拆分为

其中的乘积遍历所有极大团，

是归一化因子，称为配分函数（partition function）。在图8的网络中，按此方式得到的联合分布为

由于概率分布必须处处非负，所有的函数

都必须满足

，我们将其称为势函数（potential function）。势函数的具体形式可以通过对问题的先验知识规定。例如，势函数

会让变量

和

倾向于取值相同。

  如果势函数不仅非负，而且严格为正，我们可以用能量函数

来表示势函数，使得

。这样，联合分布就由连乘转化为指数上的求和：

再利用我们已经多次使用过的求对数的技巧，将寻找

的最大值转换为寻找

的最大值，得到

  该优化问题是对所有极大团的能量的和进行优化，且能量函数的取值和形式没有限制，相比于直接求解势函数的乘积方便很多。这一思想来自于物理中的势和势能，以及玻尔兹曼分布，感兴趣的可以自行查阅相关资料。下面，我们以图像去噪为例，展示马尔可夫网络的一个简单应用。

五、用马尔可夫网络完成图像去噪

  在k近邻算法中我们已经讲过，对于一张有意义的图像，其每个像素点的颜色和周围像素点大概率相近，并且这一关联是双向的，符合马尔可夫网络的模型。当一张图像中存在噪点时，它们与周围像素的关联大概率比真实像素弱，反映在能量函数上，其整体的能量应当较大。因此，我们可以通过设计合适的能量函数并在图像上进行优化，还原出噪点的真实颜色。

  简单起见，我们用一张只有两种颜色的图像来演示。如图9所示，上半部分是只有黑色与白色的原始图像，我们随机将其中10%的像素进行黑白反转，得到了下半部分带有噪声的图像。

图9 原始图像与带噪图像

  为了对图像添加噪声前后的状态建模，设图像上像素的真实颜色

，添加噪声后观察到的颜色

。由于不同像素的颜色是否翻转是相互独立的，某个像素的显示颜色

只和其真实颜色

有关，而与其他像素无关。对于像素间的关联，我们只考虑最简单的上下左右4个像素。这样，图像的真实颜色

与显示颜色

就构成了一个马尔可夫网络网络，其结构如图10所示。下层的蓝色节点表示图像的真实颜色，上层的橙色节点表示图像的显示颜色。从图中可以看出，该网络中极大团只有两种，一种是相邻像素的真实颜色的关联

，另一种是像素真实颜色与显示颜色的关联

。这样的网格状结构中，极大团的大小都是2。接下来，我们需要为极大团设计能量函数。考虑到这些极大团的对称性，并忽略边缘处网络结构细微的不同，我们只需要为每种极大团设计一个函数，而不需要为每个极大团分别设计。

图10 去噪任务的马尔可夫模型

  首先考虑真实像素间的关联，我们预计相邻的像素应当较为相似。由于图像中只有两种颜色-1和1，我们可以用相邻像素颜色的乘积

来判断它们是否相同。当颜色相同时，乘积为1，反之为-1。考虑到颜色相同时能量应当较低，我们用其乘积的相反数作为能量，即

其中，

是常数系数，且函数只对相邻的像素有定义。对于真实像素与对应的显示像素间的关联，我们期望像素的颜色没有被反转，显示的颜色与真实颜色相同。因此，我们可以得到与上面相似的能量函数

其中，

是另一个常数系数。记

为像素

相邻像素的集合，

为图像中像素的总数量，将能量函数对网络中所有的极大团求和，就得到总能量为

式中的第一项由于相邻像素对的能量被计算了两遍，额外乘上了

。

  由于图像中像素数量

往往较大，直接优化

较为困难。考虑到问题中变量的取值范围很小，我们每次优化一个像素

，判断改变

的值是否可以降低网络的总能量。下面，我们就在图9展示的图像上动手实现马尔可夫网络为图像去噪。

import matplotlib.image as mpimg
import matplotlib.pyplot as plt

# 读取原图
orig_img = np.array(mpimg.imread('origimg.jpg'), dtype=int)
orig_img[orig_img < 128] = -1 # 黑色设置为-1
orig_img[orig_img >= 128] = 1 # 白色设置为1

# 读取带噪图像
noisy_img = np.array(mpimg.imread('noisyimg.jpg'), dtype=int)
noisy_img[noisy_img < 128] = -1
noisy_img[noisy_img >= 128] = 1

  为了在后续过程中计算去噪的效果，我们定义一个函数来计算当前图像与原图中不一致的像素比例。

def compute_noise_rate(noisy, orig):
    err = np.sum(noisy != orig)
    return err / orig.size

init_noise_rate = compute_noise_rate(noisy_img, orig_img)
print (f'带噪图像与原图不一致的像素比例：{init_noise_rate * 100:.4f}%')

在这里插入图片描述

  由于图像在计算机中本身就是用二维数组存储，其结构与我们设计的马尔可夫网络相同，我们不需要再额外实现数据结构来存储马尔可夫网络。下面，我们就来直接实现逐像素优化的算法。在该算法中，优化每个像素时，能量的改变都是局部的。因此，我们只需要计算局部的能量变化，无须每次都重新计算整个网络的能量。

# 计算坐标(i, j)处的局部能量
def compute_energy(X, Y, i, j, alpha, beta):
    # X：当前图像
    # Y：带噪图像
    energy = -beta * X[i][j] * Y[i][j]
    # 判断坐标是否超出边界
    if i > 0:
        energy -= alpha * X[i][j] * X[i - 1][j]
    if i < X.shape[0] - 1:
        energy -= alpha * X[i][j] * X[i + 1][j]
    if j > 0:
        energy -= alpha * X[i][j] * X[i][j - 1]
    if j < X.shape[1] - 1:
        energy -= alpha * X[i][j] * X[i][j + 1]
    return energy

  最后，我们定义超参数，并将图像去噪结果随训练轮数的变化绘制出来。可以看出，随着迭代进行，图中在大范围色块内的噪点基本都消失了。

# 设置超参数
alpha = 2.1
beta = 1.0
max_iter = 5

# 逐像素优化
# 复制一份噪声图像，保持网络中的Y不变，只优化X
X = np.copy(noisy_img)
for k in range(max_iter):
    for i in range(X.shape[0]): # 枚举所有像素
        for j in range(X.shape[1]):
            # 分别计算当前像素取1和-1时的能量
            X[i, j] = 1
            pos_energy = compute_energy(X, noisy_img, i, j, alpha, beta)
            X[i, j] = -1
            neg_energy = compute_energy(X, noisy_img, i, j, alpha, beta)
            # 将该像素设置为使能量最低的值
            X[i, j] = 1 if pos_energy < neg_energy else -1
    
    # 展示图像并计算噪声率
    plt.figure()
    plt.axis('off')
    plt.imshow(X, cmap='binary_r')
    plt.show()
    noise_rate = compute_noise_rate(X, orig_img) * 100
    print(f'迭代轮数：{k}，噪声率：{noise_rate:.4f}%')

  如果图像的图案更复杂、色彩更丰富，我们也可以使用类似的方法去噪，但是通常需要我们使用先验知识，设计更复杂的网络结构和能量函数。例如考虑彩色图像中的颜色渐变、考虑更大范围像素之间的关联等。在上例中，当黑色与白色给出的能量相同时，我们默认将其设置为白色。对于更复杂的情况，我们可以为每个像素单独设置一个能量函数

，表示其颜色具有的能量，以此来规定图像整体的颜色偏好。

附：以上文中的数据集及相关资源下载地址： 链接：https://pan.quark.cn/s/bfc9cea33f73 提取码：fASw