Figure 1.
Architecture diagram of AB-CNN-BGRU-att model
Figure 2.
ALBERT model structure
Figure 3.
GRU network structure
Figure 4.
BiGRU network structure
Figure 5.
Improved TextCNN model structure
TABLE I.
experimental dataset
| Name | Training Set | Test Set | Validation Set | Category | Total |
|---|---|---|---|---|---|
| KUAKE-QIC | 6931 | 1994 | 1955 | 11 | 10880 |
| THUCNews_Title | 180000 | 10000 | 10000 | 10 | 200000 |
Figure 6.
Model validation results
Figure 7.
Comparison of Network Time
TABLE II.
comparison between bigru-att and bilstm-att
| Network Layer | Average Duration | Total Duration | Acc% | F1% |
|---|---|---|---|---|
| BiGRU-att | 2286.9s | 45738s | 90.83 | 90.64 |
| BiLSTM-att | 2422.55s | 48451s | 90.45 | 90.41 |