Adaptive feature interaction enhancement network for text classification

Read the full article See related articles

Listed in

This article is not in any list yet, why not save it to one of your lists.
Log in to save this article

Abstract

Text classification aims to establish text distinctions, which face difficulty in capturing global text semantics and local details. To address this issue, we propose an Adaptive Feature Interactive Enhancement Network (AFIENet). Specifically, AFIENet uses two branches to model the text globally and locally. The adaptive segmentation module in the local network can dynamically split the text and capture key phrases, while the global network grasps the overall central semantics. After obtaining the results from the two branches, an interaction gate is designed to evaluate the confidence of the global features and selectively fuse them with the local features effectively. Finally, the interactively enhanced features are re-input into the classifier to improve text classification performance. Experiment results show that our proposed method can effectively enhance the performance of backbone networks such as TextCNN, RNN, and Transformer with fewer parameters. AFIENet achieved an average accuracy of 3.82% and an F1-score of 3.88% improvement across the three datasets when using Transformer as the backbone network. The comparable results to MacBERT that obtained with static word vectors also reflect the applicability of the proposed method.

Article activity feed