IDEAS home Printed from https://ideas.repec.org/a/gam/jeners/v14y2021i21p7049-d666504.html
   My bibliography  Save this article

Facilitating Machine Learning Model Comparison and Explanation through a Radial Visualisation

Author

Listed:
  • Jianlong Zhou

    (Data Science Institute, University of Technology Sydney, Ultimo, NSW 2007, Australia)

  • Weidong Huang

    (TD School, University of Technology Sydney, Ultimo, NSW 2007, Australia)

  • Fang Chen

    (Data Science Institute, University of Technology Sydney, Ultimo, NSW 2007, Australia)

Abstract

Building an effective Machine Learning (ML) model for a data set is a difficult task involving various steps. One of the most important steps is to compare a substantial amount of generated ML models to find the optimal one for deployment. It is challenging to compare such models with a dynamic number of features. Comparison is more than only finding differences of ML model performance, as users are also interested in the relations between features and model performance such as feature importance for ML explanations. This paper proposes RadialNet Chart , a novel visualisation approach, to compare ML models trained with a different number of features of a given data set while revealing implicit dependent relations. In RadialNet Chart, ML models and features are represented by lines and arcs, respectively. These lines are generated effectively using a recursive function. The dependence of ML models with a dynamic number of features is encoded into the structure of visualisation, where ML models and their dependent features are directly revealed from related line connections. ML model performance information is encoded with colour and line width in RadialNet Chart. Taken together with the structure of visualisation, feature importance can be directly discerned in RadialNet Chart for ML explanations. Compared with other commonly used visualisation approaches, RadialNet Chart can help to simplify the ML model comparison process with different benefits such as the following: more efficient in terms of helping users to focus their attention to find visual elements of interest and easier to compare ML performance to find optimal ML model and discern important features visually and directly instead of through complex algorithmic calculations for ML explanations.

Suggested Citation

  • Jianlong Zhou & Weidong Huang & Fang Chen, 2021. "Facilitating Machine Learning Model Comparison and Explanation through a Radial Visualisation," Energies, MDPI, vol. 14(21), pages 1-20, October.
  • Handle: RePEc:gam:jeners:v:14:y:2021:i:21:p:7049-:d:666504
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/1996-1073/14/21/7049/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/1996-1073/14/21/7049/
    Download Restriction: no
    ---><---

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jeners:v:14:y:2021:i:21:p:7049-:d:666504. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.