ShifaMind: A Multiplicative Concept Bottleneck for Interpretable ICD-10 Coding
Not on the map. Why?
Abstract (arXiv)
Automated ICD-10 coding from clinical discharge summaries requires models that are both accurate on long-tailed multi-label classification tasks and interpretable to clinicians. Concept Bottleneck Models (CBMs) offer a principled framework for interpretability by routing predictions through human-interpretable concepts, but this transparency often comes at a cost: compressing rich clinical text representations into a narrow concept layer can restrict gradient flow and limit predictive capacity. We present ShifaMind, a concept-grounded architecture built around a Multiplicative Concept Bottleneck (MCB), which changes the form, rather than the width, of the bottleneck. Instead of projecting through a narrow concept layer, ShifaMind uses a learned multiplicative gate over a concept-grounded representation while retaining a scalar concept interface for inspection. On MIMIC-IV top-50 ICD-10 coding, ShifaMind achieves performance competitive with LAAT, the strongest baseline, across F1, AUC, and ranking metrics, while outperforming five additional ICD-coding baselines and providing concept-mediated explanations. Its substantial gains over a capacity-matched Vanilla CBM in both predictive performance and interpretability-oriented metrics highlight the importance of the bottleneck design.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 20 of its 27 references that are in the map. First posted on arXiv 8 May 2026.
Builds on
20 of 27 references.
References · 20
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
- 2016"Why Should I Trust You?": Explaining the Predictions of Any Classifiercited by 5,040
- 2013Deep Inside Convolutional Networks: Visualising Image Classification Models and Saliency Mapscited by 2,348
- 2017Towards A Rigorous Science of Interpretable Machine Learningcited by 1,400
- 2017Interpretability Beyond Feature Attribution: Quantitative Testing with Concept Activation Vectors (TCAV)cited by 1,309
- 2020Concept Bottleneck Modelscited by 1,022
- 2018This Looks Like That: Deep Learning for Interpretable Image Recognitioncited by 731
- 2020Towards Faithfully Interpretable NLP Systems: How should we define and evaluate faithfulness?cited by 590
- 2024Smarter, Better, Faster, Longer: A Modern Bidirectional Encoder for Fast, Memory Efficient, and Long Context Finetuning and Inferencecited by 527
- 2019On Completeness-aware Concept-Based Explanations in Deep Neural Networkscited by 239
- 2018Explainable Prediction of Medical Codes from Clinical Textcited by 231
- 2021Promises and Pitfalls of Black-Box Concept Learning Modelscited by 128
8 more
- 2022GlanceNets: Interpretabile, Leak-proof Concept-based Modelscited by 73
- 2023Interpretable Neural-Symbolic Concept Reasoningcited by 56
- 2022PLM-ICD: Automatic ICD Coding with Pretrained Language Modelscited by 49
- 2023Automated Medical Coding on MIMIC-III and MIMIC-IV: A Critical Review and Replicability Studycited by 48
- 2020A Label Attention Model for ICD Coding from Clinical Textcited by 47
- 2022Knowledge Injected Prompt Based Fine-tuning for Multi-label Few-shot ICD Codingcited by 40
- 2025BioClinical ModernBERT: A State-of-the-Art Long-Context Encoder for Biomedical and Clinical NLPcited by 26
- 2025A General Knowledge Injection Framework for ICD Codingcited by 7
Built on by
Not cited yet.