Orthogonal Gradient Boosting for Interpretable Additive Rule Ensembles

Fan Yang, Pierre Le Bodic, Michael Kamp, Mario Boley: Orthogonal Gradient Boosting for Interpretable Additive Rule Ensembles. In: Proceedings of the 26th International Conference on Artificial Intelligence and Statistics (AISTATS), 2024.

Abstract

Gradient boosting of prediction rules is an efficient approach to learn potentially interpretable yet accurate probabilistic models. However, actual interpretability requires to limit the number and size of the generated rules, and existing boosting variants are not designed for this purpose. Though corrective boosting refits all rule weights in each iteration to minimise prediction risk, the included rule conditions tend to be sub-optimal, because commonly used objective functions fail to anticipate this refitting. Here, we address this issue by a new objective function that measures the angle between the risk gradient vector and the projection of the condition output vector onto the orthogonal complement of the already selected conditions. This approach correctly approximates the ideal update of adding the risk gradient itself to the model and favours the inclusion of more general and thus shorter rules. As we demonstrate using a wide range of prediction tasks, this significantly improves the comprehensibility/accuracy trade-off of the fitted ensemble. Additionally, we show how objective values for related rule conditions can be computed incrementally to avoid any substantial computational overhead of the new method.

    BibTeX (Download)

    @inproceedings{yang2024orthogonal,
    title = {Orthogonal Gradient Boosting for Interpretable Additive Rule Ensembles},
    author = {Fan Yang and Pierre Le Bodic and Michael Kamp and Mario Boley},
    year  = {2024},
    date = {2024-05-02},
    urldate = {2024-05-02},
    booktitle = {Proceedings of the 26th International Conference on Artificial Intelligence and Statistics (AISTATS)},
    abstract = {Gradient boosting of prediction rules is an efficient approach to learn potentially interpretable yet accurate probabilistic models. However, actual interpretability requires to limit the number and size of the generated rules, and existing boosting variants are not designed for this purpose. Though corrective boosting refits all rule weights in each iteration to minimise prediction risk, the included rule conditions tend to be sub-optimal, because commonly used objective functions fail to anticipate this refitting. Here, we address this issue by a new objective function that measures the angle between the risk gradient vector and the projection of the condition output vector onto the orthogonal complement of the already selected conditions. This approach correctly approximates the ideal update of adding the risk gradient itself to the model and favours the inclusion of more general and thus shorter rules. As we demonstrate using a wide range of prediction tasks, this significantly improves the comprehensibility/accuracy trade-off of the fitted ensemble. Additionally, we show how objective values for related rule conditions can be computed incrementally to avoid any substantial computational overhead of the new method.},
    keywords = {complexity, explainability, interpretability, interpretable, machine learning, rule ensemble, rule mining, XAI},
    pubstate = {published},
    tppubtype = {inproceedings}
    }
    

    Leave a Reply

    Your email address will not be published. Required fields are marked *

    This site uses Akismet to reduce spam. Learn how your comment data is processed.