Multi-label classification of line chart images using convolutional neural networks


Creative Commons License

Kosemen C., Birant D.

SN APPLIED SCIENCES, vol.2, no.7, 2020 (ESCI) identifier identifier

  • Publication Type: Article / Article
  • Volume: 2 Issue: 7
  • Publication Date: 2020
  • Doi Number: 10.1007/s42452-020-3055-y
  • Journal Name: SN APPLIED SCIENCES
  • Journal Indexes: Emerging Sources Citation Index (ESCI), Scopus, INSPEC
  • Keywords: Line charts, Image classification, Multi-label classification, Convolutional neural networks, Deep learning, Machine learning, RECOGNITION, PATTERNS
  • Dokuz Eylül University Affiliated: Yes

Abstract

In this paper, we propose a new convolutional neural network (CNN) architecture to build a multi-label classifier that categorizes line chart images according to their characteristics. The class labels are organized in the form of trend property (increasing or decreasing) and functional property (linear or exponential). In the proposed method, the Canny edge detection technique is applied as a data preprocessing step to increase both the classification accuracy and training speed. In addition, two different multi-label solution approaches are compared: label powerset (LP) and binary relevance (BR) methods. The experimental studies show that the proposed LP-CNN model achieves 93.75% accuracy, while the BR-CNN model reaches 92.97% accuracy on the test set, which contains real-world line chart images. The aim of this study is to build an efficient classifier that can be used for many purposes, such as automatically captioning the chart images, providing recommendations, redesigning charts, organizing a collection of chart images and developing better search engines.