Sangam: A Confluence of Knowledge Streams

VizML: A Machine Learning Approach to Visualization Recommendation

Show simple item record

dc.contributor Massachusetts Institute of Technology. Computer Science and Artificial Intelligence Laboratory
dc.contributor Massachusetts Institute of Technology. Department of Electrical Engineering and Computer Science
dc.contributor Massachusetts Institute of Technology. Media Laboratory
dc.creator Hu, Kevin
dc.creator Bakker, Michiel A
dc.creator Li, Stephen
dc.creator Kraska, Tim
dc.creator Hidalgo, Cesar Augusto
dc.date 2022-08-04T16:51:50Z
dc.date 2021-09-20T18:21:42Z
dc.date 2022-08-04T16:51:50Z
dc.date 2019
dc.date 2021-01-11T17:23:09Z
dc.date.accessioned 2023-02-17T19:56:24Z
dc.date.available 2023-02-17T19:56:24Z
dc.identifier https://hdl.handle.net/1721.1/132290.2
dc.identifier.uri http://localhost:8080/xmlui/handle/CUHPOERS/242056
dc.description © 2019 Copyright held by the owner/author(s). Publication rights licensed to ACM. Visualization recommender systems aim to lower the barrier to exploring basic visualizations by automatically generating results for analysts to search and select, rather than manually specify. Here, we demonstrate a novel machine learning-based approach to visualization recommendation that learns visualization design choices from a large corpus of datasets and associated visualizations. First, we identify five key design choices made by analysts while creating visualizations, such as selecting a visualization type and choosing to encode a column along the X-or Y-axis. We train models to predict these design choices using one million dataset-visualization pairs collected from a popular online visualization platform. Neural networks predict these design choices with high accuracy compared to baseline models. We report and interpret feature importances from one of these baseline models. To evaluate the generalizability and uncertainty of our approach, we benchmark with a crowdsourced test set, and show that the performance of our model is comparable to human performance when predicting consensus visualization type, and exceeds that of other visualization recommender systems.
dc.format application/octet-stream
dc.language en
dc.publisher Association for Computing Machinery (ACM)
dc.relation 10.1145/3290605.3300358
dc.relation Conference on Human Factors in Computing Systems - Proceedings
dc.rights Creative Commons Attribution-Noncommercial-Share Alike
dc.rights http://creativecommons.org/licenses/by-nc-sa/4.0/
dc.source arXiv
dc.title VizML: A Machine Learning Approach to Visualization Recommendation
dc.type Article
dc.type http://purl.org/eprint/type/ConferencePaper


Files in this item

Files Size Format View
1808.04819.pdf 3.921Mb application/octet-stream View/Open

This item appears in the following Collection(s)

  • DSpace@MIT [2699]
    DSpace@MIT is a digital repository for MIT's research, including peer-reviewed articles, technical reports, working papers, theses, and more.

Show simple item record

Search DSpace


Advanced Search

Browse