一种基于最优传输的全局结构感知向量量化方法

本申请提出一种基于最优传输的全局结构感知向量量化方法，包括：通过编码器将输入数据映射到连续隐变量空间，生成连续表征；构建连续表征与码本之间的距离矩阵，并使用最优传输问题求解分配矩阵；对距离矩阵进行归一化处理，采用特殊初始化方法生成分配矩阵的初始值；通过Sinkhorn‑Knopp算法对分配矩阵进行行列归一化迭代，直至分配矩阵收敛；迭代结束后，基于分配矩阵的最大值确定量化特征，通过解码器将量化特征映射回原始数据空间，得到重建的数据结果。基于本申请提出的方案，能够确保接近100％的码本利用率和优异的数据重建结果，为图像、视频等连续数据的离散压缩重建场景提供了强力的工具。