MedCodER: A Generative AI Assistant for Medical Coding

Krishanu Das Baksi,Elijah Soba,John J. Higgins,Ravi Saini,Jaden Wood,Jane Cook,Jack Scott,Nirmala Pudota,Tim Weninger,Edward Bowen,Sanmitra Bhattacharya
2024-09-19
Abstract:Medical coding is essential for standardizing clinical data and communication but is often time-consuming and prone to errors. Traditional Natural Language Processing (NLP) methods struggle with automating coding due to the large label space, lengthy text inputs, and the absence of supporting evidence annotations that justify code selection. Recent advancements in Generative Artificial Intelligence (AI) offer promising solutions to these challenges. In this work, we introduce MedCodER, a Generative AI framework for automatic medical coding that leverages extraction, retrieval, and re-ranking techniques as core components. MedCodER achieves a micro-F1 score of 0.60 on International Classification of Diseases (ICD) code prediction, significantly outperforming state-of-the-art methods. Additionally, we present a new dataset containing medical records annotated with disease diagnoses, ICD codes, and supporting evidence texts (<a class="link-external link-https" href="https://doi.org/10.5281/zenodo.13308316" rel="external noopener nofollow">this https URL</a>). Ablation tests confirm that MedCodER's performance depends on the integration of each of its aforementioned components, as performance declines when these components are evaluated in isolation.
Computation and Language,Artificial Intelligence,Emerging Technologies,Information Retrieval,Machine Learning
What problem does this paper attempt to address?
The paper aims to address several key issues in the medical coding process: 1. **Challenges of Automatic Medical Coding**: Traditional natural language processing (NLP) methods face difficulties in automating medical coding, primarily due to the large label space, lengthy text inputs, and the lack of supporting evidence annotations to justify coding choices. 2. **Improving Accuracy and Interpretability**: The paper proposes a new generative AI framework, MedCodER, for automatic medical coding. This framework combines extraction, retrieval, and re-ranking techniques, significantly improving the performance of International Classification of Diseases (ICD) code prediction while maintaining high interpretability. 3. **Dataset Limitations**: Existing medical coding benchmark datasets (such as MIMIC) have limitations on the use of external pre-trained models and lack supporting evidence annotations, which restricts the application of generative AI solutions. Therefore, the paper also introduces a new open-source dataset that includes ICD-10 codes, diagnostic results, and supporting evidence texts to promote the development of interpretable medical coding methods. Through these efforts, the paper demonstrates the superior performance of MedCodER in ICD coding tasks and validates the effectiveness of its various components.