Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Squeezed Very Deep Convolutional Neural Networks for Text Classification

About

Most of the research in convolutional neural networks has focused on increasing network depth to improve accuracy, resulting in a massive number of parameters which restricts the trained network to platforms with memory and processing constraints. We propose to modify the structure of the Very Deep Convolutional Neural Networks (VDCNN) model to fit mobile platforms constraints and keep performance. In this paper, we evaluate the impact of Temporal Depthwise Separable Convolutions and Global Average Pooling in the network parameters, storage size, and latency. The squeezed model (SVDCNN) is between 10x and 20x smaller, depending on the network depth, maintaining a maximum size of 6MB. Regarding accuracy, the network experiences a loss between 0.4% and 1.3% and obtains lower latencies compared to the baseline model.

Andr\'ea B. Duque, Lu\~a L\'azaro J. Santos, David Mac\^edo, Cleber Zanchettin• 2019

Related benchmarks

TaskDatasetResultRank
Text ClassificationAG News (test)
Accuracy90.55
210
Text ClassificationYelp Polarity (test)
Accuracy95.26
11
News categorizationAG's News
Efficiency Ratio21
7
Showing 3 of 3 rows

Other info

Code

Follow for update