符世園，張敏行，高宇，汪璐，程耀東，3

（1.中國科學(xué)院高能物理研究所，北京 100049；2.中國科學(xué)院大學(xué)，北京 100049；3.中國科學(xué)院高能物理研究所天府宇宙線研究中心，成都 610041）

0 概述

高能同步輻射光源（High-Energy Photon Source，HEPS）是我國重大科技基礎(chǔ)設(shè)施。一期建設(shè)的實(shí)驗(yàn)線站預(yù)計每天會產(chǎn)生數(shù)百太字節(jié)的原始實(shí)驗(yàn)數(shù)據(jù)［1］，海量數(shù)據(jù)對存儲和傳輸帶來極大壓力。數(shù)據(jù)壓縮作為一種減少數(shù)據(jù)量的方法可以用于緩解該問題。實(shí)驗(yàn)線站產(chǎn)生的數(shù)據(jù)中占比最高的是硬X 射線實(shí)驗(yàn)線站產(chǎn)生的圖像數(shù)據(jù)。硬X 射線實(shí)驗(yàn)線站利用硬X 射線的穿透能力掃描樣本，產(chǎn)生16 位單通道灰度圖。在相鄰成像間隔中，樣本均圍繞自身中心軸旋轉(zhuǎn)一個極小的角度，每掃描一個樣本會產(chǎn)生成百上千張二維投影圖像。

壓縮方法按照信息有無損失可以分為有損壓縮和無損壓縮。在有損壓縮中，不相關(guān)或者不重要的數(shù)據(jù)被直接忽略，解壓后得到的數(shù)據(jù)與原始數(shù)據(jù)不同。在無損壓縮中，每個細(xì)節(jié)都會被保存，只有統(tǒng)計冗余被消除，可完整重構(gòu)原始數(shù)據(jù)。由于該成像線站采集圖像的目的是用于探索物質(zhì)內(nèi)部結(jié)構(gòu)，因此為保證數(shù)據(jù)的科學(xué)潛力，采用無損壓縮保證圖像信息的完整性。但是，通用圖像無損壓縮方法對光源圖像壓縮效果不佳。

近年來，人工智能技術(shù)得到了飛速發(fā)展。在數(shù)據(jù)無損壓縮領(lǐng)域，深度學(xué)習(xí)技術(shù)也受到了越來越多的關(guān)注，已有研究人員將Softmax 層作為壓縮的一部分與傳統(tǒng)編碼方法相結(jié)合，用于探索文本壓縮比的優(yōu)化。本文提出一種面向同步輻射光源圖像的可并行無損壓縮方法，充分利用光源圖像的相關(guān)性，以參數(shù)自適應(yīng)的可逆量化方法和非線性預(yù)測方法實(shí)現(xiàn)對光源圖像的智能化壓縮，逐步降低存儲光源圖像所需要的資源。針對算術(shù)編碼效率問題，設(shè)計概率距離量化方法提升編碼并行度，最終達(dá)到提高圖像壓縮比的同時完全保留數(shù)據(jù)科學(xué)潛力并且使其具有高并行度。

1 相關(guān)工作

1.1 通用無損壓縮方法

無損圖像壓縮方法可以分為熵編碼、預(yù)測編碼、變換編碼和基 于字典的編碼［2］4 類。Huffman 編碼［3］、算術(shù)編碼［4］等熵編碼在編碼過程中，首先通過概率估計生成編碼表，概率可通過全局統(tǒng)計生成，也可由局部信息生成，然后通過編碼表消除編碼冗余。預(yù)測編碼通過圖像空間結(jié)構(gòu)進(jìn)行像素值預(yù)測，再將預(yù)測值與真實(shí)值之間的殘差通過熵編碼壓縮，從而達(dá)到無損壓縮的目的，例如：WebP［5］基于3 個相鄰塊預(yù)測當(dāng)前塊；PNG［6］可通過可選的5 種類型的過濾器進(jìn)行預(yù)測，不同類型的過濾器對周圍的像素值進(jìn)行不同的計算得到預(yù)測值；FLIF［7］通過上下文模型結(jié)合熵編碼進(jìn)行壓縮；JPEG_LS［8］中當(dāng)前像素的預(yù)測值取決于局部邊緣方向的自適應(yīng)模型。變換編碼是將空域轉(zhuǎn)變到頻域進(jìn)行無損壓縮，例如JPEG2000［9］使用可逆的小波變換?；谧值涞木幋a是通過字符串表進(jìn)行壓縮，例如LZW 將像素分組為字符串，最后轉(zhuǎn)換成編碼。

視頻冗余除了圖像中存在的空間冗余和編碼冗余外，還包括時間冗余，所以視頻無損壓縮方法可以進(jìn)一步從時間維度上進(jìn)行預(yù)測。FFV1［10］是一個簡單高效的無損幀內(nèi)編碼器，可使用可變長度編碼或算術(shù)編碼進(jìn)行熵編碼。因此，通用壓縮方法主要包括變換、量化、預(yù)測、編碼4 個步驟，而由于這些步驟中量化過程不可逆，無損壓縮直接跳過該步驟。此外，通用圖像無損壓縮方法可以有效地利用圖像的局部結(jié)構(gòu)，而通用視頻無損壓縮方法可以進(jìn)一步消除圖像序列內(nèi)部的時間冗余，但是只能捕獲手工指定的簡單統(tǒng)計信息。為解決這兩個問題，本文結(jié)合同步輻射光源圖像的特點(diǎn)，提出一種完全可逆的分區(qū)量化方法，通過基于深度學(xué)習(xí)的無損壓縮方法提升壓縮效果。

1.2 基于深度學(xué)習(xí)的無損壓縮方法

近年來，深度學(xué)習(xí)技術(shù)在諸多任務(wù)中取得了達(dá)到或超越傳統(tǒng)通用方法的性能。由于數(shù)據(jù)量的持續(xù)爆發(fā)性增長，越來越多的研究人員將深度學(xué)習(xí)技術(shù)應(yīng)用于無損壓縮。在無損壓縮任務(wù)中最關(guān)鍵的是準(zhǔn)確的概率估計，深度學(xué)習(xí)方法主要用于預(yù)測階段，利用訓(xùn)練得到的概率模型對數(shù)據(jù)分布進(jìn)行建模，與熵編碼結(jié)合對殘差進(jìn)行無損編碼，最大化模型在真實(shí)值上的預(yù)測概率，等同于最小化使用編碼器的無損壓縮模型所獲得的比特率［11］。

在圖像無損壓縮任務(wù)中：文獻(xiàn)［11］通過建模低分辨率與高分辨率圖像的還原過程，從低分辨率圖像直接得到預(yù)測圖像的概率分布，結(jié)合算術(shù)編碼進(jìn)行壓縮；文獻(xiàn)［12］結(jié)合通用的有損圖像壓縮方法構(gòu)建無損圖像壓縮，利用CNN 對殘差分布進(jìn)行建模；文獻(xiàn)［13］通過基于流的方法將輸入圖像轉(zhuǎn)換為具有預(yù)定義分布函數(shù)的潛在表示，編碼和解碼函數(shù)必須互為可逆，并使用大型神經(jīng)網(wǎng)絡(luò)提高壓縮性能。在科學(xué)數(shù)據(jù)的壓縮任務(wù)中，深度學(xué)習(xí)方法也得到廣泛的應(yīng)用：文獻(xiàn)［14］提出一種基于神經(jīng)網(wǎng)絡(luò)的DNA 序列壓縮器，在壓縮效果上得到一定的提升；文獻(xiàn)［15］使用深度遞歸神經(jīng)網(wǎng)絡(luò)來改進(jìn)傳統(tǒng)方法并提高模型的泛化能力和預(yù)測精度。

上述基于深度學(xué)習(xí)的壓縮方法建模追求泛化性和通用性。與上述壓縮方法不同，GOYAL 等［16］在DeepZip 中提出一種基于時序網(wǎng)絡(luò)的文本無損壓縮方法，該方法針對不同的數(shù)據(jù)過擬合訓(xùn)練獨(dú)立的模型，為保證壓縮過程完全可逆，將模型與壓縮結(jié)果一并保存，即壓縮數(shù)據(jù)與模型為一對一關(guān)系，模型僅適用于訓(xùn)練數(shù)據(jù)的壓縮任務(wù)。這種對不同數(shù)據(jù)過擬合建模的方式可在其測試數(shù)據(jù)上優(yōu)化壓縮比，但是模型規(guī)模會帶來額外的存儲空間開銷。對于模型規(guī)模影響壓縮效果的問題，文獻(xiàn)［17］提出一種基于自適應(yīng)和半自適應(yīng)訓(xùn)練的新型混合架構(gòu)，無需重新訓(xùn)練模型即可壓縮新的數(shù)據(jù)。同時，基于深度學(xué)習(xí)的無損壓縮方法通常編碼速度較慢，低編碼速率限制了應(yīng)用的可行性。針對這一問題，文獻(xiàn)［18］提出一種多尺度的漸進(jìn)統(tǒng)計模型加速模型推理速度，文獻(xiàn)［19］通過流模型進(jìn)行加速，文獻(xiàn)［20］通過并行預(yù)測多個像素的概率進(jìn)行加速，文獻(xiàn)［21］通過跳過耗時較長的編碼階段進(jìn)行加速。

本文通過提出概率距離以及穩(wěn)定的計算過程替換算術(shù)編碼過程，在保證壓縮比優(yōu)化的前提下提高編碼過程的并行度。受DeepZip 啟發(fā)，結(jié)合光源圖像特點(diǎn)，以數(shù)據(jù)集為單位過擬合訓(xùn)練模型，減少模型規(guī)模對壓縮比提升的影響。

2 圖像數(shù)據(jù)集建立與相關(guān)性分析

2.1 圖像數(shù)據(jù)集建立

本文圖像數(shù)據(jù)來自上海同步輻射光源裝置，是掃描不同樣本得到的圖像序列，涵蓋了魚鱗、菌絲、小鼠腦和化石翅膀。不同圖像序列的幀數(shù)和圖像尺寸有所不同，分辨率為2 048×2 048 像素和2 048×1 200 像素兩種，均為16 位單通道灰度投影圖序列，具體信息如表1 所示。

表1 數(shù)據(jù)集信息Table 1 Dataset information

分別對上述圖像序列使用通用無損壓縮方法進(jìn)行壓縮測試，得到結(jié)果如表2 所示，壓縮結(jié)果以壓縮比（Compression Ratio，CR）衡量，壓縮比即為壓縮前文件大小與壓縮后文件大小的比值。通用無損壓縮方法達(dá)到的最優(yōu)壓縮比為1.46～1.64。壓縮比越高，說明節(jié)省的存儲空間越多，數(shù)據(jù)長期存儲的成本越低。

表2 4 種通用壓縮方法的壓縮比Table 2 CR of four general compression methods

2.2 圖像線性相關(guān)性分析

對于圖像相似度的衡量指標(biāo)，Spearman 系數(shù)［22］比SSIM［23-24］更準(zhǔn)確，因此以Spearman 系數(shù)為圖像間線性關(guān)系的衡量標(biāo)準(zhǔn)。同一樣本圖像序列內(nèi)部的時間線性相關(guān)性由相鄰圖像間線性相關(guān)性的均值計算得到。

圖像序列內(nèi)部的部分時間線性相關(guān)性通過時間差分去除。時間差分即將相鄰幀同一空間位置像素相減，該過程也可以看作將前幀作為后幀的線性預(yù)測幀，差分后得到去除部分線性冗余后的殘差數(shù)據(jù)，殘差數(shù)據(jù)可通過深度學(xué)習(xí)方法建模非線性關(guān)系進(jìn)一步壓縮。

差分前后圖像序列內(nèi)部的線性時間相關(guān)性如表3 所示，差分前圖像序列內(nèi)部具有極強(qiáng)的線性時間相關(guān)性，差分后其相關(guān)性接近于0。因此，差分可以簡單有效地去除圖像序列內(nèi)部的時間線性相關(guān)性。

表3 圖像差分前后的相關(guān)性對比Table 3 Comparison of correlation before and after image difference

3 同步輻射光源圖像智能壓縮

3.1 整體流程

本文提出一種面向同步輻射光源圖像的可并行智能無損壓縮方法，如圖1 所示，主要包括線性預(yù)測和非線性預(yù)測，用以分別去除圖像序列中的線性及非線性冗余信息。以時間差分作為去除線性冗余的線性預(yù)測方法，差分后的圖像通過分區(qū)量化方法將像素值取值范圍映射到一個更小的區(qū)間，以減小像素值所占比特位，得到images′。在非線性預(yù)測模塊中，通過預(yù)測模型及訓(xùn)練方法以加速訓(xùn)練及預(yù)測過程，定義概率距離概念以加速編碼過程，將概率距離量化流作為最終壓縮流。

圖1 整體壓縮流程Fig.1 Overall compression process

3.2 分區(qū)量化

由于光源圖像序列內(nèi)部時間線性相關(guān)性較強(qiáng)，因此通過時間差分方法去除序列內(nèi)部的大部分時間線性冗余。時間差分前圖像像素值分布如圖2（a）所示，時間差分后圖像像素值集中分布在0 和65 535 附近，該范圍中部僅有少量數(shù)據(jù)存在，如圖2（b）所示。將像素值映射到較小的數(shù)據(jù)范圍能夠縮小其所占比特位，從而達(dá)到壓縮的目的，因此本文提出可逆的分區(qū)量化方法，以保存少量未壓縮數(shù)據(jù)為代價，換取較小的數(shù)據(jù)分布范圍。

圖2 菌絲圖像差分前后的像素值分布對比Fig.2 Comparison of pixel value distribution before and after difference

該量化過程完全可逆，因此可以保證反量化后數(shù)據(jù)與量化前數(shù)據(jù)一致。分區(qū)量化與反量化過程如圖3 所示，在分區(qū)量化后，原始數(shù)據(jù)分為待壓縮數(shù)據(jù)和直接保留數(shù)據(jù)，其中直接保留數(shù)據(jù)直接按順序單獨(dú)存儲。為保證無損還原，直接保留數(shù)據(jù)以占位符的形式存在于待壓縮數(shù)據(jù)中。在反量化時，在待壓縮數(shù)據(jù)中掃描到占位符時，按順序回填直接保留數(shù)據(jù)即可。待壓縮數(shù)據(jù)量化方法為向右平移指定距離，在還原時需向左平移相同距離。

圖3 分區(qū)量化與反量化過程Fig.3 Process of partition quantization and inverse quantization

對于T×M×N的圖像序列，其中，T為幀數(shù)，M為高度，N為寬度，分區(qū)量化步驟具體如下：

1）采樣：從圖像序列中等間隔抽取一定數(shù)量樣本。

2）統(tǒng)計像素值：統(tǒng)計采樣數(shù)據(jù)集中像素值的分布得到Dict，記錄每一個像素值出現(xiàn)的次數(shù)，Dict（i）表示值為i的像素值出現(xiàn)的次數(shù)。

3）確定占位符f：由占位符確定數(shù)據(jù)的量化函數(shù)，依據(jù)式（1）確定占位符：

4）構(gòu)造量化函數(shù)F（x）：x為分區(qū)量化前的差分像素值，根據(jù)占位符構(gòu)造量化函數(shù)，約定占位符為量化后范圍末尾加1 的值。待壓縮數(shù)據(jù)后，只需向右平移指定步長（f-1）/2，直接保留數(shù)據(jù)并映射到占位符，按順序存儲到文件中，如式（2）所示：

5）量化：根據(jù)量化函數(shù)將數(shù)據(jù)量化到新的值。

分區(qū)反量化步驟具體如下：

1）確定占位符f：由于占位符的設(shè)置是新的映射范圍最大值加1，因此待壓縮數(shù)據(jù)的最大值即為占位符的值。

2）構(gòu)造反量化函數(shù)G（y）：該過程為量化函數(shù)逆過程，y為量化后的值。若數(shù)據(jù)等于占位符，則從直接保留數(shù)據(jù)中讀取一個數(shù)據(jù)，否則向左平移指定步長（f-1）/2，如式（3）所示：

3）反量化：根據(jù)反量化函數(shù)還原量化值。

在分區(qū)量化過程中，主要包括像素值統(tǒng)計、占位符確定、量化（反量化）過程，這3 個過程均可并行計算。

3.3 非線性預(yù)測

圖像序列經(jīng)過差分及量化后，已去除部分線性冗余并將像素值映射到較小范圍。非線性預(yù)測用于利用差分量化后圖像序列的非線性相關(guān)性進(jìn)一步提升壓縮比。受DeepZip 啟發(fā)，使用深度學(xué)習(xí)方法過擬合壓縮數(shù)據(jù)，學(xué)習(xí)其非線性相關(guān)性。但是該方法存在以下問題：1）由于模型規(guī)模較大，因此抵消了壓縮所節(jié)省的空間；2）一對一訓(xùn)練過程耗時；3）模型預(yù)測時間較長。對于問題1 和2，結(jié)合同步輻射光源圖像的特點(diǎn)，提出一種以數(shù)據(jù)集為單位，用少量數(shù)據(jù)訓(xùn)練模型以壓縮整體數(shù)據(jù)的方法；對于問題3，提出一種以CNN 為基礎(chǔ)架構(gòu)的概率預(yù)測模型，以提高預(yù)測的并行度。

3.3.1 訓(xùn)練與預(yù)測方法

本文將壓縮方法分為通用壓縮與專用壓縮兩類，如圖4、圖5 所示。通用方法是由專門的組織提出標(biāo)準(zhǔn)編寫而成的，對大部分圖像數(shù)據(jù)具有壓縮效果。專用方法如DeepZip，壓縮器僅對訓(xùn)練數(shù)據(jù)有效。

圖4 通用壓縮中數(shù)據(jù)與壓縮器的對應(yīng)關(guān)系Fig.4 Correspondence between data and compressors in the general compression

圖5 專用壓縮中數(shù)據(jù)與壓縮器的對應(yīng)關(guān)系Fig.5 Correspondence between data and compressors in the special compression

基于同步輻射光源圖像中一個樣本對應(yīng)一個數(shù)據(jù)集的特點(diǎn)，本文提出一種壓縮粒度介于通用與專用之間的非線性預(yù)測方法，如圖6 所示，每一個數(shù)據(jù)集即為一個圖像序列。該方法以專用壓縮的思想提升壓縮比，即對不同的圖像序列訓(xùn)練不同的模型，這是因?yàn)椴煌瑯颖旧傻膱D像序列內(nèi)容差別較大。由于同一樣本生成的圖像序列內(nèi)容非常相似，因此借鑒通用壓縮的思想加速訓(xùn)練過程，即對同一圖像序列使用統(tǒng)一的模型進(jìn)行預(yù)測。在本文的訓(xùn)練與預(yù)測方法中，以數(shù)據(jù)集為單位過擬合訓(xùn)練獨(dú)立模型，訓(xùn)練數(shù)據(jù)為樣本圖像序列中的單張圖像及其前序圖像組成的小樣本數(shù)據(jù)集，用于該樣本所有圖像的壓縮。

圖6 本文中數(shù)據(jù)與壓縮器的對應(yīng)關(guān)系Fig.6 Correspondence between data and compressors in this study

3.3.2 C-Zip 網(wǎng)絡(luò)結(jié)構(gòu)

概率預(yù)測模型C-Zip 以CNN 為基礎(chǔ)架構(gòu)，輸入為指定patch 大小，如圖7 所示。該模型主要基于TCN［25］和Octave Convolution［26］。

圖7 C-Zip 網(wǎng)絡(luò)結(jié)構(gòu)Fig.7 C-Zip network structure

TCN 是一種通過因果卷積和膨脹卷積學(xué)習(xí)序列的非線性關(guān)系的模型，僅使用CNN 模型，可大規(guī)模并行處理。本文方法利用3D-TCN 學(xué)習(xí)圖像序列中的時空特征，最后通過Fully-Connected 層加Softmax層得到編碼時使用的預(yù)測概率分布，其中3D-TOCN子模塊如圖8 所示。該子模塊結(jié)合了Octave Convolution，將輸入按通道切分為高頻和低頻，低頻數(shù)據(jù)所含信息量較少，可以在高度和寬度的維度上縮小為原來的1/2，之后分別通過3D-TCN 學(xué)習(xí)頻域內(nèi)非線性關(guān)系，通過Octave Convolution 進(jìn)行頻域間通信，最后將低頻信息通過上采樣恢復(fù)成原始大小，與高頻信息做拼接即得到最后的輸出。

圖8 3D-TOCN 子模塊結(jié)構(gòu)Fig.8 3D-TOCN submodule structure

3.4 概率距離

由于算術(shù)編碼速度慢，并行度低，影響壓縮效率，因此本文提出概率距離方法（Rank）用以增加編碼并行度。

Softmax 層的輸出根據(jù)不同的任務(wù)可以得到不同的解讀與計算，從而提取出不同的信息。Softmax的輸出經(jīng)過歸一化后可以認(rèn)為是每個值對應(yīng)的概率。對于壓縮問題，概率可直接與熵編碼進(jìn)行結(jié)合；對于多分類問題，預(yù)測類別為最大概率值對應(yīng)的類別；對于推薦問題，需對概率進(jìn)行排序。本文針對的問題是大字典數(shù)數(shù)據(jù)壓縮，對于大字典數(shù)據(jù)而言，若神經(jīng)網(wǎng)絡(luò)預(yù)測的結(jié)果較好，則大部分真實(shí)值對應(yīng)的概率距離在Top-K中，當(dāng)K的比特位數(shù)小于當(dāng)前數(shù)據(jù)的比特位數(shù)時，可結(jié)合分區(qū)量化達(dá)到壓縮效果，因此可通過概率距離結(jié)合分區(qū)量化達(dá)到壓縮的目的。

概率距離方法定義為真實(shí)值對應(yīng)的預(yù)測概率值在預(yù)測概率向量中的排序位置，針對概率值相同導(dǎo)致排序不穩(wěn)定的問題，進(jìn)一步比較字典中值的大小，保證計算過程完全可逆。

對于字典數(shù)為［0，n-1］的數(shù)據(jù)，當(dāng)前待編碼數(shù)據(jù)為v，模型輸出為預(yù)測概率向量p，p［j］為字典數(shù)j對應(yīng)的概率值，其中0≤j≤n-1，概率距離計算如式（4）及式（5）所示：

目前，在傳統(tǒng)智能壓縮方法中，將模型的輸出看作概率，結(jié)合算術(shù)編碼進(jìn)行壓縮。本文根據(jù)神經(jīng)網(wǎng)絡(luò)輸出計算得到概率距離，結(jié)合分區(qū)量化進(jìn)行壓縮，如圖9 所示。該方法計算簡單，因?yàn)閮H涉及比較大小和做統(tǒng)計的過程，而不同像素值概率距離的計算相互獨(dú)立，所以并行度高。同時，該方法的輸出可以繼續(xù)與其他編碼方法相結(jié)合，進(jìn)一步提升壓縮比。

圖9 非線性壓縮流程Fig.9 Process of nonlinear compression

4 實(shí)驗(yàn)與結(jié)果分析

4.1 分區(qū)量化分析

對不同數(shù)據(jù)集進(jìn)行時間差分，圖10 和圖11 為不同數(shù)據(jù)集中單張圖像差分前后的像素值分布對比，其中，差分前不同數(shù)據(jù)集圖像像素值分布的范圍和密度不同，差分后不同數(shù)據(jù)集圖像像素值的分布范圍均密集分布在兩端區(qū)域。因此，不同數(shù)據(jù)集時間差分后均可通過分區(qū)量化縮小像素值分布范圍。

圖10 原圖像素值分布Fig.10 Pixel value distribution of original images

圖11 差分圖像素值分布Fig.11 Pixel value distribution of difference images

對每個數(shù)據(jù)集的圖像序列抽取1/100 的數(shù)據(jù)統(tǒng)計其像素值并計算其占位符，結(jié)果如表4 所示。不同樣本的占位符有所不同，說明分區(qū)量化的計算具有數(shù)據(jù)自適應(yīng)性。量化后像素值所占比特位由16 位降低至11 或12 位，而保留數(shù)據(jù)只占原始數(shù)據(jù)的0.29%～1.97%，說明量化后數(shù)據(jù)可以達(dá)到壓縮的目的。

表4 不同數(shù)據(jù)集上的分區(qū)量化結(jié)果Table 4 Partition quantization results on different datasets

因此，分區(qū)量化可以達(dá)到將數(shù)據(jù)無損量化到較小數(shù)據(jù)范圍的目的，能夠節(jié)省20%以上的存儲空間，同時較小的數(shù)據(jù)分布范圍有利于后續(xù)基于深度學(xué)習(xí)的模型訓(xùn)練與壓縮。

4.2 非線性預(yù)測方法對比

數(shù)據(jù)集圖像序列進(jìn)行時間差分及分區(qū)量化后，通過非線性預(yù)測方法進(jìn)行概率預(yù)測，直接結(jié)合算術(shù)編碼進(jìn)行壓縮。

模型的輸入維度為3×32×32，即將圖像數(shù)據(jù)分塊處理，塊大小為32×32 像素，利用當(dāng)前幀的前向3 幀不同圖像塊預(yù)測當(dāng)前幀相應(yīng)圖像塊每一個像素值的概率分布，損失函數(shù)為交叉熵?fù)p失函數(shù)，損失連續(xù)3 個epoch 不下降或者epoch 達(dá)到16 時停止訓(xùn)練，學(xué)習(xí)率設(shè)置為0.01，使用Adam 優(yōu)化器，每3 個epoch學(xué)習(xí)率下降10%，Octave Convolution 中超參數(shù)設(shè)置為0.5。

以小鼠腦數(shù)據(jù)集為例，本文提出的C-Zip 中僅使用CNN 模型，即可達(dá)到與DeepZip 相近的壓縮比優(yōu)化效果，并且可以減少預(yù)測時間，如圖12 所示。

圖12 DeepZip 與C-Zip 壓縮比與預(yù)測時間對比Fig.12 Comparison of DeepZip and C-Zip in CR and predicting time

取數(shù)據(jù)集中少量數(shù)據(jù)過擬合訓(xùn)練模型作為整體圖像序列的預(yù)測模型，壓縮結(jié)果如表5 所示。對于專用模型壓縮以圖片為單位的原訓(xùn)練方法和本文提出的以數(shù)據(jù)集為單位的壓縮方法，兩者結(jié)果較為接近，驗(yàn)證了本文方法在光源圖像壓縮問題上的有效性。在化石翅膀數(shù)據(jù)集的結(jié)果中，原方法的壓縮比較高。在小鼠腦和魚鱗數(shù)據(jù)集的結(jié)果中，本文方法的壓縮比較高，這是因?yàn)閷?shí)際的模型訓(xùn)練過程中，考慮到實(shí)際應(yīng)用的時間限制，訓(xùn)練過程中達(dá)到指定epoch 即停止訓(xùn)練，會導(dǎo)致訓(xùn)練過程中未找到最優(yōu)解的問題，從而導(dǎo)致某些圖像壓縮效果不佳。本文方法在壓縮時隨機(jī)選取的模型若為訓(xùn)練效果好的模型，則會出現(xiàn)整體壓縮比高于原方法的結(jié)果，這是未來可以改進(jìn)的方向。但是相較于傳統(tǒng)壓縮方法，這兩種方法的壓縮比均可獲得0.5 以上的提升效果。

表5 2 種壓縮方式的壓縮比對比Table 5 Comparison of CR of two compression methods

4.3 概率距離分析

對不同數(shù)據(jù)集的圖像序列，經(jīng)過相同的時間差分與分區(qū)量化、訓(xùn)練與預(yù)測方法，僅將模型輸出改為概率距離，算術(shù)編碼改為分區(qū)量化過程，所得壓縮比結(jié)果如表6 所示。

表6 不同數(shù)據(jù)集上的概率距離量化結(jié)果Table 6 Probability distance quantization results on different datasets

不同數(shù)據(jù)集概率距離的占位符不同，量化后比特位為7～9 位，相較于原16 位數(shù)據(jù)，該部分?jǐn)?shù)據(jù)在比特位上可以節(jié)省43.75%～56.25%的存儲空間。結(jié)合保留數(shù)據(jù)占比，整體壓縮比為1.74～2.00，與使用概率距離前的壓縮方法相比，整體壓縮比有所下降。

在單個像素值的概率距離的計算過程中，僅涉及簡單的加減運(yùn)算，相較于算術(shù)編碼，該壓縮流程更易于硬件實(shí)現(xiàn)及加速。同時，不同像素值的概率距離計算過程相互獨(dú)立，后續(xù)分區(qū)量化過程同為高可并行方法。因此，對比原始壓縮流程，概率距離量化流程能夠在保證壓縮比的前提下提高并行度。

4.4 整體壓縮比對比

在本文方法中，為保證壓縮過程無損，壓縮后需要存儲的數(shù)據(jù)包含預(yù)測模型數(shù)據(jù)、差分量化后保留數(shù)據(jù)、Rank 保留數(shù)據(jù)以及Rank 量化后數(shù)據(jù)4 個部分，其中預(yù)測模型數(shù)據(jù)可忽略不計，另外3 個部分?jǐn)?shù)據(jù)大小占原數(shù)據(jù)大小比值如表7 所示，整體壓縮比為1.69～2.22。PNG、JPEG2000、JPEG_LS、FLIF 和本文方法的壓縮比對比如表8 所示，相較于傳統(tǒng)無損壓縮方法，本文方法的壓縮比提升了0.23～0.58?？梢?，本文方法在保證壓縮比優(yōu)化的前提下，具有較高的并行度。

表7 不同數(shù)據(jù)集上的整體壓縮比Table 7 Overall CR on different datasets

表8 5 種方法的壓縮比對比Table 8 Comparison of CR of five methods

5 結(jié)束語

本文提出一種面向同步輻射光源圖像的可并行智能無損壓縮方法。通過充分挖掘該類圖像序列內(nèi)部相關(guān)性，設(shè)計可逆的分區(qū)量化方法。以數(shù)據(jù)集為單位過擬合訓(xùn)練模型，作為概率預(yù)測器，并結(jié)合穩(wěn)定的概率距離計算方法，在確保圖像壓縮比的同時提高了編碼并行度。實(shí)驗(yàn)結(jié)果表明，該方法相比于傳統(tǒng)圖像無損壓縮方法對于同步輻射光源圖像具有更好的壓縮效果。下一步將細(xì)化分區(qū)量化中占位符的選擇范圍，明確占位符對壓縮效果的影響程度，形成可量化指標(biāo)，同時通過提升模型訓(xùn)練過程中尋找最優(yōu)解的速度，以獲得穩(wěn)定的壓縮比優(yōu)化效果，并且可將該方法與FPGA 等硬件相結(jié)合進(jìn)一步降低計算成本，加速壓縮過程，提升適用范圍。