Missingness Bias Calibration in Feature Attribution Explanations
Organizations: University of Pennsylvania · University of Texas at Austin
Abstract
Popular explanation methods often produce unreliable feature importance scores due to missingness bias, a systematic distortion that arises when models are probed with ablated, out-of-distribution inputs. Existing solutions treat this as a deep representational flaw that requires expensive retraining or architectural modifications. In this work, we challenge this assumption and show that missingness bias can be effectively treated as a superficial artifact of the model's output space. We introduce MCal, a lightweight post-hoc method that corrects this bias by fine-tuning a simple linear head on the outputs of a frozen base model. Surprisingly, we find this simple correction consistently reduces missingness bias and is competitive with, or even outperforms, prior heavyweight approaches across diverse medical benchmarks spanning vision, language, and tabular domains.
Figures & tables
| Dataset | Base | Replace | Retrain | Arch | TempCal | PlattCal | MCal ( ✓ ) | |
|---|---|---|---|---|---|---|---|---|
| Vision | Brain MRI | |||||||
| CheXpert | ||||||||
| BreakHis | ||||||||
| Language | MedQA | |||||||
| MedMCQA | ||||||||
| Tabular | PhysioNet |
Appendix figures & tables5 assets
Supplementary material from the paper’s appendix.
Appendix
Feature Attribution
Feature attribution estimates which input features or internal components contribute to a model’s predictions.
See all 231 papers