Multitask Balanced and Recalibrated Network for Medical Code Prediction

Wei Sun, Shaoxiong Ji, Erik Cambria, Pekka Marttinen

Tutkimustuotos: LehtiartikkeliArticleScientificvertaisarvioitu

52 Lataukset (Pure)

Abstrakti

Human coders assign standardized medical codes to clinical documents generated during patients’ hospitalization, which is error prone and labor intensive. Automated medical coding approaches have been developed using machine learning methods, such as deep neural networks. Nevertheless, automated medical coding is still challenging because of complex code association, noise in lengthy documents, and the imbalanced class problem. We propose a novel neural network, called the Multitask Balanced and Recalibrated Neural Network, to solve these issues. Significantly, the multitask learning scheme shares the relationship knowledge between different coding branches to capture code association. A recalibrated aggregation module is developed by cascading convolutional blocks to extract high-level semantic features that mitigate the impact of noise in documents. Also, the cascaded structure of the recalibrated module can benefit learning from lengthy notes. To solve the imbalanced class problem, we deploy focal loss to redistribute the attention on low- and high-frequency medical codes. Experimental results show that our proposed model outperforms competitive baselines on a real-world clinical dataset called the Medical Information Mart for Intensive Care (MIMIC-III).
AlkuperäiskieliEnglanti
Artikkeli17
Sivut1–20
Sivumäärä20
JulkaisuACM Transactions on Intelligent Systems and Technology
Vuosikerta14
Numero1
DOI - pysyväislinkit
TilaJulkaistu - 9 marrask. 2022
OKM-julkaisutyyppiA1 Alkuperäisartikkeli tieteellisessä aikakauslehdessä

Sormenjälki

Sukella tutkimusaiheisiin 'Multitask Balanced and Recalibrated Network for Medical Code Prediction'. Ne muodostavat yhdessä ainutlaatuisen sormenjäljen.

Siteeraa tätä