Neural Additive Models for Location Scale and Shape: A Framework for Interpretable Neural Regression Beyond the Mean

Anton Thielmann, René-Marcel Kruse, Thomas Kneib, Benjamin Säfken
1/27/2023
stat.MLcs.LG

Official code: AnFreTh/NAMpy (19 stars, last commit 2024-06-27).

Abstract

Deep neural networks (DNNs) have proven to be highly effective in a variety of tasks, making them the go-to method for problems requiring high-level predictive power. Despite this success, the inner workings of DNNs are often not transparent, making them difficult to interpret or understand. This lack of interpretability has led to increased research on inherently interpretable neural networks in recent years. Models such as Neural Additive Models (NAMs) achieve visual interpretability through the combination of classical statistical methods with DNNs. However, these approaches only concentrate on mean response predictions, leaving out other properties of the response distribution of the underlying data. We propose Neural Additive Models for Location Scale and Shape (NAMLSS), a modelling framework that combines the predictive power of classical deep learning models with the inherent advantages of distributional regression while maintaining the interpretability of additive models. The code is available at the following link: https://github.com/AnFreTh/NAMpy

View on arXivView PDF

Code Implementations(1)

AnFreTh/NAMpyOfficial100%
192PythonAug 15, 20232 years ago

Similar Papers

Cite this paper

@article{thielmann2023neural,
  title  = {Neural Additive Models for Location Scale and Shape: A Framework for Interpretable Neural Regression Beyond the Mean},
  author = {Anton Thielmann and René-Marcel Kruse and Thomas Kneib and Benjamin Säfken},
  year   = {2023},
  eprint = {2301.11862},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2301.11862v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2301.11862.svg)](https://codeofpaper.com/paper/2301.11862)

Discussion