LEAF-QA: Locate, Encode & Attend for Figure Question Answering

07/30/2019 ∙ by Ritwick Chaudhry, et al. ∙ adobe Carnegie Mellon University Indian Institute of Technology Kanpur IIT Roorkee The Ohio State University 0

We introduce LEAF-QA, a comprehensive dataset of 250,000 densely annotated figures/charts, constructed from real-world open data sources, along with 2 million question-answer (QA) pairs querying the structure and semantics of these charts. LEAF-QA highlights the problem of multimodal QA, which is notably different from conventional visual QA (VQA), and has recently gained interest in the community. Furthermore, LEAF-QA is significantly more complex than previous attempts at chart QA, viz. FigureQA and DVQA, which present only limited variations in chart data. LEAF-QA being constructed from real-world sources, requires a novel architecture to enable question answering. To this end, LEAF-Net, a deep architecture involving chart element localization, question and answer encoding in terms of chart elements, and an attention network is proposed. Different experiments are conducted to demonstrate the challenges of QA on LEAF-QA. The proposed architecture, LEAF-Net also considerably advances the current state-of-the-art on FigureQA and DVQA.



There are no comments yet.


page 6

This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.

1 Introduction

Charts/figures (used interchangeably hereafter) are effective and widely used representations of data in documents in almost every domain. Thus, for understanding of documents by machines, it is critical to investigate into robust technologies that can interpret charts and figures, and have the ability to answer varied user queries. With advances in deep learning technologies, it is possible to interpret images such as charts, provided a large annotated corpus is available. However, it is a challenging problem to gather large scale data consisting of charts along with annotations mainly because (a) legal and copyright issues crop up while extracting charts from different documents, (b) scraping the charts from documents doesn’t provide relevant annotations of the chart elements, as the original data values are lost when the chart for the data is produced.

Current datasets on figure question answering such as FigureQA [14] and DVQA [13] have charts generated from synthetic data which have limited ranges of data values. Further, while FigureQA has questions having only binary Yes/No answers, the answer vocabulary of DVQA has only around thousand unique answers. Such assumptions do not hold in case of charts in the real-world, where each chart can potentially have a distinct vocabulary, with words as well as data values not seen before. Furthermore, FigureQA [14] chart labels are set to the color name of the corresponding plot element in the chart (a purple line would have the legend label ‘purple’). This may cause question answering models to associate colors instead of semantics of the charts with the answers. DVQA [13] introduces diversity through randomizing legend labels, colors and axis labels, however it is limited to only bar charts. Moreover, as synthetic data is used in DVQA generation, there is a limited variation in title, axis and legend labels, charts in DVQA do not have any axis labels more than six characters long. FigureSeer [27] presents a dataset of around figures scraped from different research papers, but a set of only

figures are densely annotated using Mechanical Turk. Further, a set of heuristics are presented to extract various chart elements. However, this may not generalize well to different charts, because of considerable variations in the arrangement of chart elements. Moreover, previous datasets have fixed question templates which don’t take into consideration the possible variations in natural user queries.

To this end, LEAF-QA provides a large-scale corpus of a variety of charts, which mitigates the pressing problems in existing chart question answering datasets. Firstly, charts are extracted from varied real-world data sources, like the government census or financial data, thus avoiding biases of synthesized data, as well as providing much more complex variations than the previous attempts. Chart images are plotted as multiple visualization categories viz. bar (stacked and grouped) graphs, pie/donut charts, box plots, scatter plots and line graphs. The charts are complemented with dense annotations of the different chart elements, with bounding boxes or masks as relevant (e.g. for pie wedges). Different question-answer pairs querying the chart images are provided, covering both query types: structural and relational. Moreover, questions are randomly selected from para-phrasings of question templates, to prevent question answering models from memorizing templates. LEAF-QA consists of a corpus of training chart images and million question-answer pairs. For testing, chart images and million question-answer pairs are provided. Additionally, a novel test set generated from unseen data sources is also provided to test the generalization of learning frameworks.

LEAF-QA has a large number of unique answers, which include strings contained within the charts. Hence, current deep learning architectures for VQA with fixed answer dictionaries can not be directly applied, as the test set contains answers which are not present in the training corpus. A novel framework, LEAF-Net using detection and parsing of chart elements, followed by encoding questions and answers in terms of the detected chart elements is proposed to handle this. A comprehensive evaluation of the detection of chart elements as well as the performance of the network on different question types is detailed. A thorough comparison of LEAF-Net on architectures proposed in DVQA and FigureQA is further presented.

2 Related Work

This section describes prior art related to question answering on charts.

Figure Understanding: There has been recent interest in analyzing figures and charts, particularly to understand the type of visualization and for data extraction from the chart images. Savva et al. [26] describe algorithms to extract data from pie and bar charts, particularly to re-visualize them. Further, interactive methods for bar chart extraction have been studied [12, 25]. Cliche et al. [5] describe an object detection framework for extracting scatter plot elements. Similarly, an analysis for line plot extraction has been presented by Nair et al. [18]. There have also been attempts at indexing of figures [21, 3] for search and classification. Böschen et al. [4] and Poco et al. [20] describe methods for improving text and symbol extraction from figures. Harper et al. [8] describe a framework to restyle different kinds of visualizations, through maneuvering the data in the SVGs.

Visual Question Answering: Learning to answer questions based on natural images has been an area of extensive research in recent years. Several datasets including DAQUAR [17], COCO-QA [23], VQA [2], Visual7w [31] and MovieQA [28] have been proposed to explore different facets of question answering on natural images and videos. Correspondingly, methods using attention [29, 30, 15], neural modules [1] and compositional modeling [6] have been explored. There has been related work on question answering on synthetic data  [10, 11]. However, the current work is most related to recent work on multimodal question answering [singh2019towards, biten2019icdar], which show that current VQA do not perform well while reasoning on text in natural images, and hence, there is a need to learn image and scene text jointly for question/answering.

Object Detection:

There have been significant advances in object detection frameworks, especially networks based on the recent region-based convolutional neural networks (R-CNNs) 

[7, 9, 24], YOLO [22] and SSD [16]. As the elements in LEAF-QA charts are both rectangular boxes as well as elements like pie wedges, the Mask R-CNN [9] is used for parsing chart elements.

3 Leaf-Qa dataset

In this section, the generation of figures and question-answer pairs in LEAF-QA is explained, along with details of the training and test corpora.

3.1 Data Sources

LEAF-QA uses public data tables available from various online sources, which are listed below:

1. World Development Indicators (World Bank) 111www.datacatalog.worldbank.org/dataset/world-development-indicators 2. Gender Statistics (World Bank) 222www.datacatalog.worldbank.org/dataset/gender-statistics 3. Government of India Open Data 333www.visualize.data.gov.in 4. Commodity Trade Statistics (UN data) 444www.kaggle.com/unitednations/global-commodity-trade-statistics/data 5. US Census Data (for 2000 and 2010) 555www.kaggle.com/muonneutrino/us-census-demographic-data/data 6. Price Volume Data for Stocks and ETFs 666www.kaggle.com/borismarjanovic/price-volume-data-for-all-us-stocks-etfs

Corpus Group Bar Stack bar Pie Donut Box Line Scatter
Horz. Vert. Horz. Vert. Horz. Vert.
Training 8,588 8,562 13,726 13,414 7,008 21,206 21,012 20,961 41,935 41,710
Testing 1,674 1,757 2,690 2,736 1,445 4,221 4,167 4,203 8,392 8,336
Novel Test 626 624 639 611 1250 1250 52 58 2,500 550
Corpus Questions Answers
Relational Structural Chart Vocabulary Common Vocabulary Chart Type
Training 13,30,395 1,98,904 10,06,229 3,59,498 1,63,572
Testing 2,66,318 39,764 2,01,469 71,929 32,684
Novel Test 60,697 10,408 47,706 17,289 6,110
Table 1: Statistics of different chart types in LEAF-QA.

3.2 Data pre-processing

Real-world data tables are pre-processed to make them suitable for creating different types of charts and tackle the problem of missing data. The steps are as follows:

Merging data: Data series sharing the label space, like the x and y-label names are combined to get an integrated table. For example, from stocks database consisting of multiple tables reflecting stock prices for different companies, a single table containing information for all the companies is created. This bigger table can hence be used to make charts that compare stock prices of different companies.

Decomposing Tables: Tables representing data series of multiple indicators are split along several axes to create a pool of smaller tables. The split can be done on legend labels, x-labels or y-labels for generating varied charts. For example, the “world development indicators (World Bank)” is a table assessing various indicators of different countries for multiple years. This table is decomposed into smaller tables e.g., one representing an indicator for one country over time and another an indicator of different countries for a given year.

Handling Missing/Noisy Data

: Symbols such as N/A in columns and other missing values are imputed with a random value drawn from the normal distribution fitted to data in the column. Rows containing data aggregated from multiple rows, such as the sum of columns, are removed. Columns with complex alpha-numeric data like serial numbers or hexadecimal hashing are eliminated.

3.3 Figure/Chart Generation

From each table, multiple charts are created, based on a selection of different sets of columns, followed by a randomized selection of rows from which data is to be plotted. The chart generation is done using the Matplotlib library777www.matplotlib.org/. The different chart types are created as follows:

  • Pie charts and donut charts are created from columns having numerical data in the table with their categorical label information being present as row headers.

  • Vertical/horizontal bar, stacked bar and grouped bar graphs are created using multiple numerical columns from a table with the categorical label information taken from row headers.

  • For vertical/horizontal boxplots, statistics such as median, maximum, minimum, standard deviation are calculated for columns having numeric data.

  • Line graphs are developed using multiple rows of the table (for example, time series data).

  • Scatter plots are constructed using two columns having numeric data.

3.4 Figure/Chart Style Variations

The following steps are used to make charts more readable and introduce variations in chart elements to emulate real-world data:

Legend color palette creation: Colors of plot elements are made distinct from each other by using pre-defined color palettes having optimally distinct colors, made using an online tool888http://tools.medialab.sciences-po.fr/iwanthue/index.php.

Skew/overlap removal

: In order to prevent skewed plots, the columns which are chosen to be plotted together have values with similar order of magnitude.

Element variations: Variations are introduced in chart components to emulate varied positions/fonts/sizes as is observed in real-world data. The following features are varied: (i) Title positioning, (ii) Font Families and sizes, (iii) Marker style and line width and style, (iv) Grid style and colors, (v) Legend placement, border and position of the legend entries, (vi) Width of bars, (vii) Inner and outer radii of pies, (viii) Presence of error bars.

3.5 Annotation Generation

Bounding Box Generation: Firstly, tight bounding boxes are extracted for chart elements like axis labels, title, legend box, legend items and plot elements, using Matplotlib functions, and corresponding labels are assigned. The plot elements from different chart types like bars (horizontal and vertical, stacked), pies, scatter plot markers, lines and boxes are each defined as different classes, leading to a total of classes of chart elements.

Mask Generation: For some plot elements, bounding boxes are not representative of the actual ground truth. So masks are generated for these elements:

  • Pie/Donut Plots

    : Firstly, radius and center of the pie chart are estimated using the positions of the bounding boxes. This is then used to create a mesh representing each pie element. To approximate the circle, a polygon with vertices at every 1 degree is constructed.

  • Line Plots: A single bounding box is not sufficient to represent trends in the line plot. Hence, multiple bounding boxes, each between two consecutive x-labels, are used to represent a line plot.

3.6 Question-Answer Generation

Questions in LEAF-QA have been based upon the analytical reasoning questions in the Graduate Record Examination (GRE)999https://magoosh.com/gre/2016/gre-data-interpretation-practice/, hence reflecting real-world situations. Furthermore, questions for each type of graph vary based on the suitability of the question for that type of graph. LEAF-QA contains two types of questions: structural and relational. The structural questions test understanding of the individual elements present in the chart image, while the relational questions require a deeper understanding of the relations between different plot elements. The details of the question templates are provided in the Supplementary Material. LEAF-QA does not include value-based questions to avoid complexities due to different formats of data values (e.g. pie chart values or line plot series).

Figure 2: Sample question-answers from LEAF-QA.

Paraphrases: For each question template, paraphrases for the question are generated through Google Translate API101010https://cloud.google.com/translate/docs/, with the number of paraphrases varying from based upon the complexity of the sentence. For each question, one of the paraphrases is randomly selected and the chart specific parts in the paraphrased template are replaced by words from the chart.

Answer Types: The different answer types in LEAF-QA are described as follows. The answers which are contained in the chart are referred to as Chart Vocabulary answers. These comprise of answers to questions such as those seeking the label with the maximum value in the chart. A specialized class of answers, namely the Chart Type, deals with questions such as ”What type of graph is this?”. Finally, the Common Vocabulary answers include common words such as Yes, No, None etc and numbers for questions pertaining to counting.

3.7 Training/Test Corpus

LEAF-QA has the following corpus to train/test question answering methods:

  • Training: The training set has around chart images, million questions and million answers. Note that the number of answers is higher because some questions can have multiple answers, e.g. ”Name the years in which US GDP growth rate was higher than in 2004”. There are around unique answers, which is significantly higher than previous datasets [14, 13].

  • Testing: The testing set has around images, million questions and million answers, generated from the same data sources as training.

  • Novel Testing: A smaller testing set of images, generated from data sources not seen in the training set111111www.kaggle.com/cdc/tobacco-use,www.kaggle.com/nasa/solar-eclipses,www.kaggle.com/uciml/breast-cancer-wisconsin-data,www.kaggle.com/m4h3ndr4n5p4rk3r/total-number-of-road-kills-in-india-from-20092015, is provided to test the generalization of question answering models.

The detailed break-up of charts and the question and answer types in each corpus is shown in Table 1.

Figure 3: Sample Detection output of the trained Mask R-CNN on charts from the test and novel test sets
Figure 4: The proposed LEAF-Net architecture for learning to answer questions on LEAF-QA. The question is encoded in terms of the chart elements which are located using the the Mask R-CNN and the Oracle/OCR. This encoded question and the chart image is used to calculate the spatial attention and the final answer, encoded in terms of the detected chart elements.

4 LEAF-Net baseline

In this section, a novel architecture, LEAF-Net (shown in Figure 4) is described to address the challenges of question answering on LEAF-QA. While visual question answering is a challenging task, LEAF-QA has a further challenge to enable multimodal question answering with no fixed vocabulary. LEAF-Net enables this through an integrated pipeline consisting of three stages - chart parsing, question and answer encoding and attention-based learning. Each of these is described in details below:

4.1 Chart Parsing

A Mask-RCNN network [9]

using a Resnet-101 backbone has been used to detect and classify elements of the provided chart images. The Matterport implementation

121212www.github.com/matterport/Mask_RCNN is used for the network. The training corpus of images along with the bounding boxes/masks and class labels are fed to the network for optimization. The training is done with a batch size of over NVIDA K80 GPUs, with a learning rate of , for over epochs. Some sample detection results from the trained model on the test and novel test sets are shown in Figure 3. Detailed evaluation is discussed in Section 5.

4.2 Question-Answer Encoding

As the question and answers in LEAF-QA contain words from chart specific vocabularies, it is necessary to encode them in a consistent way to enable learning by question answering algorithms. For this the following steps are taken:

Chart Text Extraction: Parsed chart components from the previous step, corresponding to text elements, are processed to extract the string inside them. During the training phase, text string from the ground truth box with the maximum overlap (min IOU 0.5) is attached to the detected box. During inference, text strings are extracted using an oracle (assuming ground truth is available) or fully automated using OCR131313https://opensource.google.com/projects/tesseract. LEAF-Net has been tested with both oracle and OCR baselines.

Text string ordering: Each text string is assigned a tuple (Element Type, Element Order) where element type is the detected class of the text string. Also, to maintain consistency in value axis of horizontal plots, the element types for x-label and y-label strings are reversed. This is determined based on variation in the horizontal and vertical lengths of the detected bars/boxes. Ordering of each text element type is done as follows: x-labels are sorted left to right and y-labels from bottom to top. The detected legend text elements are sorted top-to-bottom, left-to-right. Pie labels and pie values are sorted clockwise.

Question Encoding: Questions for a chart are then encoded in terms of the text strings which are extracted before. To identify the correspondence, the question string is matched with the list of extracted text strings, longest first. After each positive match, the matched part of the question is replaced by a specific rare word in the GloVe dictionary [19]. The rare word is pre-defined in terms of one-to-one mapping from the tuple (Element Type, Element Order) corresponding to the matched text string. The process of obtaining the ordering is detailed in the previous paragraph. The encoded question string is in turn fed to the LSTM network.

Answer Encoding: The answer text is encoded into a fixed

dimensional vector, with one dimension each for chart text elements, chart types, yes/no/none/negative/positive and numeric values (1-15 in this case). The text elements are ordered similarly as for the question, but using the actual ground truth boxes and annotations, for the purpose of training. The answer is encoded as a 1-0 vector, wherein the dimensions representing the answer text are set to 1.

For example in the pie chart in Figure 2, the question “Which country has the minimum value for 1983?” has the string “1983”, which is the legend title in the chart. The string is replaced with the rare word mapped to the legend title. Similarly the answer, which is “Antigua and Barbuda”, is the first legend label, and therefore the answer vector has a in the index reserved for the first legend label.

4.3 Attention Network

Stacked attention networks (SANs) [15, 30] have shown strong performance on the task of question answering on natural images. In LEAF-Net, the encoded question-answer pairs and chart images are used to train the attention network. The chart images are resized to and pre-processed using Resnet-152141414https://github.com/KaimingHe/deep-residual-networks to a encoding. This is concatenated with the output of the LSTM layer applied on the question encoding, and fed to spatial attention network, followed by a fully connected network. Finally, the cross-entropy loss between answer vector and the output of fully connected layer is minimized using an Adam optimizer. The network is trained using NVDIA K80 GPUs for epochs with a batch size of , an initial learning rate of , a half-life of iterations over the training corpus of images and million question-answers.

4.4 Inference

At inference time, the chart image is fed to the Mask-RCNN and the output detections are used to encode the question being asked. The encoded question and the image are then fed to the attention network to get the output vector. The answer is taken to be the chart string corresponding to the highest valued dimension in the output vector. Note that even for questions having multiple answers, the current inference produces only one answer. Multiple answer generation would be a future direction.

5 Evaluation

A thorough analysis of the performance of LEAF-Net along with different baselines for LEAF-QA are described in this section.

5.1 Chart Parsing Evaluation

Metric Calculation

: The detection performance is computed in terms of precision and recall for each chart element type, averaged over all the charts. Following prior art 

[9, 24], we start with the predicted mask with the highest confidence score and the closest unassigned ground truth mask is matched if the mask IoU is greater than . The precision is then taken as the ratio of matched predicted masks to total predicted masks, and the recall is taken as the ratio of number of matched predictions to the number of ground truth masks, and then averaged over charts.

Analysis: The detection performance for various text elements is reported in Table 2. Analysis of the detection performance of different element types is discussed below:

  • Title text: Chart titles and axis titles are detected with high precision for both test and novel test. Chart titles show high recall.

  • Axis Labels: Axis labels are detected with high precision and recall across both the sets.

  • Legend: Legend boxes are detected accurately, while the detection scores for legend title, label and preview elements are lower. This reflects the variation in legends across charts.

  • Pies: While pie values and labels have high recall, precision is lower. This may be because of the varied positioning and sizes of these elements.

5.2 Question Answering Evaluation

Evaluation Metric: The Question Answering performance is reported in terms of the accuracy of predicting an answer string present in the set of ground truth answers, given that there can be multiple correct answers. For the current baseline, getting one right answer out of multiple possible answers is considered a success. The performance has been broken down by the type of questions and further by the type of answers that the question seeks, and has been shown in Table 3.

Baselines: The following baselines are reported for evaluation:

  • QUES: For QUES baseline, the question string is passed through an LSTM layer and the loss is optimized with the answer vector. This does not take the chart image as input.

  • IMG: The chart image is re-sized and passed through SAN, with the loss being optimized as before. This assumes no question input.

  • QUES-ENC: For QUES-ENC, the question string is encoded using the output of chart parsing, which is then passed through an LSTM and optimized with the answer vector. This inherits the parsing structure from the chart image, but the further attention network is not utilized.

  • LEAF-Net: For LEAF-Net, results using both OCR and Oracle are provided.

Analysis: The performance break-down has been shown in Table 3. It can be seen that QUES, QUES-Enc and IMG baselines do not generalize well across question sets, however LEAF-Net gives a consistent performance. This ablation study shows that LEAF-QA answers are not biased, and also highlights the need for robust methods taking both the chart image and question into account to generalize well. There is a drop in performance of LEAF-Net on the novel test set as compared to the test set for certain question types. While this highlights the difficulty of the novel test set, it also calls for future work on studying adaptation. A detailed analysis of LEAF-Net performance is discussed below:

  • Structural Questions: Questions pertaining to chart type have high accuracy across test and novel test for LEAF-Net. For common vocabulary answers, there is a fall in accuracy on the novel test, however both Oracle and OCR show similar performance.

  • Relational Questions: Relational questions with common vocabulary answers show consistent performance for both Oracle and OCR. However, there is an overall decrease in performance for more complex relational questions compared to structural ones. For chart vocabulary answers, there is a fall in accuracy for the OCR case, as it involves parsing longer chart labels.

Element Test Novel Test Element Test Novel Test
Title 0.99 0.99 0.99 0.99 Legend Box 0.97 0.99 0.95 0.99
X-Axis Title 0.99 0.63 0.99 0.68 Legend Title 0.79 0.90 0.49 0.30
Y-Axis Title 0.99 0.65 0.99 0.68 Legend Label 0.91 0.99 0.85 0.99
X-Axis Label 0.99 0.98 1.00 0.99 Legend Preview 0.84 0.66 0.82 0.68
Y-Axis Label 0.99 0.98 0.99 0.99 Pie Values 0.81 0.99 0.79 0.99
Pie Label 0.51 0.99 0.47 0.99
Table 2: Class-wise precision (P) and recall (R) scores for text elements in test and novel test charts using Mask-RCNN.
Baselines Structural Relational Overall
Common Vocab Ans Chart Type Common Vocab Ans Chart Vocab Ans
Test Set
QUES 55.21 21.02 50.69 0.0 18.85
QUES-ENC (Oracle) 51.03 21.02 40.11 41.52 40.43
IMG (Oracle) 4.92 0.59 13.65 30.52 22.29
LEAF-Net (Oracle) 90.21 99.89 68.80 60.77 67.42
LEAF-Net (OCR) 90.09 99.89 65.83 42.89 55.94
Novel Test Set
QUES 38.49 6.26 41.94 0.0 14.18
QUES-ENC (Oracle) 39.06 6.26 43.72 39.45 38.60
IMG 3.52 0.0 5.72 28.55 19.05
LEAF-Net (Oracle) 72.28 96.33 62.99 55.78 61.33
LEAF-Net (OCR) 72.20 96.33 61.25 47.26 55.73
Table 3: Question Answering accuracy for different question and answer types on the test and novel test.
Baselines Test-Familiar Test-Novel
Structure Data Reasoning Overall Structure Data Reasoning Overall
SAN-VQA[13] 94.71 18.78 37.29 36.04 94.82 18.92 37.25 36.14
MOM[13] 94.71 38.20 40.99 45.03 94.82 29.14 39.26 40.90
SANDY (Oracle)[13] 96.47 65.40 44.03 56.48 96.42 65.55 44.09 56.62
SANDY (OCR)[13] 96.47 37.82 41.50 45.77 96.42 37.78 41.49 45.81
LEAF-Net (Oracle) 98.42 81.25 61.38 72.72 98.47 81.32 61.59 72.89
LEAF-Net (OCR) 98.35 74.64 57.96 68.73 98.39 74.11 58.07 68.67
Table 4: Comparison of Question Answering accuracy on DVQA[13] dataset.

Sample Runs of LEAF-Net: Performance of the model on some example questions from the test set is shown as follows:

In the first chart in Figure 5, the question seeks the box having the maximum median. Both the Oracle and OCR based models predict the answer index corresponding to the label as the answer. The Oracle thus predicts the correct string answer but the OCR predicts one extra spurious character and thus makes an incorrect prediction.

In the second example shown in Figure 5, the question involves comparing the values for a specific data series out of the three series plotted in the chart. Both the networks correctly predict the right index (corresponding to the Label) as the answer, and also recover the correct strings, thus making correct predictions.

Figure 5: Some examples of predictions using LEAF-Net (Oracle) and LEAF-Net (OCR).
Method Accuracy
Relation Net[14]
Table 5: Validation accuracy on FigureQA [14].

6 Comparisons on DVQA and FigureQA

To evaluate the performance of LEAF-Net, it has been tested on the existing figure question answering datasets, FigureQA[14] and DVQA[13]. For FigureQA, LEAF-Net has been trained with a fixed yes/no answer dictionary, without encoding, since the answers are not contained in the charts.

For DVQA, LEAF-Net has been evaluated against their enlisted baselines, namely SAN, MOM and SANDY. The Mask-RCNN, trained on LEAF-QA, has been fine-tuned with the ground truth boxes of DVQA’s bar graphs, and the question-answer pairs have been encoded with the chart specific strings.

As shown in Tables 45, LEAF-Net significantly outperforms the prior work on both the datasets. A detailed analysis of comparison with SANDY [13] is as below:

Overall performance: LEAF-Net significantly outperforms SANDY on all question types, particularly on complex data and reasoning questions. This is due to the consistent chart element-wise encoding in LEAF-Net which ensures generalizability on charts with varied vocabularies.

OCR: There is only a slight drop in OCR performance from the oracle performance in LEAF-Net as compared to a drastic reduction in SANDY. This is because chart elements are being precisely localized in LEAF-Net which leads to improved OCR, as compared to running OCR on the entire chart. This is also evident from the significantly better performance of LEAF-Net (OCR) than SANDY (Oracle).

7 Conclusion and Future Direction

This paper presents LEAF-QA, a comprehensive dataset of charts of varied types constructed from different open real-world sources, along with question answer pairs. The charts have been randomized for visual aspects and the questions have been paraphrased to avoid models from memorizing templates. The dataset is augmented with a novel test set, constructed from unseen data sources to test the generalizability of question answering models. The charts are complemented by dense annotations including masks for each plot element. A strong baseline, LEAF-Net comprising of chart parsing, question and answer encoding in terms of chart elements followed by an attention network is proposed. Ablation studies show that LEAF-Net performs high consistently across different question types. Further, LEAF-Net advances start-of-the-art on DVQA and FigureQA datasets.

LEAF-QA is an advancement towards reasoning from figures and charts emulating real-world data. However, there is still significant scope for future work. Firstly, it would be pertinent to study how users query charts/figures and incorporate it into the current question-answer framework. Further, LEAF-QA presents a complex reasoning task, and would require exploration of novel deep learning architectures to advance the field. Lastly, it would be useful to incorporate evaluation metrics from information retrieval literature to enable a more holistic view of the question answering performance.


  • [1] J. Andreas, M. Rohrbach, T. Darrell, and D. Klein. Neural module networks. In

    Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition

    , pages 39–48, 2016.
  • [2] S. Antol, A. Agrawal, J. Lu, M. Mitchell, D. Batra, C. Lawrence Zitnick, and D. Parikh. Vqa: Visual question answering. In Proceedings of the IEEE international conference on computer vision, pages 2425–2433, 2015.
  • [3] L. Battle, P. Duan, Z. Miranda, D. Mukusheva, R. Chang, and M. Stonebraker. Beagle: Automated extraction and interpretation of visualizations from the web. In Proceedings of the 2018 CHI Conference on Human Factors in Computing Systems, page 594. ACM, 2018.
  • [4] F. Böschen and A. Scherp. Multi-oriented text extraction from information graphics. In Proceedings of the 2015 ACM Symposium on Document Engineering, pages 35–38. ACM, 2015.
  • [5] M. Cliche, D. Rosenberg, D. Madeka, and C. Yee. Scatteract: Automated extraction of data from scatter plots. In

    Joint European Conference on Machine Learning and Knowledge Discovery in Databases

    , pages 135–150. Springer, 2017.
  • [6] A. Fukui, D. H. Park, D. Yang, A. Rohrbach, T. Darrell, and M. Rohrbach. Multimodal compact bilinear pooling for visual question answering and visual grounding. In

    Proceedings of the 2016 Conference on Empirical Methods in Natural Language Processing

    , pages 457–468, 2016.
  • [7] R. Girshick. Fast r-cnn. In Proceedings of the IEEE international conference on computer vision, pages 1440–1448, 2015.
  • [8] J. Harper and M. Agrawala. Deconstructing and restyling d3 visualizations. In Proceedings of the 27th annual ACM symposium on User interface software and technology, pages 253–262. ACM, 2014.
  • [9] K. He, G. Gkioxari, P. Dollár, and R. Girshick. Mask r-cnn. In Computer Vision (ICCV), 2017 IEEE International Conference on, pages 2980–2988. IEEE, 2017.
  • [10] J. Johnson, B. Hariharan, L. van der Maaten, L. Fei-Fei, C. L. Zitnick, and R. Girshick. Clevr: A diagnostic dataset for compositional language and elementary visual reasoning. In Computer Vision and Pattern Recognition (CVPR), 2017 IEEE Conference on, pages 1988–1997. IEEE, 2017.
  • [11] J. Johnson, B. Hariharan, L. van der Maaten, J. Hoffman, L. Fei-Fei, C. Lawrence Zitnick, and R. Girshick. Inferring and executing programs for visual reasoning. In The IEEE International Conference on Computer Vision (ICCV), Oct 2017.
  • [12] D. Jung, W. Kim, H. Song, J.-i. Hwang, B. Lee, B. Kim, and J. Seo. Chartsense: Interactive data extraction from chart images. In Proceedings of the 2017 CHI Conference on Human Factors in Computing Systems, pages 6706–6717. ACM, 2017.
  • [13] K. Kafle, S. Cohen, B. Price, and C. Kanan.

    Dvqa: Understanding data visualizations via question answering.

    In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pages 5648–5656, 2018.
  • [14] S. E. Kahou, A. Atkinson, V. Michalski, Ákos Kádár, A. Trischler, and Y. Bengio. FigureQA: An annotated figure dataset for visual reasoning, 2018.
  • [15] V. Kazemi and A. Elqursh. Show, ask, attend, and answer: A strong baseline for visual question answering, 2017.
  • [16] W. Liu, D. Anguelov, D. Erhan, C. Szegedy, S. Reed, C.-Y. Fu, and A. C. Berg. Ssd: Single shot multibox detector. In European conference on computer vision, pages 21–37. Springer, 2016.
  • [17] M. Malinowski and M. Fritz. A multi-world approach to question answering about real-world scenes based on uncertain input. In Advances in neural information processing systems, pages 1682–1690, 2014.
  • [18] R. R. Nair, N. Sankaran, I. Nwogu, and V. Govindaraju. Automated analysis of line plots in documents. In Document Analysis and Recognition (ICDAR), 2015 13th International Conference on, pages 796–800. IEEE, 2015.
  • [19] J. Pennington, R. Socher, and C. Manning. Glove: Global vectors for word representation. In Proceedings of the 2014 conference on empirical methods in natural language processing (EMNLP), pages 1532–1543, 2014.
  • [20] J. Poco and J. Heer. Reverse-engineering visualizations: Recovering visual encodings from chart images. In Computer Graphics Forum, volume 36, pages 353–363. Wiley Online Library, 2017.
  • [21] S. Ray Choudhury and C. L. Giles. An architecture for information extraction from figures in digital libraries. In Proceedings of the 24th International Conference on World Wide Web, pages 667–672. ACM, 2015.
  • [22] J. Redmon, S. Divvala, R. Girshick, and A. Farhadi. You only look once: Unified, real-time object detection. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 779–788, 2016.
  • [23] M. Ren, R. Kiros, and R. Zemel. Exploring models and data for image question answering. In Advances in neural information processing systems, pages 2953–2961, 2015.
  • [24] S. Ren, K. He, R. Girshick, and J. Sun. Faster r-cnn: Towards real-time object detection with region proposal networks. In Advances in neural information processing systems, pages 91–99, 2015.
  • [25] A. Rohatgi. Webplotdigitizer, 2011.
  • [26] M. Savva, N. Kong, A. Chhajta, L. Fei-Fei, M. Agrawala, and J. Heer. Revision: Automated classification, analysis and redesign of chart images. In Proceedings of the 24th annual ACM symposium on User interface software and technology, pages 393–402. ACM, 2011.
  • [27] N. Siegel, Z. Horvitz, R. Levin, S. Divvala, and A. Farhadi. Figureseer: Parsing result-figures in research papers. In European Conference on Computer Vision, pages 664–680. Springer, 2016.
  • [28] M. Tapaswi, Y. Zhu, R. Stiefelhagen, A. Torralba, R. Urtasun, and S. Fidler. Movieqa: Understanding stories in movies through question-answering. In The IEEE Conference on Computer Vision and Pattern Recognition (CVPR), June 2016.
  • [29] K. Xu, J. Ba, R. Kiros, K. Cho, A. Courville, R. Salakhudinov, R. Zemel, and Y. Bengio. Show, attend and tell: Neural image caption generation with visual attention. In International conference on machine learning, pages 2048–2057, 2015.
  • [30] Z. Yang, X. He, J. Gao, L. Deng, and A. Smola. Stacked attention networks for image question answering. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pages 21–29, 2016.
  • [31] Y. Zhu, O. Groth, M. Bernstein, and L. Fei-Fei. Visual7w: Grounded question answering in images. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pages 4995–5004, 2016.