Improving Model Safety by Targeted Error Correction
Not on the map. Why?
Abstract (arXiv)
The widespread adoption of machine learning in critical applications demands techniques to mitigate high-consequence errors. Our method utilizes a dual-classifier GBDT pipeline to distinguish routine human-like errors from high-risk non-human misclassifications. Evaluated across three domains, animal breed classification, skin lesion diagnosis (ISIC 2018), and prostate histopathology (SICAPv2), our framework demonstrates robust safety improvements. To address real-world deployment concerns, our results confirm the pipeline introduces negligible inference latency (1.60% overhead for the animal dataset, 1.84% for ISIC, and 1.70% for SICAPv2) while outperforming traditional Maximum Class Probability (MCP) baselines in correction precision. Our conservative correction strategy successfully reduced dangerous non-human errors by 34.1% in ISIC and 12.57% in SICAPv2, improving super-class diagnostic safety to 90.41% and 92.13% respectively. This proves that safety-critical reliability can be substantially enhanced post-hoc without expensive model retraining. keywords: Error Analysis, Post-hoc Correction, Trustworthy AI.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 19 of its 33 references that are in the map. First posted on arXiv 4 May 2026.
Builds on
19 of 33 references.
References · 19
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2016"Why Should I Trust You?": Explaining the Predictions of Any Classifiercited by 5,040
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
- 2020Shortcut Learning in Deep Neural Networkscited by 1,928
- 2018ImageNet-trained CNNs are biased towards texture; increasing shape bias improves accuracy and robustnesscited by 1,430
- 2019Skin Lesion Analysis Toward Melanoma Detection 2018: A Challenge Hosted by the International Skin Imaging Collaboration (ISIC)cited by 657
- 2019SelectiveNet: A Deep Neural Network with an Integrated Reject Optioncited by 333
- 2018To Trust Or Not To Trust A Classifiercited by 205
7 more
- 2021Going Deeper through the Gleason Scoring Scale: An Automatic end-to-end System for Histology Prostate Grading and Cribriform Pattern Detectioncited by 66
- 2023Concrete Problems in AI Safety, Revisitedcited by 21
- 2026Beyond the Mean: Distribution-Aware Loss Functions for Bimodal Regressioncited by 2
- 2026Face Density as a Proxy for Data Complexity: Quantifying the Hardness of Instance Countcited by 2
- 2025A Comparison of Human and Machine Learning Errors in Face Recognitioncited by 1
- 2025Human Response to Decision Support in Face Matching: The Influence of Task Difficulty and Machine Accuracycited by 1
- 2026Risk-Calibrated Learning: Minimizing Fatal Errors in Medical AIcited by 1
Built on by
Not cited yet.