Subgroup mining for performance analysis of regression models

被引:4
|
作者
Pimentel, Joao [1 ]
Azevedo, Paulo J. [1 ,2 ]
Torgo, Luis [1 ,3 ]
机构
[1] Univ Minho, Dept Informat, Braga, Portugal
[2] INESCT TEC, P-4050190 Porto, Portugal
[3] Dalhousie Univ, Fac Comp Sci, Halifax, NS, Canada
关键词
interpretability; machine learning; performance; regression;
D O I
10.1111/exsy.13118
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Machine learning algorithms have shown several advantages compared to humans, namely in terms of the scale of data that can be analysed, delivering high speed and precision. However, it is not always possible to understand how algorithms work. As a result of the complexity of some algorithms, users started to feel the need to ask for explanations, boosting the relevance of Explainable Artificial Intelligence. This field aims to explain and interpret models with the use of specific analytical methods that usually analyse how their predicted values and/or errors behave. While prediction analysis is widely studied, performance analysis has limitations for regression models. This paper proposes a rule-based approach, Error Distribution Rules (EDRs), to uncover atypical error regions, while considering multivariate feature interactions without size restrictions. Extracting EDRs is a form of subgroup mining. EDRs are model agnostic and a drill-down technique to evaluate regression models, which consider multivariate interactions between predictors. EDRs uncover regions of the input space with deviating performance providing an interpretable description of these regions. They can be regarded as a complementary tool to the standard reporting of the expected average predictive performance. Moreover, by providing interpretable descriptions of these specific regions, EDRs allow end users to understand the dangers of using regression tools for some specific cases that fall on these regions, that is, they improve the accountability of models. The performance of several models from different problems was studied, showing that our proposal allows the analysis of many situations and direct model comparison. In order to facilitate the examination of rules, two visualization tools based on boxplots and density plots were implemented. A network visualization tool is also provided to rapidly check interactions of every feature condition. An additional tool is provided by using a grid of boxplots, where comparison between quartiles of every distribution with a reference is performed. Based on this comparison, an extrapolation of counterfactual examples to regression was also implemented. A set of examples is described, including a setting where regression models performance is compared in detail using EDRs. Specifically, the error difference between two models in a dataset is studied by deriving rules highlighting regions of the input space where model performance difference is unexpected. The application of visual tools is illustrated using EDRs examples derived from public available datasets. Also, case studies illustrating the specialization of subgroups, identification of counter factual subgroups and detecting unanticipated complex models are presented. This paper extends the state of the art by providing a method to derive explanations for model performance instead of explanations for model predictions.
引用
收藏
页数:20
相关论文
共 50 条
  • [21] Enhancing project performance by developing multiple regression analysis and risk analysis models for interface
    Weshah, Nesreen
    El-Ghandour, Wael
    Falls, Lynne Cowe
    Jergeas, George
    CANADIAN JOURNAL OF CIVIL ENGINEERING, 2014, 41 (11) : 929 - 944
  • [22] Subgroup analysis of linear models with measurement error
    Le, Yuan
    Bai, Yang
    Qin, Guoyou
    CANADIAN JOURNAL OF STATISTICS-REVUE CANADIENNE DE STATISTIQUE, 2024, 52 (01): : 26 - 42
  • [23] Bayesian models for subgroup analysis in clinical trials
    Jones, Hayley E.
    Ohlssen, David I.
    Neuenschwander, Beat
    Racine, Amy
    Branson, Michael
    CLINICAL TRIALS, 2011, 8 (02) : 129 - 143
  • [24] Predictive performance of linear regression models
    Ozkale, M. Revan
    STATISTICAL PAPERS, 2015, 56 (02) : 531 - 567
  • [25] Predictive performance of linear regression models
    M. Revan Özkale
    Statistical Papers, 2015, 56 : 531 - 567
  • [26] Survival analysis and regression models
    Brandon George
    Samantha Seals
    Inmaculada Aban
    Journal of Nuclear Cardiology, 2014, 21 : 686 - 694
  • [27] DISPERSION MODELS IN REGRESSION ANALYSIS
    Song, Peter X. -K.
    PAKISTAN JOURNAL OF STATISTICS, 2009, 25 (04): : 529 - 551
  • [28] Survival analysis and regression models
    George, Brandon
    Seals, Samantha
    Aban, Inmaculada
    JOURNAL OF NUCLEAR CARDIOLOGY, 2014, 21 (04) : 686 - 694
  • [29] A Comparative Analysis on Prediction Performance of Regression Models during Machining of Composite Materials
    Bhattacharya, Shibaprasad
    Kalita, Kanak
    Cep, Robert
    Chakraborty, Shankar
    MATERIALS, 2021, 14 (21)
  • [30] Development of performance prediction models in flexible pavement using regression analysis method
    Sung-Hee Kim
    Nakseok Kim
    KSCE Journal of Civil Engineering, 2006, 10 (2) : 91 - 96