ELRT: Efficient Low-Rank Training for Compact Convolutional Neural Networks

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Sui, Yang, Yin, Miao, Gong, Yu, Xiao, Jinqi, Phan, Huy, Yuan, Bo
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914645991751680
author Sui, Yang
Yin, Miao
Gong, Yu
Xiao, Jinqi
Phan, Huy
Yuan, Bo
author_facet Sui, Yang
Yin, Miao
Gong, Yu
Xiao, Jinqi
Phan, Huy
Yuan, Bo
contents Low-rank compression, a popular model compression technique that produces compact convolutional neural networks (CNNs) with low rankness, has been well-studied in the literature. On the other hand, low-rank training, as an alternative way to train low-rank CNNs from scratch, has been exploited little yet. Unlike low-rank compression, low-rank training does not need pre-trained full-rank models, and the entire training phase is always performed on the low-rank structure, bringing attractive benefits for practical applications. However, the existing low-rank training solutions still face several challenges, such as a considerable accuracy drop and/or still needing to update full-size models during the training. In this paper, we perform a systematic investigation on low-rank CNN training. By identifying the proper low-rank format and performance-improving strategy, we propose ELRT, an efficient low-rank training solution for high-accuracy, high-compactness, low-rank CNN models. Our extensive evaluation results for training various CNNs on different datasets demonstrate the effectiveness of ELRT.
format Preprint
id arxiv_https___arxiv_org_abs_2401_10341
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ELRT: Efficient Low-Rank Training for Compact Convolutional Neural Networks
Sui, Yang
Yin, Miao
Gong, Yu
Xiao, Jinqi
Phan, Huy
Yuan, Bo
Computer Vision and Pattern Recognition
Artificial Intelligence
Low-rank compression, a popular model compression technique that produces compact convolutional neural networks (CNNs) with low rankness, has been well-studied in the literature. On the other hand, low-rank training, as an alternative way to train low-rank CNNs from scratch, has been exploited little yet. Unlike low-rank compression, low-rank training does not need pre-trained full-rank models, and the entire training phase is always performed on the low-rank structure, bringing attractive benefits for practical applications. However, the existing low-rank training solutions still face several challenges, such as a considerable accuracy drop and/or still needing to update full-size models during the training. In this paper, we perform a systematic investigation on low-rank CNN training. By identifying the proper low-rank format and performance-improving strategy, we propose ELRT, an efficient low-rank training solution for high-accuracy, high-compactness, low-rank CNN models. Our extensive evaluation results for training various CNNs on different datasets demonstrate the effectiveness of ELRT.
title ELRT: Efficient Low-Rank Training for Compact Convolutional Neural Networks
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2401.10341