Abstract:Background: Text Matching (TM) is a fundamental task of natural language processing widely used in many application systems such as information retrieval, automatic question answering, machine translation, dialogue system, reading comprehension, etc. In recent years, a large number of deep learning neural networks have been applied to TM, and have refreshed benchmarks of TM repeatedly. Among the deep learning neural networks, convolutional neural network (CNN) is one of the most popular networks, which suffers from difficulties in dealing with small samples and keeping relative structures of features. In this paper, we propose a novel deep learning architecture based on capsule network for TM, called CapsTM, where capsule network is a new type of neural network architecture proposed to address some of the short comings of CNN and shows great potential in many tasks. Methods: CapsTM is a five-layer neural network, including an input layer, a representation layer, an aggregation layer, a capsule layer and a prediction layer. In CapsTM, two pieces of text are first individually converted into sequences of embeddings and are further transformed by a highway network in the input layer. Then, Bidirectional Long Short-Term Memory (BiLSTM) is used to represent each piece of text and attention-based interaction matrix is used to represent interactive information of the two pieces of text in the representation layer. Subsequently, the two kinds of representations are fused together by BiLSTM in the aggregation layer, and are further represented with capsules (vectors) in the capsule layer. Finally, the prediction layer is a connected network used for classification. CapsTM is an extension of ESIM by adding a capsule layer before the prediction layer. Results: We construct a corpus of Chinese medical question matching, which contains 36,360 question pairs. This corpus is randomly split into three parts: a training set of 32,360 question pairs, a development set of 2000 question pairs and a test set of 2000 question pairs. On this corpus, we conduct a series of experiments to evaluate the proposed CapsTM and compare it with other state-of-the-art methods. CapsTM achieves the highest F-score of 0.8666. Conclusion: The experimental results demonstrate that CapsTM is effective for Chinese medical question matching and outperforms other state-of-the-art methods for comparison.

Capsule-Transformer for Neural Machine Translation

Improving Multi-Head Attention with Capsule Networks

Towards Linear Time Neural Machine Translation with Capsule Networks

Towards Linear Time Neural Machine Translation with Capsule Networks.

Cross Aggregation of Multi-head Attention for Neural Machine Translation.

MO-Transformer: Extract High-level Relationship Between Words for Neural Machine Translation

Adaptive Capsule Network

A lightweight capsule network via channel-space decoupling and self-attention routing

Hybrid Gromov-Wasserstein Embedding for Capsule Learning

Accelerating Transformer for Neural Machine Translation.

Improving the Robustness of Capsule Networks to Image Affine Transformations

Syntax-based Transformer for Neural Machine Translation

X-Transformer: A Machine Translation Model Enhanced by the Self-Attention Mechanism

Attention routing between capsules

Transformer-Dw: A Transformer Network With Dynamic And Weighted Head

A Transformer-Based Capsule Network for 3D Part–Whole Relationship Learning

An Improved Capsule Network Based on Capsule Filter Routing

Multi-Task Learning with Capsule Networks.

RS-CapsNet: an Advanced Capsule Network.

CapsTM: capsule network for Chinese medical text matching

Spiking CapsNet: A spiking neural network with a biologically plausible routing rule between capsules