Making Clinical Language Models Auditable: Concept-Guided Fine-Tuning for Robust Prediction

Jin Mu, Guanhua Chen


Abstract
Clinical language models can achieve strong in-hospital accuracy yet fail under deployment shifts because they exploit note-specific artifacts (e.g., templates, separators, boilerplate) that do not reflect patient state. We propose CAST (Concept-guided Artifact Suppression Tuning), an SAE-based framework for auditable clinical text classification. CAST uses Sparse Autoencoders to expose sparse, human-auditable features from intermediate Transformer activations, labels SAE latents with an LLM-assisted interpretation pipeline and ICD-10 retrieval constraints, suppresses verified artifact latents via residual subtraction during fine-tuning, and provides post-hoc perconcept attributions for auditing model decisions. On MIMIC-IV discharge-note mortality prediction, CAST improves over its corresponding fine-tuned encoder baselines and remains competitive with strong LLM baselines, while producing a feature-level audit trail of the clinical concepts that support each prediction and the artifact concepts suppressed during training.
Anthology ID:
2026.emnlp-main.628
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
14429–14446
Language:
URL:
https://aclanthology.org/2026.emnlp-main.628/
DOI:
Bibkey:
Cite (ACL):
Jin Mu and Guanhua Chen. 2026. Making Clinical Language Models Auditable: Concept-Guided Fine-Tuning for Robust Prediction. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 14429–14446, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Making Clinical Language Models Auditable: Concept-Guided Fine-Tuning for Robust Prediction (Mu & Chen, EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.628.pdf