VizML: A Machine Learning Approach to Visualization Recommendation
© 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 n...
| Main Authors: | , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Association for Computing Machinery (ACM)
2021
|
| Online Access: | https://hdl.handle.net/1721.1/132290 |
| _version_ | 1826217563868102656 |
|---|---|
| author | Hu, Kevin Bakker, Michiel A Li, Stephen Kraska, Tim Hidalgo, César |
| author_facet | Hu, Kevin Bakker, Michiel A Li, Stephen Kraska, Tim Hidalgo, César |
| author_sort | Hu, Kevin |
| collection | MIT |
| 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. |
| first_indexed | 2024-09-23T17:05:40Z |
| format | Article |
| id | mit-1721.1/132290 |
| institution | Massachusetts Institute of Technology |
| language | English |
| last_indexed | 2024-09-23T17:05:40Z |
| publishDate | 2021 |
| publisher | Association for Computing Machinery (ACM) |
| record_format | dspace |
| spelling | mit-1721.1/1322902021-09-21T03:01:57Z VizML: A Machine Learning Approach to Visualization Recommendation Hu, Kevin Bakker, Michiel A Li, Stephen Kraska, Tim Hidalgo, César © 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. 2021-09-20T18:21:42Z 2021-09-20T18:21:42Z 2021-01-11T17:23:09Z Article http://purl.org/eprint/type/ConferencePaper https://hdl.handle.net/1721.1/132290 en 10.1145/3290605.3300358 Conference on Human Factors in Computing Systems - Proceedings Creative Commons Attribution-Noncommercial-Share Alike http://creativecommons.org/licenses/by-nc-sa/4.0/ application/pdf Association for Computing Machinery (ACM) arXiv |
| spellingShingle | Hu, Kevin Bakker, Michiel A Li, Stephen Kraska, Tim Hidalgo, César VizML: A Machine Learning Approach to Visualization Recommendation |
| title | VizML: A Machine Learning Approach to Visualization Recommendation |
| title_full | VizML: A Machine Learning Approach to Visualization Recommendation |
| title_fullStr | VizML: A Machine Learning Approach to Visualization Recommendation |
| title_full_unstemmed | VizML: A Machine Learning Approach to Visualization Recommendation |
| title_short | VizML: A Machine Learning Approach to Visualization Recommendation |
| title_sort | vizml a machine learning approach to visualization recommendation |
| url | https://hdl.handle.net/1721.1/132290 |
| work_keys_str_mv | AT hukevin vizmlamachinelearningapproachtovisualizationrecommendation AT bakkermichiela vizmlamachinelearningapproachtovisualizationrecommendation AT listephen vizmlamachinelearningapproachtovisualizationrecommendation AT kraskatim vizmlamachinelearningapproachtovisualizationrecommendation AT hidalgocesar vizmlamachinelearningapproachtovisualizationrecommendation |