English

Description-based Label Attention Classifier for Explainable ICD-9 Classification

Machine Learning 2021-09-27 v1 Information Retrieval

Abstract

ICD-9 coding is a relevant clinical billing task, where unstructured texts with information about a patient's diagnosis and treatments are annotated with multiple ICD-9 codes. Automated ICD-9 coding is an active research field, where CNN- and RNN-based model architectures represent the state-of-the-art approaches. In this work, we propose a description-based label attention classifier to improve the model explainability when dealing with noisy texts like clinical notes. We evaluate our proposed method with different transformer-based encoders on the MIMIC-III-50 dataset. Our method achieves strong results together with augmented explainablilty.

Keywords

Cite

@article{arxiv.2109.12026,
  title  = {Description-based Label Attention Classifier for Explainable ICD-9 Classification},
  author = {Malte Feucht and Zhiliang Wu and Sophia Althammer and Volker Tresp},
  journal= {arXiv preprint arXiv:2109.12026},
  year   = {2021}
}

Comments

Accepted at the Workshop on Noisy User-generated Text (W-NUT) at EMNLP 2021