sunlabuiuc · Sterilistic · Apr 18, 2026 · Apr 18, 2026 · Apr 19, 2026 · Apr 19, 2026
diff --git a/docs/api/datasets.rst b/docs/api/datasets.rst
@@ -238,6 +238,7 @@ Available Datasets
     datasets/pyhealth.datasets.BMDHSDataset
     datasets/pyhealth.datasets.COVID19CXRDataset
     datasets/pyhealth.datasets.ChestXray14Dataset
+    datasets/pyhealth.datasets.CheXpertPlusDataset
     datasets/pyhealth.datasets.PhysioNetDeIDDataset
     datasets/pyhealth.datasets.TUABDataset
     datasets/pyhealth.datasets.TUEVDataset

diff --git a/docs/api/datasets/pyhealth.datasets.CheXpertPlusDataset.rst b/docs/api/datasets/pyhealth.datasets.CheXpertPlusDataset.rst
@@ -0,0 +1,12 @@
+pyhealth.datasets.CheXpertPlusDataset
+====================================
+
+CheXpert Plus dataset wrapper for report-driven chest X-ray knowledge graph
+extraction. For more information see the dataset page
+`here <https://stanfordaimi.azurewebsites.net/datasets/5158c524-d3ab-4e02-96e9-6ee9efc110a1>`_
+and paper `here <https://arxiv.org/abs/2405.19111>`_.
+
+.. autoclass:: pyhealth.datasets.CheXpertPlusDataset
+    :members:
+    :undoc-members:
+    :show-inheritance:
diff --git a/docs/api/models.rst b/docs/api/models.rst
@@ -177,6 +177,7 @@ API Reference
     models/pyhealth.models.GNN
     models/pyhealth.models.Transformer
     models/pyhealth.models.TransformersModel
+    models/pyhealth.models.ReXKGModel
     models/pyhealth.models.TransformerDeID
     models/pyhealth.models.RETAIN
     models/pyhealth.models.GAMENet

diff --git a/docs/api/models/pyhealth.models.ReXKGModel.rst b/docs/api/models/pyhealth.models.ReXKGModel.rst
@@ -0,0 +1,10 @@
+pyhealth.models.ReXKGModel
+===========================
+
+ReXKG model for span-based radiology NER, pairwise relation extraction,
+and downstream knowledge graph construction.
+
+.. autoclass:: pyhealth.models.ReXKGModel
+    :members:
+    :undoc-members:
+    :show-inheritance:
diff --git a/docs/api/tasks.rst b/docs/api/tasks.rst
@@ -226,6 +226,7 @@ Available Tasks
     ChestX-ray14 Binary Classification <tasks/pyhealth.tasks.ChestXray14BinaryClassification>
     De-Identification NER <tasks/pyhealth.tasks.DeIDNERTask>
     ChestX-ray14 Multilabel Classification <tasks/pyhealth.tasks.ChestXray14MultilabelClassification>
+    Radiology KG Extraction <tasks/pyhealth.tasks.RadiologyKGExtractionTask>
     Variant Classification (ClinVar) <tasks/pyhealth.tasks.VariantClassificationClinVar>
     Mutation Pathogenicity (COSMIC) <tasks/pyhealth.tasks.MutationPathogenicityPrediction>
     Cancer Survival Prediction (TCGA) <tasks/pyhealth.tasks.CancerSurvivalPrediction>

diff --git a/docs/api/tasks/pyhealth.tasks.RadiologyKGExtractionTask.rst b/docs/api/tasks/pyhealth.tasks.RadiologyKGExtractionTask.rst
@@ -0,0 +1,9 @@
+pyhealth.tasks.RadiologyKGExtractionTask
+========================================
+
+Task for ReXKG-style entity and relation extraction from radiology reports.
+
+.. autoclass:: pyhealth.tasks.RadiologyKGExtractionTask
+    :members:
+    :undoc-members:
+    :show-inheritance:
diff --git a/docs/tutorials.rst b/docs/tutorials.rst
@@ -218,6 +218,8 @@ These examples are located in ``examples/cxr/``.
      - VAE for chest X-ray image generation
    * - ``cxr/ChestXray-image-generation-GAN.ipynb``
      - GAN for chest X-ray image generation (notebook)
+   * - ``cxr/chexpert_plus_rexkg.ipynb``
+     - Radiology KG extraction with CheXpert Plus (ReXKG pipeline, notebook)
 
 Interpretability
 ----------------