atlas

Research map / Label-efficient and robust learning / Recognition and long-tail

Interpretable Generalized Additive Models for Datasets with Missing Values

NeurIPS 2024 · Label-efficient and robust learning › Recognition and long-tail · approach: Feature attribution methods · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes M-GAM, a sparse generalized additive model that handles missing values by adding missingness indicators and missingness-by-threshold interaction terms, with l0 regularization for sparsity, avoiding imputation.

Problem

Maintaining interpretability of models on datasets with missing feature values; imputation complicates feature-to-label mappings, while naive missingness indicators and interactions blow up the number of terms and cause overfitting.

Approach

Extend sparse GAMs with missingness indicators and missingness-interaction terms that adjust other features' shape functions when a feature is missing, and use l0 regularization (FastSparse) to keep the model sparse, rather than imputing values.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add missingness interaction terms to M-GAMM-GAM · FICO and Breast Cancer with synthetic MAR missingnessaccuracy (FICO); AUC (Breast Cancer)Helped
Add missingness interaction terms rather than use indicators only (ablation)M-GAM · FICO and Breast Cancer with synthetic MAR missingnessaccuracy (FICO); AUC (Breast Cancer)Helped
Use M-GAM instead of impute-then-predictM-GAM · FICO, Breast Cancer, MIMIC, and Pharyngitis with naturally occurring missingnessaccuracy (FICO, MIMIC, Pharyngitis); AUC (Breast Cancer)No clear effect
Use M-GAM with missingness interactions instead of impute-then-predictM-GAM · FICO, Breast Cancer, MIMIC, and PharyngitisruntimeHelped
Use 32 evenly spaced quantiles instead of fewer quantiles (ablation)M-GAM · FICOaccuracyHelped
Vary the number of evenly spaced quantiles among 4, 8, 16, and 32 (ablation)M-GAM · FICOaccuracyNo clear effect
Increase MICE imputations from 1 to 5 (ablation)MICE impute-then-predict classifiers · FICOaccuracyHelped
Increase MICE imputations beyond 5 (ablation)MICE impute-then-predict classifiers · FICOaccuracyNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: FICO (Explainable ML Challenge) Breast Cancer (Razavi et al., 2018) MIMIC-III Pharyngitis (Miyagi, 2023) CKD (UCI) Heart Disease (UCI)

Benchmarks: FICO Breast Cancer MIMIC Pharyngitis CKD Heart Disease

Metrics: test accuracy test AUC number of nonzero coefficients runtime in seconds

Baselines: MICE MIWAE MissForest GAM w/ mean-value imputation SMIM XGBoost random forest decision tree

Builds on: FastSparse (Liu et al., 2022) Van Ness et al. (2023) missingness indicators (SMIM) multiple imputation (MICE) Le Morvan et al. (2021)

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy