ICD Coding from Clinical Text Using Multi-Filter Residual Convolutional Neural Network

Fei Li, Hong Yu

Proceedings of the AAAI Conference on Artificial Intelligence · 2020 · 169 citations · 28 references

DOIFull text

Open access

Concepts

TL;DR

Automated ICD coding reduces billing time and labor, yet prior single‑layer convolutional models struggle with the variable length and grammar of clinical text, limiting representation quality. This study introduces a Multi‑Filter Residual Convolutional Neural Network (MultiResCNN) to improve ICD code prediction. MultiResCNN employs multi‑filter convolutions to capture patterns of varying lengths and residual connections to enlarge the receptive field, and is evaluated on the MIMIC dataset. The model surpasses state‑of‑the‑art baselines on MIMIC‑III’s full code set (4 of 6 metrics) and outperforms all existing methods on MIMIC‑III top‑50 and MIMIC‑II full code sets across all metrics, with code released at the provided GitHub link.

Abstract

Automated ICD coding, which assigns the International Classification of Disease codes to patient visits, has attracted much research attention since it can save time and labor for billing. The previous state-of-the-art model utilized one convolutional layer to build document representations for predicting ICD codes. However, the lengths and grammar of text fragments, which are closely related to ICD coding, vary a lot in different documents. Therefore, a flat and fixed-length convolutional architecture may not be capable of learning good document representations. In this paper, we proposed a Multi-Filter Residual Convolutional Neural Network (MultiResCNN) for ICD coding. The innovations of our model are two-folds: it utilizes a multi-filter convolutional layer to capture various text patterns with different lengths and a residual convolutional layer to enlarge the receptive field. We evaluated the effectiveness of our model on the widely-used MIMIC dataset. On the full code set of MIMIC-III, our model outperformed the state-of-the-art model in 4 out of 6 evaluation metrics. On the top-50 code set of MIMIC-III and the full code set of MIMIC-II, our model outperformed all the existing and state-of-the-art models in all evaluation metrics. The code is available at https://github.com/foxlf823/Multi-Filter-Residual-Convolutional-Neural-Network.

References

28