基于迭代剪枝VGGNet的火星圖像分類(lèi)

摘要： VGGNet能提供高精度的火星圖像分類(lèi)，但需消耗大量?jī)?nèi)存資源。鑒于器載計(jì)算機(jī)內(nèi)存資源有限，為解決這一矛盾，本文提出了基于迭代剪枝VGGNet的火星圖像分類(lèi)方法。首先，采用遷移學(xué)習(xí)訓(xùn)練網(wǎng)絡(luò)的連通性，以便評(píng)估神經(jīng)元的重要性；其次，通過(guò)迭代剪枝方法修剪不重要的神經(jīng)元，以便將全連接層的參數(shù)量和內(nèi)存占用量減少；最后，采用K-means++聚類(lèi)實(shí)現(xiàn)權(quán)重參數(shù)的量化，利用霍夫曼編碼壓縮迭代剪枝...