mlsanityData to 7 Oct 2026

ShifaMind: A Multiplicative Concept Bottleneck for Interpretable ICD-10 Coding

arXiv 2605.08482 ↗First posted 8 May 2026cs.LG cs.CLcited by 0

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

  1. 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
  2. 2016"Why Should I Trust You?": Explaining the Predictions of Any Classifiercited by 5,040
  3. 2013Deep Inside Convolutional Networks: Visualising Image Classification Models and Saliency Mapscited by 2,348
  4. 2017Towards A Rigorous Science of Interpretable Machine Learningcited by 1,400
  5. 2017Interpretability Beyond Feature Attribution: Quantitative Testing with Concept Activation Vectors (TCAV)cited by 1,309
  6. 2020Concept Bottleneck Modelscited by 1,022
  7. 2018This Looks Like That: Deep Learning for Interpretable Image Recognitioncited by 731
  8. 2020Towards Faithfully Interpretable NLP Systems: How should we define and evaluate faithfulness?cited by 590
  9. 2024Smarter, Better, Faster, Longer: A Modern Bidirectional Encoder for Fast, Memory Efficient, and Long Context Finetuning and Inferencecited by 527
  10. 2019On Completeness-aware Concept-Based Explanations in Deep Neural Networkscited by 239
  11. 2018Explainable Prediction of Medical Codes from Clinical Textcited by 231
  12. 2021Promises and Pitfalls of Black-Box Concept Learning Modelscited by 128
8 more
  1. 2022GlanceNets: Interpretabile, Leak-proof Concept-based Modelscited by 73
  2. 2023Interpretable Neural-Symbolic Concept Reasoningcited by 56
  3. 2022PLM-ICD: Automatic ICD Coding with Pretrained Language Modelscited by 49
  4. 2023Automated Medical Coding on MIMIC-III and MIMIC-IV: A Critical Review and Replicability Studycited by 48
  5. 2020A Label Attention Model for ICD Coding from Clinical Textcited by 47
  6. 2022Knowledge Injected Prompt Based Fine-tuning for Multi-label Few-shot ICD Codingcited by 40
  7. 2025BioClinical ModernBERT: A State-of-the-Art Long-Context Encoder for Biomedical and Clinical NLPcited by 26
  8. 2025A General Knowledge Injection Framework for ICD Codingcited by 7

Built on by

Not cited yet.