GRACE: Graph Neural Networks for Locus-of-Care Prediction under Extreme Class Imbalance
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2025
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866912829731241984 |
|---|---|
| author | Kumar, Subham Shukla, Lekhansh Mukherjee, Animesh Rudra, Koustav Shivaprakash, Prakrithi |
| author_facet | Kumar, Subham Shukla, Lekhansh Mukherjee, Animesh Rudra, Koustav Shivaprakash, Prakrithi |
| contents | Determining the appropriate locus of care for addiction patients is one of the most critical clinical decisions that affects patient treatment outcomes and effective use of resources. With a lack of sufficient specialized treatment resources, such as inpatient beds or staff, there is an unmet need to develop an automated framework for the same. Current decision-making approaches suffer from severe class imbalances in addiction datasets. To address this limitation, we propose a novel graph neural network (GRACE) framework that formalizes locus of care prediction as a structured learning problem. In addition, we propose a new approach of obtaining an unbiased meta-graph to train a GNN to overcome the class imbalance problem. Experimental results with real-world data show an improvement of 11-35% in terms of the F1 score of the minority class over competitive baselines. Further, if we jointly finetune the base embedding fed into GRACE as input together with the rest of the GNN component of GRACE, there is a remarkable boost of 15.8% in performance. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2510_20671 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | GRACE: Graph Neural Networks for Locus-of-Care Prediction under Extreme Class Imbalance Kumar, Subham Shukla, Lekhansh Mukherjee, Animesh Rudra, Koustav Shivaprakash, Prakrithi Machine Learning Artificial Intelligence Determining the appropriate locus of care for addiction patients is one of the most critical clinical decisions that affects patient treatment outcomes and effective use of resources. With a lack of sufficient specialized treatment resources, such as inpatient beds or staff, there is an unmet need to develop an automated framework for the same. Current decision-making approaches suffer from severe class imbalances in addiction datasets. To address this limitation, we propose a novel graph neural network (GRACE) framework that formalizes locus of care prediction as a structured learning problem. In addition, we propose a new approach of obtaining an unbiased meta-graph to train a GNN to overcome the class imbalance problem. Experimental results with real-world data show an improvement of 11-35% in terms of the F1 score of the minority class over competitive baselines. Further, if we jointly finetune the base embedding fed into GRACE as input together with the rest of the GNN component of GRACE, there is a remarkable boost of 15.8% in performance. |
| title | GRACE: Graph Neural Networks for Locus-of-Care Prediction under Extreme Class Imbalance |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2510.20671 |