SlowGuess/NAACL_2021_OCR huggingface dataset:
Yinya Huang $^{1*}$  Meng Fang $^{2}$  Yu Cao $^{3}$  Liwei Wang $^{4}$  Xiaodan Liang $^{1\dagger}$
Tencent Robotics X
$^{3}$ School of Computer Science, The University of Sydney
<sup>4</sup>The Chinese University of Hong Kong
yinya.huang@hotmail, mfang@tencent.com,
ycao8647@uni.sydney.edu.au,
lwwang@cse.cuhk.edu.hk, xdliang328@gmail.com
Recent QA with logical reasoning questions requires passage-level relations among the sentences. However, current approaches still focus on sentence-level relations interacting among tokens. In this work, we explore aggregating passage-level clues for solving logical reasoning QA by using discourse-based information. We propose a discourse-aware graph network (DAGN) that reasons relying on the discourse structure of the texts. The model encodes discourse information as a graph with elementary discourse units (EDUs) and discourse relations, and learns the discourse-aware features via a graph network for downstream QA tasks. Experiments are conducted on two logical reasoning QA datasets, ReClor and LogiQA, and our proposed DAGN achieves competitive results. The source code is available at https://github.com/EleanorH/DAGN.
A variety of QA datasets have promoted the development of reading comprehensions, for instance, SQuAD (Rajpurkar et al., 2016), HotpotQA (Yang et al., 2018), DROP (Dua et al., 2019), and so on. Recently, QA datasets with more complicated reasoning types, i.e., logical reasoning, are also introduced, such as ReClor (Yu et al., 2020) and LogiQA (Liu et al., 2020). The logical questions are taken from standardized exams such as GMAT and LSAT, and require QA models to read complicated argument passages and identify logical relationships therein. For example, selecting a correct assumption that supports an argument, or finding out a claim that weakens an argument in a passage. Such logical reasoning is beyond the capability of most of the previous QA models which focus on reasoning with entities or numerical keywords.
A main challenge for the QA models is to uncover the logical structures under passages, such as identifying claims or hypotheses, or pointing out flaws in arguments. To achieve this, the QA models should first be aware of logical units, which can be sentences or clauses or other meaningful text spans, then identify the logical relationships between the units. However, the logical structures are usually hidden and difficult to be extracted, and most datasets do not provide such logical structure annotations.
An intuitive idea for unwrapping such logical information is using discourse relations. For instance, as a conjunction, "because" indicates a causal relationship, whereas "if" indicates a hypothetical relationship. However, such discourse-based information is seldom considered in logical reasoning tasks. Modeling logical structures is still lacking in logical reasoning tasks, while current opened methods use contextual pre-trained models (Yu et al., 2020). Besides, previous graph-based methods (Ran et al., 2019; Chen et al., 2020a) that construct entity-based graphs are not suitable for logical reasoning tasks because of different reasoning units.
In this paper, we propose a new approach to solve logical reasoning QA tasks by incorporating discourse-based information. First, we construct discourse structures. We use discourse relations from the Penn Discourse TreeBank 2.0 (PDTB 2.0) (Prasad et al., 2008) as delimiters to split texts into elementary discourse units (EDUs). A logic graph is constructed in which EDUs are nodes and discourse relations are edges. Then, we propose a Discourse-Aware Graph Network (DAGN) for learning high-level discourse features to represent passages. The discourse features are incorporated with the contextual token features from pre-trained language models. With the enhanced features, DAGN predicts answers to logical questions. Our experiments show that DAGN surpasses current opened methods on two recent logical rea
*This work was done during Yinya Huang's internship in Tencent with L. Wang and M. Fang.
† Corresponding Author: Xiaodan Liang.
Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, pages 5848-5855 June 6-11, 2021. ©2021 Association for Computational Linguistics
(1)
Figure 1: The architecture of our proposed method with an example below.
Context:	Theoretically, analog systems are superior to digital systems. A signal in a pure analog system can be infinitely detailed, while digital systems cannot produce signals that are more precise than their digital units. With this theoretical advantage there is a practical disadvantage. Since there is no limit on the potential detail of the signal, the duplication of an analog representation allows tiny variations from the original, which are errors.
Question:	The statements above, if true, most strongly support which one of the following?
Option	Digital systems are the best information systems because error cannot occur in the emission of digital signals.
soning QA datasets, ReClor and LogiQA. Our main contributions are three-fold:
- We propose to construct logic graphs from texts by using discourse relations as edges and elementary discourse units as nodes.
- We obtain discourse features via graph neural networks to facilitate logical reasoning in QA models.
- We show the effectiveness of using logic graph and feature enhancement by noticeable improvements on two datasets, ReClor and LogiQA.
Our intuition is to explicitly use discourse-based information to mimic the human reasoning process for logical reasoning questions. The questions are in multiple choices format, which means given a triplet (context, question, answer options), models answer the question by selecting the correct answer option. Our framework is shown in Figure 1. We first construct a discourse-based logic graph from the raw text. Then we conduct reasoning via graph networks to learn and update the discourse-based features, which are incorporated with the contextual token embeddings for downstream answer prediction.
2.1 Graph Construction
Our discourse-based logic graph is constructed via two steps: delimiting text into elementary discourse units (EDUs) and forming the graph using their relations as edges, as illustrated in Figure 1(1).
Discourse Units Delimitation It is studied that clause-like text spans delimited by discourse relations can be discourse units that reveal the rhetorical structure of texts (Mann and Thompson, 1988; Prasad et al., 2008). We further observe that such discourse units are essential units in logical reasoning, such as being assumptions or opinions. As the example shown in Figure 1, the "while" in the context indicates a comparison between the attributes of "pure analog system" and that of "digital systems". The "because" in the option provides evidence "error cannot occur in the emission of digital signals" to the claim "digital systems are the best information systems".
We use PDTB 2.0 (Prasad et al., 2008) to help drawing discourse relations. PDTB 2.0 contains discourse relations that are manually annotated on the 1 million Wall Street Journal (WSJ) corpus and are broadly characterized into "Explicit" and "Implicit" connectives. The former apparently presents in sentences such as discourse adverbial "instead" or subordinating conjunction "because", whereas the latter are inferred by annotators between successive pairs of text spans split by punctuation marks
such as “.” or “;”. We simply take all the “Explicit” connectives as well as common punctuation marks to form our discourse delimiter library (details are given in Appendix A), with which we delimit the texts into EDUs. For each data sample, we segment the context and options, ignoring the question since the question usually does not carry logical content.
Discourse Graph Construction We define the discourse-based graphs with EDUs as nodes, the "Explicit" connectives as well as the punctuation marks as two types of edges. We assume that each connective or punctuation mark connects the EDUs before and after it. For example, the option sentence in Figure 1 is delimited into two EDUs,  $\mathrm{EDU}_7 =$  "digital systems are the best information systems" and  $\mathrm{EDU}_8 =$  "error cannot occur in the emission of digital signals" by the connective  $r =$  "because". Then the returned triplets are  $(\mathrm{EDU}_7,r,\mathrm{EDU}_8)$  and  $(\mathrm{EDU}_8,r,\mathrm{EDU}_7)$ . For each data sample with the context and multiple answer options, we separately construct graphs corresponding to each option, with EDUs in the same context and every single option. The graph for the single option  $k$  is denoted by  $\mathcal{G}^k = (\mathcal{V}^k,\mathcal{E}^k)$ .
2.2 Discourse-Aware Graph Network
We present the Discourse-Aware Graph Network (DAGN) that uses the constructed graph to exploit discourse-based information for answering logical questions. It consists of three main components: an EDU encoding module, a graph reasoning module, and an answer prediction module. The former two are demonstrated in Figure 1(2), whereas the final component is in Figure 1(3).
EDU Encoding An EDU span embedding is obtained from its token embeddings. There are two steps. First, similar to previous works (Yu et al., 2020; Liu et al., 2020), we encode such input sequence “
 context 
 question | option 
” into contextual token embeddings with pre-trained language models, where 
 and 
 are the special tokens for RoBERTa (Liu et al., 2019) model, and | | denotes concatenation. Second, given the token embedding sequence {t1, t2, ..., tL}, the n-th EDU embedding is obtained by e_n = ∑_{l ∈ S_n} t_l, where Sn is the set of token indices belonging to n-th EDU.
Graph Reasoning After EDU encoding, DAGN performs reasoning over the discourse graph. Inspired by previous graph-based models (Ran et al.,
2019; Chen et al., 2020a), we also learn graph node representations to obtain higher-level features. However, we consider different graph construction and encoding. Specifically, let  $\mathcal{G}^k = (\mathcal{V}^k,\mathcal{E}^k)$  denote a graph corresponding to the  $k$ -th option in answer choices. For each node  $v_{i}\in \mathbf{V}$ , the node embedding  $\mathbf{v}_i$  is initialized with the corresponding EDU embedding  $\mathbf{e}_i$ .  $\mathcal{N}_i = \{j|(v_j,v_i)\in \mathcal{E}^k\}$  indicates the neighbors of node  $v_{i}$ .  $\mathbf{W}^{r_{ji}}$  is the adjacency matrix for one of the two edge types, where  $r_E$  indicates graph edges corresponding to the explicit connectives, and  $r_I$  indicates graph edges corresponding to punctuation marks.
The model first calculates weight  $\alpha_{i}$  for each node with a linear transformation and a sigmoid function  $\alpha_{i} = \sigma (\mathbf{W}^{\alpha}(\mathbf{v}_{i}) + b^{\alpha})$ , then conducts message propagation with the weights:
$$
\tilde {\mathbf {v}} _ {i} = \frac {1}{| \mathcal {N} _ {i} |} \left(\sum_ {j \in \mathcal {N} _ {i}} \alpha_ {j} \mathbf {W} ^ {r _ {j i}} \mathbf {v} _ {j}\right), r _ {j i} \in \left\{r _ {E}, r _ {I} \right\} \tag {1}
$$
where  $\tilde{\mathbf{v}}_i$  is the message representation of node  $v_{i}$ $\alpha_{j}$  and  $\mathbf{V}_j$  are the weight and the node embedding of  $v_{j}$  respectively.
After the message propagation, the node representations are updated with the initial node embeddings and the message representations by
$$
\mathbf {v} _ {i} ^ {\prime} = \operatorname {R e L U} \left(\mathbf {W} ^ {u} \mathbf {v} _ {i} + \tilde {\mathbf {v}} _ {i} + \mathbf {b} ^ {u}\right), \tag {2}
$$
where  $\mathbf{W}^u$  and  $\mathbf{b}^u$  are weight and bias respectively. The updated node representations  $\mathbf{v}_i'$  will be used to enhance the contextual token embedding via summation in corresponding positions. Thus  $\mathbf{t}_l' = \mathbf{t}_l + \mathbf{v}_n'$ , where  $l \in S_n$  and  $S_n$  is the corresponding token indices set for  $n$ -th EDU.
Answer Prediction The probabilities of options are obtained by feeding the discourse-enhanced token embeddings into the answer prediction module. The model is end-to-end trained using cross entropy loss. Specifically, the embedding sequence first goes through a layer normalization (Ba et al., 2016), then a bidirectional GRU (Cho et al., 2014). The output embeddings are then added to the input ones as the residual structure (He et al., 2016). We finally obtain the encoded sequence after another layer normalization on the added embeddings.
We then merge the high-level discourse features and the low-level token features. Specifically, the variant-length encoded context sequence, question-and-option sequence are pooled via weighted summation wherein the weights are softmax results of
Methods	Dev	Test	Test-E	Test-H
BERT-Large	53.80	49.80	72.00	32.30
XLNet-Large	62.00	56.00	75.70	40.50
RoBERTa-Large	62.60	55.60	75.50	40.00
DAGN	65.20	58.20	76.14	44.11
DAGN (Aug)	65.80	58.30	75.91	44.46
* The results are taken from the ReClor paper.
* DAGN ranks the 1st on the public ReClor leaderboard<sup>1</sup> until 17th Nov., 2020 before submitting it to NAACL. Until now, we find that several better results appeared in the leaderboard and they are not opened.
Table 1: Experimental results (accuracy %) of DAGN compared with baseline models on ReClor dataset. Test-E = Test-EASY, Test-H = Test-HARD.
a linear transformation of the sequence, resulting in single feature vectors separately. We concatenate them with “ $\langle s \rangle$ ” embedding from the backbone pre-trained model, and feed the new vector into a two-layer perceptron with a GELU activation (Hendrycks and Gimpel, 2016) to get the output features for classification.
We evaluate the performance of DAGN on two logical reasoning datasets, ReClor (Yu et al., 2020) and LogiQA (Liu et al., 2020), and conduct ablation study on graph construction and graph network. The implementation details are shown in Appendix B.
ReClor contains 6,138 questions modified from standardized tests such as GMAT and LSAT, which are split into train / dev / test sets with 4,638 / 500 / 1,000 samples respectively. The training set and the development set are available. The test set is blind and hold-out, and split into an EASY subset and a HARD subset according to the performance of BERT-base model (Devlin et al., 2019). The test results are obtained by submitting the test predictions to the leaderboard. LogiQA consists of 8,678 questions that are collected from National Civil Servants Examinations of China and manually translated into English by professionals. The dataset is randomly split into train / dev / test sets with 7,376 / 651 / 651 samples respectively. Both datasets contain multiple logical reasoning types.
The experimental results are shown in Tables 1 and 2. Since there is no public method for both datasets, we compare DAGN with the baseline
Methods	Dev	Test
BERT-Large	34.10	31.03
RoBERTa-Large	35.02	35.33
DAGN	35.48	38.71
DAGN (Aug)	36.87	39.32
Table 2: Experimental results (accuracy %) of DAGN compared with baseline models on LogiQA dataset.
Methods	Dev
DAGN	65.20
ablation on nodes
DAGN - clause nodes	64.40
DAGN - sentence nodes	64.40
ablation on edges
DAGN - single edge type	64.80
DAGN - fully connected edges	61.60
ablation on graph reasoning
DAGN w/o graph module	64.00
Table 3: Ablation study results (accuracy %) on ReClor development set.
models. As for DAGN, we fine-tune RoBERTa-Large as the backbone. DAGN (Aug) is a variant that augments the graph features.
DAGN reaches  $58.20\%$  of test accuracy on ReClor. DAGN (Aug) reaches  $58.30\%$ , therein  $75.91\%$  on EASY subset, and  $44.46\%$  on HARD subset. Compared with RoBERTa-Large, the improvement on the HARD subset is remarkably  $4.46\%$ . This indicates that the incorporated discourse-based information supplements the shortcoming of the baseline model, and that the discourse features are beneficial for such logical reasoning. Besides, DAGN and DAGN (Aug) also outperform the baseline models on LogiQA, especially showing  $4.01\%$  improvement over RoBERTa-Large on the test set.
We conduct ablation study on graph construction details as well as the graph reasoning module. The results are reported in Table 3.
Varied Graph Nodes We first use clauses or sentences in substitution for EDUs as graph nodes. For clause nodes, we simply remove "Explicit" connectives during discourse unit delimitation. So that the texts are just delimited by punctuation marks. For sentence nodes, we further reduce the delimiter library to solely period (":"). Using the modified graphs with clause nodes or coarser sentence nodes, the accuracy of DAGN drops to  $64.40\%$ . This indicates that clause or sentence nodes carry
https://bit.ly/2UOQfaS
less discourse information and act poorly as logical reasoning units.
Varied Graph Edges We make two changes of the edges: (1) modifying the edge type, (2) modifying the edge linking. For edge type, all edges are regarded as a single type. For edge linking, we ignore discourse relations and connect every pair of nodes, turning the graph into fully-connected. The resulting accuracies drop to  $64.80\%$  and  $61.60\%$  respectively. It is proved that in the graph we built, edges link EDUs in reasonable manners, which properly indicates the logical relations.
Ablation on Graph Reasoning We remove the graph module from DAGN and give a comparison. This model solely contains an extra prediction module than the baseline. The performance on ReClor dev set is between the baseline model and DAGN. Therefore, despite the prediction module benefits the accuracy, the lack of graph reasoning leads to the absence of discourse features and degenerates the performance. It demonstrates the necessity of discourse-based structure in logical reasoning.
Recent datasets for reading comprehension tend to be more complicated and require models' capability of reasoning. For instance, HotpotQA (Yang et al., 2018), WikiHop (Welbl et al., 2018), OpenBookQA (Mihaylov et al., 2018), and MultiRC (Khashabi et al., 2018) require the models to have multi-hop reasoning. DROP (Dua et al., 2019) and MA-TACO (Zhou et al., 2019) need the models to have numerical reasoning. WIQA (Tandon et al., 2019) and CosmosQA (Huang et al., 2019) require causal reasoning that the models can understand the counterfactual hypothesis or find out the cause-effect relationships in events. However, the logical reasoning datasets (Yu et al., 2020; Liu et al., 2020) require the models to have the logical reasoning capability of uncovering the inner logic of texts.
Deep neural networks are used for reasoning-driven RC. Evidence-based methods (Madaan et al., 2020; Huang et al., 2020; Rajagopal et al., 2020) generate explainable evidence from a given context as the backup of reasoning. Graph-based methods (Qiu et al., 2019; De Cao et al., 2019; Cao et al., 2019; Ran et al., 2019; Chen et al., 2020b; Xu et al., 2020b; Zhang et al., 2020) explicitly model the reasoning process with constructed graphs, then learn and update features through message passing
based on graphs. There are also other methods such as neuro-symbolic models (Saha et al., 2021) and adversarial training (Pereira et al., 2020). Our paper uses a graph-based model. However, for uncovering logical relations, graph nodes and edges are customized with discourse information.
Discourse information provides a high-level understanding of texts and hence is beneficial for many of the natural language tasks, for instance, text summarization (Cohan et al., 2018; Joty et al., 2019; Xu et al., 2020a; Feng et al., 2020), neural machine translation (Voita et al., 2018), and coherent text generation (Wang et al., 2020; Bosselut et al., 2018). There are also discourse-based applications for reading comprehension. DISCERN (Gao et al., 2020) segments texts into EDUs and learns interactive EDU features. Mihaylov and Frank (2019) provide additional discourse-based annotations and encodes them with discourse-aware self-attention models. Unlike previous works, DAGN first uses discourse relations as graph edges connecting EDUs for texts, then learns the discourse features via message passing with graph neural networks.
In this paper, we introduce a Discourse-Aware Graph Network (DAGN) to addressing logical reasoning QA tasks. We first treat elementary discourse units (EDUs) that are split by discourse relations as basic reasoning units. We then build discourse-based logic graphs with EDUs as nodes and discourse relations as edges. DAGN then learns the discourse-based features and enhances them with contextual token embeddings. DAGN reaches competitive performances on two recent logical reasoning datasets ReClor and LogiQA.
The authors would like to thank Wenge Liu, Jianheng Tang, Guanlin Li and Wei Wang for their support and useful discussions. This work was supported in part by National Natural Science Foundation of China (NSFC) under Grant No.U19A2073 and No.61976233, Guangdong Province Basic and Applied Basic Research (Regional Joint Fund-Key) Grant No.2019B1515120039, Shenzhen Basic Research Project (Project No. JCYJ20190807154211365), Zhijiang Lab's Open Fund (No. 2020AA3AB14) and CSIG Young Fellow Support Fund.
Jimmy Lei Ba, Jamie Ryan Kiros, and Geoffrey E Hinton. 2016. Layer normalization. stat, 1050:21.
Antoine Bosselut, Asli Celikyilmaz, Xiaodong He, Jianfeng Gao, Po-Sen Huang, and Yejin Choi. 2018. Discourse-aware neural rewards for coherent text generation. In Proceedings of the 2018 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 1 (Long Papers), pages 173-184.
Yu Cao, Meng Fang, and Dacheng Tao. 2019. Bag: Bi-directional attention entity graph convolutional network for multi-hop reasoning question answering. In Proceedings of the 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 1 (Long and Short Papers), pages 357-362.
Kunlong Chen, Weidi Xu, Xingyi Cheng, Zou Xiaochuan, Yuyu Zhang, Le Song, Taifeng Wang, Yuan Qi, and Wei Chu. 2020a. Question directed graph attention network for numerical reasoning over text. In Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP), pages 6759-6768.
Kunlong Chen, Weidi Xu, Xingyi Cheng, Zou Xiaochuan, Yuyu Zhang, Le Song, Taifeng Wang, Yuan Qi, and Wei Chu. 2020b. Question directed graph attention network for numerical reasoning over text. In Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP), pages 6759-6768, Online. Association for Computational Linguistics.
Kyunghyun Cho, Bart van Merrienboer, Dzmitry Bah-danau, and Yoshua Bengio. 2014. On the properties of neural machine translation: Encoder-decoder approaches. In Proceedings of SSST-8, Eighth Workshop on Syntax, Semantics and Structure in Statistical Translation, pages 103-111.
Arman Cohan, Franck Dernoncourt, Doo Soon Kim, Trung Bui, Seokhwan Kim, Walter Chang, and Nazli Goharian. 2018. A discourse-aware attention model for abstractive summarization of long documents. In Proceedings of the 2018 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 2 (Short Papers), pages 615-621, New Orleans, Louisiana. Association for Computational Linguistics.
Nicola De Cao, Wilker Aziz, and Ivan Titov. 2019. Question answering by reasoning across documents with graph convolutional networks. In Proceedings of the 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 1 (Long and Short Papers), pages 2306-2317.