Marsbildklassifikation basierend auf iterativem Pruning von VGGNet

LIU Meng ,  

LIU Jin ,  

YIN Li-jun ,  

KANG Zhi-wei ,  

MA Xin ,  

摘要

VGGNet kann eine hochpräzise Marsbildklassifikation bieten, benötigt jedoch erhebliche Speicherressourcen. Angesichts der begrenzten Speicherkapazität an Bord wurde in dieser Arbeit eine Methode zur Marsbildklassifikation basierend auf iterativem Pruning von VGGNet vorgeschlagen, um diesen Konflikt zu lösen. Zunächst wird die Netzwerkverbindung mittels Transferlernen trainiert, um die Bedeutung der Neuronen zu bewerten; zweitens werden mittels iterativem Pruning unwichtige Neuronen beschnitten, um die Anzahl der Parameter und den Speicherverbrauch der voll verbundenen Schichten zu reduzieren; schließlich wird die Gewichtungsquantisierung durch K-means++-Clustering durchgeführt, und die Gewichtsparameter des VGGNet nach Pruning und Quantisierung werden mittels Huffman-Codierung komprimiert, um Speicherbedarf und Floating-Point-Operationen zu verringern. Zusätzlich werden fünf Datenaugmentationsmethoden verwendet, um die Daten zu erweitern und das Problem des Klassenungleichgewichts zu lösen. Die Experimentalergebnisse zeigen, dass das komprimierte VGGNet-Modell 62,63 MB Speicher, 150,6 MFlops und eine Genauigkeit von 96,15 % benötigt. Im Vergleich zu leichten Bildklassifikationsalgorithmen wie ShuffleNet, MobileNet und EfficientNet bietet das vorgeschlagene Modell eine bessere Leistung.

关键词

Bildklassifikation; konvolutionales neuronales Netzwerk; iteratives Verfahren; Clustering-Algorithmus; VGGNet

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website