Horses to Zebras: Ontology-Guided Data Augmentation and Synthesis for ICD-9 Coding

Matúš Falis, Hang Dong, Alexandra Birch, Beatrice Alex


Abstract
Medical document coding is the process of assigning labels from a structured label space (ontology – e.g., ICD-9) to medical documents. This process is laborious, costly, and error-prone. In recent years, efforts have been made to automate this process with neural models. The label spaces are large (in the order of thousands of labels) and follow a big-head long-tail label distribution, giving rise to few-shot and zero-shot scenarios. Previous efforts tried to address these scenarios within the model, leading to improvements on rare labels, but worse results on frequent ones. We propose data augmentation and synthesis techniques in order to address these scenarios. We further introduce an analysis technique for this setting inspired by confusion matrices. This analysis technique points to the positive impact of data augmentation and synthesis, but also highlights more general issues of confusion within families of codes, and underprediction.
Anthology ID:
2022.bionlp-1.39
Volume:
Proceedings of the 21st Workshop on Biomedical Language Processing
Month:
May
Year:
2022
Address:
Dublin, Ireland
Editors:
Dina Demner-Fushman, Kevin Bretonnel Cohen, Sophia Ananiadou, Junichi Tsujii
Venue:
BioNLP
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
389–401
Language:
URL:
https://aclanthology.org/2022.bionlp-1.39
DOI:
10.18653/v1/2022.bionlp-1.39
Bibkey:
Cite (ACL):
Matúš Falis, Hang Dong, Alexandra Birch, and Beatrice Alex. 2022. Horses to Zebras: Ontology-Guided Data Augmentation and Synthesis for ICD-9 Coding. In Proceedings of the 21st Workshop on Biomedical Language Processing, pages 389–401, Dublin, Ireland. Association for Computational Linguistics.
Cite (Informal):
Horses to Zebras: Ontology-Guided Data Augmentation and Synthesis for ICD-9 Coding (Falis et al., BioNLP 2022)
Copy Citation:
PDF:
https://aclanthology.org/2022.bionlp-1.39.pdf
Video:
 https://aclanthology.org/2022.bionlp-1.39.mp4
Data
MIMIC-III