Li Fei, Yu Hong
Department of Computer Science, University of Massachusetts Lowell, Lowell, MA, United States.
Center for Healthcare Organization and Implementation Research, Bedford Veterans Affairs Medical Center, Bedford, MA, United States.
Proc AAAI Conf Artif Intell. 2020 Feb;34(5):8180-8187. doi: 10.1609/aaai.v34i05.6331. Epub 2020 Apr 3.
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 -Filter idual onvolutional eural etwork (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.
自动ICD编码,即将疾病国际分类代码分配给患者就诊记录,因其可为计费节省时间和人力,已吸引了大量研究关注。先前的最先进模型利用一个卷积层来构建文档表示以预测ICD代码。然而,与ICD编码密切相关的文本片段的长度和语法在不同文档中差异很大。因此,一个扁平且固定长度的卷积架构可能无法学习到良好的文档表示。在本文中,我们提出了一种用于ICD编码的多滤波器残差卷积神经网络(MultiResCNN)。我们模型的创新之处有两点:它利用多滤波器卷积层来捕获不同长度的各种文本模式,并利用残差卷积层来扩大感受野。我们在广泛使用的MIMIC数据集上评估了我们模型的有效性。在MIMIC - III的完整代码集上,我们的模型在6个评估指标中的4个上优于最先进模型。在MIMIC - III的前50个代码集和MIMIC - II的完整代码集上,我们的模型在所有评估指标上均优于所有现有和最先进的模型。代码可在https://github.com/foxlf823/Multi - Filter - Residual - Convolutional - Neural - Network获取。