（24）张量网络图像生成与压缩

最新推荐文章于 2023-05-30 23:07:24 发布

计算机量子狗

最新推荐文章于 2023-05-30 23:07:24 发布

阅读量407

点赞数

分类专栏：冉仕举视频学习

本文链接：https://blog.csdn.net/weixin_49883619/article/details/111941613

版权

冉仕举视频学习专栏收录该内容

25 篇文章 37 订阅

订阅专栏

1.非监督张量网络机器学习

当获得量子态 $|\psi)$ 后，不但可求出像素的联合概率密度, 还可计算条件概率。
以图像修复为例：

1.1求 $P\left(\left\{x_{n}^{(B]}\right\} \mid\left\{x_{m}^{[A]}\right\}\right)$

考虑已知 $|\psi\rangle$ 与图片中的部分像素 (记为 $\left.\left\{x_{m}^{[A]}\right\}\right),$ 则剩余未知像素 $\left(\right.$ 记为 $\left.\left\{x_{n}^{[B]}\right\}\right)$ 的概率分布可由如下条件概率给出（注： $\left\{x_{m}^{[A]}\right\} \cup\left\{x_{n}^{[B]}\right\}$ 给出整张图片）：
$P\left(\left\{x_{n}^{[B]}\right\} \mid\left\{x_{m}^{[A]}\right\}\right)=\left(\prod_{\otimes n}\left\langle x_{n}^{[B]} \mid \tilde{\psi}\right\rangle\right)^{2}$ 量子态 $|\widetilde{\psi}\rangle$ 为通过对 $|\psi\rangle$ 的投影测量获得
$|\tilde{\psi}\rangle=\frac{1}{Z} \prod_{\otimes m}\left\langle x_{m}^{[A]} \mid \psi\right\rangle$ 其中 $Z$ 为归一化系数。 $|\tilde{\psi}\rangle$ 即为 $\left\{x_{n}^{[B]}\right\}$ 对应的量子概率模型，其包含的qubit数与 $\left\{x_{n}^{[B]}\right\}$ 中像素个数一致。
在这里插入图片描述

在等概率的先验分布前提下，上述条件概率的定义与之前子系统概率公式自洽。根据概率公式：
$\begin{aligned}P\left(\left\{x_{n}^{[B]}\right\}\right)&=\sum_{\left\{x_{m}^{(A)}\right\}} P\left(\left\{x_{m}^{[A]}\right\} \cup\left\{x_{n}^{[B]}\right\}\right)\\&=\sum_{\left\{x_{m}^{(A)}\right\}} P\left(\left\{x_{n}^{[B]}\right\} \mid\left\{x_{m}^{[A]}\right\}\right) P\left(\left\{x_{m}^{[A]}\right\}\right)\end{aligned}$ 先验分布近似为等概率分布的前提下，即： $P\left(\left\{x_{m}^{[A]}\right\}\right) = \frac{1}{Z}$ 有:
$P\left(\left\{x_{n}^{[B]}\right\}\right)=\frac{1}{Z} \sum_{\left\{x_{m}^{[A]}\right\}} P\left(\left\{x_{n}^{[B]}\right\} \mid\left\{x_{m}^{[A]}\right\}\right)$ 注： $\operatorname{Tr}_{\left\{x_{m}^{[\mathrm{A}]}\right\}}=\sum_{\left\{x_{m}^{(A)}\right\}}$
因为
$P\left(\left\{x_{n}^{[B]}\right\} \mid\left\{x_{m}^{[A]}\right\}\right)=\left(\prod_{\otimes n}\left\langle x_{n}^{[B]} \mid \tilde{\psi}\right\rangle\right)^{2}$ $|\tilde{\psi}\rangle=\frac{1}{Z} \prod_{\otimes m}\left\langle x_{m}^{[A]} \mid \psi\right\rangle$ 代入条件概率公式 $P\left(\left\{x_{n}^{[B]}\right\} \right)=\frac{1}{Z}\left(\Pi_{\otimes n}\left\langle x_{n}^{[B]}\right| \Pi_{\otimes m}\left\langle x_{m}^{[A]} \mid \psi\right\rangle\right)^{2}$
即可得到概率公式：
$\begin{aligned}\mathrm{P}\left(\left\{x_{n}^{[\mathrm{B}]}\right\}\right)&=\prod_{\otimes n} \operatorname{Tr}_{\left\{x_{m}^{[\mathrm{A}]}\right\}}\left\langle x_{n}^{[\mathrm{B}]} \mid \varphi\right\rangle\left\langle\varphi \mid x_{n}^{[\mathrm{B}]}\right\rangle\\&=\prod_{\otimes n}\left\langle x_{n}^{[\mathrm{B}]} \operatorname{Tr}_{\left\{x_{m}^{[\mathrm{A}]}\right\}}|\varphi\rangle\langle\varphi \mid x_{n}^{[\mathrm{B}]}\right\rangle\\&=\prod_{\otimes n}\left\langle x_{n}^{[\mathrm{B}]} \mid \hat{\rho}^{[B]}\mid x_{n}^{[\mathrm{B}]}\right\rangle \end{aligned}$ 注： $\hat{\rho}^{[B]}=\operatorname{Tr}_{\left\{x_{m}^{[\mathrm{A}]}\right\}}|\varphi\rangle\langle\varphi\mid$
注：由线性可以将 $\operatorname{Tr}_{\left\{x_{m}^{[\mathrm{A}]}\right\}}$ 移入

我们当然可以根据 $P\left(\left\{x_{n}^{(B]}\right\} \mid\left\{x_{m}^{[A]}\right\}\right)$ 进行蒙特卡洛采样, 从而根据已知像素来生成未知像素 $\left\{z_{l}\right\},$ 但是该概率分布中包含的可能性个数会随着未知像素个数指数上升，因此在一般情况下，我们不会直接通过 $P\left(\left\{x_{n}^{[B]}\right\} \mid\left\{x_{m}^{[A]}\right\}\right)$ 进行生成。

1.2逐点生成的方法

可采用如下逐点生成的方法避免指数级复杂的 $P\left(\left\{x_{n}^{[B]}\right\} \mid\left\{x_{m}^{[A]}\right\}\right)$ ：
(1) 通过| $\psi\rangle$ 与已知像素 ${x_{m}^{[A]}\}$ , 利用投影公式计算描述未知像素对应的量子态, 记为 $|\tilde{\psi}^{(0)}\rangle$
(2) 通过| $\tilde{\psi}^{(t-1)}\rangle$ 计算第t个未知像素 $x_{t}^{[B]}$ 对应的qubit的约化密度矩阵 $\hat{\rho}^{[t-1]}($ 其系数为 $2\times2$ 厄密矩阵），计算该像素的概率分布 $P(x)=\left\langle x\left|\hat{\rho}^{[t-1]}\right| x\right\rangle,$ 并根据该概率分布进行采样, 生成该像素 $x_{t}^{[B]} ;$
(3) 如果仍有未知像素，则根据生成的像素对| $\tilde{\psi}^{(t-1)}\rangle$ 进行投影得到 $|\tilde{\psi}^{(t)}\rangle=$ $\frac{1}{z}\langle x_{t}^{[B]} \mid \tilde{\psi}^{(t-1)}\rangle,$ 并执行步骤 (2)
注：(Phys. Rev. $\times 8$ , $031012 (2018))$

1.3最概然图片（量子平均图）

即使仅知道 $|\psi\rangle$ 且不知道任何像素，也可以通过逐点生成法生成图片
每次通过约化密度矩阵计算出单个像素概率分布后, 可通过随机采样生成该像素。
可以在获得单像素概率分布后, 计算最概然的像素值作为生成的像素，采用这种方法时，每个量子态| $\boldsymbol{\psi}\rangle$ 仅生成一张最概然图片（出现在| $\boldsymbol{\psi}\rangle$ 中概率最大的那张图片），被称为量子平均图。
在这里插入图片描述
经典平均和量子平均之间的差别：

1.4张量网络压缩感知

利用最概然生成法，我们可以对图片进行压缩采样，即对于目标图片，通过保留尽量少的像素，利用量子态恢复重构出原始图片，这被称为张量网络压缩感知（Phys. Rev. Research 2, 033293 (2020) )
其中采用的采样方法被称为纠缠次序采样协议 (entanglement-ordered sampling protocol, EOSP)，核心思想是利用纠缠嫡 （即量子版本的信息嫡) 衡量不同像素携带的信息量大小，以此采样出信息量（纠缠熵）大的像素。
在这里插入图片描述
注：采样的方法不依赖于图片具体样子

2.监督性张量网络机器学习

在从量子概率的角度理解了非监督张量网络机器学习后，监督性张量网络机器学习可以看作是非监督的自然推广，这点是不同于神经网络等非概率机器学习模型的。

以分类任务为例：
我们需要使用张量网络建立从数据（特征量 ${x\}$ ) 到分类标签 K的函数映射 $f,$ 而对于概率模型而言，该映射可以由条件概率P给出：
$f:\{x\} \rightarrow \kappa \Rightarrow P(\kappa \mid\{x\})$ 那么, 我们可以考虑利用张量网络建立该条件概率，一种常用的方法采用非监督学习的方法先获得| $\breve{\psi}\rangle$ 并计算联合概率分布，再通过投影计算获得条件概率。
$|\breve{\psi}\rangle$ 中含有 $(L + 1)$ 个qubit, 其中 $L$ 个qubit对应于图片像素, 一个qubit对应于分类标签, 因此, MPS表示中应含有 $(L + 1)$ 个物理指标。
在这里插入图片描述
注： $|\breve{\psi}\rangle$ 中包含 $L$ 个对应于像素的指标 (朝上)
与一个对应于分类标签的指标 (朝下)

进行监督性学习的具体方法是：
(1) 将分类标签 $\kappa$ 也当作是特征量, 利用训练集进行非监督机器学习, 训练量子态| $\breve{\psi}\rangle ;$
(2) 利用投影计算条件概率 :
$P(\kappa \mid\{x\})=(\langle\kappa|\prod_{\otimes n}\langle x_{n} \mid \breve{\psi}\rangle)^{2}（只对像素值的qubit进行投影）$ 分类结果即为最概然的标签值 $\operatorname{argmax } _ { \kappa } P ( \kappa | \{ x \} )$ 。
关于非监督张量网络机器学习，并不一定要按照条件概率建立映射，例如当矩阵乘积态不满足量子态对应的归一化条件时，条件概率的图景也就丢失了（量子态归一化条件是张量网络量子概率可解释性的核心！），上述算法失效，而矩阵乘积态（或称tensor-train模型）也就仅仅表示了从特征量到分类标签的映射 $f:\{x\} \rightarrow \boldsymbol{\kappa}_{\circ}$ 此时，可直接使用梯度下降法训练模型，可采用自动微分技术（arXiv:1906.06329）。
比较有意思的是, 舍去量子态归一化条件的张量网络分类精度要高于量子概率性的张量网络，这大概是由于后者既可作为生成模型又可作为分类器而不够“专一”所致吧。

3.张量网络与经典概率模型

张量网络不但可以作为非概率性的映射进行监督性机器学习，也可作为量子态进行量子概率性机器学习，还可作为经典概率模型进行概率性机器学习, 其中两个例子为受限玻尔兹曼机 （restricted Boltzmann machine）与贝叶斯张量网络 (Bayesian tensor network)
在这里插入图片描述