Pith. sign in

REVIEW 5 cited by

Pre-training of Graph Augmented Transformers for Medication Recommendation

Not yet reviewed by Pith; the record is open.

Re-run · record.json Download PDF Read on arXiv ↗

This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.

SPECIMEN: schema-true, not a live event

T0 review · schema-true

One-sentence machine reading of the paper's core claim.

pith:XXXXXXXX · record.json · timestamp

arxiv 1906.00346 v2 pith:JRE5ID6B submitted 2019-06-02 cs.AI cs.CLcs.LG

Junyuan Shang , Tengfei Ma , Cao Xiao , Jimeng Sun This is my paper · ORCID
classification cs.AIcs.CLcs.LG
keywords medicationpatientsrecommendationrepresentationvisitencoderehrsg-bert
verification ladder T0 review T1 audit T2 compute T3 formal
0 comments
read the original abstract

Medication recommendation is an important healthcare application. It is commonly formulated as a temporal prediction task. Hence, most existing works only utilize longitudinal electronic health records (EHRs) from a small number of patients with multiple visits ignoring a large number of patients with a single visit (selection bias). Moreover, important hierarchical knowledge such as diagnosis hierarchy is not leveraged in the representation learning process. To address these challenges, we propose G-BERT, a new model to combine the power of Graph Neural Networks (GNNs) and BERT (Bidirectional Encoder Representations from Transformers) for medical code representation and medication recommendation. We use GNNs to represent the internal hierarchical structures of medical codes. Then we integrate the GNN representation into a transformer-based visit encoder and pre-train it on EHR data from patients only with a single visit. The pre-trained visit encoder and representation are then fine-tuned for downstream predictive tasks on longitudinal EHRs from patients with multiple visits. G-BERT is the first to bring the language model pre-training schema into the healthcare domain and it achieved state-of-the-art performance on the medication recommendation task.

Discussion (0). Continue with ORCID to comment.

Forward citations

Cited by 5 Pith papers

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score. Full citation record

  1. Fishing Out Free Riders: Shapley-Based Reward Attribution for Parallel Reasoning via Reinforcement Learning

    cs.AI 2026-07 conditional novelty 6.0 of 10

    Rewarding each parallel reasoning path by Monte-Carlo-Shapley marginal contribution, scored by a generative reward model, lifts Pass@16 on AIME24/AIME25/AMC23 by 4-90% relative over Parallel-R1 with a fifth of the tra...

  2. Multi-Ontology Integration with Dual-Axis Propagation for Medical Concept Representation

    cs.AI 2025-08 conditional novelty 6.0 of 10

    LINKO integrates multiple medical ontologies with dual-axis graph propagation and LLM-based initialization, improving diagnosis prediction on MIMIC-III and MIMIC-IV.

  3. Foundation Models for Clinical Records at Health System Scale

    cs.LG 2025-07 conditional novelty 6.0 of 10

    A decoder-only Transformer pretrained with next-visit multi-label prediction on EHRs achieves zero-shot dementia and knee OA forecasting comparable to a fully fine-tuned BERT baseline.

  4. SAFER: A Calibrated Risk-Aware Multimodal Recommendation Model for Dynamic Treatment Regimes

    cs.LG 2025-06 reject novelty 5.0 of 10

    SAFER combines tabular EHR and clinical notes to make treatment recommendations with a claimed conformal FDR guarantee, but the proof and evaluation do not support the formal assurances.

  5. A Comprehensive Survey of Electronic Health Record Modeling: From Deep Learning Approaches to Large Language Models

    cs.LG 2025-07 reject novelty 4.0 of 10

    A survey that taxonomizes EHR modeling research into data-centric, architectural, learning-focused, multimodal, and LLM-based categories, with datasets and metrics.

Pith tools