Towards Understanding Attention-based Reasoning through Graph Structures in Medical Codes Classification

Noon Goldstein, Saadullah Amin, Günter Neumann


Abstract
A common approach to automatically assigning diagnostic and procedural clinical codes to health records is to solve the task as a multi-label classification problem. Difficulties associated with this task stem from domain knowledge requirements, long document texts, large and imbalanced label space, reflecting the breadth and dependencies between medical diagnoses and procedures. Decisions in the healthcare domain also need to demonstrate sound reasoning, both when they are correct and when they are erroneous. Existing works address some of these challenges by incorporating external knowledge, which can be encoded into a graph-structured format. Incorporating graph structures on the output label space or between the input document and output label spaces have shown promising results in medical codes classification. Limited focus has been put on utilizing graph-based representation on the input document space. To partially bridge this gap, we represent clinical texts as graph-structured data through the UMLS Metathesaurus; we explore implicit graph representation through pre-trained knowledge graph embeddings and explicit domain-knowledge guided encoding of document concepts and relational information through graph neural networks. Our findings highlight the benefits of pre-trained knowledge graph embeddings in understanding model’s attention-based reasoning. In contrast, transparent domain knowledge guidance in graph encoder approaches is overshadowed by performance loss. Our qualitative analysis identifies limitations that contribute to prediction errors.
Anthology ID:
2024.textgraphs-1.6
Volume:
Proceedings of TextGraphs-17: Graph-based Methods for Natural Language Processing
Month:
August
Year:
2024
Address:
Bangkok, Thailand
Editors:
Dmitry Ustalov, Yanjun Gao, Alexander Pachenko, Elena Tutubalina, Irina Nikishina, Arti Ramesh, Andrey Sakhovskiy, Ricardo Usbeck, Gerald Penn, Marco Valentino
Venues:
TextGraphs | WS
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
78–92
Language:
URL:
https://aclanthology.org/2024.textgraphs-1.6
DOI:
Bibkey:
Cite (ACL):
Noon Goldstein, Saadullah Amin, and Günter Neumann. 2024. Towards Understanding Attention-based Reasoning through Graph Structures in Medical Codes Classification. In Proceedings of TextGraphs-17: Graph-based Methods for Natural Language Processing, pages 78–92, Bangkok, Thailand. Association for Computational Linguistics.
Cite (Informal):
Towards Understanding Attention-based Reasoning through Graph Structures in Medical Codes Classification (Goldstein et al., TextGraphs-WS 2024)
Copy Citation:
PDF:
https://aclanthology.org/2024.textgraphs-1.6.pdf